@_daniel_jiang/ai-host 0.1.0

package/src/index.ts

@@ -0,0 +1,410 @@
+import { IframeBridge } from './iframe'
+import {
+  buildActionLaunchPayload,
+  buildActionWidgetRequest,
+  clearHostPageStore,
+  dismissCurrentPageHint,
+  dispatchRegisteredPageRefresh,
+  getCurrentPageId,
+  getCurrentPagePrimaryAction,
+  getRegisteredPages,
+  isDirectHostAction,
+  registerHostPage,
+  serializePageActions,
+  setCurrentHostPage,
+  setHintUiListener,
+  replaceHostContext as applyReplaceHostContextStore,
+  getHostContext as readHostContextStore,
+  unregisterHostPage,
+} from './store'
+import type {
+  BootstrapResult,
+  HostInitOptions,
+  HostPageContext,
+  HostPrefilledContext,
+  OpenWorkflowOptions,
+  RegisterPageOptions,
+} from './types'
+import type { ChildToParentMessage } from './messages'
+
+export type {
+  BootstrapResult,
+  HostActionKind,
+  HostAnchorRef,
+  HostHintPlacement,
+  HostInitOptions,
+  HostPageAction,
+  HostPageContext,
+  HostPrefilledContext,
+  HostPrefilledPatients,
+  HostRefreshPayload,
+  OpenWorkflowOptions,
+  RegisterPageOptions,
+} from './types'
+export { AGENT_HOST_MSG, isAgentHostMessage } from './messages'
+export type { WidgetPayload, SerializedPageAction } from './messages'
+
+let _init: HostInitOptions = {}
+let _bridge: IframeBridge | null = null
+let _cachedToken = ''
+let _cachedRefreshToken = ''
+let _initPromise: Promise<void> | null = null
+let _hubRunningCount = 0
+
+function resolveAgentOrigin(): string {
+  const origin = (_init.agentOrigin || '').replace(/\/$/, '')
+  if (!origin) {
+    throw new Error(
+      'Agent 前端地址未配置：请在 init({ agentOrigin }) 传入，或由宿主 BFF bootstrap 返回 agent_web_origin',
+    )
+  }
+  return origin
+}
+
+function applyBootstrap(boot: BootstrapResult) {
+  if (boot.agent_web_origin) {
+    _init.agentOrigin = boot.agent_web_origin.replace(/\/$/, '')
+  }
+  if (boot.agent_token) {
+    _cachedToken = boot.agent_token
+  }
+  if (boot.refresh_token) {
+    _cachedRefreshToken = boot.refresh_token
+  }
+  _init.onBootstrap?.(boot)
+}
+
+async function ensureToken(): Promise<string> {
+  if (_init.getToken) {
+    _cachedToken = await _init.getToken()
+    return _cachedToken
+  }
+  if (_cachedToken) return _cachedToken
+  if (_init.bootstrap) {
+    const boot = await _init.bootstrap()
+    applyBootstrap(boot)
+    if (_cachedToken) return _cachedToken
+  }
+  throw new Error('未连接 Agent：请配置 bootstrap 或 getToken')
+}
+
+function refreshLauncherHint() {
+  if (!_init.agentOrigin) return
+  try {
+    ensureBridge().updateLauncherHint(getCurrentPagePrimaryAction())
+  } catch {
+    /* agentOrigin 尚未就绪 */
+  }
+}
+
+function ensureBridge(): IframeBridge {
+  const origin = resolveAgentOrigin()
+  if (!_bridge) {
+    setHintUiListener(refreshLauncherHint)
+    _bridge = new IframeBridge(origin, handleChildMessage, {
+      onHintUse: (pageId, actionId) => {
+        void launchPageAction(pageId, actionId)
+      },
+      onHintDismiss: () => {
+        dismissCurrentPageHint()
+        refreshLauncherHint()
+      },
+    })
+    if (_init.showHub !== false) {
+      _bridge.mountHub()
+    }
+  }
+  return _bridge
+}
+
+function syncHubIframeState() {
+  if (!_bridge || !_init.agentOrigin) return
+  for (const { pageId, actions } of getRegisteredPages()) {
+    _bridge.sendToHub({
+      channel: 'agent-host',
+      type: 'register-page',
+      pageId,
+      actions: serializePageActions(actions),
+    })
+  }
+  _bridge.sendToHub({
+    channel: 'agent-host',
+    type: 'set-page',
+    pageId: getCurrentPageId(),
+  })
+  refreshLauncherHint()
+}
+
+function handleChildMessage(msg: ChildToParentMessage) {
+  if (msg.type === 'ready' && msg.surface === 'hub') {
+    void ensureToken()
+      .then((token) => {
+        _bridge?.sendToHub({
+          channel: 'agent-host',
+          type: 'init',
+          token,
+          refreshToken: _cachedRefreshToken || undefined,
+        })
+        syncHubIframeState()
+      })
+      .catch(() => {})
+  }
+  if (msg.type === 'started') {
+    _bridge?.closeWidget()
+    _bridge?.openHubPanel()
+    _bridge?.sendToHub({
+      channel: 'agent-host',
+      type: 'task-started',
+      taskId: msg.taskId,
+      sessionId: msg.sessionId,
+      workflow: msg.workflow,
+      label: msg.label || msg.workflow,
+      formValues: msg.formValues,
+      prefilled: msg.prefilled,
+      hostContext: msg.hostContext,
+      hubContinue: msg.hubContinue,
+    })
+  }
+  if (msg.type === 'complete') {
+    // 成功回填；失败/取消也刷新，便于宿主关掉 OCR loading 等 UI
+    if (msg.status === 'success' || msg.status === 'failed' || msg.status === 'cancelled') {
+      const refreshKey = msg.hostContext && (msg.hostContext as { refresh?: unknown }).refresh
+      dispatchRegisteredPageRefresh(refreshKey, {
+        taskId: msg.taskId,
+        result: msg.result,
+        hostContext: msg.hostContext,
+        apply: msg.apply,
+      })
+    }
+    _init.onTaskComplete?.({
+      taskId: msg.taskId,
+      status: msg.status,
+      hostContext: msg.hostContext,
+      result: msg.result,
+      apply: msg.apply,
+    })
+    _bridge?.closeWidget()
+  }
+  if (msg.type === 'close') {
+    _bridge?.closeWidget()
+  }
+  if (msg.type === 'open-widget') {
+    void openWidgetIframe(msg.payload)
+  }
+  if (msg.type === 'open-action') {
+    void launchPageAction(msg.pageId, msg.actionId)
+  }
+  if (msg.type === 'hub-stats') {
+    _hubRunningCount = msg.running
+    _bridge?.updateLauncherStats(msg.running, msg.waiting)
+    refreshLauncherHint()
+  }
+  if (msg.type === 'request-host-context') {
+    void respondHostContextRequest()
+  }
+}
+
+async function respondHostContextRequest() {
+  if (!_bridge || !_init.agentOrigin) return
+  let raw: HostPageContext = {}
+  try {
+    if (_init.getHostContext) {
+      raw = (await _init.getHostContext()) || {}
+    } else {
+      raw = readHostContextStore()
+    }
+  } catch {
+    raw = readHostContextStore()
+  }
+  const context = applyReplaceHostContextStore(raw)
+  _bridge.sendToHub({
+    channel: 'agent-host',
+    type: 'set-host-context',
+    context,
+  })
+}
+
+async function launchPageAction(pageId: string, actionId: string) {
+  const packed = buildActionLaunchPayload(pageId, actionId)
+  if (!packed) return
+  if (isDirectHostAction(packed.action)) {
+    await startDirectRun(packed)
+    return
+  }
+  const req = buildActionWidgetRequest(pageId, actionId)
+  if (req) await openWidgetIframe(req)
+}
+
+async function startDirectRun(packed: NonNullable<ReturnType<typeof buildActionLaunchPayload>>) {
+  await ensureInitialized()
+  if (packed.action.workflow && !packed.action.skill && _init.prepareOpenWidget) {
+    await _init.prepareOpenWidget()
+  }
+  const token = await ensureToken()
+  const bridge = ensureBridge()
+  bridge.openHubPanel()
+  bridge.sendToHub({
+    channel: 'agent-host',
+    type: 'init',
+    token,
+    refreshToken: _cachedRefreshToken || undefined,
+  })
+  bridge.sendToHub({
+    channel: 'agent-host',
+    type: 'run-direct',
+    workflow: packed.action.workflow || packed.action.skill || '',
+    title: packed.action.label,
+    host: packed.action.host || _init.host,
+    skill: packed.action.skill,
+    noTools: packed.action.noTools === true,
+    prefilled: packed.prefilled,
+    hostContext: packed.hostContext,
+  })
+}
+
+async function openWidgetIframe(payload: OpenWorkflowOptions & { workflow: string }) {
+  await ensureInitialized()
+  if (_init.prepareOpenWidget) {
+    await _init.prepareOpenWidget()
+  }
+  const bridge = ensureBridge()
+  const token = await ensureToken()
+  bridge.openWidget(
+    {
+      workflow: payload.workflow,
+      title: payload.title || payload.label,
+      host: payload.host || _init.host,
+      prefilled: payload.prefilled,
+      hostContext: payload.hostContext,
+      sessionId: payload.sessionId,
+      preferSandboxFiles: payload.preferSandboxFiles,
+    },
+    token,
+    _cachedRefreshToken || undefined,
+  )
+}
+
+async function ensureInitialized() {
+  if (_initPromise) await _initPromise
+}
+
+async function runInit(opts: HostInitOptions) {
+  _init = { showHub: true, ...opts }
+  if (_init.bootstrap) {
+    const boot = await _init.bootstrap()
+    applyBootstrap(boot)
+  }
+  // 配置了 bootstrap 时必须有 agent_web_origin，禁止回退到宿主 origin（避免 Hub 指到 EDC 自己）
+  if (!_init.agentOrigin) {
+    return
+  }
+  if (_init.showHub !== false && typeof document !== 'undefined') {
+    try {
+      ensureBridge()
+      const token = await ensureToken()
+      _bridge?.sendToHub({
+        channel: 'agent-host',
+        type: 'init',
+        token,
+        refreshToken: _cachedRefreshToken || undefined,
+      })
+    } catch {
+      /* 未登录或 bootstrap 失败：不挂 Hub */
+    }
+  }
+}
+
+export const AgentHost = {
+  async init(opts: HostInitOptions): Promise<void> {
+    if (_initPromise) await _initPromise
+    _initPromise = runInit(opts).finally(() => {
+      _initPromise = null
+    })
+    await _initPromise
+  },
+
+  async getToken(): Promise<string> {
+    await ensureInitialized()
+    return ensureToken()
+  },
+
+  async registerPage(
+    pageId: string,
+    actions: Parameters<typeof registerHostPage>[1],
+    options: RegisterPageOptions = {},
+  ) {
+    await ensureInitialized()
+    registerHostPage(pageId, actions, options)
+    if (_init.agentOrigin) {
+      ensureBridge().sendToHub({
+        channel: 'agent-host',
+        type: 'register-page',
+        pageId,
+        actions: serializePageActions(actions),
+      })
+    }
+    refreshLauncherHint()
+  },
+
+  /** 最近一次按需回写的宿主上下文（系统 / 名称 / 描述 / 菜单） */
+  getHostContext() {
+    return readHostContextStore()
+  },
+
+  async unregisterPage(pageId: string) {
+    await ensureInitialized()
+    unregisterHostPage(pageId)
+    if (_init.agentOrigin) {
+      ensureBridge().sendToHub({ channel: 'agent-host', type: 'unregister-page', pageId })
+      ensureBridge().sendToHub({ channel: 'agent-host', type: 'set-page', pageId: null })
+    }
+    refreshLauncherHint()
+  },
+
+  async setCurrentPage(pageId: string | null) {
+    await ensureInitialized()
+    setCurrentHostPage(pageId)
+    if (_init.agentOrigin) {
+      ensureBridge().sendToHub({ channel: 'agent-host', type: 'set-page', pageId })
+    }
+    refreshLauncherHint()
+  },
+
+  async openAction(pageId: string, actionId: string) {
+    const packed = buildActionLaunchPayload(pageId, actionId)
+    if (!packed) throw new Error('当前操作不可用')
+    await launchPageAction(pageId, actionId)
+  },
+
+  async openWorkflow(workflowName: string, opts: OpenWorkflowOptions = {}) {
+    await openWidgetIframe({
+      workflow: workflowName,
+      ...opts,
+      host: opts.host || _init.host,
+      title: opts.title || opts.label || workflowName,
+    })
+  },
+
+  /**
+   * 兼容保留。场景能否启动由宿主 `action.enabled` 判断，SDK 不识别具体 workflow/skill slug。
+   */
+  canOpen(_workflowName?: string, _opts?: { prefilled?: HostPrefilledContext & Record<string, unknown> }): boolean {
+    return true
+  },
+
+  destroy() {
+    _bridge?.destroy()
+    _bridge = null
+    _cachedToken = ''
+    _cachedRefreshToken = ''
+    _init = {}
+    _initPromise = null
+    _hubRunningCount = 0
+    setHintUiListener(null)
+    clearHostPageStore()
+  },
+}
+
+if (typeof window !== 'undefined') {
+  ;(window as unknown as { AgentHost: typeof AgentHost }).AgentHost = AgentHost
+}

