skybridge 0.0.0-dev.f76ccdc → 0.0.0-dev.f792261

package/dist/web/bridges/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../src/web/bridges/types.ts"],"names":[],"mappings":"","sourcesContent":["import type { CallToolResult } from \"@modelcontextprotocol/sdk/types.js\";\nimport type { useSyncExternalStore } from \"react\";\nimport type { ViewHostType } from \"../../server/index.js\";\n\nexport type SkybridgeProperties = {\n  hostType: ViewHostType;\n  serverUrl: string;\n};\n\ndeclare global {\n  interface Window {\n    skybridge: SkybridgeProperties;\n  }\n}\n\nexport type CallToolArgs = Record<string, unknown> | null;\n\nexport type CallToolResponse = {\n  content: CallToolResult[\"content\"];\n  structuredContent: NonNullable<CallToolResult[\"structuredContent\"]>;\n  isError: NonNullable<CallToolResult[\"isError\"]>;\n  meta?: CallToolResult[\"_meta\"];\n};\n\nexport type DisplayMode = \"pip\" | \"inline\" | \"fullscreen\" | \"modal\";\nexport type RequestDisplayMode = Exclude<DisplayMode, \"modal\">;\n\nexport type Theme = \"light\" | \"dark\";\n\nexport type DeviceType = \"mobile\" | \"tablet\" | \"desktop\" | \"unknown\";\n\nexport type SafeAreaInsets = {\n  top: number;\n  right: number;\n  bottom: number;\n  left: number;\n};\n\nexport type SafeArea = {\n  insets: SafeAreaInsets;\n};\n\nexport type UserAgent = {\n  device: {\n    type: DeviceType;\n  };\n  capabilities: {\n    hover: boolean;\n    touch: boolean;\n  };\n};\n\nexport interface HostContext {\n  theme: Theme;\n  locale: string;\n  displayMode: DisplayMode;\n  safeArea: SafeArea;\n  maxHeight: number | undefined;\n  userAgent: UserAgent;\n  toolInput: Record<string, unknown> | null;\n  toolOutput: Record<string, unknown> | null;\n  toolResponseMetadata: Record<string, unknown> | null;\n  display: {\n    mode: DisplayMode;\n    params?: Record<string, unknown>;\n  };\n  viewState: Record<string, unknown> | null;\n}\n\nexport type Subscribe = Parameters<typeof useSyncExternalStore>[0];\n\nexport interface Bridge<Context> {\n  subscribe(key: keyof Context): Subscribe;\n  subscribe(keys: readonly (keyof Context)[]): Subscribe;\n  getSnapshot<K extends keyof Context>(key: K): Context[K] | undefined;\n}\n\nexport type HostContextStore<K extends keyof HostContext> = {\n  subscribe: Subscribe;\n  getSnapshot: () => HostContext[K];\n};\n\nexport type ViewState = Record<string, unknown>;\n\nexport type SetViewStateAction =\n  | ViewState\n  | ((prevState: ViewState | null) => ViewState);\n\nexport type FileMetadata = {\n  fileId: string;\n  fileName?: string;\n  mimeType?: string;\n};\n\nexport type UploadFileOptions = { library?: boolean };\n\nexport type RequestModalOptions = {\n  title?: string;\n  params?: Record<string, unknown>;\n  template?: string;\n  anchor?: { top?: number; left?: number; width?: number; height?: number };\n};\n\nexport type OpenExternalOptions = {\n  redirectUrl?: false;\n};\n\nexport type SendFollowUpMessageOptions = { scrollToBottom?: boolean };\n\nexport interface Adaptor {\n  getHostContextStore<K extends keyof HostContext>(key: K): HostContextStore<K>;\n  callTool<\n    ToolArgs extends CallToolArgs = null,\n    ToolResponse extends CallToolResponse = CallToolResponse,\n  >(name: string, args: ToolArgs): Promise<ToolResponse>;\n  requestDisplayMode(mode: RequestDisplayMode): Promise<{\n    mode: RequestDisplayMode;\n  }>;\n  sendFollowUpMessage(\n    prompt: string,\n    options?: SendFollowUpMessageOptions,\n  ): Promise<void>;\n  openExternal(href: string, options?: OpenExternalOptions): void;\n  setViewState(stateOrUpdater: SetViewStateAction): Promise<void>;\n  uploadFile(file: File, options?: UploadFileOptions): Promise<FileMetadata>;\n  getFileDownloadUrl(file: FileMetadata): Promise<{ downloadUrl: string }>;\n  selectFiles(): Promise<FileMetadata[]>;\n  openModal(options: RequestModalOptions): void;\n  setOpenInAppUrl(href: string): Promise<void>;\n}\n"]}

package/dist/web/bridges/use-host-context.d.ts

@@ -0,0 +1,2 @@
+import type { HostContext } from "./types.js";
+export declare const useHostContext: <K extends keyof HostContext>(key: K) => HostContext[K];

package/dist/web/bridges/use-host-context.js

@@ -0,0 +1,8 @@
+import { useSyncExternalStore } from "react";
+import { getAdaptor } from "./get-adaptor.js";
+export const useHostContext = (key) => {
+    const adaptor = getAdaptor();
+    const store = adaptor.getHostContextStore(key);
+    return useSyncExternalStore(store.subscribe, store.getSnapshot);
+};
+//# sourceMappingURL=use-host-context.js.map

package/dist/web/bridges/use-host-context.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"use-host-context.js","sourceRoot":"","sources":["../../../src/web/bridges/use-host-context.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,oBAAoB,EAAE,MAAM,OAAO,CAAC;AAC7C,OAAO,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAG9C,MAAM,CAAC,MAAM,cAAc,GAAG,CAC5B,GAAM,EACU,EAAE;IAClB,MAAM,OAAO,GAAG,UAAU,EAAE,CAAC;IAC7B,MAAM,KAAK,GAAG,OAAO,CAAC,mBAAmB,CAAC,GAAG,CAAC,CAAC;IAE/C,OAAO,oBAAoB,CAAC,KAAK,CAAC,SAAS,EAAE,KAAK,CAAC,WAAW,CAAC,CAAC;AAClE,CAAC,CAAC","sourcesContent":["import { useSyncExternalStore } from \"react\";\nimport { getAdaptor } from \"./get-adaptor.js\";\nimport type { HostContext } from \"./types.js\";\n\nexport const useHostContext = <K extends keyof HostContext>(\n  key: K,\n): HostContext[K] => {\n  const adaptor = getAdaptor();\n  const store = adaptor.getHostContextStore(key);\n\n  return useSyncExternalStore(store.subscribe, store.getSnapshot);\n};\n"]}

package/dist/web/components/modal-provider.d.ts

@@ -0,0 +1,4 @@
+import { type ReactNode } from "react";
+export declare function ModalProvider({ children }: {
+    children: ReactNode;
+}): import("react/jsx-runtime").JSX.Element;

package/dist/web/components/modal-provider.js

@@ -0,0 +1,45 @@
+import { jsx as _jsx, Fragment as _Fragment, jsxs as _jsxs } from "react/jsx-runtime";
+import { useEffect, useSyncExternalStore } from "react";
+import { McpAppAdaptor } from "../bridges/index.js";
+const modalStyles = `
+.sb-modal-backdrop {
+  position: fixed;
+  inset: 0;
+  background: rgba(0, 0, 0, 0.5);
+  z-index: 9998;
+}
+.sb-modal-container {
+  border-radius: 12px;
+  position: fixed;
+  inset: 0;
+  margin: auto;
+  width: fit-content;
+  height: fit-content;
+  background: white;
+  z-index: 9999;
+}
+`;
+export function ModalProvider({ children }) {
+    const adaptor = McpAppAdaptor.getInstance();
+    const { mode } = useSyncExternalStore(adaptor.getHostContextStore("display").subscribe, adaptor.getHostContextStore("display").getSnapshot);
+    const isOpen = mode === "modal";
+    const handleBackdropClick = (e) => {
+        if (e.target === e.currentTarget) {
+            adaptor.closeModal();
+        }
+    };
+    useEffect(() => {
+        if (!isOpen) {
+            return;
+        }
+        const handler = (e) => {
+            if (e.key === "Escape") {
+                adaptor.closeModal();
+            }
+        };
+        document.addEventListener("keydown", handler);
+        return () => document.removeEventListener("keydown", handler);
+    }, [isOpen, adaptor]);
+    return (_jsxs(_Fragment, { children: [_jsx("style", { children: modalStyles }), isOpen && (_jsx("div", { role: "dialog", className: "sb-modal-backdrop", onClick: handleBackdropClick })), _jsx("div", { className: isOpen ? "sb-modal-container" : undefined, children: children })] }));
+}
+//# sourceMappingURL=modal-provider.js.map

