import { mount, waitMountApp } from "./ui"
import {
  chatDocsPanel,
  pipLauncher,
  pipLoading,
  pipWindow,
  sidebarAddon,
} from "@/store/content"
import {
  ContentEventType,
  FrameMessageType,
  MessageType,
  ServiceFunc,
  WebviewFunc,
  WindowName,
} from "@/types"
import Copilot from "./Copilot.vue"
import {
  dispatchContentEvent,
  addContentEventListener,
  removeContentEventListener,
} from "@/content/event"
import { messageInvoke } from "@/utils/invoke"
import { getPageIcon } from "@/utils/dom"
// import { PipEventName } from "@/types/pip"

function handleMessage(
  message: any,
  sender: chrome.runtime.MessageSender,
  sendResponse: (res: any) => void
) {
  console.log("[Content]", message.type, message, sender)
  switch (message?.type) {
    case MessageType.pip:
      dispatchContentEvent({
        type: ContentEventType.pip,
        detail: message.options,
      })
      break
    case MessageType.pipLaunch:
      pipLauncher.visible = true
      break
    case MessageType.hiContent:
      chrome.runtime.sendMessage({
        type: MessageType.contentHere,
      })
      sendResponse({ type: MessageType.contentHere })
      break
    case MessageType.showChatDocs:
      chatDocsPanel.visible = true
      break
    // case MessageType.openContentSidebar:
    //   sidebarAddon.visible = true
    //   sidebarAddon.collapse = false
    //   break
    case MessageType.invokeResponse:
      messageInvoke.handleResMsg(message)
      break
    case MessageType.invokeRequest:
      messageInvoke.handleReqMsg(message).then((result) => {
        if (result) {
          chrome.runtime.sendMessage({
            type: MessageType.invokeResponse,
            ...result,
          })
        }
      })
      break
  }
}

async function handlePipEvent(event: any) {
  const win = await new Promise<Window | null>((r) => {
    const docPip = window.documentPictureInPicture
    const handleEnter = () => {
      r(docPip.window)
      docPip?.removeEventListener("enter", handleEnter)
    }
    docPip?.addEventListener("enter", handleEnter)
  })

  console.log("content pip event: ", event)

  if (win) {
    pipWindow.window = win
    mount(Copilot, win.document)

    await new Promise<void>((r) => {
      const handlePipLoaded = (e: Event) => {
        console.log("load", e)
        r()
        removeContentEventListener(ContentEventType.pipLoaded, handlePipLoaded)
      }
      addContentEventListener(ContentEventType.pipLoaded, handlePipLoaded)
    })

    const window = await messageInvoke.invoke({
      func: ServiceFunc.getPipWindow,
      args: [
        {
          width: win.outerWidth,
          height: win.outerHeight,
        },
      ],
    })

    const tab = await messageInvoke.invoke({
      func: ServiceFunc.getMyTab,
      args: [],
    })

    pipWindow.windowsWindow = window
    chrome.storage.local.set({
      pipWindow: {
        windowId: window.id,
        tabId: tab.id,
        icon: getPageIcon(),
      },
    })

    win.addEventListener("pagehide", () => {
      chrome.storage.local.set({
        pipWindow: null,
      })
    })
  }
}

async function handlePipLoadedEvent(e: Event) {
  console.log("e: ", e)
  pipLoading.splashScreen = false
  pipLoading.isLoading = false
  const win = window.documentPictureInPicture.window
  if (win) {
    mount(Copilot, win.document)
  }
}

async function handlePopLoadDocEvent(e: CustomEvent | Event) {
  pipLoading.isLoading = true
}

function run() {
  chrome.runtime?.onMessage.addListener(handleMessage)
  addContentEventListener(ContentEventType.pip, handlePipEvent)
  addContentEventListener(ContentEventType.pipLoaded, handlePipLoadedEvent)
  addContentEventListener(ContentEventType.pipLoad, handlePopLoadDocEvent)
  waitMountApp()
}

async function postPageInfo() {
  await new Promise((r) => setTimeout(r, 200))
  window.parent?.postMessage(
    {
      type: FrameMessageType.pageInfo,
      url: location.href,
      title: document.title || location.host,
      icon: getPageIcon(),
    },
    chrome.runtime.getURL("")
  )
}

function handleFrameMessage(e: MessageEvent) {
  if (!e.data || typeof e.data !== "object") return
  const type = e.data.type
  switch (type) {
    case FrameMessageType.escapeLoad:
      return dispatchContentEvent({
        type: ContentEventType.escapeLoad,
        detail: { url: e.data.url },
      })
    case FrameMessageType.invokeRequest:
      handleWebviewInvokeRequest(e.data, e.source as Window)
      break
  }
}

async function handleWebviewInvokeRequest(message: any, source: Window) {
  const { key, func, args } = message
  let result = null
  let error = null
  try {
    switch (func) {
      case WebviewFunc.reload:
        location.reload()
        break
      case WebviewFunc.goBack:
        history.back()
        break
      case WebviewFunc.goForward:
        history.forward()
        break
    }
  } catch (err) {
    console.error("invoke error: ", err)
    error = err
  }
  console.log("invoke response: ", result, error)
  source.postMessage(
    {
      type: FrameMessageType.invokeResponse,
      key,
      success: !error,
      payload: !error ? result : error,
    },
    chrome.runtime.getURL("")
  )
}

if (window.top === window) {
  run()
}

import { setupMcpClient } from "./mcp-injector"

// webview
if (window.top !== window && window.name.startsWith(WindowName.webview)) {
  window.addEventListener("message", handleFrameMessage)
  window.parent?.postMessage(
    {
      type: FrameMessageType.frameReady,
      url: location.href,
    },
    chrome.runtime.getURL("")
  )

  run()
  postPageInfo()
  
  // Connect to the MCP Server proxy
  setupMcpClient().catch(console.error)
}

// dev
if (location.host == chrome.runtime.id && location.hash == "#copilot") {
  pipWindow.window = window
  mount(Copilot, window.document)
}
