skybridge 0.0.0-dev.d9ee984 → 0.0.0-dev.db1ee0e
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/LICENSE +21 -674
- package/README.md +123 -1
- package/dist/src/server/devtoolsStaticServer.d.ts +15 -0
- package/dist/src/server/devtoolsStaticServer.js +38 -0
- package/dist/src/server/devtoolsStaticServer.js.map +1 -0
- package/dist/src/server/index.d.ts +5 -0
- package/dist/{server → src/server}/index.js +1 -1
- package/dist/src/server/index.js.map +1 -0
- package/dist/src/server/inferUtilityTypes.d.ts +64 -0
- package/dist/src/server/inferUtilityTypes.js +2 -0
- package/dist/src/server/inferUtilityTypes.js.map +1 -0
- package/dist/src/server/server.d.ts +73 -0
- package/dist/src/server/server.js +74 -0
- package/dist/src/server/server.js.map +1 -0
- package/dist/src/server/templateHelper.d.ts +15 -0
- package/dist/src/server/templateHelper.js +30 -0
- package/dist/src/server/templateHelper.js.map +1 -0
- package/dist/src/server/templates/development.hbs +12 -0
- package/dist/src/server/templates/production.hbs +6 -0
- package/dist/{server → src/server}/widgetsDevServer.d.ts +5 -2
- package/dist/{server → src/server}/widgetsDevServer.js +8 -5
- package/dist/src/server/widgetsDevServer.js.map +1 -0
- package/dist/src/test/utils.d.ts +135 -0
- package/dist/src/test/utils.js +242 -0
- package/dist/src/test/utils.js.map +1 -0
- package/dist/src/test/widget.test.d.ts +1 -0
- package/dist/src/test/widget.test.js +134 -0
- package/dist/src/test/widget.test.js.map +1 -0
- package/dist/src/web/bridges/index.d.ts +2 -0
- package/dist/src/web/bridges/index.js +3 -0
- package/dist/src/web/bridges/index.js.map +1 -0
- package/dist/src/web/bridges/use-apps-sdk-bridge.d.ts +2 -0
- package/dist/{web/use-openai-global.js → src/web/bridges/use-apps-sdk-bridge.js} +9 -5
- package/dist/src/web/bridges/use-apps-sdk-bridge.js.map +1 -0
- package/dist/src/web/bridges/use-mcp-app-bridge.d.ts +24 -0
- package/dist/src/web/bridges/use-mcp-app-bridge.js +123 -0
- package/dist/src/web/bridges/use-mcp-app-bridge.js.map +1 -0
- package/dist/src/web/bridges/use-mcp-app-bridge.test.d.ts +1 -0
- package/dist/src/web/bridges/use-mcp-app-bridge.test.js +64 -0
- package/dist/src/web/bridges/use-mcp-app-bridge.test.js.map +1 -0
- package/dist/src/web/create-store.d.ts +3 -0
- package/dist/src/web/create-store.js +25 -0
- package/dist/src/web/create-store.js.map +1 -0
- package/dist/src/web/create-store.test.d.ts +1 -0
- package/dist/src/web/create-store.test.js +70 -0
- package/dist/src/web/create-store.test.js.map +1 -0
- package/dist/src/web/data-llm.d.ts +14 -0
- package/dist/src/web/data-llm.js +68 -0
- package/dist/src/web/data-llm.js.map +1 -0
- package/dist/src/web/data-llm.test.d.ts +1 -0
- package/dist/src/web/data-llm.test.js +76 -0
- package/dist/src/web/data-llm.test.js.map +1 -0
- package/dist/src/web/generate-helpers.d.ts +115 -0
- package/dist/src/web/generate-helpers.js +111 -0
- package/dist/src/web/generate-helpers.js.map +1 -0
- package/dist/src/web/generate-helpers.test-d.d.ts +1 -0
- package/dist/src/web/generate-helpers.test-d.js +206 -0
- package/dist/src/web/generate-helpers.test-d.js.map +1 -0
- package/dist/src/web/generate-helpers.test.d.ts +1 -0
- package/dist/src/web/generate-helpers.test.js +17 -0
- package/dist/src/web/generate-helpers.test.js.map +1 -0
- package/dist/src/web/helpers/state.d.ts +7 -0
- package/dist/src/web/helpers/state.js +40 -0
- package/dist/src/web/helpers/state.js.map +1 -0
- package/dist/src/web/helpers/state.test.d.ts +1 -0
- package/dist/src/web/helpers/state.test.js +53 -0
- package/dist/src/web/helpers/state.test.js.map +1 -0
- package/dist/src/web/hooks/index.d.ts +12 -0
- package/dist/src/web/hooks/index.js +13 -0
- package/dist/src/web/hooks/index.js.map +1 -0
- package/dist/src/web/hooks/use-call-tool.d.ts +100 -0
- package/dist/src/web/hooks/use-call-tool.js +60 -0
- package/dist/src/web/hooks/use-call-tool.js.map +1 -0
- package/dist/src/web/hooks/use-call-tool.test-d.d.ts +1 -0
- package/dist/src/web/hooks/use-call-tool.test-d.js +104 -0
- package/dist/src/web/hooks/use-call-tool.test-d.js.map +1 -0
- package/dist/src/web/hooks/use-call-tool.test.d.ts +1 -0
- package/dist/src/web/hooks/use-call-tool.test.js +165 -0
- package/dist/src/web/hooks/use-call-tool.test.js.map +1 -0
- package/dist/src/web/hooks/use-display-mode.d.ts +4 -0
- package/dist/src/web/hooks/use-display-mode.js +7 -0
- package/dist/src/web/hooks/use-display-mode.js.map +1 -0
- package/dist/src/web/hooks/use-display-mode.test.d.ts +1 -0
- package/dist/src/web/hooks/use-display-mode.test.js +40 -0
- package/dist/src/web/hooks/use-display-mode.test.js.map +1 -0
- package/dist/src/web/hooks/use-files.d.ts +10 -0
- package/dist/src/web/hooks/use-files.js +7 -0
- package/dist/src/web/hooks/use-files.js.map +1 -0
- package/dist/src/web/hooks/use-files.test.d.ts +1 -0
- package/dist/src/web/hooks/use-files.test.js +29 -0
- package/dist/src/web/hooks/use-files.test.js.map +1 -0
- package/dist/src/web/hooks/use-locale.d.ts +1 -0
- package/dist/src/web/hooks/use-locale.js +5 -0
- package/dist/src/web/hooks/use-locale.js.map +1 -0
- package/dist/src/web/hooks/use-locale.test.d.ts +1 -0
- package/dist/src/web/hooks/use-locale.test.js +21 -0
- package/dist/src/web/hooks/use-locale.test.js.map +1 -0
- package/dist/src/web/hooks/use-open-external.d.ts +1 -0
- package/dist/src/web/hooks/use-open-external.js +6 -0
- package/dist/src/web/hooks/use-open-external.js.map +1 -0
- package/dist/src/web/hooks/use-open-external.test.d.ts +1 -0
- package/dist/src/web/hooks/use-open-external.test.js +24 -0
- package/dist/src/web/hooks/use-open-external.test.js.map +1 -0
- package/dist/src/web/hooks/use-openai-global.d.ts +3 -0
- package/dist/src/web/hooks/use-openai-global.js +6 -0
- package/dist/src/web/hooks/use-openai-global.js.map +1 -0
- package/dist/src/web/hooks/use-request-modal.d.ts +6 -0
- package/dist/src/web/hooks/use-request-modal.js +9 -0
- package/dist/src/web/hooks/use-request-modal.js.map +1 -0
- package/dist/src/web/hooks/use-request-modal.test.d.ts +1 -0
- package/dist/src/web/hooks/use-request-modal.test.js +24 -0
- package/dist/src/web/hooks/use-request-modal.test.js.map +1 -0
- package/dist/src/web/hooks/use-send-follow-up-message.d.ts +1 -0
- package/dist/src/web/hooks/use-send-follow-up-message.js +11 -0
- package/dist/src/web/hooks/use-send-follow-up-message.js.map +1 -0
- package/dist/src/web/hooks/use-theme.d.ts +1 -0
- package/dist/src/web/hooks/use-theme.js +5 -0
- package/dist/src/web/hooks/use-theme.js.map +1 -0
- package/dist/src/web/hooks/use-theme.test.d.ts +1 -0
- package/dist/src/web/hooks/use-theme.test.js +26 -0
- package/dist/src/web/hooks/use-theme.test.js.map +1 -0
- package/dist/src/web/hooks/use-tool-info.d.ts +25 -0
- package/dist/src/web/hooks/use-tool-info.js +20 -0
- package/dist/src/web/hooks/use-tool-info.js.map +1 -0
- package/dist/src/web/hooks/use-tool-info.test-d.d.ts +1 -0
- package/dist/src/web/hooks/use-tool-info.test-d.js +73 -0
- package/dist/src/web/hooks/use-tool-info.test-d.js.map +1 -0
- package/dist/src/web/hooks/use-tool-info.test.d.ts +1 -0
- package/dist/src/web/hooks/use-tool-info.test.js +59 -0
- package/dist/src/web/hooks/use-tool-info.test.js.map +1 -0
- package/dist/src/web/hooks/use-user-agent.d.ts +1 -0
- package/dist/src/web/hooks/use-user-agent.js +5 -0
- package/dist/src/web/hooks/use-user-agent.js.map +1 -0
- package/dist/src/web/hooks/use-user-agent.test.d.ts +1 -0
- package/dist/src/web/hooks/use-user-agent.test.js +31 -0
- package/dist/src/web/hooks/use-user-agent.test.js.map +1 -0
- package/dist/src/web/hooks/use-widget-state.d.ts +4 -0
- package/dist/src/web/hooks/use-widget-state.js +32 -0
- package/dist/src/web/hooks/use-widget-state.js.map +1 -0
- package/dist/src/web/hooks/use-widget-state.test.d.ts +1 -0
- package/dist/src/web/hooks/use-widget-state.test.js +60 -0
- package/dist/src/web/hooks/use-widget-state.test.js.map +1 -0
- package/dist/src/web/index.d.ts +8 -0
- package/dist/src/web/index.js +9 -0
- package/dist/src/web/index.js.map +1 -0
- package/dist/src/web/mount-widget.js +19 -0
- package/dist/src/web/mount-widget.js.map +1 -0
- package/dist/src/web/plugin/data-llm.test.d.ts +1 -0
- package/dist/src/web/plugin/data-llm.test.js +81 -0
- package/dist/src/web/plugin/data-llm.test.js.map +1 -0
- package/dist/src/web/plugin/plugin.d.ts +2 -0
- package/dist/src/web/plugin/plugin.js +39 -0
- package/dist/src/web/plugin/plugin.js.map +1 -0
- package/dist/src/web/plugin/transform-data-llm.d.ts +12 -0
- package/dist/src/web/plugin/transform-data-llm.js +93 -0
- package/dist/src/web/plugin/transform-data-llm.js.map +1 -0
- package/dist/src/web/plugin/transform-data-llm.test.d.ts +1 -0
- package/dist/src/web/plugin/transform-data-llm.test.js +81 -0
- package/dist/src/web/plugin/transform-data-llm.test.js.map +1 -0
- package/dist/src/web/proxy.d.ts +1 -0
- package/dist/src/web/proxy.js +53 -0
- package/dist/src/web/proxy.js.map +1 -0
- package/dist/{web → src/web}/types.d.ts +56 -18
- package/dist/{web → src/web}/types.js +1 -0
- package/dist/src/web/types.js.map +1 -0
- package/dist/vitest.config.d.ts +2 -0
- package/dist/vitest.config.js +8 -0
- package/dist/vitest.config.js.map +1 -0
- package/package.json +36 -13
- package/dist/server/index.d.ts +0 -3
- package/dist/server/index.js.map +0 -1
- package/dist/server/middleware.d.ts +0 -3
- package/dist/server/middleware.js +0 -47
- package/dist/server/middleware.js.map +0 -1
- package/dist/server/server.d.ts +0 -12
- package/dist/server/server.js +0 -70
- package/dist/server/server.js.map +0 -1
- package/dist/server/widgetsDevServer.js.map +0 -1
- package/dist/web/index.d.ts +0 -4
- package/dist/web/index.js +0 -5
- package/dist/web/index.js.map +0 -1
- package/dist/web/mount-widget.js +0 -10
- package/dist/web/mount-widget.js.map +0 -1
- package/dist/web/types.js.map +0 -1
- package/dist/web/use-openai-global.d.ts +0 -2
- package/dist/web/use-openai-global.js.map +0 -1
- package/dist/web/use-tool-output.d.ts +0 -3
- package/dist/web/use-tool-output.js +0 -5
- package/dist/web/use-tool-output.js.map +0 -1
- /package/dist/{web → src/web}/mount-widget.d.ts +0 -0
|
@@ -0,0 +1,123 @@
|
|
|
1
|
+
import { useSyncExternalStore } from "react";
|
|
2
|
+
const LATEST_PROTOCOL_VERSION = "2025-11-21";
|
|
3
|
+
export class McpAppBridge {
|
|
4
|
+
context = null;
|
|
5
|
+
listeners = new Map();
|
|
6
|
+
pendingRequests = new Map();
|
|
7
|
+
nextId = 1;
|
|
8
|
+
initialized;
|
|
9
|
+
appInitializationOptions;
|
|
10
|
+
requestTimeout;
|
|
11
|
+
constructor(options, requestTimeout = 10_000) {
|
|
12
|
+
this.requestTimeout = requestTimeout;
|
|
13
|
+
this.initialized = false;
|
|
14
|
+
this.appInitializationOptions = {
|
|
15
|
+
appInfo: options.appInfo,
|
|
16
|
+
appCapabilities: {},
|
|
17
|
+
protocolVersion: LATEST_PROTOCOL_VERSION,
|
|
18
|
+
};
|
|
19
|
+
}
|
|
20
|
+
subscribe = (key) => (onChange) => {
|
|
21
|
+
this.listeners.set(key, new Set([...(this.listeners.get(key) || []), onChange]));
|
|
22
|
+
this.init();
|
|
23
|
+
return () => this.listeners.get(key)?.delete(onChange);
|
|
24
|
+
};
|
|
25
|
+
cleanup() {
|
|
26
|
+
window.removeEventListener("message", this.handleMessage);
|
|
27
|
+
this.pendingRequests.forEach((request) => {
|
|
28
|
+
clearTimeout(request.timeout);
|
|
29
|
+
});
|
|
30
|
+
this.pendingRequests.clear();
|
|
31
|
+
this.listeners.clear();
|
|
32
|
+
}
|
|
33
|
+
request({ method, params, }) {
|
|
34
|
+
const id = this.nextId++;
|
|
35
|
+
const { promise, resolve, reject } = Promise.withResolvers();
|
|
36
|
+
this.pendingRequests.set(id, {
|
|
37
|
+
resolve: resolve,
|
|
38
|
+
reject,
|
|
39
|
+
timeout: setTimeout(() => {
|
|
40
|
+
reject(new Error("Request timed out"));
|
|
41
|
+
this.pendingRequests.delete(id);
|
|
42
|
+
}, this.requestTimeout),
|
|
43
|
+
});
|
|
44
|
+
window.parent.postMessage({ jsonrpc: "2.0", id, method, params }, "*");
|
|
45
|
+
return promise;
|
|
46
|
+
}
|
|
47
|
+
emit(key) {
|
|
48
|
+
this.listeners.get(key)?.forEach((listener) => {
|
|
49
|
+
listener();
|
|
50
|
+
});
|
|
51
|
+
}
|
|
52
|
+
setContext(context) {
|
|
53
|
+
if (context == null)
|
|
54
|
+
return;
|
|
55
|
+
this.context = context;
|
|
56
|
+
for (const key of Object.keys(context)) {
|
|
57
|
+
this.emit(key);
|
|
58
|
+
}
|
|
59
|
+
}
|
|
60
|
+
init() {
|
|
61
|
+
if (this.initialized)
|
|
62
|
+
return;
|
|
63
|
+
this.initialized = true;
|
|
64
|
+
if (typeof window === "undefined" || window.parent === window) {
|
|
65
|
+
return;
|
|
66
|
+
}
|
|
67
|
+
window.addEventListener("message", this.handleMessage);
|
|
68
|
+
this.connect();
|
|
69
|
+
}
|
|
70
|
+
handleMessage = (event) => {
|
|
71
|
+
const data = event.data;
|
|
72
|
+
if (data?.jsonrpc !== "2.0")
|
|
73
|
+
return;
|
|
74
|
+
const request = this.pendingRequests.get(data.id);
|
|
75
|
+
if (request) {
|
|
76
|
+
clearTimeout(request.timeout);
|
|
77
|
+
this.pendingRequests.delete(data.id);
|
|
78
|
+
if (data.error) {
|
|
79
|
+
request.reject(new Error(data.error.message));
|
|
80
|
+
return;
|
|
81
|
+
}
|
|
82
|
+
request.resolve(data.result);
|
|
83
|
+
return;
|
|
84
|
+
}
|
|
85
|
+
if (data.method === "ui/notifications/host-context-changed") {
|
|
86
|
+
this.setContext({ ...this.context, ...data.params });
|
|
87
|
+
}
|
|
88
|
+
};
|
|
89
|
+
async connect() {
|
|
90
|
+
try {
|
|
91
|
+
const result = await this.request({
|
|
92
|
+
method: "ui/initialize",
|
|
93
|
+
params: this.appInitializationOptions,
|
|
94
|
+
});
|
|
95
|
+
this.setContext(result.hostContext);
|
|
96
|
+
this.notify({ method: "ui/notifications/initialized" });
|
|
97
|
+
}
|
|
98
|
+
catch (err) {
|
|
99
|
+
console.error(err);
|
|
100
|
+
}
|
|
101
|
+
}
|
|
102
|
+
notify(notification) {
|
|
103
|
+
window.parent.postMessage({ jsonrpc: "2.0", ...notification }, "*");
|
|
104
|
+
}
|
|
105
|
+
}
|
|
106
|
+
let instance = null;
|
|
107
|
+
const defaultOptions = {
|
|
108
|
+
appInfo: { name: "skybridge-app", version: "0.0.1" },
|
|
109
|
+
};
|
|
110
|
+
export function getMcpHost(options, requestTimeout) {
|
|
111
|
+
if (instance && (options || requestTimeout)) {
|
|
112
|
+
console.warn("getMcpHost: options and requestTimeout ignored, instance already exists");
|
|
113
|
+
}
|
|
114
|
+
if (!instance) {
|
|
115
|
+
instance = new McpAppBridge({ ...defaultOptions, ...options }, requestTimeout);
|
|
116
|
+
}
|
|
117
|
+
return instance;
|
|
118
|
+
}
|
|
119
|
+
export function useMcpAppBridge(key, options, requestTimeout) {
|
|
120
|
+
const host = getMcpHost(options, requestTimeout);
|
|
121
|
+
return useSyncExternalStore(host.subscribe(key), () => host.context?.[key]);
|
|
122
|
+
}
|
|
123
|
+
//# sourceMappingURL=use-mcp-app-bridge.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"use-mcp-app-bridge.js","sourceRoot":"","sources":["../../../../src/web/bridges/use-mcp-app-bridge.ts"],"names":[],"mappings":"AAOA,OAAO,EAAE,oBAAoB,EAAE,MAAM,OAAO,CAAC;AAa7C,MAAM,uBAAuB,GAAG,YAAY,CAAC;AAE7C,MAAM,OAAO,YAAY;IAChB,OAAO,GAA4B,IAAI,CAAC;IACvC,SAAS,GAAG,IAAI,GAAG,EAA2C,CAAC;IAC/D,eAAe,GAAG,IAAI,GAAG,EAAmC,CAAC;IAC7D,MAAM,GAAG,CAAC,CAAC;IACX,WAAW,CAAU;IACrB,wBAAwB,CAAmC;IAC3D,cAAc,CAAS;IAE/B,YACE,OAAoC,EACpC,iBAAyB,MAAM;QAE/B,IAAI,CAAC,cAAc,GAAG,cAAc,CAAC;QACrC,IAAI,CAAC,WAAW,GAAG,KAAK,CAAC;QACzB,IAAI,CAAC,wBAAwB,GAAG;YAC9B,OAAO,EAAE,OAAO,CAAC,OAAO;YACxB,eAAe,EAAE,EAAE;YACnB,eAAe,EAAE,uBAAuB;SACzC,CAAC;IACJ,CAAC;IAED,SAAS,GAAG,CAAC,GAA2B,EAAE,EAAE,CAAC,CAAC,QAAoB,EAAE,EAAE;QACpE,IAAI,CAAC,SAAS,CAAC,GAAG,CAChB,GAAG,EACH,IAAI,GAAG,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,EAAE,QAAQ,CAAC,CAAC,CACxD,CAAC;QACF,IAAI,CAAC,IAAI,EAAE,CAAC;QACZ,OAAO,GAAG,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,MAAM,CAAC,QAAQ,CAAC,CAAC;IACzD,CAAC,CAAC;IAEK,OAAO;QACZ,MAAM,CAAC,mBAAmB,CAAC,SAAS,EAAE,IAAI,CAAC,aAAa,CAAC,CAAC;QAC1D,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;YACvC,YAAY,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QAChC,CAAC,CAAC,CAAC;QACH,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,CAAC;QAC7B,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC;IACzB,CAAC;IAEO,OAAO,CAAoD,EACjE,MAAM,EACN,MAAM,GACJ;QACF,MAAM,EAAE,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;QACzB,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,aAAa,EAAK,CAAC;QAChE,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,EAAE,EAAE;YAC3B,OAAO,EAAE,OAAmC;YAC5C,MAAM;YACN,OAAO,EAAE,UAAU,CAAC,GAAG,EAAE;gBACvB,MAAM,CAAC,IAAI,KAAK,CAAC,mBAAmB,CAAC,CAAC,CAAC;gBACvC,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;YAClC,CAAC,EAAE,IAAI,CAAC,cAAc,CAAC;SACxB,CAAC,CAAC;QACH,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,EAAE,GAAG,CAAC,CAAC;QAEvE,OAAO,OAAO,CAAC;IACjB,CAAC;IAEO,IAAI,CAAC,GAA2B;QACtC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,OAAO,CAAC,CAAC,QAAQ,EAAE,EAAE;YAC5C,QAAQ,EAAE,CAAC;QACb,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,UAAU,CAAC,OAAgC;QACjD,IAAI,OAAO,IAAI,IAAI;YAAE,OAAO;QAC5B,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC;YACvC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACjB,CAAC;IACH,CAAC;IAEO,IAAI;QACV,IAAI,IAAI,CAAC,WAAW;YAAE,OAAO;QAC7B,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;QAExB,IAAI,OAAO,MAAM,KAAK,WAAW,IAAI,MAAM,CAAC,MAAM,KAAK,MAAM,EAAE,CAAC;YAC9D,OAAO;QACT,CAAC;QAED,MAAM,CAAC,gBAAgB,CAAC,SAAS,EAAE,IAAI,CAAC,aAAa,CAAC,CAAC;QACvD,IAAI,CAAC,OAAO,EAAE,CAAC;IACjB,CAAC;IAEO,aAAa,GAAG,CAAC,KAAmB,EAAE,EAAE;QAC9C,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC;QACxB,IAAI,IAAI,EAAE,OAAO,KAAK,KAAK;YAAE,OAAO;QAEpC,MAAM,OAAO,GAAG,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAClD,IAAI,OAAO,EAAE,CAAC;YACZ,YAAY,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;YAC9B,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YACrC,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;gBACf,OAAO,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC;gBAC9C,OAAO;YACT,CAAC;YAED,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAC7B,OAAO;QACT,CAAC;QAED,IAAI,IAAI,CAAC,MAAM,KAAK,uCAAuC,EAAE,CAAC;YAC5D,IAAI,CAAC,UAAU,CAAC,EAAE,GAAG,IAAI,CAAC,OAAO,EAAE,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;QACvD,CAAC;IACH,CAAC,CAAC;IAEM,KAAK,CAAC,OAAO;QACnB,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAG/B;gBACA,MAAM,EAAE,eAAe;gBACvB,MAAM,EAAE,IAAI,CAAC,wBAAwB;aACtC,CAAC,CAAC;YAEH,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;YACpC,IAAI,CAAC,MAAM,CAAC,EAAE,MAAM,EAAE,8BAA8B,EAAE,CAAC,CAAC;QAC1D,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QACrB,CAAC;IACH,CAAC;IAEO,MAAM,CAAC,YAA0C;QACvD,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,GAAG,YAAY,EAAE,EAAE,GAAG,CAAC,CAAC;IACtE,CAAC;CACF;AAED,IAAI,QAAQ,GAAwB,IAAI,CAAC;AAEzC,MAAM,cAAc,GAAgC;IAClD,OAAO,EAAE,EAAE,IAAI,EAAE,eAAe,EAAE,OAAO,EAAE,OAAO,EAAE;CACrD,CAAC;AAEF,MAAM,UAAU,UAAU,CACxB,OAA8C,EAC9C,cAAuB;IAEvB,IAAI,QAAQ,IAAI,CAAC,OAAO,IAAI,cAAc,CAAC,EAAE,CAAC;QAC5C,OAAO,CAAC,IAAI,CACV,yEAAyE,CAC1E,CAAC;IACJ,CAAC;IACD,IAAI,CAAC,QAAQ,EAAE,CAAC;QACd,QAAQ,GAAG,IAAI,YAAY,CACzB,EAAE,GAAG,cAAc,EAAE,GAAG,OAAO,EAAE,EACjC,cAAc,CACf,CAAC;IACJ,CAAC;IAED,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,MAAM,UAAU,eAAe,CAC7B,GAAM,EACN,OAA8C,EAC9C,cAAuB;IAEvB,MAAM,IAAI,GAAG,UAAU,CAAC,OAAO,EAAE,cAAc,CAAC,CAAC;IAEjD,OAAO,oBAAoB,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC;AAC9E,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
export {};
|
|
@@ -0,0 +1,64 @@
|
|
|
1
|
+
import { act, renderHook, waitFor } from "@testing-library/react";
|
|
2
|
+
import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
|
|
3
|
+
describe("useMcpAppBridge", () => {
|
|
4
|
+
let mockPostMessage;
|
|
5
|
+
let mcpAppModule;
|
|
6
|
+
const importModule = async () => {
|
|
7
|
+
return import("./use-mcp-app-bridge.js");
|
|
8
|
+
};
|
|
9
|
+
beforeEach(async () => {
|
|
10
|
+
vi.resetModules();
|
|
11
|
+
mcpAppModule = await importModule();
|
|
12
|
+
mockPostMessage = vi.fn();
|
|
13
|
+
Object.defineProperty(window, "parent", {
|
|
14
|
+
value: { postMessage: mockPostMessage },
|
|
15
|
+
writable: true,
|
|
16
|
+
configurable: true,
|
|
17
|
+
});
|
|
18
|
+
});
|
|
19
|
+
afterEach(() => {
|
|
20
|
+
vi.clearAllMocks();
|
|
21
|
+
});
|
|
22
|
+
it("should return the theme value from host context", async () => {
|
|
23
|
+
const { useMcpAppBridge } = mcpAppModule;
|
|
24
|
+
const { result } = renderHook(() => useMcpAppBridge("theme"));
|
|
25
|
+
const initCall = mockPostMessage.mock.calls.find((call) => call[0].method === "ui/initialize");
|
|
26
|
+
if (initCall) {
|
|
27
|
+
act(() => {
|
|
28
|
+
window.dispatchEvent(new MessageEvent("message", {
|
|
29
|
+
data: {
|
|
30
|
+
jsonrpc: "2.0",
|
|
31
|
+
id: initCall[0].id,
|
|
32
|
+
result: {
|
|
33
|
+
protocolVersion: "2025-06-18",
|
|
34
|
+
hostInfo: { name: "test-host", version: "1.0.0" },
|
|
35
|
+
hostCapabilities: {},
|
|
36
|
+
hostContext: { theme: "light" },
|
|
37
|
+
},
|
|
38
|
+
},
|
|
39
|
+
}));
|
|
40
|
+
});
|
|
41
|
+
}
|
|
42
|
+
await waitFor(() => {
|
|
43
|
+
expect(result.current).toBe("light");
|
|
44
|
+
});
|
|
45
|
+
});
|
|
46
|
+
it("should reject the request after timeout", async () => {
|
|
47
|
+
vi.useFakeTimers();
|
|
48
|
+
const consoleErrorSpy = vi
|
|
49
|
+
.spyOn(console, "error")
|
|
50
|
+
.mockImplementation(() => { });
|
|
51
|
+
const { useMcpAppBridge, getMcpHost } = mcpAppModule;
|
|
52
|
+
const bridge = getMcpHost({ appInfo: { name: "test", version: "1.0.0" } }, 100);
|
|
53
|
+
renderHook(() => useMcpAppBridge("theme"));
|
|
54
|
+
expect(mockPostMessage).toHaveBeenCalledWith(expect.objectContaining({ method: "ui/initialize" }), "*");
|
|
55
|
+
await act(async () => {
|
|
56
|
+
await vi.advanceTimersByTimeAsync(100);
|
|
57
|
+
});
|
|
58
|
+
expect(consoleErrorSpy).toHaveBeenCalledWith(new Error("Request timed out"));
|
|
59
|
+
consoleErrorSpy.mockRestore();
|
|
60
|
+
vi.useRealTimers();
|
|
61
|
+
bridge.cleanup();
|
|
62
|
+
});
|
|
63
|
+
});
|
|
64
|
+
//# sourceMappingURL=use-mcp-app-bridge.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"use-mcp-app-bridge.test.js","sourceRoot":"","sources":["../../../../src/web/bridges/use-mcp-app-bridge.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,GAAG,EAAE,UAAU,EAAE,OAAO,EAAE,MAAM,wBAAwB,CAAC;AAClE,OAAO,EAAE,SAAS,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAEzE,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC/B,IAAI,eAAyC,CAAC;IAC9C,IAAI,YAAsD,CAAC;IAE3D,MAAM,YAAY,GAAG,KAAK,IAAI,EAAE;QAC9B,OAAO,MAAM,CAAC,yBAAyB,CAAC,CAAC;IAC3C,CAAC,CAAC;IAEF,UAAU,CAAC,KAAK,IAAI,EAAE;QACpB,EAAE,CAAC,YAAY,EAAE,CAAC;QAClB,YAAY,GAAG,MAAM,YAAY,EAAE,CAAC;QACpC,eAAe,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QAC1B,MAAM,CAAC,cAAc,CAAC,MAAM,EAAE,QAAQ,EAAE;YACtC,KAAK,EAAE,EAAE,WAAW,EAAE,eAAe,EAAE;YACvC,QAAQ,EAAE,IAAI;YACd,YAAY,EAAE,IAAI;SACnB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;QAC/D,MAAM,EAAE,eAAe,EAAE,GAAG,YAAY,CAAC;QACzC,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC,CAAC;QAE9D,MAAM,QAAQ,GAAG,eAAe,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAC9C,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,MAAM,KAAK,eAAe,CAC7C,CAAC;QAEF,IAAI,QAAQ,EAAE,CAAC;YACb,GAAG,CAAC,GAAG,EAAE;gBACP,MAAM,CAAC,aAAa,CAClB,IAAI,YAAY,CAAC,SAAS,EAAE;oBAC1B,IAAI,EAAE;wBACJ,OAAO,EAAE,KAAK;wBACd,EAAE,EAAE,QAAQ,CAAC,CAAC,CAAC,CAAC,EAAE;wBAClB,MAAM,EAAE;4BACN,eAAe,EAAE,YAAY;4BAC7B,QAAQ,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,OAAO,EAAE;4BACjD,gBAAgB,EAAE,EAAE;4BACpB,WAAW,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE;yBAChC;qBACF;iBACF,CAAC,CACH,CAAC;YACJ,CAAC,CAAC,CAAC;QACL,CAAC;QAED,MAAM,OAAO,CAAC,GAAG,EAAE;YACjB,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACvC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;QACvD,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,MAAM,eAAe,GAAG,EAAE;aACvB,KAAK,CAAC,OAAO,EAAE,OAAO,CAAC;aACvB,kBAAkB,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;QAEhC,MAAM,EAAE,eAAe,EAAE,UAAU,EAAE,GAAG,YAAY,CAAC;QACrD,MAAM,MAAM,GAAG,UAAU,CACvB,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,EAAE,EAC/C,GAAG,CACJ,CAAC;QAEF,UAAU,CAAC,GAAG,EAAE,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC,CAAC;QAE3C,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAC1C,MAAM,CAAC,gBAAgB,CAAC,EAAE,MAAM,EAAE,eAAe,EAAE,CAAC,EACpD,GAAG,CACJ,CAAC;QAEF,MAAM,GAAG,CAAC,KAAK,IAAI,EAAE;YACnB,MAAM,EAAE,CAAC,wBAAwB,CAAC,GAAG,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAC1C,IAAI,KAAK,CAAC,mBAAmB,CAAC,CAC/B,CAAC;QAEF,eAAe,CAAC,WAAW,EAAE,CAAC;QAC9B,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,MAAM,CAAC,OAAO,EAAE,CAAC;IACnB,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1,3 @@
|
|
|
1
|
+
import { type StateCreator } from "zustand";
|
|
2
|
+
import type { UnknownObject } from "./types.js";
|
|
3
|
+
export declare function createStore<State extends UnknownObject>(storeCreator: StateCreator<State, [], [], State>, defaultState?: State | (() => State)): import("zustand").UseBoundStore<import("zustand").StoreApi<State>>;
|
|
@@ -0,0 +1,25 @@
|
|
|
1
|
+
import { create } from "zustand";
|
|
2
|
+
import { getInitialState, injectWidgetContext, serializeState, } from "./helpers/state.js";
|
|
3
|
+
export function createStore(storeCreator, defaultState) {
|
|
4
|
+
const initialState = getInitialState(defaultState);
|
|
5
|
+
const store = create()((...args) => {
|
|
6
|
+
const baseStore = storeCreator(...args);
|
|
7
|
+
if (initialState !== null) {
|
|
8
|
+
return { ...baseStore, ...initialState };
|
|
9
|
+
}
|
|
10
|
+
return baseStore;
|
|
11
|
+
});
|
|
12
|
+
store.subscribe((state) => {
|
|
13
|
+
if (window.openai) {
|
|
14
|
+
const serializedState = serializeState(state);
|
|
15
|
+
if (serializedState !== null && serializedState !== undefined) {
|
|
16
|
+
const stateToPersist = injectWidgetContext(serializedState);
|
|
17
|
+
if (stateToPersist !== null) {
|
|
18
|
+
window.openai.setWidgetState(stateToPersist);
|
|
19
|
+
}
|
|
20
|
+
}
|
|
21
|
+
}
|
|
22
|
+
});
|
|
23
|
+
return store;
|
|
24
|
+
}
|
|
25
|
+
//# sourceMappingURL=create-store.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"create-store.js","sourceRoot":"","sources":["../../../src/web/create-store.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAqB,MAAM,SAAS,CAAC;AACpD,OAAO,EACL,eAAe,EACf,mBAAmB,EACnB,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,KAAK,CAAC,SAAS,CAAC,CAAC,KAAY,EAAE,EAAE;QAC/B,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;YAClB,MAAM,eAAe,GAAG,cAAc,CAAC,KAAK,CAAC,CAAC;YAC9C,IAAI,eAAe,KAAK,IAAI,IAAI,eAAe,KAAK,SAAS,EAAE,CAAC;gBAC9D,MAAM,cAAc,GAAG,mBAAmB,CAAC,eAAwB,CAAC,CAAC;gBACrE,IAAI,cAAc,KAAK,IAAI,EAAE,CAAC;oBAC5B,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,cAAc,CAAC,CAAC;gBAC/C,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,OAAO,KAAK,CAAC;AACf,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
export {};
|
|
@@ -0,0 +1,70 @@
|
|
|
1
|
+
import { afterEach, beforeEach, describe, expect, it, vi, } from "vitest";
|
|
2
|
+
import { createStore } from "./create-store.js";
|
|
3
|
+
import { WIDGET_CONTEXT_KEY } from "./data-llm.js";
|
|
4
|
+
describe("createStore", () => {
|
|
5
|
+
let OpenaiMock;
|
|
6
|
+
beforeEach(() => {
|
|
7
|
+
OpenaiMock = {
|
|
8
|
+
widgetState: null,
|
|
9
|
+
setWidgetState: vi.fn().mockResolvedValue(undefined),
|
|
10
|
+
};
|
|
11
|
+
vi.stubGlobal("openai", OpenaiMock);
|
|
12
|
+
});
|
|
13
|
+
afterEach(() => {
|
|
14
|
+
vi.unstubAllGlobals();
|
|
15
|
+
vi.resetAllMocks();
|
|
16
|
+
});
|
|
17
|
+
it("should create a store without default state", () => {
|
|
18
|
+
const storeCreator = () => ({
|
|
19
|
+
count: 0,
|
|
20
|
+
});
|
|
21
|
+
const store = createStore(storeCreator);
|
|
22
|
+
expect(store.getState()).toEqual({ count: 0 });
|
|
23
|
+
});
|
|
24
|
+
it("should create a store with default state", () => {
|
|
25
|
+
const storeCreator = () => ({
|
|
26
|
+
count: 0,
|
|
27
|
+
name: "initial",
|
|
28
|
+
});
|
|
29
|
+
const defaultState = { count: 5, name: "default" };
|
|
30
|
+
const store = createStore(storeCreator, defaultState);
|
|
31
|
+
expect(store.getState()).toEqual({ count: 5, name: "default" });
|
|
32
|
+
});
|
|
33
|
+
it("should initialize from window.openai.widgetState when available", () => {
|
|
34
|
+
const storeCreator = () => ({
|
|
35
|
+
count: 0,
|
|
36
|
+
name: "initial",
|
|
37
|
+
});
|
|
38
|
+
const windowState = { count: 20, name: "window" };
|
|
39
|
+
OpenaiMock.widgetState = windowState;
|
|
40
|
+
const store = createStore(storeCreator);
|
|
41
|
+
expect(store.getState()).toEqual({ count: 20, name: "window" });
|
|
42
|
+
});
|
|
43
|
+
it("should persist state changes to window.openai.setWidgetState", async () => {
|
|
44
|
+
const storeCreator = (set) => ({
|
|
45
|
+
count: 0,
|
|
46
|
+
increment: () => set((state) => ({ count: state.count + 1 })),
|
|
47
|
+
});
|
|
48
|
+
const store = createStore(storeCreator);
|
|
49
|
+
store.getState().increment();
|
|
50
|
+
await vi.waitFor(() => {
|
|
51
|
+
expect(OpenaiMock.setWidgetState).toHaveBeenCalled();
|
|
52
|
+
});
|
|
53
|
+
const callArgs = OpenaiMock.setWidgetState.mock.calls[0]?.[0];
|
|
54
|
+
expect(callArgs).toEqual({ count: 1 });
|
|
55
|
+
});
|
|
56
|
+
it("should filter widget context from initial state", () => {
|
|
57
|
+
const storeCreator = () => ({
|
|
58
|
+
count: 0,
|
|
59
|
+
});
|
|
60
|
+
const windowState = {
|
|
61
|
+
count: 5,
|
|
62
|
+
[WIDGET_CONTEXT_KEY]: "context-value",
|
|
63
|
+
};
|
|
64
|
+
OpenaiMock.widgetState = windowState;
|
|
65
|
+
const store = createStore(storeCreator);
|
|
66
|
+
expect(store.getState()).toEqual({ count: 5 });
|
|
67
|
+
expect(store.getState()[WIDGET_CONTEXT_KEY]).toBeUndefined();
|
|
68
|
+
});
|
|
69
|
+
});
|
|
70
|
+
//# sourceMappingURL=create-store.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"create-store.test.js","sourceRoot":"","sources":["../../../src/web/create-store.test.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,SAAS,EACT,UAAU,EACV,QAAQ,EACR,MAAM,EACN,EAAE,EAEF,EAAE,GACH,MAAM,QAAQ,CAAC;AAEhB,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,EAAE,kBAAkB,EAAE,MAAM,eAAe,CAAC;AAEnD,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;IAC3B,IAAI,UAGH,CAAC;IAEF,UAAU,CAAC,GAAG,EAAE;QACd,UAAU,GAAG;YACX,WAAW,EAAE,IAAI;YACjB,cAAc,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;SACrD,CAAC;QACF,EAAE,CAAC,UAAU,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;IACtC,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,gBAAgB,EAAE,CAAC;QACtB,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;QAErD,MAAM,YAAY,GAA+C,GAAG,EAAE,CAAC,CAAC;YACtE,KAAK,EAAE,CAAC;SACT,CAAC,CAAC;QAEH,MAAM,KAAK,GAAG,WAAW,CAAC,YAAY,CAAC,CAAC;QAExC,MAAM,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC;IACjD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;QAElD,MAAM,YAAY,GAA+C,GAAG,EAAE,CAAC,CAAC;YACtE,KAAK,EAAE,CAAC;YACR,IAAI,EAAE,SAAS;SAChB,CAAC,CAAC;QACH,MAAM,YAAY,GAAG,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;QAEnD,MAAM,KAAK,GAAG,WAAW,CAAC,YAAY,EAAE,YAAY,CAAC,CAAC;QAEtD,MAAM,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,CAAC;IAClE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iEAAiE,EAAE,GAAG,EAAE;QAEzE,MAAM,YAAY,GAA+C,GAAG,EAAE,CAAC,CAAC;YACtE,KAAK,EAAE,CAAC;YACR,IAAI,EAAE,SAAS;SAChB,CAAC,CAAC;QACH,MAAM,WAAW,GAAG,EAAE,KAAK,EAAE,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;QAClD,UAAU,CAAC,WAAW,GAAG,WAAW,CAAC;QAErC,MAAM,KAAK,GAAG,WAAW,CAAC,YAAY,CAAC,CAAC;QAExC,MAAM,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAC;IAClE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8DAA8D,EAAE,KAAK,IAAI,EAAE;QAE5E,MAAM,YAAY,GAA+C,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;YACzE,KAAK,EAAE,CAAC;YACR,SAAS,EAAE,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,GAAG,CAAC,EAAE,CAAC,CAAC;SAC9D,CAAC,CAAC;QAEH,MAAM,KAAK,GAAG,WAAW,CAAC,YAAY,CAAC,CAAC;QACxC,KAAK,CAAC,QAAQ,EAAE,CAAC,SAAS,EAAE,CAAC;QAE7B,MAAM,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE;YACpB,MAAM,CAAC,UAAU,CAAC,cAAc,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACvD,CAAC,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,UAAU,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QAC9D,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;QAEzD,MAAM,YAAY,GAA+C,GAAG,EAAE,CAAC,CAAC;YACtE,KAAK,EAAE,CAAC;SACT,CAAC,CAAC;QACH,MAAM,WAAW,GAAG;YAClB,KAAK,EAAE,CAAC;YACR,CAAC,kBAAkB,CAAC,EAAE,eAAe;SACtC,CAAC;QACF,UAAU,CAAC,WAAW,GAAG,WAAW,CAAC;QAErC,MAAM,KAAK,GAAG,WAAW,CAAC,YAAY,CAAC,CAAC;QAExC,MAAM,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC;QAC/C,MAAM,CACH,KAAK,CAAC,QAAQ,EAA8B,CAAC,kBAAkB,CAAC,CAClE,CAAC,aAAa,EAAE,CAAC;IACpB,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1,14 @@
|
|
|
1
|
+
import { type ReactNode } from "react";
|
|
2
|
+
export type DataLLMContent = string;
|
|
3
|
+
export interface DataLLMNode {
|
|
4
|
+
id: string;
|
|
5
|
+
parentId: string | null;
|
|
6
|
+
content: string | null;
|
|
7
|
+
}
|
|
8
|
+
export declare const WIDGET_CONTEXT_KEY: "__widget_context";
|
|
9
|
+
interface DataLLMProps {
|
|
10
|
+
content: DataLLMContent | null | undefined;
|
|
11
|
+
children?: ReactNode;
|
|
12
|
+
}
|
|
13
|
+
export declare function DataLLM({ content, children }: DataLLMProps): import("react/jsx-runtime").JSX.Element;
|
|
14
|
+
export {};
|
|
@@ -0,0 +1,68 @@
|
|
|
1
|
+
import { jsx as _jsx } from "react/jsx-runtime";
|
|
2
|
+
import { createContext, useContext, useEffect, useId, } from "react";
|
|
3
|
+
export const WIDGET_CONTEXT_KEY = "__widget_context";
|
|
4
|
+
const nodes = new Map();
|
|
5
|
+
function setNode(node) {
|
|
6
|
+
nodes.set(node.id, node);
|
|
7
|
+
onChange();
|
|
8
|
+
}
|
|
9
|
+
function removeNode(id) {
|
|
10
|
+
nodes.delete(id);
|
|
11
|
+
onChange();
|
|
12
|
+
}
|
|
13
|
+
function onChange() {
|
|
14
|
+
const description = getLLMDescriptionString();
|
|
15
|
+
window.openai.setWidgetState({
|
|
16
|
+
...window.openai.widgetState,
|
|
17
|
+
[WIDGET_CONTEXT_KEY]: description,
|
|
18
|
+
});
|
|
19
|
+
}
|
|
20
|
+
const ParentIdContext = createContext(null);
|
|
21
|
+
export function DataLLM({ content, children }) {
|
|
22
|
+
const parentId = useContext(ParentIdContext);
|
|
23
|
+
const id = useId();
|
|
24
|
+
useEffect(() => {
|
|
25
|
+
if (content) {
|
|
26
|
+
setNode({
|
|
27
|
+
id,
|
|
28
|
+
parentId,
|
|
29
|
+
content,
|
|
30
|
+
});
|
|
31
|
+
}
|
|
32
|
+
else {
|
|
33
|
+
removeNode(id);
|
|
34
|
+
}
|
|
35
|
+
return () => {
|
|
36
|
+
removeNode(id);
|
|
37
|
+
};
|
|
38
|
+
}, [id, parentId, content]);
|
|
39
|
+
return (_jsx(ParentIdContext.Provider, { value: id, children: children }));
|
|
40
|
+
}
|
|
41
|
+
function getLLMDescriptionString() {
|
|
42
|
+
const byParent = new Map();
|
|
43
|
+
for (const node of Array.from(nodes.values())) {
|
|
44
|
+
const key = node.parentId ?? null;
|
|
45
|
+
if (!byParent.has(key))
|
|
46
|
+
byParent.set(key, []);
|
|
47
|
+
byParent.get(key)?.push(node);
|
|
48
|
+
}
|
|
49
|
+
for (const list of byParent.values()) {
|
|
50
|
+
list.sort((a, b) => a.id.localeCompare(b.id));
|
|
51
|
+
}
|
|
52
|
+
const lines = [];
|
|
53
|
+
function traverseTree(parentId, depth) {
|
|
54
|
+
const children = byParent.get(parentId);
|
|
55
|
+
if (!children)
|
|
56
|
+
return;
|
|
57
|
+
for (const child of children) {
|
|
58
|
+
if (child.content?.trim()) {
|
|
59
|
+
const indent = " ".repeat(depth);
|
|
60
|
+
lines.push(`${indent}- ${child.content.trim()}`);
|
|
61
|
+
}
|
|
62
|
+
traverseTree(child.id, depth + 1);
|
|
63
|
+
}
|
|
64
|
+
}
|
|
65
|
+
traverseTree(null, 0);
|
|
66
|
+
return lines.join("\n");
|
|
67
|
+
}
|
|
68
|
+
//# sourceMappingURL=data-llm.js.map
|
|
@@ -0,0 +1 @@
|
|
|
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;AAUf,MAAM,CAAC,MAAM,kBAAkB,GAAG,kBAA2B,CAAC;AAE9D,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,CAAC,MAAM,CAAC,cAAc,CAAC;QAC3B,GAAG,MAAM,CAAC,MAAM,CAAC,WAAW;QAC5B,CAAC,kBAAkB,CAAC,EAAE,WAAW;KAClC,CAAC,CAAC;AACL,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;YAAE,QAAQ,CAAC,GAAG,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;QAC9C,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;YAAE,OAAO;QAEtB,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"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
export {};
|
|
@@ -0,0 +1,76 @@
|
|
|
1
|
+
import { jsx as _jsx, jsxs as _jsxs } from "react/jsx-runtime";
|
|
2
|
+
import { render } from "@testing-library/react";
|
|
3
|
+
import { afterEach, beforeEach, describe, expect, it, vi, } from "vitest";
|
|
4
|
+
import { DataLLM } from "./data-llm.js";
|
|
5
|
+
describe("DataLLM", () => {
|
|
6
|
+
let OpenaiMock;
|
|
7
|
+
beforeEach(() => {
|
|
8
|
+
OpenaiMock = {
|
|
9
|
+
widgetState: {},
|
|
10
|
+
setWidgetState: vi.fn(),
|
|
11
|
+
};
|
|
12
|
+
// Use Object.defineProperty to ensure it persists
|
|
13
|
+
Object.defineProperty(globalThis, "openai", {
|
|
14
|
+
value: OpenaiMock,
|
|
15
|
+
writable: true,
|
|
16
|
+
configurable: true,
|
|
17
|
+
});
|
|
18
|
+
// Also set on window for browser-like environment
|
|
19
|
+
if (typeof window !== "undefined") {
|
|
20
|
+
Object.defineProperty(window, "openai", {
|
|
21
|
+
value: OpenaiMock,
|
|
22
|
+
writable: true,
|
|
23
|
+
configurable: true,
|
|
24
|
+
});
|
|
25
|
+
}
|
|
26
|
+
vi.stubGlobal("openai", OpenaiMock);
|
|
27
|
+
});
|
|
28
|
+
afterEach(() => {
|
|
29
|
+
vi.resetAllMocks();
|
|
30
|
+
// Keep the mock available for React cleanup, but reset it
|
|
31
|
+
if (typeof window !== "undefined" && window.openai) {
|
|
32
|
+
window.openai.setWidgetState = vi.fn();
|
|
33
|
+
window.openai.widgetState = {};
|
|
34
|
+
}
|
|
35
|
+
});
|
|
36
|
+
it("should register a node with content and call setWidgetState", () => {
|
|
37
|
+
render(_jsx(DataLLM, { content: "Test content", children: _jsx("div", { children: "Child" }) }));
|
|
38
|
+
expect(OpenaiMock.setWidgetState).toHaveBeenCalled();
|
|
39
|
+
const callArgs = OpenaiMock.setWidgetState.mock.calls[0]?.[0];
|
|
40
|
+
expect(callArgs).toHaveProperty("__widget_context");
|
|
41
|
+
expect(callArgs?.__widget_context).toContain("- Test content");
|
|
42
|
+
});
|
|
43
|
+
it("should preserve existing widgetState when updating context", () => {
|
|
44
|
+
OpenaiMock.widgetState = { existingKey: "existingValue" };
|
|
45
|
+
render(_jsx(DataLLM, { content: "Test content", children: _jsx("div", { children: "Child" }) }));
|
|
46
|
+
const callArgs = OpenaiMock.setWidgetState.mock.calls[0]?.[0];
|
|
47
|
+
expect(callArgs).toHaveProperty("existingKey", "existingValue");
|
|
48
|
+
expect(callArgs).toHaveProperty("__widget_context");
|
|
49
|
+
});
|
|
50
|
+
it("should handle deeply nested DataLLM components", () => {
|
|
51
|
+
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" }) }) })] }));
|
|
52
|
+
const callArgs = OpenaiMock.setWidgetState.mock.calls[OpenaiMock.setWidgetState.mock.calls.length - 1]?.[0];
|
|
53
|
+
const context = callArgs?.__widget_context;
|
|
54
|
+
expect(context).toContain("- Level 1");
|
|
55
|
+
expect(context).toContain(" - Level 2A");
|
|
56
|
+
expect(context).toContain(" - Level 2B");
|
|
57
|
+
expect(context).toContain(" - Level 3");
|
|
58
|
+
});
|
|
59
|
+
it("should update context when content changes", () => {
|
|
60
|
+
const { rerender } = render(_jsx(DataLLM, { content: "Initial content", children: _jsx("div", { children: "Child" }) }));
|
|
61
|
+
const initialCalls = OpenaiMock.setWidgetState.mock.calls.length;
|
|
62
|
+
rerender(_jsx(DataLLM, { content: "Updated content", children: _jsx("div", { children: "Child" }) }));
|
|
63
|
+
expect(OpenaiMock.setWidgetState.mock.calls.length).toBeGreaterThan(initialCalls);
|
|
64
|
+
const lastCallArgs = OpenaiMock.setWidgetState.mock.calls[OpenaiMock.setWidgetState.mock.calls.length - 1]?.[0];
|
|
65
|
+
expect(lastCallArgs?.__widget_context).toContain("- Updated content");
|
|
66
|
+
});
|
|
67
|
+
it("should remove node and update context when component unmounts", () => {
|
|
68
|
+
const { unmount } = render(_jsx(DataLLM, { content: "Content to remove", children: _jsx("div", { children: "Child" }) }));
|
|
69
|
+
const callsBeforeUnmount = OpenaiMock.setWidgetState.mock.calls.length;
|
|
70
|
+
unmount();
|
|
71
|
+
expect(OpenaiMock.setWidgetState.mock.calls.length).toBeGreaterThan(callsBeforeUnmount);
|
|
72
|
+
const lastCallArgs = OpenaiMock.setWidgetState.mock.calls[OpenaiMock.setWidgetState.mock.calls.length - 1]?.[0];
|
|
73
|
+
expect(lastCallArgs?.__widget_context).not.toContain("Content to remove");
|
|
74
|
+
});
|
|
75
|
+
});
|
|
76
|
+
//# sourceMappingURL=data-llm.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"data-llm.test.js","sourceRoot":"","sources":["../../../src/web/data-llm.test.tsx"],"names":[],"mappings":";AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,wBAAwB,CAAC;AAChD,OAAO,EACL,SAAS,EACT,UAAU,EACV,QAAQ,EACR,MAAM,EACN,EAAE,EAEF,EAAE,GACH,MAAM,QAAQ,CAAC;AAChB,OAAO,EAAE,OAAO,EAAE,MAAM,eAAe,CAAC;AAExC,QAAQ,CAAC,SAAS,EAAE,GAAG,EAAE;IACvB,IAAI,UAA0D,CAAC;IAE/D,UAAU,CAAC,GAAG,EAAE;QACd,UAAU,GAAG;YACX,WAAW,EAAE,EAAE;YACf,cAAc,EAAE,EAAE,CAAC,EAAE,EAAE;SACxB,CAAC;QACF,kDAAkD;QAClD,MAAM,CAAC,cAAc,CAAC,UAAU,EAAE,QAAQ,EAAE;YAC1C,KAAK,EAAE,UAAU;YACjB,QAAQ,EAAE,IAAI;YACd,YAAY,EAAE,IAAI;SACnB,CAAC,CAAC;QACH,kDAAkD;QAClD,IAAI,OAAO,MAAM,KAAK,WAAW,EAAE,CAAC;YAClC,MAAM,CAAC,cAAc,CAAC,MAAM,EAAE,QAAQ,EAAE;gBACtC,KAAK,EAAE,UAAU;gBACjB,QAAQ,EAAE,IAAI;gBACd,YAAY,EAAE,IAAI;aACnB,CAAC,CAAC;QACL,CAAC;QACD,EAAE,CAAC,UAAU,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;IACtC,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,0DAA0D;QAC1D,IAAI,OAAO,MAAM,KAAK,WAAW,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;YACnD,MAAM,CAAC,MAAM,CAAC,cAAc,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YACvC,MAAM,CAAC,MAAM,CAAC,WAAW,GAAG,EAAE,CAAC;QACjC,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6DAA6D,EAAE,GAAG,EAAE;QACrE,MAAM,CACJ,KAAC,OAAO,IAAC,OAAO,EAAC,cAAc,YAC7B,kCAAgB,GACR,CACX,CAAC;QAEF,MAAM,CAAC,UAAU,CAAC,cAAc,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACrD,MAAM,QAAQ,GAAG,UAAU,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QAC9D,MAAM,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,kBAAkB,CAAC,CAAC;QACpD,MAAM,CAAC,QAAQ,EAAE,gBAAgB,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;IACjE,CAAC,CAAC,CAAC;IACH,EAAE,CAAC,4DAA4D,EAAE,GAAG,EAAE;QACpE,UAAU,CAAC,WAAW,GAAG,EAAE,WAAW,EAAE,eAAe,EAAE,CAAC;QAE1D,MAAM,CACJ,KAAC,OAAO,IAAC,OAAO,EAAC,cAAc,YAC7B,kCAAgB,GACR,CACX,CAAC;QAEF,MAAM,QAAQ,GAAG,UAAU,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QAC9D,MAAM,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,aAAa,EAAE,eAAe,CAAC,CAAC;QAChE,MAAM,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,kBAAkB,CAAC,CAAC;IACtD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;QACxD,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;QAEF,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,CAAC;QACT,MAAM,OAAO,GAAG,QAAQ,EAAE,gBAA0B,CAAC;QACrD,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;QACvC,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;QAC1C,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;QAC1C,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC;IAC7C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;QACpD,MAAM,EAAE,QAAQ,EAAE,GAAG,MAAM,CACzB,KAAC,OAAO,IAAC,OAAO,EAAC,iBAAiB,YAChC,kCAAgB,GACR,CACX,CAAC;QAEF,MAAM,YAAY,GAAG,UAAU,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC;QAEjE,QAAQ,CACN,KAAC,OAAO,IAAC,OAAO,EAAC,iBAAiB,YAChC,kCAAgB,GACR,CACX,CAAC;QAEF,MAAM,CAAC,UAAU,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,eAAe,CACjE,YAAY,CACb,CAAC;QACF,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,CAAC;QACT,MAAM,CAAC,YAAY,EAAE,gBAAgB,CAAC,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAC;IACxE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+DAA+D,EAAE,GAAG,EAAE;QACvE,MAAM,EAAE,OAAO,EAAE,GAAG,MAAM,CACxB,KAAC,OAAO,IAAC,OAAO,EAAC,mBAAmB,YAClC,kCAAgB,GACR,CACX,CAAC;QAEF,MAAM,kBAAkB,GAAG,UAAU,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC;QAEvE,OAAO,EAAE,CAAC;QAEV,MAAM,CAAC,UAAU,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,eAAe,CACjE,kBAAkB,CACnB,CAAC;QACF,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,CAAC;QACT,MAAM,CAAC,YAAY,EAAE,gBAAgB,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAC;IAC5E,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1,115 @@
|
|
|
1
|
+
import type { InferTools, ToolInput, ToolOutput, ToolResponseMetadata } from "../server/index.js";
|
|
2
|
+
import { type CallToolAsyncFn, type CallToolFn, type CallToolState } from "./hooks/use-call-tool.js";
|
|
3
|
+
import { type ToolState } from "./hooks/use-tool-info.js";
|
|
4
|
+
import type { CallToolResponse, Objectify, Prettify } from "./types.js";
|
|
5
|
+
type TypedCallToolResponse<TOutput> = CallToolResponse & {
|
|
6
|
+
structuredContent: TOutput;
|
|
7
|
+
};
|
|
8
|
+
type TypedCallToolReturn<TInput, TOutput> = Prettify<CallToolState<TypedCallToolResponse<TOutput>> & {
|
|
9
|
+
callTool: CallToolFn<TInput, TypedCallToolResponse<TOutput>>;
|
|
10
|
+
callToolAsync: CallToolAsyncFn<TInput, TypedCallToolResponse<TOutput>>;
|
|
11
|
+
}>;
|
|
12
|
+
type TypedToolInfoReturn<TInput, TOutput, TResponseMetadata> = ToolState<Objectify<TInput>, Objectify<TOutput>, Objectify<TResponseMetadata>>;
|
|
13
|
+
/**
|
|
14
|
+
* Creates typed versions of skybridge hooks with full type inference
|
|
15
|
+
* for tool names, inputs, and outputs.
|
|
16
|
+
*
|
|
17
|
+
* This is the recommended way to use skybridge hooks in your widgets.
|
|
18
|
+
* Set this up once in a dedicated file and export the typed hooks for use across your app.
|
|
19
|
+
*
|
|
20
|
+
* @typeParam ServerType - The type of your McpServer instance (use `typeof server`).
|
|
21
|
+
* Must be a server instance created with method chaining.
|
|
22
|
+
* TypeScript will validate that tools can be inferred from this type.
|
|
23
|
+
*
|
|
24
|
+
* @example
|
|
25
|
+
* ```typescript
|
|
26
|
+
* // server/src/index.ts
|
|
27
|
+
* const server = new McpServer({ name: "my-app", version: "1.0" }, {})
|
|
28
|
+
* .widget("search-voyage", {}, {
|
|
29
|
+
* inputSchema: { destination: z.string() },
|
|
30
|
+
* outputSchema: { results: z.array(z.string()) },
|
|
31
|
+
* }, async ({ destination }) => {
|
|
32
|
+
* return { content: [{ type: "text", text: `Found trips to ${destination}` }] };
|
|
33
|
+
* });
|
|
34
|
+
*
|
|
35
|
+
* export type AppType = typeof server;
|
|
36
|
+
* ```
|
|
37
|
+
*
|
|
38
|
+
* @example
|
|
39
|
+
* ```typescript
|
|
40
|
+
* // web/src/skybridge.ts (one-time setup)
|
|
41
|
+
* import type { AppType } from "../server";
|
|
42
|
+
* import { generateHelpers } from "skybridge/web";
|
|
43
|
+
*
|
|
44
|
+
* export const { useCallTool, useToolInfo } = generateHelpers<AppType>();
|
|
45
|
+
* ```
|
|
46
|
+
*
|
|
47
|
+
* @example
|
|
48
|
+
* ```typescript
|
|
49
|
+
* // web/src/widgets/search.tsx (usage)
|
|
50
|
+
* import { useCallTool, useToolInfo } from "../skybridge";
|
|
51
|
+
*
|
|
52
|
+
* export function SearchWidget() {
|
|
53
|
+
* const { callTool, data } = useCallTool("search-voyage");
|
|
54
|
+
* // ^ autocomplete for tool names
|
|
55
|
+
* callTool({ destination: "Spain" });
|
|
56
|
+
* // ^ autocomplete for input fields
|
|
57
|
+
*
|
|
58
|
+
* const toolInfo = useToolInfo<"search-voyage">();
|
|
59
|
+
* // ^ autocomplete for widget names
|
|
60
|
+
* // toolInfo.input is typed based on widget input schema
|
|
61
|
+
* // toolInfo.output.structuredContent is typed based on widget output schema
|
|
62
|
+
* }
|
|
63
|
+
* ```
|
|
64
|
+
*/
|
|
65
|
+
export declare function generateHelpers<ServerType = never>(): {
|
|
66
|
+
/**
|
|
67
|
+
* Typed version of `useCallTool` that provides autocomplete for tool names
|
|
68
|
+
* and type inference for inputs and outputs.
|
|
69
|
+
*
|
|
70
|
+
* @param name - The name of the widget to call. Autocompletes based on your server's widget registry.
|
|
71
|
+
* @returns A hook with typed `callTool` function and `data` property.
|
|
72
|
+
*
|
|
73
|
+
* @example
|
|
74
|
+
* ```typescript
|
|
75
|
+
* const { callTool, data, isPending } = useCallTool("search-voyage");
|
|
76
|
+
* // TypeScript knows callTool expects { destination: string }
|
|
77
|
+
* callTool({ destination: "Spain" });
|
|
78
|
+
*
|
|
79
|
+
* // data.structuredContent is typed based on your outputSchema
|
|
80
|
+
* if (data) {
|
|
81
|
+
* console.log(data.structuredContent.results);
|
|
82
|
+
* }
|
|
83
|
+
* ```
|
|
84
|
+
*/
|
|
85
|
+
useCallTool: <ToolName extends keyof InferTools<ServerType> & string>(name: ToolName) => TypedCallToolReturn<ToolInput<ServerType, ToolName>, ToolOutput<ServerType, ToolName>>;
|
|
86
|
+
/**
|
|
87
|
+
* Typed version of `useToolInfo` that provides autocomplete for widget names
|
|
88
|
+
* and type inference for inputs, outputs, and responseMetadata.
|
|
89
|
+
*
|
|
90
|
+
* @typeParam K - The name of the widget. Autocompletes based on your server's widget registry.
|
|
91
|
+
* @returns A discriminated union with `status: "pending" | "success"` that narrows correctly.
|
|
92
|
+
*
|
|
93
|
+
* @example
|
|
94
|
+
* ```typescript
|
|
95
|
+
* const toolInfo = useToolInfo<"search-voyage">();
|
|
96
|
+
* // toolInfo.input is typed as { destination: string; ... }
|
|
97
|
+
* // toolInfo.output is typed as { results: Array<...>; ... }
|
|
98
|
+
* // toolInfo.responseMetadata is typed based on _meta in callback return
|
|
99
|
+
* // toolInfo.status narrows correctly: "pending" | "success"
|
|
100
|
+
*
|
|
101
|
+
* if (toolInfo.isPending) {
|
|
102
|
+
* // TypeScript knows output and responseMetadata are undefined here
|
|
103
|
+
* console.log(toolInfo.input.destination);
|
|
104
|
+
* }
|
|
105
|
+
*
|
|
106
|
+
* if (toolInfo.isSuccess) {
|
|
107
|
+
* // TypeScript knows output and responseMetadata are defined here
|
|
108
|
+
* console.log(toolInfo.output.results);
|
|
109
|
+
* console.log(toolInfo.responseMetadata);
|
|
110
|
+
* }
|
|
111
|
+
* ```
|
|
112
|
+
*/
|
|
113
|
+
useToolInfo: <ToolName extends keyof InferTools<ServerType> & string>() => TypedToolInfoReturn<ToolInput<ServerType, ToolName>, ToolOutput<ServerType, ToolName>, ToolResponseMetadata<ServerType, ToolName>>;
|
|
114
|
+
};
|
|
115
|
+
export {};
|