package/dist/web/components/modal-provider.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"modal-provider.js","sourceRoot":"","sources":["../../../src/web/components/modal-provider.tsx"],"names":[],"mappings":";AAAA,OAAO,EAAkB,SAAS,EAAE,oBAAoB,EAAE,MAAM,OAAO,CAAC;AACxE,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AAEpD,MAAM,WAAW,GAAG;;;;;;;;;;;;;;;;;CAiBnB,CAAC;AAEF,MAAM,UAAU,aAAa,CAAC,EAAE,QAAQ,EAA2B;IACjE,MAAM,OAAO,GAAG,aAAa,CAAC,WAAW,EAAE,CAAC;IAE5C,MAAM,EAAE,IAAI,EAAE,GAAG,oBAAoB,CACnC,OAAO,CAAC,mBAAmB,CAAC,SAAS,CAAC,CAAC,SAAS,EAChD,OAAO,CAAC,mBAAmB,CAAC,SAAS,CAAC,CAAC,WAAW,CACnD,CAAC;IACF,MAAM,MAAM,GAAG,IAAI,KAAK,OAAO,CAAC;IAEhC,MAAM,mBAAmB,GAAG,CAAC,CAAmB,EAAE,EAAE;QAClD,IAAI,CAAC,CAAC,MAAM,KAAK,CAAC,CAAC,aAAa,EAAE,CAAC;YACjC,OAAO,CAAC,UAAU,EAAE,CAAC;QACvB,CAAC;IACH,CAAC,CAAC;IAEF,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,OAAO;QACT,CAAC;QACD,MAAM,OAAO,GAAG,CAAC,CAAgB,EAAE,EAAE;YACnC,IAAI,CAAC,CAAC,GAAG,KAAK,QAAQ,EAAE,CAAC;gBACvB,OAAO,CAAC,UAAU,EAAE,CAAC;YACvB,CAAC;QACH,CAAC,CAAC;QACF,QAAQ,CAAC,gBAAgB,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;QAC9C,OAAO,GAAG,EAAE,CAAC,QAAQ,CAAC,mBAAmB,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;IAChE,CAAC,EAAE,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC;IAEtB,OAAO,CACL,8BACE,0BAAQ,WAAW,GAAS,EAC3B,MAAM,IAAI,CAET,cACE,IAAI,EAAC,QAAQ,EACb,SAAS,EAAC,mBAAmB,EAC7B,OAAO,EAAE,mBAAmB,GAC5B,CACH,EACD,cAAK,SAAS,EAAE,MAAM,CAAC,CAAC,CAAC,oBAAoB,CAAC,CAAC,CAAC,SAAS,YACtD,QAAQ,GACL,IACL,CACJ,CAAC;AACJ,CAAC","sourcesContent":["import { type ReactNode, useEffect, useSyncExternalStore } from \"react\";\nimport { McpAppAdaptor } from \"../bridges/index.js\";\n\nconst modalStyles = `\n.sb-modal-backdrop {\n  position: fixed;\n  inset: 0;\n  background: rgba(0, 0, 0, 0.5);\n  z-index: 9998;\n}\n.sb-modal-container {\n  border-radius: 12px;\n  position: fixed;\n  inset: 0;\n  margin: auto;\n  width: fit-content;\n  height: fit-content;\n  background: white;\n  z-index: 9999;\n}\n`;\n\nexport function ModalProvider({ children }: { children: ReactNode }) {\n  const adaptor = McpAppAdaptor.getInstance();\n\n  const { mode } = useSyncExternalStore(\n    adaptor.getHostContextStore(\"display\").subscribe,\n    adaptor.getHostContextStore(\"display\").getSnapshot,\n  );\n  const isOpen = mode === \"modal\";\n\n  const handleBackdropClick = (e: React.MouseEvent) => {\n    if (e.target === e.currentTarget) {\n      adaptor.closeModal();\n    }\n  };\n\n  useEffect(() => {\n    if (!isOpen) {\n      return;\n    }\n    const handler = (e: KeyboardEvent) => {\n      if (e.key === \"Escape\") {\n        adaptor.closeModal();\n      }\n    };\n    document.addEventListener(\"keydown\", handler);\n    return () => document.removeEventListener(\"keydown\", handler);\n  }, [isOpen, adaptor]);\n\n  return (\n    <>\n      <style>{modalStyles}</style>\n      {isOpen && (\n        // biome-ignore lint/a11y/useKeyWithClickEvents: backdrop isn't focusable\n        <div\n          role=\"dialog\"\n          className=\"sb-modal-backdrop\"\n          onClick={handleBackdropClick}\n        />\n      )}\n      <div className={isOpen ? \"sb-modal-container\" : undefined}>\n        {children}\n      </div>\n    </>\n  );\n}\n"]}

package/dist/web/create-store.d.ts

@@ -0,0 +1,3 @@
+import { type StateCreator } from "zustand";
+import type { UnknownObject } from "./types.js";
+export declare function createStore<State extends UnknownObject>(storeCreator: StateCreator<State, [], [], State>, defaultState?: State | (() => State)): import("zustand").UseBoundStore<import("zustand").StoreApi<State>>;

package/dist/web/create-store.js

@@ -0,0 +1,38 @@
+import { dequal } from "dequal/lite";
+import { create } from "zustand";
+import { getAdaptor } from "./bridges/index.js";
+import { filterViewContext, getInitialState, injectViewContext, serializeState, } from "./helpers/state.js";
+export function createStore(storeCreator, defaultState) {
+    const initialState = getInitialState(defaultState);
+    const store = create()((...args) => {
+        const baseStore = storeCreator(...args);
+        if (initialState !== null) {
+            return { ...baseStore, ...initialState };
+        }
+        return baseStore;
+    });
+    // Bidirectional sync between the Zustand store and the adaptor's viewState.
+    // Store changes persist to the host; external viewState changes rehydrate the store.
+    store.subscribe((state) => {
+        const serializedState = serializeState(state);
+        if (serializedState !== null && serializedState !== undefined) {
+            const stateToPersist = injectViewContext(serializedState);
+            if (stateToPersist !== null) {
+                getAdaptor().setViewState(stateToPersist);
+            }
+        }
+    });
+    const viewStateStore = getAdaptor().getHostContextStore("viewState");
+    viewStateStore.subscribe(() => {
+        const externalState = viewStateStore.getSnapshot();
+        if (externalState !== null) {
+            const filtered = filterViewContext(externalState);
+            const current = serializeState(store.getState());
+            if (!dequal(filtered, current)) {
+                store.setState(filtered);
+            }
+        }
+    });
+    return store;
+}
+//# sourceMappingURL=create-store.js.map

package/dist/web/create-store.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"create-store.js","sourceRoot":"","sources":["../../src/web/create-store.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AACrC,OAAO,EAAE,MAAM,EAAqB,MAAM,SAAS,CAAC;AACpD,OAAO,EAAE,UAAU,EAAE,MAAM,oBAAoB,CAAC;AAChD,OAAO,EACL,iBAAiB,EACjB,eAAe,EACf,iBAAiB,EACjB,cAAc,GACf,MAAM,oBAAoB,CAAC;AAG5B,MAAM,UAAU,WAAW,CACzB,YAAgD,EAChD,YAAoC;IAEpC,MAAM,YAAY,GAAG,eAAe,CAAC,YAAY,CAAC,CAAC;IAEnD,MAAM,KAAK,GAAG,MAAM,EAAS,CAC3B,CAAC,GAAG,IAAoD,EAAE,EAAE;QAC1D,MAAM,SAAS,GAAG,YAAY,CAAC,GAAG,IAAI,CAAC,CAAC;QAExC,IAAI,YAAY,KAAK,IAAI,EAAE,CAAC;YAC1B,OAAO,EAAE,GAAG,SAAS,EAAE,GAAG,YAAY,EAAE,CAAC;QAC3C,CAAC;QAED,OAAO,SAAS,CAAC;IACnB,CAAC,CACF,CAAC;IAEF,4EAA4E;IAC5E,qFAAqF;IACrF,KAAK,CAAC,SAAS,CAAC,CAAC,KAAY,EAAE,EAAE;QAC/B,MAAM,eAAe,GAAG,cAAc,CAAC,KAAK,CAAC,CAAC;QAC9C,IAAI,eAAe,KAAK,IAAI,IAAI,eAAe,KAAK,SAAS,EAAE,CAAC;YAC9D,MAAM,cAAc,GAAG,iBAAiB,CAAC,eAAwB,CAAC,CAAC;YACnE,IAAI,cAAc,KAAK,IAAI,EAAE,CAAC;gBAC5B,UAAU,EAAE,CAAC,YAAY,CAAC,cAAc,CAAC,CAAC;YAC5C,CAAC;QACH,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,MAAM,cAAc,GAAG,UAAU,EAAE,CAAC,mBAAmB,CAAC,WAAW,CAAC,CAAC;IACrE,cAAc,CAAC,SAAS,CAAC,GAAG,EAAE;QAC5B,MAAM,aAAa,GAAG,cAAc,CAAC,WAAW,EAAE,CAAC;QACnD,IAAI,aAAa,KAAK,IAAI,EAAE,CAAC;YAC3B,MAAM,QAAQ,GAAG,iBAAiB,CAAC,aAAa,CAAU,CAAC;YAC3D,MAAM,OAAO,GAAG,cAAc,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAU,CAAC;YAC1D,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,OAAO,CAAC,EAAE,CAAC;gBAC/B,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;YAC3B,CAAC;QACH,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,OAAO,KAAK,CAAC;AACf,CAAC","sourcesContent":["import { dequal } from \"dequal/lite\";\nimport { create, type StateCreator } from \"zustand\";\nimport { getAdaptor } from \"./bridges/index.js\";\nimport {\n  filterViewContext,\n  getInitialState,\n  injectViewContext,\n  serializeState,\n} from \"./helpers/state.js\";\nimport type { UnknownObject } from \"./types.js\";\n\nexport function createStore<State extends UnknownObject>(\n  storeCreator: StateCreator<State, [], [], State>,\n  defaultState?: State | (() => State),\n) {\n  const initialState = getInitialState(defaultState);\n\n  const store = create<State>()(\n    (...args: Parameters<StateCreator<State, [], [], State>>) => {\n      const baseStore = storeCreator(...args);\n\n      if (initialState !== null) {\n        return { ...baseStore, ...initialState };\n      }\n\n      return baseStore;\n    },\n  );\n\n  // Bidirectional sync between the Zustand store and the adaptor's viewState.\n  // Store changes persist to the host; external viewState changes rehydrate the store.\n  store.subscribe((state: State) => {\n    const serializedState = serializeState(state);\n    if (serializedState !== null && serializedState !== undefined) {\n      const stateToPersist = injectViewContext(serializedState as State);\n      if (stateToPersist !== null) {\n        getAdaptor().setViewState(stateToPersist);\n      }\n    }\n  });\n\n  const viewStateStore = getAdaptor().getHostContextStore(\"viewState\");\n  viewStateStore.subscribe(() => {\n    const externalState = viewStateStore.getSnapshot();\n    if (externalState !== null) {\n      const filtered = filterViewContext(externalState) as State;\n      const current = serializeState(store.getState()) as State;\n      if (!dequal(filtered, current)) {\n        store.setState(filtered);\n      }\n    }\n  });\n\n  return store;\n}\n"]}

