import { app, BrowserWindow, dialog, ipcMain, shell } from 'electron'
import { mkdirSync } from 'node:fs'
import { userInfo } from 'node:os'
import { AgentService } from './pi/agent-service'
import { getPluginHost } from './plugins/plugin-host'
import { getBrowserController } from './browser/browser-controller'
import { detectBrowserSources, importBrowserProfile } from './browser/browser-profile'

/**
 * 注册 IPC 通道并把 agent 事件转发到所有渲染窗口。
 * 返回 service 实例供入口进程使用。
 */
export function registerIpc(): AgentService {
  const service = new AgentService()

  service.onEvent((event) => {
    for (const win of BrowserWindow.getAllWindows()) {
      if (!win.isDestroyed()) {
        win.webContents.send('agent:event', event)
      }
    }
  })

  ipcMain.handle('agent:send', (_event, text: unknown, images: unknown) => {
    if (typeof text !== 'string' || !text.trim()) {
      throw new Error('empty prompt')
    }
    const sendImages = Array.isArray(images)
      ? images.filter(
          (img): img is { mimeType: string; data: string } =>
            typeof img === 'object' &&
            img !== null &&
            typeof (img as { mimeType?: unknown }).mimeType === 'string' &&
            typeof (img as { data?: unknown }).data === 'string'
        )
      : undefined
    return service.send(text, sendImages)
  })

  ipcMain.handle('agent:steer', (_event, text: unknown) => {
    if (typeof text !== 'string' || !text.trim()) {
      throw new Error('empty steer text')
    }
    return service.steer(text)
  })

  ipcMain.handle('agent:stop', () => service.stop())

  ipcMain.handle('agent:state', () =>
    service.getState().catch((err: unknown) => {
      throw new Error(err instanceof Error ? err.message : String(err))
    })
  )

  // ---- 会话管理 ----
  ipcMain.handle('sessions:list', () =>
    service.listSessions().catch((err: unknown) => {
      throw new Error(err instanceof Error ? err.message : String(err))
    })
  )

  ipcMain.handle('sessions:new', () =>
    service.newSession().catch((err: unknown) => {
      throw new Error(err instanceof Error ? err.message : String(err))
    })
  )

  ipcMain.handle('sessions:open', (_event, path: unknown) => {
    if (typeof path !== 'string' || !path.trim()) throw new Error('invalid session path')
    return service.openSession(path).catch((err: unknown) => {
      throw new Error(err instanceof Error ? err.message : String(err))
    })
  })

  ipcMain.handle('sessions:rename', (_event, name: unknown) => {
    if (typeof name !== 'string' || !name.trim()) throw new Error('invalid session name')
    return service.renameSession(name.trim()).catch((err: unknown) => {
      throw new Error(err instanceof Error ? err.message : String(err))
    })
  })

  ipcMain.handle('sessions:delete', (_event, path: unknown) => {
    if (typeof path !== 'string' || !path.trim()) throw new Error('invalid session path')
    return service.deleteSession(path).catch((err: unknown) => {
      throw new Error(err instanceof Error ? err.message : String(err))
    })
  })

  // ---- 模型与凭据 ----
  const pass = (fn: () => Promise<unknown>): Promise<unknown> =>
    fn().catch((err: unknown) => {
      throw new Error(err instanceof Error ? err.message : String(err))
    })

  ipcMain.handle('models:catalog', () => pass(() => service.getModelCatalog()))

  // ---- 自定义 API 提供商（models.json）----
  ipcMain.handle('models:custom:list', () => service.listCustomProviders())

  ipcMain.handle('models:custom:save', (_event, config: unknown) => {
    if (typeof config !== 'object' || config === null) throw new Error('invalid custom provider')
    const c = config as Record<string, unknown>
    if (
      typeof c.id !== 'string' ||
      typeof c.baseUrl !== 'string' ||
      typeof c.apiKey !== 'string' ||
      typeof c.api !== 'string' ||
      !Array.isArray(c.models)
    ) {
      throw new Error('invalid custom provider')
    }
    const normalized = {
      id: c.id,
      name: typeof c.name === 'string' ? c.name : c.id,
      baseUrl: c.baseUrl,
      apiKey: c.apiKey,
      api: c.api,
      models: c.models.filter((m): m is string => typeof m === 'string')
    }
    return pass(() => service.saveCustomProvider(normalized))
  })

  ipcMain.handle('models:custom:delete', (_event, id: unknown) => {
    if (typeof id !== 'string' || !id.trim()) throw new Error('invalid provider id')
    return pass(() => service.deleteCustomProvider(id))
  })

  ipcMain.handle('models:set', (_event, providerId: unknown, modelId: unknown) => {
    if (typeof providerId !== 'string' || typeof modelId !== 'string') throw new Error('invalid model')
    return pass(() => service.setModel(providerId, modelId))
  })

  ipcMain.handle('auth:setKey', (_event, providerId: unknown, apiKey: unknown) => {
    if (typeof providerId !== 'string' || typeof apiKey !== 'string' || !apiKey.trim()) {
      throw new Error('invalid api key')
    }
    return pass(() => service.setApiKey(providerId, apiKey.trim()))
  })

  ipcMain.handle('auth:logout', (_event, providerId: unknown) => {
    if (typeof providerId !== 'string') throw new Error('invalid provider')
    return pass(() => service.logoutProvider(providerId))
  })

  // ---- 思考等级 / 压缩 / 分支摘要（pi 原生能力） ----
  ipcMain.handle('thinking:set', (_event, level: unknown) => {
    if (typeof level !== 'string') throw new Error('invalid thinking level')
    return pass(() => service.setThinkingLevel(level as never))
  })

  ipcMain.handle('session:compact', (_event, instructions: unknown) => {
    if (instructions !== undefined && instructions !== null && typeof instructions !== 'string') {
      throw new Error('invalid compaction instructions')
    }
    return pass(() => service.compactSession(instructions || undefined))
  })

  ipcMain.handle('session:compact:abort', () => pass(() => service.abortCompaction()))

  ipcMain.handle('session:autoCompact', (_event, enabled: unknown) => {
    if (typeof enabled !== 'boolean') throw new Error('invalid flag')
    return pass(() => service.setAutoCompactionEnabled(enabled))
  })

  ipcMain.handle('session:forkPoints', () => pass(() => service.listForkPoints()))

  ipcMain.handle('session:navigate', (_event, entryId: unknown, options: unknown) => {
    if (typeof entryId !== 'string' || !entryId.trim()) throw new Error('invalid entry id')
    const o = (options ?? {}) as { summarize?: unknown; customInstructions?: unknown }
    if (o.summarize !== undefined && typeof o.summarize !== 'boolean') throw new Error('invalid navigate options')
    if (o.customInstructions !== undefined && typeof o.customInstructions !== 'string') {
      throw new Error('invalid navigate options')
    }
    return pass(() =>
      service.navigateTree(entryId, {
        ...(o.summarize === true ? { summarize: true } : {}),
        ...(typeof o.customInstructions === 'string' ? { customInstructions: o.customInstructions } : {})
      })
    )
  })

  // 会话末尾分叉为新会话（pi fork，原会话保留）
  ipcMain.handle('session:fork', () => pass(() => service.forkSession()))

  // 从会话树上的锚点分叉为新会话（'at' 含该条为止 / 'before' 到其父条目为止）
  ipcMain.handle('session:forkAt', (_event, entryId: unknown, position: unknown) => {
    if (typeof entryId !== 'string' || !entryId.trim()) throw new Error('invalid entry id')
    if (position !== 'at' && position !== 'before') throw new Error('invalid fork position')
    return pass(() => service.forkSessionAt(entryId, position))
  })

  // ---- 应用设置 ----
  ipcMain.handle('settings:get', () => service.getSettings())

  // 当前生效的系统提示词（pi 原生 + 技能注入，只读展示）
  ipcMain.handle('settings:getSystemPrompt', () => service.getSystemPrompt())

  ipcMain.handle('settings:setTheme', (_event, id: unknown) => {
    if (typeof id !== 'string' || !id.trim()) throw new Error('invalid theme id')
    service.setThemeColor(id.trim())
  })

  ipcMain.handle('settings:setThemeMode', (_event, mode: unknown) => {
    if (mode !== 'light' && mode !== 'dark' && mode !== 'system') throw new Error('invalid theme mode')
    service.setThemeMode(mode)
  })

  // 启用/禁用单个技能或命令（重建运行时使发现阶段过滤生效）
  ipcMain.handle('settings:setResourceEnabled', (_event, kind: unknown, name: unknown, enabled: unknown) => {
    if (kind !== 'skill' && kind !== 'command') throw new Error('invalid resource kind')
    if (typeof name !== 'string' || !name.trim() || name.length > 200) throw new Error('invalid resource name')
    if (typeof enabled !== 'boolean') throw new Error('invalid enabled flag')
    return service.setResourceEnabled(kind, name, enabled)
  })

  // ---- 工作区 ----
  ipcMain.handle('workspaces:list', () =>
    service.listWorkspaces().catch((err: unknown) => {
      throw new Error(err instanceof Error ? err.message : String(err))
    })
  )

  ipcMain.handle('workspaces:create', (_event, name: unknown) => {
    if (typeof name !== 'string' || !name.trim()) throw new Error('invalid workspace name')
    return service.createWorkspace(name).catch((err: unknown) => {
      throw new Error(err instanceof Error ? err.message : String(err))
    })
  })

  ipcMain.handle('workspaces:open', (_event, path: unknown) => {
    if (typeof path !== 'string' || !path.trim()) throw new Error('invalid workspace path')
    return service.openWorkspace(path).catch((err: unknown) => {
      throw new Error(err instanceof Error ? err.message : String(err))
    })
  })

  // 调起系统目录选择器；取消返回 null
  ipcMain.handle('workspaces:browse', async () => {
    const win = BrowserWindow.getFocusedWindow() ?? BrowserWindow.getAllWindows()[0]
    const result = await dialog.showOpenDialog(win, {
      title: '选择工作区目录',
      defaultPath: service.workspacesRoot,
      properties: ['openDirectory', 'createDirectory']
    })
    if (result.canceled || result.filePaths.length === 0) return null
    return result.filePaths[0]
  })

  // ---- 工具审批 ----
  ipcMain.handle('approval:resolve', (_event, id: unknown, approved: unknown) => {
    if (typeof id !== 'string' || typeof approved !== 'boolean') throw new Error('invalid approval')
    // settle 内部经 onSettle 订阅发出 approval_settled，此处无需重复广播
    service.approvals.settle(id, approved)
    return Promise.resolve()
  })

  ipcMain.handle('app:versions', () => ({
    app: process.env.npm_package_version ?? '0.0.0',
    electron: process.versions.electron,
    node: process.versions.node,
    chrome: process.versions.chrome,
    user: userInfo().username
  }))

  // ---- Chatbox 资源（/ 命令 + 技能、@ 工作区文件、+ 附件）----
  ipcMain.handle('chat:resources', () =>
    service.listChatResources().catch((err: unknown) => {
      throw new Error(err instanceof Error ? err.message : String(err))
    })
  )

  ipcMain.handle('queue:set', (_event, steering: unknown, followUp: unknown) => {
    const norm = (value: unknown): string[] =>
      Array.isArray(value) ? value.filter((item): item is string => typeof item === 'string') : []
    return service.setQueue(norm(steering), norm(followUp))
  })

  ipcMain.handle('permissions:set', (_event, mode: unknown) => {
    if (mode !== 'confirm' && mode !== 'full') throw new Error('invalid permission mode')
    service.setPermissionMode(mode)
  })

  // ---- 设置页：技能 / 命令详情 ----
  ipcMain.handle('resources:details', () =>
    service.listResourceDetails().catch((err: unknown) => {
      throw new Error(err instanceof Error ? err.message : String(err))
    })
  )

  // 在系统文件管理器中打开（技能/命令来源文件所在目录）
  ipcMain.handle('shell:openPath', (_event, target: unknown) => {
    if (typeof target !== 'string' || !target.trim()) throw new Error('invalid path')
    return shell.openPath(target)
  })

  // 打开资源目录：不存在时先创建（skills/prompts 目录通常尚未建立）
  ipcMain.handle('shell:openDirectory', (_event, target: unknown) => {
    if (typeof target !== 'string' || !target.trim()) throw new Error('invalid path')
    try {
      mkdirSync(target, { recursive: true })
    } catch {
      // 已存在等情况忽略，交给 openPath 反馈
    }
    return shell.openPath(target).then((result) => {
      // shell.openPath 失败时以字符串返回错误信息，成功返回空串
      if (result) throw new Error(result)
    })
  })

  ipcMain.handle('files:list', () =>
    service.listWorkspaceFiles().catch((err: unknown) => {
      throw new Error(err instanceof Error ? err.message : String(err))
    })
  )

  ipcMain.handle('attachments:pick', async (event) => {
    const win = BrowserWindow.fromWebContents(event.sender) ?? BrowserWindow.getAllWindows()[0]
    return service.pickAttachments(win ?? undefined)
  })

  // ---- 内置受管浏览器 ----
  // 控制器构造会调 session.fromPartition，必须等 app ready；
  // IPC 处理器在渲染层就绪后才可能被调用，惰性取用是安全的
  void app.whenReady().then(() => {
    getBrowserController().onState((state) => {
      for (const win of BrowserWindow.getAllWindows()) {
        if (!win.isDestroyed()) win.webContents.send('browser:event', { type: 'state', state })
      }
    })
  })

  const browserPass = <T,>(fn: () => T): T => {
    try {
      return fn()
    } catch (err) {
      throw new Error(err instanceof Error ? err.message : String(err))
    }
  }

  ipcMain.handle('browser:getState', () => browserPass(() => getBrowserController().getState()))
  ipcMain.handle('browser:navigate', (_e, input: unknown) =>
    browserPass(() => getBrowserController().navigateDisplay(typeof input === 'string' ? input : ''))
  )
  ipcMain.handle('browser:goBack', () => browserPass(() => getBrowserController().goBack()))
  ipcMain.handle('browser:goForward', () => browserPass(() => getBrowserController().goForward()))
  ipcMain.handle('browser:reload', () => browserPass(() => getBrowserController().reload()))
  ipcMain.handle('browser:newTab', (_e, url: unknown) =>
    browserPass(() => getBrowserController().newTab(typeof url === 'string' && url ? url : undefined, false))
  )
  ipcMain.handle('browser:selectTab', (_e, tabId: unknown) =>
    browserPass(() => getBrowserController().selectTab(typeof tabId === 'string' ? tabId : ''))
  )
  ipcMain.handle('browser:closeTab', (_e, tabId: unknown) =>
    browserPass(() => getBrowserController().closeTab(typeof tabId === 'string' ? tabId : undefined))
  )
  ipcMain.handle('browser:close', () => browserPass(() => getBrowserController().close()))
  // 渲染层冻结帧：菜单弹出需暂时隐藏原生视图前，截当前页面画面做占位
  ipcMain.handle('browser:captureTab', (_e, tabId: unknown) =>
    browserPass(() => getBrowserController().captureTabDataUrl(typeof tabId === 'string' && tabId ? tabId : undefined))
  )
  ipcMain.handle('browser:acknowledgeRisk', () => browserPass(() => getBrowserController().acknowledgeRisk()))
  // 渲染层 BrowserSlot 量出的占位矩形（原生 WebContentsView 对齐到 DOM 布局）
  ipcMain.handle('browser:layout', (_e, layout: unknown) => {
    const l = layout as { tabId?: unknown; visible?: unknown; revision?: unknown; bounds?: unknown }
    const { tabId, visible, revision } = l
    const bounds = l.bounds as { x?: unknown; y?: unknown; width?: unknown; height?: unknown } | undefined
    const x = bounds?.x
    const y = bounds?.y
    const width = bounds?.width
    const height = bounds?.height
    if (
      typeof tabId !== 'string' ||
      typeof visible !== 'boolean' ||
      typeof revision !== 'number' ||
      typeof x !== 'number' ||
      typeof y !== 'number' ||
      typeof width !== 'number' ||
      typeof height !== 'number'
    ) {
      throw new Error('invalid browser layout')
    }
    return browserPass(() => getBrowserController().applyLayout({ tabId, visible, revision, bounds: { x, y, width, height } }))
  })

  // ---- 本地浏览器配置导入 ----
  ipcMain.handle('browser:detectSources', () => browserPass(() => detectBrowserSources()))
  ipcMain.handle('browser:importProfile', (_e, sourceId: unknown, profileDir: unknown) => {
    if (typeof sourceId !== 'string' || typeof profileDir !== 'string') throw new Error('invalid import source')
    return browserPass(() => importBrowserProfile(sourceId, profileDir))
  })

  // ---- 插件系统 ----
  void app.whenReady().then(() => {
    // 插件启停/重载会重建 agent 运行时（extensionFactories 重跑）
    getPluginHost().setRuntimeRebuilder(() => service.rebuildRuntime())
  })

  ipcMain.handle('plugins:list', () => getPluginHost().list())
  ipcMain.handle('plugins:uiState', () => getPluginHost().uiState())
  ipcMain.handle('plugins:setEnabled', (_e, id: unknown, enabled: unknown) => {
    if (typeof id !== 'string' || typeof enabled !== 'boolean') throw new Error('invalid plugin toggle')
    return getPluginHost().setEnabled(id, enabled)
  })
  ipcMain.handle('plugins:reload', () => getPluginHost().reload())
  ipcMain.handle('plugins:openView', (_e, view: unknown, params: unknown) => {
    if (typeof view !== 'string' || !view) throw new Error('invalid view')
    getPluginHost().openView(view, params)
  })
  // 插件页（iframe）经渲染层桥转发来的状态胶囊写请求；pluginId 键控防越权
  ipcMain.handle('plugins:setStatus', (_e, pluginId: unknown, itemId: unknown, item: unknown) => {
    if (typeof pluginId !== 'string' || typeof itemId !== 'string') throw new Error('invalid status item')
    getPluginHost().setStatusItem(pluginId, itemId, (item ?? null) as never)
    return Promise.resolve()
  })

  return service
}