package/src/messages.ts

@@ -0,0 +1,102 @@
+/** postMessage 契约：宿主页面 ↔ Agent 嵌入 iframe */
+
+export const AGENT_HOST_MSG = 'agent-host' as const
+
+export type WidgetPayload = {
+  workflow: string
+  title?: string
+  host?: string
+  prefilled?: Record<string, unknown>
+  hostContext?: Record<string, unknown>
+  /** 复用已有会话沙盒（续聊工作流） */
+  sessionId?: string
+  /** 文件字段优先从会话沙盒选择，而非本地上传 */
+  preferSandboxFiles?: boolean
+}
+
+export type SerializedPageAction = {
+  id: string
+  workflow: string
+  label: string
+  host?: string
+  hint?: string
+  hostContext?: Record<string, unknown>
+  prefilled?: Record<string, unknown>
+  /** registerPage 时在宿主侧求值，供 iframe hint 展示/点击判断 */
+  enabled?: boolean
+  /** 缺省视为 widget；direct 由 Hub 无弹窗启动 */
+  kind?: 'widget' | 'direct'
+  skill?: string
+  noTools?: boolean
+}
+
+export type ParentToChildMessage =
+  | { channel: typeof AGENT_HOST_MSG; type: 'init'; token: string; refreshToken?: string }
+  | { channel: typeof AGENT_HOST_MSG; type: 'register-page'; pageId: string; actions: SerializedPageAction[] }
+  | { channel: typeof AGENT_HOST_MSG; type: 'unregister-page'; pageId: string }
+  | { channel: typeof AGENT_HOST_MSG; type: 'set-page'; pageId: string | null }
+  /** 宿主对 request-host-context 的回写（整份上下文） */
+  | { channel: typeof AGENT_HOST_MSG; type: 'set-host-context'; context: Record<string, unknown> }
+  | { channel: typeof AGENT_HOST_MSG; type: 'open-widget'; payload: WidgetPayload }
+  | { channel: typeof AGENT_HOST_MSG; type: 'open-hub-panel' }
+  | { channel: typeof AGENT_HOST_MSG; type: 'close-hub-panel' }
+  | {
+      channel: typeof AGENT_HOST_MSG
+      type: 'run-direct'
+      workflow: string
+      title?: string
+      host?: string
+      skill?: string
+      noTools?: boolean
+      prefilled?: Record<string, unknown>
+      hostContext?: Record<string, unknown>
+    }
+  | {
+      channel: typeof AGENT_HOST_MSG
+      type: 'task-started'
+      taskId: string
+      sessionId?: string
+      workflow: string
+      label?: string
+      formValues?: Record<string, unknown>
+      prefilled?: Record<string, unknown>
+      hostContext?: Record<string, unknown>
+      hubContinue?: boolean
+    }
+
+export type ChildToParentMessage =
+  | { channel: typeof AGENT_HOST_MSG; type: 'ready'; surface: 'hub' | 'widget' }
+  | { channel: typeof AGENT_HOST_MSG; type: 'hub-panel-close' }
+  | { channel: typeof AGENT_HOST_MSG; type: 'hub-panel-width'; expanded: boolean }
+  | { channel: typeof AGENT_HOST_MSG; type: 'hub-stats'; running: number; waiting: number; done: number }
+  | { channel: typeof AGENT_HOST_MSG; type: 'started'; taskId: string; sessionId?: string; workflow: string; label?: string; formValues?: Record<string, unknown>; prefilled?: Record<string, unknown>; hostContext?: Record<string, unknown>; hubContinue?: boolean }
+  | {
+      channel: typeof AGENT_HOST_MSG
+      type: 'complete'
+      taskId: string
+      status: string
+      hostContext?: Record<string, unknown>
+      result?: string
+      apply?: Record<string, unknown>
+    }
+  | { channel: typeof AGENT_HOST_MSG; type: 'close' }
+  | { channel: typeof AGENT_HOST_MSG; type: 'open-widget'; payload: WidgetPayload }
+  | { channel: typeof AGENT_HOST_MSG; type: 'open-action'; pageId: string; actionId: string }
+  | { channel: typeof AGENT_HOST_MSG; type: 'widget-resize'; height: number }
+  /** Hub 按需向宿主索取当前上下文（配合 init.getHostContext） */
+  | { channel: typeof AGENT_HOST_MSG; type: 'request-host-context' }
+
+export function isAgentHostMessage(data: unknown): data is ParentToChildMessage | ChildToParentMessage {
+  return (
+    typeof data === 'object' &&
+    data !== null &&
+    (data as { channel?: string }).channel === AGENT_HOST_MSG &&
+    typeof (data as { type?: string }).type === 'string'
+  )
+}
+
+/** postMessage 仅接受 plain object；剥离 Vue reactive / 函数等不可克隆值 */
+export function cloneForPostMessage<T>(value: T): T {
+  if (value === undefined || value === null) return value
+  return JSON.parse(JSON.stringify(value)) as T
+}