package/dist/web/create-store.test.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/web/create-store.test.js

@@ -0,0 +1,129 @@
+import { cleanup } from "@testing-library/react";
+import { afterEach, beforeEach, describe, expect, it, vi, } from "vitest";
+import { McpAppAdaptor } from "./bridges/mcp-app/adaptor.js";
+import { McpAppBridge } from "./bridges/mcp-app/bridge.js";
+import { createStore } from "./create-store.js";
+import { VIEW_CONTEXT_KEY } from "./data-llm.js";
+import { getMcpAppHostPostMessageMock, MockResizeObserver, } from "./hooks/test/utils.js";
+describe("createStore", () => {
+    afterEach(() => {
+        vi.unstubAllGlobals();
+        vi.resetAllMocks();
+    });
+    describe("apps-sdk mode", () => {
+        let OpenaiMock;
+        beforeEach(() => {
+            OpenaiMock = {
+                widgetState: null,
+                setWidgetState: vi.fn().mockResolvedValue(undefined),
+            };
+            vi.stubGlobal("openai", OpenaiMock);
+            vi.stubGlobal("skybridge", { hostType: "apps-sdk" });
+        });
+        it("should create a store without default state", () => {
+            const storeCreator = () => ({
+                count: 0,
+            });
+            const store = createStore(storeCreator);
+            expect(store.getState()).toEqual({ count: 0 });
+        });
+        it("should create a store with default state", () => {
+            const storeCreator = () => ({
+                count: 0,
+                name: "initial",
+            });
+            const defaultState = { count: 5, name: "default" };
+            const store = createStore(storeCreator, defaultState);
+            expect(store.getState()).toEqual({ count: 5, name: "default" });
+        });
+        it("should initialize from window.openai.widgetState when available", () => {
+            const storeCreator = () => ({
+                count: 0,
+                name: "initial",
+            });
+            const windowState = { count: 20, name: "window" };
+            OpenaiMock.widgetState = { modelContent: windowState };
+            const store = createStore(storeCreator);
+            expect(store.getState()).toEqual({ count: 20, name: "window" });
+        });
+        it("should persist state changes to window.openai.setWidgetState", async () => {
+            const storeCreator = (set) => ({
+                count: 0,
+                increment: () => set((state) => ({ count: state.count + 1 })),
+            });
+            const store = createStore(storeCreator);
+            store.getState().increment();
+            await vi.waitFor(() => {
+                expect(OpenaiMock.setWidgetState).toHaveBeenCalled();
+            });
+            const callArgs = OpenaiMock.setWidgetState.mock.calls[0]?.[0];
+            expect(callArgs).toEqual({
+                modelContent: { count: 1 },
+                privateContent: {},
+            });
+        });
+        it("should filter view context from initial state", () => {
+            const storeCreator = () => ({
+                count: 0,
+            });
+            const windowState = {
+                count: 5,
+                [VIEW_CONTEXT_KEY]: "context-value",
+            };
+            OpenaiMock.widgetState = { modelContent: windowState };
+            const store = createStore(storeCreator);
+            expect(store.getState()).toEqual({ count: 5 });
+            expect(store.getState()[VIEW_CONTEXT_KEY]).toBeUndefined();
+        });
+    });
+    describe("mcp-app mode", () => {
+        beforeEach(() => {
+            vi.stubGlobal("skybridge", { hostType: "mcp-app" });
+            vi.stubGlobal("ResizeObserver", MockResizeObserver);
+        });
+        afterEach(async () => {
+            cleanup();
+            McpAppBridge.resetInstance();
+            McpAppAdaptor.resetInstance();
+        });
+        it("should initialize with null viewState", () => {
+            const adaptor = McpAppAdaptor.getInstance();
+            const viewState = adaptor.getHostContextStore("viewState").getSnapshot();
+            expect(viewState).toBeNull();
+        });
+        it("should create a store with default state when no persisted state exists", () => {
+            const storeCreator = () => ({
+                count: 0,
+            });
+            const store = createStore(storeCreator);
+            expect(store.getState()).toEqual({ count: 0 });
+        });
+        it("should update in-memory state via setViewState", async () => {
+            const adaptor = McpAppAdaptor.getInstance();
+            vi.spyOn(adaptor, "setViewState").mockResolvedValue(undefined);
+            const storeCreator = (set) => ({
+                count: 0,
+                increment: () => set((state) => ({ count: state.count + 1 })),
+            });
+            const store = createStore(storeCreator);
+            store.getState().increment();
+            await vi.waitFor(() => {
+                expect(adaptor.setViewState).toHaveBeenCalledWith({ count: 1 });
+            });
+        });
+        it("should notify listeners when view state changes", async () => {
+            const postMessageMock = getMcpAppHostPostMessageMock();
+            vi.stubGlobal("parent", { postMessage: postMessageMock });
+            const adaptor = McpAppAdaptor.getInstance();
+            const listener = vi.fn();
+            adaptor.getHostContextStore("viewState").subscribe(listener);
+            await adaptor.setViewState({ count: 42 });
+            expect(postMessageMock).toHaveBeenCalledWith(expect.objectContaining({ method: "ui/update-model-context" }), "*");
+            expect(listener).toHaveBeenCalled();
+            expect(adaptor.getHostContextStore("viewState").getSnapshot()).toEqual({
+                count: 42,
+            });
+        });
+    });
+});
+//# sourceMappingURL=create-store.test.js.map

package/dist/web/create-store.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"create-store.test.js","sourceRoot":"","sources":["../../src/web/create-store.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,wBAAwB,CAAC;AACjD,OAAO,EACL,SAAS,EACT,UAAU,EACV,QAAQ,EACR,MAAM,EACN,EAAE,EAEF,EAAE,GACH,MAAM,QAAQ,CAAC;AAEhB,OAAO,EAAE,aAAa,EAAE,MAAM,8BAA8B,CAAC;AAC7D,OAAO,EAAE,YAAY,EAAE,MAAM,6BAA6B,CAAC;AAC3D,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,EAAE,gBAAgB,EAAE,MAAM,eAAe,CAAC;AACjD,OAAO,EACL,4BAA4B,EAC5B,kBAAkB,GACnB,MAAM,uBAAuB,CAAC;AAE/B,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;IAC3B,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,gBAAgB,EAAE,CAAC;QACtB,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,IAAI,UAGH,CAAC;QAEF,UAAU,CAAC,GAAG,EAAE;YACd,UAAU,GAAG;gBACX,WAAW,EAAE,IAAI;gBACjB,cAAc,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;aACrD,CAAC;YACF,EAAE,CAAC,UAAU,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;YACpC,EAAE,CAAC,UAAU,CAAC,WAAW,EAAE,EAAE,QAAQ,EAAE,UAAU,EAAE,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;YAErD,MAAM,YAAY,GAA+C,GAAG,EAAE,CAAC,CAAC;gBACtE,KAAK,EAAE,CAAC;aACT,CAAC,CAAC;YAEH,MAAM,KAAK,GAAG,WAAW,CAAC,YAAY,CAAC,CAAC;YAExC,MAAM,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC;QACjD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;YAElD,MAAM,YAAY,GAA+C,GAAG,EAAE,CAAC,CAAC;gBACtE,KAAK,EAAE,CAAC;gBACR,IAAI,EAAE,SAAS;aAChB,CAAC,CAAC;YACH,MAAM,YAAY,GAAG,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;YAEnD,MAAM,KAAK,GAAG,WAAW,CAAC,YAAY,EAAE,YAAY,CAAC,CAAC;YAEtD,MAAM,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,CAAC;QAClE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iEAAiE,EAAE,GAAG,EAAE;YAEzE,MAAM,YAAY,GAA+C,GAAG,EAAE,CAAC,CAAC;gBACtE,KAAK,EAAE,CAAC;gBACR,IAAI,EAAE,SAAS;aAChB,CAAC,CAAC;YACH,MAAM,WAAW,GAAG,EAAE,KAAK,EAAE,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;YAClD,UAAU,CAAC,WAAW,GAAG,EAAE,YAAY,EAAE,WAAW,EAAE,CAAC;YAEvD,MAAM,KAAK,GAAG,WAAW,CAAC,YAAY,CAAC,CAAC;YAExC,MAAM,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAC;QAClE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8DAA8D,EAAE,KAAK,IAAI,EAAE;YAE5E,MAAM,YAAY,GAA+C,CAC/D,GAAG,EACH,EAAE,CAAC,CAAC;gBACJ,KAAK,EAAE,CAAC;gBACR,SAAS,EAAE,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,GAAG,CAAC,EAAE,CAAC,CAAC;aAC9D,CAAC,CAAC;YAEH,MAAM,KAAK,GAAG,WAAW,CAAC,YAAY,CAAC,CAAC;YACxC,KAAK,CAAC,QAAQ,EAAE,CAAC,SAAS,EAAE,CAAC;YAE7B,MAAM,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE;gBACpB,MAAM,CAAC,UAAU,CAAC,cAAc,CAAC,CAAC,gBAAgB,EAAE,CAAC;YACvD,CAAC,CAAC,CAAC;YAEH,MAAM,QAAQ,GAAG,UAAU,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;YAC9D,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC;gBACvB,YAAY,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE;gBAC1B,cAAc,EAAE,EAAE;aACnB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;YAEvD,MAAM,YAAY,GAA+C,GAAG,EAAE,CAAC,CAAC;gBACtE,KAAK,EAAE,CAAC;aACT,CAAC,CAAC;YACH,MAAM,WAAW,GAAG;gBAClB,KAAK,EAAE,CAAC;gBACR,CAAC,gBAAgB,CAAC,EAAE,eAAe;aACpC,CAAC;YACF,UAAU,CAAC,WAAW,GAAG,EAAE,YAAY,EAAE,WAAW,EAAE,CAAC;YAEvD,MAAM,KAAK,GAAG,WAAW,CAAC,YAAY,CAAC,CAAC;YAExC,MAAM,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC;YAC/C,MAAM,CACH,KAAK,CAAC,QAAQ,EAA8B,CAAC,gBAAgB,CAAC,CAChE,CAAC,aAAa,EAAE,CAAC;QACpB,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;QAC5B,UAAU,CAAC,GAAG,EAAE;YACd,EAAE,CAAC,UAAU,CAAC,WAAW,EAAE,EAAE,QAAQ,EAAE,SAAS,EAAE,CAAC,CAAC;YACpD,EAAE,CAAC,UAAU,CAAC,gBAAgB,EAAE,kBAAkB,CAAC,CAAC;QACtD,CAAC,CAAC,CAAC;QAEH,SAAS,CAAC,KAAK,IAAI,EAAE;YACnB,OAAO,EAAE,CAAC;YACV,YAAY,CAAC,aAAa,EAAE,CAAC;YAC7B,aAAa,CAAC,aAAa,EAAE,CAAC;QAChC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;YAC/C,MAAM,OAAO,GAAG,aAAa,CAAC,WAAW,EAAE,CAAC;YAC5C,MAAM,SAAS,GAAG,OAAO,CAAC,mBAAmB,CAAC,WAAW,CAAC,CAAC,WAAW,EAAE,CAAC;YAEzE,MAAM,CAAC,SAAS,CAAC,CAAC,QAAQ,EAAE,CAAC;QAC/B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yEAAyE,EAAE,GAAG,EAAE;YAEjF,MAAM,YAAY,GAA+C,GAAG,EAAE,CAAC,CAAC;gBACtE,KAAK,EAAE,CAAC;aACT,CAAC,CAAC;YAEH,MAAM,KAAK,GAAG,WAAW,CAAC,YAAY,CAAC,CAAC;YAExC,MAAM,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC;QACjD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;YAC9D,MAAM,OAAO,GAAG,aAAa,CAAC,WAAW,EAAE,CAAC;YAC5C,EAAE,CAAC,KAAK,CAAC,OAAO,EAAE,cAAc,CAAC,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;YAG/D,MAAM,YAAY,GAA+C,CAC/D,GAAG,EACH,EAAE,CAAC,CAAC;gBACJ,KAAK,EAAE,CAAC;gBACR,SAAS,EAAE,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,GAAG,CAAC,EAAE,CAAC,CAAC;aAC9D,CAAC,CAAC;YAEH,MAAM,KAAK,GAAG,WAAW,CAAC,YAAY,CAAC,CAAC;YACxC,KAAK,CAAC,QAAQ,EAAE,CAAC,SAAS,EAAE,CAAC;YAE7B,MAAM,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE;gBACpB,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC;YAClE,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;YAC/D,MAAM,eAAe,GAAG,4BAA4B,EAAE,CAAC;YACvD,EAAE,CAAC,UAAU,CAAC,QAAQ,EAAE,EAAE,WAAW,EAAE,eAAe,EAAE,CAAC,CAAC;YAE1D,MAAM,OAAO,GAAG,aAAa,CAAC,WAAW,EAAE,CAAC;YAC5C,MAAM,QAAQ,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YAEzB,OAAO,CAAC,mBAAmB,CAAC,WAAW,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;YAC7D,MAAM,OAAO,CAAC,YAAY,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;YAE1C,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAC1C,MAAM,CAAC,gBAAgB,CAAC,EAAE,MAAM,EAAE,yBAAyB,EAAE,CAAC,EAC9D,GAAG,CACJ,CAAC;YACF,MAAM,CAAC,QAAQ,CAAC,CAAC,gBAAgB,EAAE,CAAC;YACpC,MAAM,CAAC,OAAO,CAAC,mBAAmB,CAAC,WAAW,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC,OAAO,CAAC;gBACrE,KAAK,EAAE,EAAE;aACV,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import { cleanup } from \"@testing-library/react\";\nimport {\n  afterEach,\n  beforeEach,\n  describe,\n  expect,\n  it,\n  type Mock,\n  vi,\n} from \"vitest\";\nimport type { StateCreator } from \"zustand\";\nimport { McpAppAdaptor } from \"./bridges/mcp-app/adaptor.js\";\nimport { McpAppBridge } from \"./bridges/mcp-app/bridge.js\";\nimport { createStore } from \"./create-store.js\";\nimport { VIEW_CONTEXT_KEY } from \"./data-llm.js\";\nimport {\n  getMcpAppHostPostMessageMock,\n  MockResizeObserver,\n} from \"./hooks/test/utils.js\";\n\ndescribe(\"createStore\", () => {\n  afterEach(() => {\n    vi.unstubAllGlobals();\n    vi.resetAllMocks();\n  });\n\n  describe(\"apps-sdk mode\", () => {\n    let OpenaiMock: {\n      widgetState: unknown;\n      setWidgetState: Mock;\n    };\n\n    beforeEach(() => {\n      OpenaiMock = {\n        widgetState: null,\n        setWidgetState: vi.fn().mockResolvedValue(undefined),\n      };\n      vi.stubGlobal(\"openai\", OpenaiMock);\n      vi.stubGlobal(\"skybridge\", { hostType: \"apps-sdk\" });\n    });\n\n    it(\"should create a store without default state\", () => {\n      type TestState = { count: number };\n      const storeCreator: StateCreator<TestState, [], [], TestState> = () => ({\n        count: 0,\n      });\n\n      const store = createStore(storeCreator);\n\n      expect(store.getState()).toEqual({ count: 0 });\n    });\n\n    it(\"should create a store with default state\", () => {\n      type TestState = { count: number; name: string };\n      const storeCreator: StateCreator<TestState, [], [], TestState> = () => ({\n        count: 0,\n        name: \"initial\",\n      });\n      const defaultState = { count: 5, name: \"default\" };\n\n      const store = createStore(storeCreator, defaultState);\n\n      expect(store.getState()).toEqual({ count: 5, name: \"default\" });\n    });\n\n    it(\"should initialize from window.openai.widgetState when available\", () => {\n      type TestState = { count: number; name: string };\n      const storeCreator: StateCreator<TestState, [], [], TestState> = () => ({\n        count: 0,\n        name: \"initial\",\n      });\n      const windowState = { count: 20, name: \"window\" };\n      OpenaiMock.widgetState = { modelContent: windowState };\n\n      const store = createStore(storeCreator);\n\n      expect(store.getState()).toEqual({ count: 20, name: \"window\" });\n    });\n\n    it(\"should persist state changes to window.openai.setWidgetState\", async () => {\n      type TestState = { count: number; increment: () => void };\n      const storeCreator: StateCreator<TestState, [], [], TestState> = (\n        set,\n      ) => ({\n        count: 0,\n        increment: () => set((state) => ({ count: state.count + 1 })),\n      });\n\n      const store = createStore(storeCreator);\n      store.getState().increment();\n\n      await vi.waitFor(() => {\n        expect(OpenaiMock.setWidgetState).toHaveBeenCalled();\n      });\n\n      const callArgs = OpenaiMock.setWidgetState.mock.calls[0]?.[0];\n      expect(callArgs).toEqual({\n        modelContent: { count: 1 },\n        privateContent: {},\n      });\n    });\n\n    it(\"should filter view context from initial state\", () => {\n      type TestState = { count: number };\n      const storeCreator: StateCreator<TestState, [], [], TestState> = () => ({\n        count: 0,\n      });\n      const windowState = {\n        count: 5,\n        [VIEW_CONTEXT_KEY]: \"context-value\",\n      };\n      OpenaiMock.widgetState = { modelContent: windowState };\n\n      const store = createStore(storeCreator);\n\n      expect(store.getState()).toEqual({ count: 5 });\n      expect(\n        (store.getState() as Record<string, unknown>)[VIEW_CONTEXT_KEY],\n      ).toBeUndefined();\n    });\n  });\n\n  describe(\"mcp-app mode\", () => {\n    beforeEach(() => {\n      vi.stubGlobal(\"skybridge\", { hostType: \"mcp-app\" });\n      vi.stubGlobal(\"ResizeObserver\", MockResizeObserver);\n    });\n\n    afterEach(async () => {\n      cleanup();\n      McpAppBridge.resetInstance();\n      McpAppAdaptor.resetInstance();\n    });\n\n    it(\"should initialize with null viewState\", () => {\n      const adaptor = McpAppAdaptor.getInstance();\n      const viewState = adaptor.getHostContextStore(\"viewState\").getSnapshot();\n\n      expect(viewState).toBeNull();\n    });\n\n    it(\"should create a store with default state when no persisted state exists\", () => {\n      type TestState = { count: number };\n      const storeCreator: StateCreator<TestState, [], [], TestState> = () => ({\n        count: 0,\n      });\n\n      const store = createStore(storeCreator);\n\n      expect(store.getState()).toEqual({ count: 0 });\n    });\n\n    it(\"should update in-memory state via setViewState\", async () => {\n      const adaptor = McpAppAdaptor.getInstance();\n      vi.spyOn(adaptor, \"setViewState\").mockResolvedValue(undefined);\n\n      type TestState = { count: number; increment: () => void };\n      const storeCreator: StateCreator<TestState, [], [], TestState> = (\n        set,\n      ) => ({\n        count: 0,\n        increment: () => set((state) => ({ count: state.count + 1 })),\n      });\n\n      const store = createStore(storeCreator);\n      store.getState().increment();\n\n      await vi.waitFor(() => {\n        expect(adaptor.setViewState).toHaveBeenCalledWith({ count: 1 });\n      });\n    });\n\n    it(\"should notify listeners when view state changes\", async () => {\n      const postMessageMock = getMcpAppHostPostMessageMock();\n      vi.stubGlobal(\"parent\", { postMessage: postMessageMock });\n\n      const adaptor = McpAppAdaptor.getInstance();\n      const listener = vi.fn();\n\n      adaptor.getHostContextStore(\"viewState\").subscribe(listener);\n      await adaptor.setViewState({ count: 42 });\n\n      expect(postMessageMock).toHaveBeenCalledWith(\n        expect.objectContaining({ method: \"ui/update-model-context\" }),\n        \"*\",\n      );\n      expect(listener).toHaveBeenCalled();\n      expect(adaptor.getHostContextStore(\"viewState\").getSnapshot()).toEqual({\n        count: 42,\n      });\n    });\n  });\n});\n"]}

package/dist/web/data-llm.d.ts

@@ -0,0 +1,14 @@
+import { type ReactNode } from "react";
+export type DataLLMContent = string;
+export interface DataLLMNode {
+    id: string;
+    parentId: string | null;
+    content: string | null;
+}
+export declare const VIEW_CONTEXT_KEY: "__view_context";
+interface DataLLMProps {
+    content: DataLLMContent | null | undefined;
+    children?: ReactNode;
+}
+export declare function DataLLM({ content, children }: DataLLMProps): import("react/jsx-runtime").JSX.Element;
+export {};

package/dist/web/data-llm.js

@@ -0,0 +1,72 @@
+import { jsx as _jsx } from "react/jsx-runtime";
+import { createContext, useContext, useEffect, useId, } from "react";
+import { getAdaptor } from "./bridges/index.js";
+export const VIEW_CONTEXT_KEY = "__view_context";
+const nodes = new Map();
+function setNode(node) {
+    nodes.set(node.id, node);
+    onChange();
+}
+function removeNode(id) {
+    nodes.delete(id);
+    onChange();
+}
+function onChange() {
+    const description = getLLMDescriptionString();
+    const adaptor = getAdaptor();
+    adaptor.setViewState((prevState) => ({
+        ...prevState,
+        [VIEW_CONTEXT_KEY]: description,
+    }));
+}
+const ParentIdContext = createContext(null);
+export function DataLLM({ content, children }) {
+    const parentId = useContext(ParentIdContext);
+    const id = useId();
+    useEffect(() => {
+        if (content) {
+            setNode({
+                id,
+                parentId,
+                content,
+            });
+        }
+        else {
+            removeNode(id);
+        }
+        return () => {
+            removeNode(id);
+        };
+    }, [id, parentId, content]);
+    return (_jsx(ParentIdContext.Provider, { value: id, children: children }));
+}
+function getLLMDescriptionString() {
+    const byParent = new Map();
+    for (const node of Array.from(nodes.values())) {
+        const key = node.parentId ?? null;
+        if (!byParent.has(key)) {
+            byParent.set(key, []);
+        }
+        byParent.get(key)?.push(node);
+    }
+    for (const list of byParent.values()) {
+        list.sort((a, b) => a.id.localeCompare(b.id));
+    }
+    const lines = [];
+    function traverseTree(parentId, depth) {
+        const children = byParent.get(parentId);
+        if (!children) {
+            return;
+        }
+        for (const child of children) {
+            if (child.content?.trim()) {
+                const indent = "  ".repeat(depth);
+                lines.push(`${indent}- ${child.content.trim()}`);
+            }
+            traverseTree(child.id, depth + 1);
+        }
+    }
+    traverseTree(null, 0);
+    return lines.join("\n");
+}
+//# sourceMappingURL=data-llm.js.map

package/dist/web/data-llm.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"data-llm.js","sourceRoot":"","sources":["../../src/web/data-llm.tsx"],"names":[],"mappings":";AAAA,OAAO,EACL,aAAa,EAEb,UAAU,EACV,SAAS,EACT,KAAK,GACN,MAAM,OAAO,CAAC;AACf,OAAO,EAAE,UAAU,EAAE,MAAM,oBAAoB,CAAC;AAUhD,MAAM,CAAC,MAAM,gBAAgB,GAAG,gBAAyB,CAAC;AAE1D,MAAM,KAAK,GAAG,IAAI,GAAG,EAAuB,CAAC;AAE7C,SAAS,OAAO,CAAC,IAAiB;IAChC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC;IACzB,QAAQ,EAAE,CAAC;AACb,CAAC;AAED,SAAS,UAAU,CAAC,EAAU;IAC5B,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;IACjB,QAAQ,EAAE,CAAC;AACb,CAAC;AAED,SAAS,QAAQ;IACf,MAAM,WAAW,GAAG,uBAAuB,EAAE,CAAC;IAC9C,MAAM,OAAO,GAAG,UAAU,EAAE,CAAC;IAC7B,OAAO,CAAC,YAAY,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;QACnC,GAAG,SAAS;QACZ,CAAC,gBAAgB,CAAC,EAAE,WAAW;KAChC,CAAC,CAAC,CAAC;AACN,CAAC;AAED,MAAM,eAAe,GAAG,aAAa,CAAgB,IAAI,CAAC,CAAC;AAO3D,MAAM,UAAU,OAAO,CAAC,EAAE,OAAO,EAAE,QAAQ,EAAgB;IACzD,MAAM,QAAQ,GAAG,UAAU,CAAC,eAAe,CAAC,CAAC;IAC7C,MAAM,EAAE,GAAG,KAAK,EAAE,CAAC;IAEnB,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,OAAO,EAAE,CAAC;YACZ,OAAO,CAAC;gBACN,EAAE;gBACF,QAAQ;gBACR,OAAO;aACR,CAAC,CAAC;QACL,CAAC;aAAM,CAAC;YACN,UAAU,CAAC,EAAE,CAAC,CAAC;QACjB,CAAC;QAED,OAAO,GAAG,EAAE;YACV,UAAU,CAAC,EAAE,CAAC,CAAC;QACjB,CAAC,CAAC;IACJ,CAAC,EAAE,CAAC,EAAE,EAAE,QAAQ,EAAE,OAAO,CAAC,CAAC,CAAC;IAE5B,OAAO,CACL,KAAC,eAAe,CAAC,QAAQ,IAAC,KAAK,EAAE,EAAE,YAAG,QAAQ,GAA4B,CAC3E,CAAC;AACJ,CAAC;AAED,SAAS,uBAAuB;IAC9B,MAAM,QAAQ,GAAG,IAAI,GAAG,EAAgC,CAAC;IACzD,KAAK,MAAM,IAAI,IAAI,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,EAAE,CAAC;QAC9C,MAAM,GAAG,GAAG,IAAI,CAAC,QAAQ,IAAI,IAAI,CAAC;QAClC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;YACvB,QAAQ,CAAC,GAAG,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;QACxB,CAAC;QACD,QAAQ,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC;IAChC,CAAC;IAED,KAAK,MAAM,IAAI,IAAI,QAAQ,CAAC,MAAM,EAAE,EAAE,CAAC;QACrC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IAChD,CAAC;IAED,MAAM,KAAK,GAAa,EAAE,CAAC;IAE3B,SAAS,YAAY,CAAC,QAAuB,EAAE,KAAa;QAC1D,MAAM,QAAQ,GAAG,QAAQ,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACxC,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,OAAO;QACT,CAAC;QAED,KAAK,MAAM,KAAK,IAAI,QAAQ,EAAE,CAAC;YAC7B,IAAI,KAAK,CAAC,OAAO,EAAE,IAAI,EAAE,EAAE,CAAC;gBAC1B,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;gBAClC,KAAK,CAAC,IAAI,CAAC,GAAG,MAAM,KAAK,KAAK,CAAC,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YACnD,CAAC;YACD,YAAY,CAAC,KAAK,CAAC,EAAE,EAAE,KAAK,GAAG,CAAC,CAAC,CAAC;QACpC,CAAC;IACH,CAAC;IAED,YAAY,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;IAEtB,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC1B,CAAC","sourcesContent":["import {\n  createContext,\n  type ReactNode,\n  useContext,\n  useEffect,\n  useId,\n} from \"react\";\nimport { getAdaptor } from \"./bridges/index.js\";\n\nexport type DataLLMContent = string;\n\nexport interface DataLLMNode {\n  id: string;\n  parentId: string | null;\n  content: string | null;\n}\n\nexport const VIEW_CONTEXT_KEY = \"__view_context\" as const;\n\nconst nodes = new Map<string, DataLLMNode>();\n\nfunction setNode(node: DataLLMNode) {\n  nodes.set(node.id, node);\n  onChange();\n}\n\nfunction removeNode(id: string) {\n  nodes.delete(id);\n  onChange();\n}\n\nfunction onChange() {\n  const description = getLLMDescriptionString();\n  const adaptor = getAdaptor();\n  adaptor.setViewState((prevState) => ({\n    ...prevState,\n    [VIEW_CONTEXT_KEY]: description,\n  }));\n}\n\nconst ParentIdContext = createContext<string | null>(null);\n\ninterface DataLLMProps {\n  content: DataLLMContent | null | undefined;\n  children?: ReactNode;\n}\n\nexport function DataLLM({ content, children }: DataLLMProps) {\n  const parentId = useContext(ParentIdContext);\n  const id = useId();\n\n  useEffect(() => {\n    if (content) {\n      setNode({\n        id,\n        parentId,\n        content,\n      });\n    } else {\n      removeNode(id);\n    }\n\n    return () => {\n      removeNode(id);\n    };\n  }, [id, parentId, content]);\n\n  return (\n    <ParentIdContext.Provider value={id}>{children}</ParentIdContext.Provider>\n  );\n}\n\nfunction getLLMDescriptionString(): string {\n  const byParent = new Map<string | null, DataLLMNode[]>();\n  for (const node of Array.from(nodes.values())) {\n    const key = node.parentId ?? null;\n    if (!byParent.has(key)) {\n      byParent.set(key, []);\n    }\n    byParent.get(key)?.push(node);\n  }\n\n  for (const list of byParent.values()) {\n    list.sort((a, b) => a.id.localeCompare(b.id));\n  }\n\n  const lines: string[] = [];\n\n  function traverseTree(parentId: string | null, depth: number) {\n    const children = byParent.get(parentId);\n    if (!children) {\n      return;\n    }\n\n    for (const child of children) {\n      if (child.content?.trim()) {\n        const indent = \"  \".repeat(depth);\n        lines.push(`${indent}- ${child.content.trim()}`);\n      }\n      traverseTree(child.id, depth + 1);\n    }\n  }\n\n  traverseTree(null, 0);\n\n  return lines.join(\"\\n\");\n}\n"]}

package/dist/web/data-llm.test.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/web/data-llm.test.js

@@ -0,0 +1,142 @@
+import { jsx as _jsx, jsxs as _jsxs, Fragment as _Fragment } from "react/jsx-runtime";
+import { cleanup, render } from "@testing-library/react";
+import { afterEach, beforeEach, describe, expect, it, vi, } from "vitest";
+import { McpAppAdaptor, McpAppBridge } from "./bridges/mcp-app/index.js";
+import { DataLLM } from "./data-llm.js";
+import { getMcpAppHostPostMessageMock, MockResizeObserver, } from "./hooks/test/utils.js";
+describe("DataLLM", () => {
+    afterEach(() => {
+        // Clean up React components BEFORE unstubbing globals
+        cleanup();
+        vi.unstubAllGlobals();
+        vi.resetAllMocks();
+    });
+    describe("apps-sdk mode", () => {
+        let OpenaiMock;
+        beforeEach(() => {
+            OpenaiMock = {
+                widgetState: { modelContent: {} },
+                setWidgetState: vi.fn(),
+            };
+            // Use Object.defineProperty to ensure it persists
+            Object.defineProperty(globalThis, "openai", {
+                value: OpenaiMock,
+                writable: true,
+                configurable: true,
+            });
+            // Also set on window for browser-like environment
+            if (typeof window !== "undefined") {
+                Object.defineProperty(window, "openai", {
+                    value: OpenaiMock,
+                    writable: true,
+                    configurable: true,
+                });
+            }
+            vi.stubGlobal("openai", OpenaiMock);
+            vi.stubGlobal("skybridge", { hostType: "apps-sdk" });
+        });
+        afterEach(() => {
+            // Keep the mock available for React cleanup, but reset it
+            if (typeof window !== "undefined" && window.openai) {
+                window.openai.setWidgetState = vi.fn();
+                window.openai.widgetState = { modelContent: {}, privateContent: {} };
+            }
+        });
+        it("should register a node with content and call setViewState", () => {
+            render(_jsx(DataLLM, { content: "Test content", children: _jsx("div", { children: "Child" }) }));
+            expect(OpenaiMock.setWidgetState).toHaveBeenCalled();
+            const callArgs = OpenaiMock.setWidgetState.mock.calls[0]?.[0]?.modelContent;
+            expect(callArgs).toHaveProperty("__view_context");
+            expect(callArgs?.__view_context).toContain("- Test content");
+        });
+        it("should preserve existing viewState when updating context", () => {
+            OpenaiMock.widgetState = {
+                modelContent: { existingKey: "existingValue" },
+            };
+            render(_jsx(DataLLM, { content: "Test content", children: _jsx("div", { children: "Child" }) }));
+            const callArgs = OpenaiMock.setWidgetState.mock.calls[0]?.[0]?.modelContent;
+            expect(callArgs).toHaveProperty("existingKey", "existingValue");
+            expect(callArgs).toHaveProperty("__view_context");
+        });
+        it("should handle deeply nested DataLLM components", () => {
+            render(_jsxs(DataLLM, { content: "Level 1", children: [_jsx(DataLLM, { content: "Level 2A" }), _jsx(DataLLM, { content: "Level 2B", children: _jsx(DataLLM, { content: "Level 3", children: _jsx("div", { children: "Content" }) }) })] }));
+            const callArgs = OpenaiMock.setWidgetState.mock.calls[OpenaiMock.setWidgetState.mock.calls.length - 1]?.[0]?.modelContent;
+            const context = callArgs?.__view_context;
+            expect(context).toContain("- Level 1");
+            expect(context).toContain("  - Level 2A");
+            expect(context).toContain("  - Level 2B");
+            expect(context).toContain("    - Level 3");
+        });
+        it("should update context when content changes", () => {
+            const { rerender } = render(_jsx(DataLLM, { content: "Initial content", children: _jsx("div", { children: "Child" }) }));
+            const initialCalls = OpenaiMock.setWidgetState.mock.calls.length;
+            rerender(_jsx(DataLLM, { content: "Updated content", children: _jsx("div", { children: "Child" }) }));
+            expect(OpenaiMock.setWidgetState.mock.calls.length).toBeGreaterThan(initialCalls);
+            const lastCallArgs = OpenaiMock.setWidgetState.mock.calls[OpenaiMock.setWidgetState.mock.calls.length - 1]?.[0]?.modelContent;
+            expect(lastCallArgs?.__view_context).toContain("- Updated content");
+        });
+        it("should remove node and update context when component unmounts", () => {
+            const { unmount } = render(_jsx(DataLLM, { content: "Content to remove", children: _jsx("div", { children: "Child" }) }));
+            const callsBeforeUnmount = OpenaiMock.setWidgetState.mock.calls.length;
+            unmount();
+            expect(OpenaiMock.setWidgetState.mock.calls.length).toBeGreaterThan(callsBeforeUnmount);
+            const lastCallArgs = OpenaiMock.setWidgetState.mock.calls[OpenaiMock.setWidgetState.mock.calls.length - 1]?.[0]?.modelContent;
+            expect(lastCallArgs?.__view_context).not.toContain("Content to remove");
+        });
+    });
+    describe("mcp-app mode", () => {
+        let postMessageMock;
+        beforeEach(() => {
+            vi.stubGlobal("skybridge", { hostType: "mcp-app" });
+            vi.stubGlobal("ResizeObserver", MockResizeObserver);
+            postMessageMock = getMcpAppHostPostMessageMock();
+            vi.stubGlobal("parent", { postMessage: postMessageMock });
+        });
+        afterEach(() => {
+            cleanup();
+            McpAppBridge.resetInstance();
+            McpAppAdaptor.resetInstance();
+        });
+        it("should register a node and update view state via adaptor", async () => {
+            const adaptor = McpAppAdaptor.getInstance();
+            render(_jsx(DataLLM, { content: "Test content", children: _jsx("div", { children: "Child" }) }));
+            await vi.waitFor(() => {
+                const viewState = adaptor
+                    .getHostContextStore("viewState")
+                    .getSnapshot();
+                expect(viewState?.__view_context).toContain("- Test content");
+            });
+        });
+        it("should preserve existing view state when updating context", async () => {
+            const adaptor = McpAppAdaptor.getInstance();
+            await adaptor.setViewState({ existingKey: "existingValue" });
+            render(_jsx(DataLLM, { content: "Test content", children: _jsx("div", { children: "Child" }) }));
+            await vi.waitFor(() => {
+                const viewState = adaptor
+                    .getHostContextStore("viewState")
+                    .getSnapshot();
+                expect(viewState?.existingKey).toBe("existingValue");
+                expect(viewState?.__view_context).toContain("- Test content");
+            });
+        });
+        it("should handle multiple DataLLM components sharing state through adaptor", async () => {
+            const adaptor = McpAppAdaptor.getInstance();
+            render(_jsxs(_Fragment, { children: [_jsx(DataLLM, { content: "First component", children: _jsx("div", { children: "First" }) }), _jsx(DataLLM, { content: "Second component", children: _jsx("div", { children: "Second" }) })] }));
+            await vi.waitFor(() => {
+                const viewState = adaptor
+                    .getHostContextStore("viewState")
+                    .getSnapshot();
+                const context = viewState?.__view_context;
+                expect(context).toContain("- First component");
+                expect(context).toContain("- Second component");
+            });
+        });
+        it("should call ui/update-model-context when view state changes", async () => {
+            render(_jsx(DataLLM, { content: "Test content", children: _jsx("div", { children: "Child" }) }));
+            await vi.waitFor(() => {
+                expect(postMessageMock).toHaveBeenCalledWith(expect.objectContaining({ method: "ui/update-model-context" }), "*");
+            });
+        });
+    });
+});
+//# sourceMappingURL=data-llm.test.js.map

package/dist/web/data-llm.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"data-llm.test.js","sourceRoot":"","sources":["../../src/web/data-llm.test.tsx"],"names":[],"mappings":";AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,wBAAwB,CAAC;AACzD,OAAO,EACL,SAAS,EACT,UAAU,EACV,QAAQ,EACR,MAAM,EACN,EAAE,EAEF,EAAE,GACH,MAAM,QAAQ,CAAC;AAChB,OAAO,EAAE,aAAa,EAAE,YAAY,EAAE,MAAM,4BAA4B,CAAC;AACzE,OAAO,EAAE,OAAO,EAAE,MAAM,eAAe,CAAC;AACxC,OAAO,EACL,4BAA4B,EAC5B,kBAAkB,GACnB,MAAM,uBAAuB,CAAC;AAE/B,QAAQ,CAAC,SAAS,EAAE,GAAG,EAAE;IACvB,SAAS,CAAC,GAAG,EAAE;QACb,sDAAsD;QACtD,OAAO,EAAE,CAAC;QACV,EAAE,CAAC,gBAAgB,EAAE,CAAC;QACtB,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,IAAI,UAA0D,CAAC;QAE/D,UAAU,CAAC,GAAG,EAAE;YACd,UAAU,GAAG;gBACX,WAAW,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE;gBACjC,cAAc,EAAE,EAAE,CAAC,EAAE,EAAE;aACxB,CAAC;YACF,kDAAkD;YAClD,MAAM,CAAC,cAAc,CAAC,UAAU,EAAE,QAAQ,EAAE;gBAC1C,KAAK,EAAE,UAAU;gBACjB,QAAQ,EAAE,IAAI;gBACd,YAAY,EAAE,IAAI;aACnB,CAAC,CAAC;YACH,kDAAkD;YAClD,IAAI,OAAO,MAAM,KAAK,WAAW,EAAE,CAAC;gBAClC,MAAM,CAAC,cAAc,CAAC,MAAM,EAAE,QAAQ,EAAE;oBACtC,KAAK,EAAE,UAAU;oBACjB,QAAQ,EAAE,IAAI;oBACd,YAAY,EAAE,IAAI;iBACnB,CAAC,CAAC;YACL,CAAC;YACD,EAAE,CAAC,UAAU,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;YACpC,EAAE,CAAC,UAAU,CAAC,WAAW,EAAE,EAAE,QAAQ,EAAE,UAAU,EAAE,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC;QAEH,SAAS,CAAC,GAAG,EAAE;YACb,0DAA0D;YAC1D,IAAI,OAAO,MAAM,KAAK,WAAW,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;gBACnD,MAAM,CAAC,MAAM,CAAC,cAAc,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;gBACvC,MAAM,CAAC,MAAM,CAAC,WAAW,GAAG,EAAE,YAAY,EAAE,EAAE,EAAE,cAAc,EAAE,EAAE,EAAE,CAAC;YACvE,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2DAA2D,EAAE,GAAG,EAAE;YACnE,MAAM,CACJ,KAAC,OAAO,IAAC,OAAO,EAAC,cAAc,YAC7B,kCAAgB,GACR,CACX,CAAC;YAEF,MAAM,CAAC,UAAU,CAAC,cAAc,CAAC,CAAC,gBAAgB,EAAE,CAAC;YACrD,MAAM,QAAQ,GACZ,UAAU,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,YAAY,CAAC;YAC7D,MAAM,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,gBAAgB,CAAC,CAAC;YAClD,MAAM,CAAC,QAAQ,EAAE,cAAc,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;QAC/D,CAAC,CAAC,CAAC;QACH,EAAE,CAAC,0DAA0D,EAAE,GAAG,EAAE;YAClE,UAAU,CAAC,WAAW,GAAG;gBACvB,YAAY,EAAE,EAAE,WAAW,EAAE,eAAe,EAAE;aAC/C,CAAC;YAEF,MAAM,CACJ,KAAC,OAAO,IAAC,OAAO,EAAC,cAAc,YAC7B,kCAAgB,GACR,CACX,CAAC;YAEF,MAAM,QAAQ,GACZ,UAAU,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,YAAY,CAAC;YAC7D,MAAM,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,aAAa,EAAE,eAAe,CAAC,CAAC;YAChE,MAAM,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,gBAAgB,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;YACxD,MAAM,CACJ,MAAC,OAAO,IAAC,OAAO,EAAC,SAAS,aACxB,KAAC,OAAO,IAAC,OAAO,EAAC,UAAU,GAAG,EAC9B,KAAC,OAAO,IAAC,OAAO,EAAC,UAAU,YACzB,KAAC,OAAO,IAAC,OAAO,EAAC,SAAS,YACxB,oCAAkB,GACV,GACF,IACF,CACX,CAAC;YAEF,MAAM,QAAQ,GACZ,UAAU,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAClC,UAAU,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAChD,EAAE,CAAC,CAAC,CAAC,EAAE,YAAY,CAAC;YACvB,MAAM,OAAO,GAAG,QAAQ,EAAE,cAAwB,CAAC;YACnD,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;YACvC,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;YAC1C,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;YAC1C,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC;QAC7C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;YACpD,MAAM,EAAE,QAAQ,EAAE,GAAG,MAAM,CACzB,KAAC,OAAO,IAAC,OAAO,EAAC,iBAAiB,YAChC,kCAAgB,GACR,CACX,CAAC;YAEF,MAAM,YAAY,GAAG,UAAU,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC;YAEjE,QAAQ,CACN,KAAC,OAAO,IAAC,OAAO,EAAC,iBAAiB,YAChC,kCAAgB,GACR,CACX,CAAC;YAEF,MAAM,CAAC,UAAU,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,eAAe,CACjE,YAAY,CACb,CAAC;YACF,MAAM,YAAY,GAChB,UAAU,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAClC,UAAU,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAChD,EAAE,CAAC,CAAC,CAAC,EAAE,YAAY,CAAC;YACvB,MAAM,CAAC,YAAY,EAAE,cAAc,CAAC,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAC;QACtE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+DAA+D,EAAE,GAAG,EAAE;YACvE,MAAM,EAAE,OAAO,EAAE,GAAG,MAAM,CACxB,KAAC,OAAO,IAAC,OAAO,EAAC,mBAAmB,YAClC,kCAAgB,GACR,CACX,CAAC;YAEF,MAAM,kBAAkB,GAAG,UAAU,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC;YAEvE,OAAO,EAAE,CAAC;YAEV,MAAM,CAAC,UAAU,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,eAAe,CACjE,kBAAkB,CACnB,CAAC;YACF,MAAM,YAAY,GAChB,UAAU,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAClC,UAAU,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAChD,EAAE,CAAC,CAAC,CAAC,EAAE,YAAY,CAAC;YACvB,MAAM,CAAC,YAAY,EAAE,cAAc,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAC;QAC1E,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;QAC5B,IAAI,eAAgE,CAAC;QAErE,UAAU,CAAC,GAAG,EAAE;YACd,EAAE,CAAC,UAAU,CAAC,WAAW,EAAE,EAAE,QAAQ,EAAE,SAAS,EAAE,CAAC,CAAC;YACpD,EAAE,CAAC,UAAU,CAAC,gBAAgB,EAAE,kBAAkB,CAAC,CAAC;YACpD,eAAe,GAAG,4BAA4B,EAAE,CAAC;YACjD,EAAE,CAAC,UAAU,CAAC,QAAQ,EAAE,EAAE,WAAW,EAAE,eAAe,EAAE,CAAC,CAAC;QAC5D,CAAC,CAAC,CAAC;QAEH,SAAS,CAAC,GAAG,EAAE;YACb,OAAO,EAAE,CAAC;YACV,YAAY,CAAC,aAAa,EAAE,CAAC;YAC7B,aAAa,CAAC,aAAa,EAAE,CAAC;QAChC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;YACxE,MAAM,OAAO,GAAG,aAAa,CAAC,WAAW,EAAE,CAAC;YAE5C,MAAM,CACJ,KAAC,OAAO,IAAC,OAAO,EAAC,cAAc,YAC7B,kCAAgB,GACR,CACX,CAAC;YAEF,MAAM,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE;gBACpB,MAAM,SAAS,GAAG,OAAO;qBACtB,mBAAmB,CAAC,WAAW,CAAC;qBAChC,WAAW,EAAE,CAAC;gBACjB,MAAM,CAAC,SAAS,EAAE,cAAc,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;YAChE,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;YACzE,MAAM,OAAO,GAAG,aAAa,CAAC,WAAW,EAAE,CAAC;YAC5C,MAAM,OAAO,CAAC,YAAY,CAAC,EAAE,WAAW,EAAE,eAAe,EAAE,CAAC,CAAC;YAE7D,MAAM,CACJ,KAAC,OAAO,IAAC,OAAO,EAAC,cAAc,YAC7B,kCAAgB,GACR,CACX,CAAC;YAEF,MAAM,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE;gBACpB,MAAM,SAAS,GAAG,OAAO;qBACtB,mBAAmB,CAAC,WAAW,CAAC;qBAChC,WAAW,EAAE,CAAC;gBACjB,MAAM,CAAC,SAAS,EAAE,WAAW,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;gBACrD,MAAM,CAAC,SAAS,EAAE,cAAc,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;YAChE,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yEAAyE,EAAE,KAAK,IAAI,EAAE;YACvF,MAAM,OAAO,GAAG,aAAa,CAAC,WAAW,EAAE,CAAC;YAE5C,MAAM,CACJ,8BACE,KAAC,OAAO,IAAC,OAAO,EAAC,iBAAiB,YAChC,kCAAgB,GACR,EACV,KAAC,OAAO,IAAC,OAAO,EAAC,kBAAkB,YACjC,mCAAiB,GACT,IACT,CACJ,CAAC;YAEF,MAAM,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE;gBACpB,MAAM,SAAS,GAAG,OAAO;qBACtB,mBAAmB,CAAC,WAAW,CAAC;qBAChC,WAAW,EAAE,CAAC;gBACjB,MAAM,OAAO,GAAG,SAAS,EAAE,cAAwB,CAAC;gBACpD,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAC;gBAC/C,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,oBAAoB,CAAC,CAAC;YAClD,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6DAA6D,EAAE,KAAK,IAAI,EAAE;YAC3E,MAAM,CACJ,KAAC,OAAO,IAAC,OAAO,EAAC,cAAc,YAC7B,kCAAgB,GACR,CACX,CAAC;YAEF,MAAM,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE;gBACpB,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAC1C,MAAM,CAAC,gBAAgB,CAAC,EAAE,MAAM,EAAE,yBAAyB,EAAE,CAAC,EAC9D,GAAG,CACJ,CAAC;YACJ,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import { cleanup, render } from \"@testing-library/react\";\nimport {\n  afterEach,\n  beforeEach,\n  describe,\n  expect,\n  it,\n  type Mock,\n  vi,\n} from \"vitest\";\nimport { McpAppAdaptor, McpAppBridge } from \"./bridges/mcp-app/index.js\";\nimport { DataLLM } from \"./data-llm.js\";\nimport {\n  getMcpAppHostPostMessageMock,\n  MockResizeObserver,\n} from \"./hooks/test/utils.js\";\n\ndescribe(\"DataLLM\", () => {\n  afterEach(() => {\n    // Clean up React components BEFORE unstubbing globals\n    cleanup();\n    vi.unstubAllGlobals();\n    vi.resetAllMocks();\n  });\n\n  describe(\"apps-sdk mode\", () => {\n    let OpenaiMock: { widgetState: unknown; setWidgetState: Mock };\n\n    beforeEach(() => {\n      OpenaiMock = {\n        widgetState: { modelContent: {} },\n        setWidgetState: vi.fn(),\n      };\n      // Use Object.defineProperty to ensure it persists\n      Object.defineProperty(globalThis, \"openai\", {\n        value: OpenaiMock,\n        writable: true,\n        configurable: true,\n      });\n      // Also set on window for browser-like environment\n      if (typeof window !== \"undefined\") {\n        Object.defineProperty(window, \"openai\", {\n          value: OpenaiMock,\n          writable: true,\n          configurable: true,\n        });\n      }\n      vi.stubGlobal(\"openai\", OpenaiMock);\n      vi.stubGlobal(\"skybridge\", { hostType: \"apps-sdk\" });\n    });\n\n    afterEach(() => {\n      // Keep the mock available for React cleanup, but reset it\n      if (typeof window !== \"undefined\" && window.openai) {\n        window.openai.setWidgetState = vi.fn();\n        window.openai.widgetState = { modelContent: {}, privateContent: {} };\n      }\n    });\n\n    it(\"should register a node with content and call setViewState\", () => {\n      render(\n        <DataLLM content=\"Test content\">\n          <div>Child</div>\n        </DataLLM>,\n      );\n\n      expect(OpenaiMock.setWidgetState).toHaveBeenCalled();\n      const callArgs =\n        OpenaiMock.setWidgetState.mock.calls[0]?.[0]?.modelContent;\n      expect(callArgs).toHaveProperty(\"__view_context\");\n      expect(callArgs?.__view_context).toContain(\"- Test content\");\n    });\n    it(\"should preserve existing viewState when updating context\", () => {\n      OpenaiMock.widgetState = {\n        modelContent: { existingKey: \"existingValue\" },\n      };\n\n      render(\n        <DataLLM content=\"Test content\">\n          <div>Child</div>\n        </DataLLM>,\n      );\n\n      const callArgs =\n        OpenaiMock.setWidgetState.mock.calls[0]?.[0]?.modelContent;\n      expect(callArgs).toHaveProperty(\"existingKey\", \"existingValue\");\n      expect(callArgs).toHaveProperty(\"__view_context\");\n    });\n\n    it(\"should handle deeply nested DataLLM components\", () => {\n      render(\n        <DataLLM content=\"Level 1\">\n          <DataLLM content=\"Level 2A\" />\n          <DataLLM content=\"Level 2B\">\n            <DataLLM content=\"Level 3\">\n              <div>Content</div>\n            </DataLLM>\n          </DataLLM>\n        </DataLLM>,\n      );\n\n      const callArgs =\n        OpenaiMock.setWidgetState.mock.calls[\n          OpenaiMock.setWidgetState.mock.calls.length - 1\n        ]?.[0]?.modelContent;\n      const context = callArgs?.__view_context as string;\n      expect(context).toContain(\"- Level 1\");\n      expect(context).toContain(\"  - Level 2A\");\n      expect(context).toContain(\"  - Level 2B\");\n      expect(context).toContain(\"    - Level 3\");\n    });\n\n    it(\"should update context when content changes\", () => {\n      const { rerender } = render(\n        <DataLLM content=\"Initial content\">\n          <div>Child</div>\n        </DataLLM>,\n      );\n\n      const initialCalls = OpenaiMock.setWidgetState.mock.calls.length;\n\n      rerender(\n        <DataLLM content=\"Updated content\">\n          <div>Child</div>\n        </DataLLM>,\n      );\n\n      expect(OpenaiMock.setWidgetState.mock.calls.length).toBeGreaterThan(\n        initialCalls,\n      );\n      const lastCallArgs =\n        OpenaiMock.setWidgetState.mock.calls[\n          OpenaiMock.setWidgetState.mock.calls.length - 1\n        ]?.[0]?.modelContent;\n      expect(lastCallArgs?.__view_context).toContain(\"- Updated content\");\n    });\n\n    it(\"should remove node and update context when component unmounts\", () => {\n      const { unmount } = render(\n        <DataLLM content=\"Content to remove\">\n          <div>Child</div>\n        </DataLLM>,\n      );\n\n      const callsBeforeUnmount = OpenaiMock.setWidgetState.mock.calls.length;\n\n      unmount();\n\n      expect(OpenaiMock.setWidgetState.mock.calls.length).toBeGreaterThan(\n        callsBeforeUnmount,\n      );\n      const lastCallArgs =\n        OpenaiMock.setWidgetState.mock.calls[\n          OpenaiMock.setWidgetState.mock.calls.length - 1\n        ]?.[0]?.modelContent;\n      expect(lastCallArgs?.__view_context).not.toContain(\"Content to remove\");\n    });\n  });\n\n  describe(\"mcp-app mode\", () => {\n    let postMessageMock: ReturnType<typeof getMcpAppHostPostMessageMock>;\n\n    beforeEach(() => {\n      vi.stubGlobal(\"skybridge\", { hostType: \"mcp-app\" });\n      vi.stubGlobal(\"ResizeObserver\", MockResizeObserver);\n      postMessageMock = getMcpAppHostPostMessageMock();\n      vi.stubGlobal(\"parent\", { postMessage: postMessageMock });\n    });\n\n    afterEach(() => {\n      cleanup();\n      McpAppBridge.resetInstance();\n      McpAppAdaptor.resetInstance();\n    });\n\n    it(\"should register a node and update view state via adaptor\", async () => {\n      const adaptor = McpAppAdaptor.getInstance();\n\n      render(\n        <DataLLM content=\"Test content\">\n          <div>Child</div>\n        </DataLLM>,\n      );\n\n      await vi.waitFor(() => {\n        const viewState = adaptor\n          .getHostContextStore(\"viewState\")\n          .getSnapshot();\n        expect(viewState?.__view_context).toContain(\"- Test content\");\n      });\n    });\n\n    it(\"should preserve existing view state when updating context\", async () => {\n      const adaptor = McpAppAdaptor.getInstance();\n      await adaptor.setViewState({ existingKey: \"existingValue\" });\n\n      render(\n        <DataLLM content=\"Test content\">\n          <div>Child</div>\n        </DataLLM>,\n      );\n\n      await vi.waitFor(() => {\n        const viewState = adaptor\n          .getHostContextStore(\"viewState\")\n          .getSnapshot();\n        expect(viewState?.existingKey).toBe(\"existingValue\");\n        expect(viewState?.__view_context).toContain(\"- Test content\");\n      });\n    });\n\n    it(\"should handle multiple DataLLM components sharing state through adaptor\", async () => {\n      const adaptor = McpAppAdaptor.getInstance();\n\n      render(\n        <>\n          <DataLLM content=\"First component\">\n            <div>First</div>\n          </DataLLM>\n          <DataLLM content=\"Second component\">\n            <div>Second</div>\n          </DataLLM>\n        </>,\n      );\n\n      await vi.waitFor(() => {\n        const viewState = adaptor\n          .getHostContextStore(\"viewState\")\n          .getSnapshot();\n        const context = viewState?.__view_context as string;\n        expect(context).toContain(\"- First component\");\n        expect(context).toContain(\"- Second component\");\n      });\n    });\n\n    it(\"should call ui/update-model-context when view state changes\", async () => {\n      render(\n        <DataLLM content=\"Test content\">\n          <div>Child</div>\n        </DataLLM>,\n      );\n\n      await vi.waitFor(() => {\n        expect(postMessageMock).toHaveBeenCalledWith(\n          expect.objectContaining({ method: \"ui/update-model-context\" }),\n          \"*\",\n        );\n      });\n    });\n  });\n});\n"]}