skybridge 0.0.0-dev.ff49e24 → 0.0.0

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.
Files changed (169) hide show
  1. package/LICENSE +21 -674
  2. package/dist/src/server/devtoolsStaticServer.d.ts +15 -0
  3. package/dist/src/server/devtoolsStaticServer.js +43 -0
  4. package/dist/src/server/devtoolsStaticServer.js.map +1 -0
  5. package/dist/src/server/index.d.ts +3 -0
  6. package/dist/src/server/index.js +1 -0
  7. package/dist/src/server/index.js.map +1 -1
  8. package/dist/src/server/inferUtilityTypes.d.ts +64 -0
  9. package/dist/src/server/inferUtilityTypes.js +2 -0
  10. package/dist/src/server/inferUtilityTypes.js.map +1 -0
  11. package/dist/src/server/server.d.ts +59 -9
  12. package/dist/src/server/server.js +26 -9
  13. package/dist/src/server/server.js.map +1 -1
  14. package/dist/src/server/templateHelper.d.ts +9 -8
  15. package/dist/src/server/templateHelper.js +5 -12
  16. package/dist/src/server/templateHelper.js.map +1 -1
  17. package/dist/src/server/templates/development.hbs +2 -1
  18. package/dist/src/server/templates/production.hbs +3 -2
  19. package/dist/src/server/widgetsDevServer.js +3 -3
  20. package/dist/src/server/widgetsDevServer.js.map +1 -1
  21. package/dist/src/test/utils.d.ts +97 -1
  22. package/dist/src/test/utils.js +182 -3
  23. package/dist/src/test/utils.js.map +1 -1
  24. package/dist/src/test/widget.test.js +45 -19
  25. package/dist/src/test/widget.test.js.map +1 -1
  26. package/dist/src/web/create-store.d.ts +3 -0
  27. package/dist/src/web/create-store.js +25 -0
  28. package/dist/src/web/create-store.js.map +1 -0
  29. package/dist/src/web/create-store.test.js +70 -0
  30. package/dist/src/web/create-store.test.js.map +1 -0
  31. package/dist/src/web/data-llm.d.ts +14 -0
  32. package/dist/src/web/data-llm.js +68 -0
  33. package/dist/src/web/data-llm.js.map +1 -0
  34. package/dist/src/web/data-llm.test.js +76 -0
  35. package/dist/src/web/data-llm.test.js.map +1 -0
  36. package/dist/src/web/generate-helpers.d.ts +115 -0
  37. package/dist/src/web/generate-helpers.js +111 -0
  38. package/dist/src/web/generate-helpers.js.map +1 -0
  39. package/dist/src/web/generate-helpers.test-d.d.ts +1 -0
  40. package/dist/src/web/generate-helpers.test-d.js +180 -0
  41. package/dist/src/web/generate-helpers.test-d.js.map +1 -0
  42. package/dist/src/web/generate-helpers.test.d.ts +1 -0
  43. package/dist/src/web/generate-helpers.test.js +17 -0
  44. package/dist/src/web/generate-helpers.test.js.map +1 -0
  45. package/dist/src/web/helpers/state.d.ts +7 -0
  46. package/dist/src/web/helpers/state.js +40 -0
  47. package/dist/src/web/helpers/state.js.map +1 -0
  48. package/dist/src/web/helpers/state.test.d.ts +1 -0
  49. package/dist/src/web/helpers/state.test.js +53 -0
  50. package/dist/src/web/helpers/state.test.js.map +1 -0
  51. package/dist/src/web/hooks/index.d.ts +12 -0
  52. package/dist/src/web/hooks/index.js +13 -0
  53. package/dist/src/web/hooks/index.js.map +1 -0
  54. package/dist/src/web/hooks/use-call-tool.d.ts +100 -0
  55. package/dist/src/web/{use-call-tool.js → hooks/use-call-tool.js} +26 -10
  56. package/dist/src/web/hooks/use-call-tool.js.map +1 -0
  57. package/dist/src/web/hooks/use-call-tool.test-d.d.ts +1 -0
  58. package/dist/src/web/hooks/use-call-tool.test-d.js +104 -0
  59. package/dist/src/web/hooks/use-call-tool.test-d.js.map +1 -0
  60. package/dist/src/web/hooks/use-call-tool.test.d.ts +1 -0
  61. package/dist/src/web/hooks/use-call-tool.test.js +165 -0
  62. package/dist/src/web/hooks/use-call-tool.test.js.map +1 -0
  63. package/dist/src/web/hooks/use-display-mode.d.ts +4 -0
  64. package/dist/src/web/hooks/use-display-mode.js +7 -0
  65. package/dist/src/web/hooks/use-display-mode.js.map +1 -0
  66. package/dist/src/web/hooks/use-display-mode.test.d.ts +1 -0
  67. package/dist/src/web/hooks/use-display-mode.test.js +40 -0
  68. package/dist/src/web/hooks/use-display-mode.test.js.map +1 -0
  69. package/dist/src/web/hooks/use-files.d.ts +10 -0
  70. package/dist/src/web/hooks/use-files.js +7 -0
  71. package/dist/src/web/hooks/use-files.js.map +1 -0
  72. package/dist/src/web/hooks/use-files.test.d.ts +1 -0
  73. package/dist/src/web/hooks/use-files.test.js +29 -0
  74. package/dist/src/web/hooks/use-files.test.js.map +1 -0
  75. package/dist/src/web/hooks/use-locale.d.ts +1 -0
  76. package/dist/src/web/hooks/use-locale.js +5 -0
  77. package/dist/src/web/hooks/use-locale.js.map +1 -0
  78. package/dist/src/web/hooks/use-locale.test.d.ts +1 -0
  79. package/dist/src/web/hooks/use-locale.test.js +21 -0
  80. package/dist/src/web/hooks/use-locale.test.js.map +1 -0
  81. package/dist/src/web/hooks/use-open-external.d.ts +1 -0
  82. package/dist/src/web/hooks/use-open-external.js +6 -0
  83. package/dist/src/web/hooks/use-open-external.js.map +1 -0
  84. package/dist/src/web/hooks/use-open-external.test.d.ts +1 -0
  85. package/dist/src/web/hooks/use-open-external.test.js +24 -0
  86. package/dist/src/web/hooks/use-open-external.test.js.map +1 -0
  87. package/dist/src/web/hooks/use-openai-global.d.ts +2 -0
  88. package/dist/src/web/{use-openai-global.js → hooks/use-openai-global.js} +7 -3
  89. package/dist/src/web/hooks/use-openai-global.js.map +1 -0
  90. package/dist/src/web/hooks/use-request-modal.d.ts +6 -0
  91. package/dist/src/web/hooks/use-request-modal.js +9 -0
  92. package/dist/src/web/hooks/use-request-modal.js.map +1 -0
  93. package/dist/src/web/hooks/use-request-modal.test.d.ts +1 -0
  94. package/dist/src/web/hooks/use-request-modal.test.js +24 -0
  95. package/dist/src/web/hooks/use-request-modal.test.js.map +1 -0
  96. package/dist/src/web/hooks/use-send-follow-up-message.d.ts +1 -0
  97. package/dist/src/web/hooks/use-send-follow-up-message.js +11 -0
  98. package/dist/src/web/hooks/use-send-follow-up-message.js.map +1 -0
  99. package/dist/src/web/hooks/use-theme.d.ts +1 -0
  100. package/dist/src/web/hooks/use-theme.js +5 -0
  101. package/dist/src/web/hooks/use-theme.js.map +1 -0
  102. package/dist/src/web/hooks/use-theme.test.d.ts +1 -0
  103. package/dist/src/web/hooks/use-theme.test.js +26 -0
  104. package/dist/src/web/hooks/use-theme.test.js.map +1 -0
  105. package/dist/src/web/hooks/use-tool-info.d.ts +25 -0
  106. package/dist/src/web/hooks/use-tool-info.js +20 -0
  107. package/dist/src/web/hooks/use-tool-info.js.map +1 -0
  108. package/dist/src/web/hooks/use-tool-info.test-d.d.ts +1 -0
  109. package/dist/src/web/hooks/use-tool-info.test-d.js +73 -0
  110. package/dist/src/web/hooks/use-tool-info.test-d.js.map +1 -0
  111. package/dist/src/web/hooks/use-tool-info.test.d.ts +1 -0
  112. package/dist/src/web/hooks/use-tool-info.test.js +59 -0
  113. package/dist/src/web/hooks/use-tool-info.test.js.map +1 -0
  114. package/dist/src/web/hooks/use-user-agent.d.ts +1 -0
  115. package/dist/src/web/hooks/use-user-agent.js +5 -0
  116. package/dist/src/web/hooks/use-user-agent.js.map +1 -0
  117. package/dist/src/web/hooks/use-user-agent.test.d.ts +1 -0
  118. package/dist/src/web/hooks/use-user-agent.test.js +31 -0
  119. package/dist/src/web/hooks/use-user-agent.test.js.map +1 -0
  120. package/dist/src/web/hooks/use-widget-state.d.ts +4 -0
  121. package/dist/src/web/hooks/use-widget-state.js +32 -0
  122. package/dist/src/web/hooks/use-widget-state.js.map +1 -0
  123. package/dist/src/web/hooks/use-widget-state.test.d.ts +1 -0
  124. package/dist/src/web/hooks/use-widget-state.test.js +60 -0
  125. package/dist/src/web/hooks/use-widget-state.test.js.map +1 -0
  126. package/dist/src/web/index.d.ts +6 -6
  127. package/dist/src/web/index.js +6 -6
  128. package/dist/src/web/index.js.map +1 -1
  129. package/dist/src/web/mount-widget.js +5 -0
  130. package/dist/src/web/mount-widget.js.map +1 -1
  131. package/dist/src/web/plugin/data-llm.test.d.ts +1 -0
  132. package/dist/src/web/plugin/data-llm.test.js +81 -0
  133. package/dist/src/web/plugin/data-llm.test.js.map +1 -0
  134. package/dist/src/web/{plugin.js → plugin/plugin.js} +7 -4
  135. package/dist/src/web/plugin/plugin.js.map +1 -0
  136. package/dist/src/web/plugin/transform-data-llm.d.ts +12 -0
  137. package/dist/src/web/plugin/transform-data-llm.js +93 -0
  138. package/dist/src/web/plugin/transform-data-llm.js.map +1 -0
  139. package/dist/src/web/plugin/transform-data-llm.test.d.ts +1 -0
  140. package/dist/src/web/plugin/transform-data-llm.test.js +81 -0
  141. package/dist/src/web/plugin/transform-data-llm.test.js.map +1 -0
  142. package/dist/src/web/proxy.d.ts +1 -0
  143. package/dist/src/web/proxy.js +48 -0
  144. package/dist/src/web/proxy.js.map +1 -0
  145. package/dist/src/web/types.d.ts +47 -17
  146. package/dist/src/web/types.js +1 -0
  147. package/dist/src/web/types.js.map +1 -1
  148. package/dist/vitest.config.js +0 -1
  149. package/dist/vitest.config.js.map +1 -1
  150. package/package.json +20 -14
  151. package/README.md +0 -198
  152. package/dist/src/test/setup.js +0 -9
  153. package/dist/src/test/setup.js.map +0 -1
  154. package/dist/src/web/plugin.js.map +0 -1
  155. package/dist/src/web/use-call-tool.d.ts +0 -54
  156. package/dist/src/web/use-call-tool.js.map +0 -1
  157. package/dist/src/web/use-call-tool.test.js +0 -66
  158. package/dist/src/web/use-call-tool.test.js.map +0 -1
  159. package/dist/src/web/use-openai-global.d.ts +0 -2
  160. package/dist/src/web/use-openai-global.js.map +0 -1
  161. package/dist/src/web/use-tool-output.d.ts +0 -3
  162. package/dist/src/web/use-tool-output.js +0 -5
  163. package/dist/src/web/use-tool-output.js.map +0 -1
  164. package/dist/src/web/use-tool-response-metadata.d.ts +0 -3
  165. package/dist/src/web/use-tool-response-metadata.js +0 -5
  166. package/dist/src/web/use-tool-response-metadata.js.map +0 -1
  167. /package/dist/src/{test/setup.d.ts → web/create-store.test.d.ts} +0 -0
  168. /package/dist/src/web/{use-call-tool.test.d.ts → data-llm.test.d.ts} +0 -0
  169. /package/dist/src/web/{plugin.d.ts → plugin/plugin.d.ts} +0 -0
@@ -1,5 +1,6 @@
1
1
  import { vi } from "vitest";
2
- import { McpServer, McpServer as McpServerBase } from "../server/server.js";
2
+ import * as z from "zod";
3
+ import { McpServer } from "../server/server.js";
3
4
  /**
4
5
  * Creates a real McpServer instance for testing
5
6
  */
@@ -10,14 +11,192 @@ export function createMockMcpServer() {
10
11
  version: "0.0.1",
11
12
  }, { capabilities: {} });
12
13
  // Mock the underlying methods to track calls
13
- const mockResource = vi.spyOn(server, "resource");
14
+ const mockRegisterResource = vi.spyOn(server, "registerResource");
14
15
  const mockRegisterTool = vi.spyOn(server, "registerTool");
15
16
  return {
16
17
  server,
17
- mockResource,
18
+ mockRegisterResource,
18
19
  mockRegisterTool,
19
20
  };
20
21
  }
22
+ export function createTestServer() {
23
+ return new McpServer({ name: "test-app", version: "1.0.0" }, {})
24
+ .registerWidget("search-voyage", {}, {
25
+ description: "Search for voyages",
26
+ inputSchema: {
27
+ destination: z.string(),
28
+ departureDate: z.string().optional(),
29
+ maxPrice: z.number().optional(),
30
+ },
31
+ outputSchema: {
32
+ results: z.array(z.object({
33
+ id: z.string(),
34
+ name: z.string(),
35
+ price: z.number(),
36
+ })),
37
+ totalCount: z.number(),
38
+ },
39
+ }, async ({ destination }) => {
40
+ return {
41
+ content: [{ type: "text", text: `Found trips to ${destination}` }],
42
+ structuredContent: {
43
+ results: [{ id: "1", name: "Trip", price: 1000 }],
44
+ totalCount: 1,
45
+ },
46
+ };
47
+ })
48
+ .registerWidget("get-trip-details", {}, {
49
+ description: "Get trip details",
50
+ inputSchema: {
51
+ tripId: z.string(),
52
+ },
53
+ outputSchema: {
54
+ name: z.string(),
55
+ description: z.string(),
56
+ images: z.array(z.string()),
57
+ },
58
+ }, async ({ tripId }) => {
59
+ return {
60
+ content: [{ type: "text", text: `Details for ${tripId}` }],
61
+ structuredContent: {
62
+ name: "Trip",
63
+ description: "A great trip",
64
+ images: ["image1.jpg"],
65
+ },
66
+ };
67
+ })
68
+ .registerWidget("no-input-widget", {}, {
69
+ description: "Widget with no input",
70
+ inputSchema: {},
71
+ outputSchema: {},
72
+ }, async () => {
73
+ return {
74
+ content: [{ type: "text", text: "No input needed" }],
75
+ structuredContent: {},
76
+ };
77
+ })
78
+ .registerWidget("inferred-output-widget", {}, {
79
+ description: "Widget with output inferred from callback",
80
+ inputSchema: {
81
+ query: z.string(),
82
+ },
83
+ }, async ({ query }) => {
84
+ return {
85
+ content: [{ type: "text", text: `Query: ${query}` }],
86
+ structuredContent: {
87
+ inferredResults: [{ id: "inferred-1", score: 0.95 }],
88
+ inferredCount: 1,
89
+ },
90
+ };
91
+ })
92
+ .registerTool("calculate-price", {
93
+ description: "Calculate trip price",
94
+ inputSchema: {
95
+ tripId: z.string(),
96
+ passengers: z.number(),
97
+ },
98
+ outputSchema: {
99
+ totalPrice: z.number(),
100
+ currency: z.string(),
101
+ },
102
+ }, async ({ tripId, passengers }) => {
103
+ return {
104
+ content: [{ type: "text", text: `Price for ${tripId}` }],
105
+ structuredContent: {
106
+ totalPrice: 1000 * passengers,
107
+ currency: "USD",
108
+ },
109
+ };
110
+ })
111
+ .registerTool("inferred-tool", {
112
+ description: "Tool with output inferred from callback",
113
+ inputSchema: {
114
+ itemId: z.string(),
115
+ },
116
+ }, async ({ itemId }) => {
117
+ return {
118
+ content: [{ type: "text", text: `Item: ${itemId}` }],
119
+ structuredContent: {
120
+ itemDetails: { name: "Inferred Item", available: true },
121
+ fetchedAt: "2024-01-01",
122
+ },
123
+ };
124
+ })
125
+ .registerWidget("widget-with-metadata", {}, {
126
+ description: "Widget that returns response metadata",
127
+ inputSchema: {
128
+ resourceId: z.string(),
129
+ },
130
+ }, async ({ resourceId }) => {
131
+ return {
132
+ content: [{ type: "text", text: `Resource: ${resourceId}` }],
133
+ structuredContent: {
134
+ data: { id: resourceId, loaded: true },
135
+ },
136
+ _meta: {
137
+ requestId: "req-123",
138
+ timestamp: 1704067200000,
139
+ cached: false,
140
+ },
141
+ };
142
+ })
143
+ .registerTool("tool-with-metadata", {
144
+ description: "Tool that returns response metadata",
145
+ inputSchema: {
146
+ query: z.string(),
147
+ },
148
+ }, async ({ query }) => {
149
+ return {
150
+ content: [{ type: "text", text: `Query: ${query}` }],
151
+ structuredContent: {
152
+ results: [query],
153
+ },
154
+ _meta: {
155
+ executionTime: 150,
156
+ source: "cache",
157
+ },
158
+ };
159
+ })
160
+ .registerWidget("widget-with-mixed-returns", {}, {
161
+ description: "Widget with mixed return paths (some with _meta, some without)",
162
+ inputSchema: {
163
+ shouldSucceed: z.boolean(),
164
+ },
165
+ }, async ({ shouldSucceed }) => {
166
+ if (!shouldSucceed) {
167
+ // Error path - no _meta
168
+ return {
169
+ content: [{ type: "text", text: "Error occurred" }],
170
+ structuredContent: { error: "Something went wrong" },
171
+ };
172
+ }
173
+ // Success path - has _meta
174
+ return {
175
+ content: [{ type: "text", text: "Success" }],
176
+ structuredContent: { data: "result" },
177
+ _meta: {
178
+ processedAt: 1704067200000,
179
+ region: "eu-west-1",
180
+ },
181
+ };
182
+ });
183
+ }
184
+ export function createMinimalTestServer() {
185
+ return new McpServer({ name: "test-app", version: "1.0.0" }, {}).registerWidget("search-voyage", {}, {
186
+ description: "Search for voyages",
187
+ inputSchema: {
188
+ destination: z.string(),
189
+ },
190
+ outputSchema: {
191
+ results: z.array(z.object({ id: z.string() })),
192
+ },
193
+ }, async ({ destination }) => {
194
+ return {
195
+ content: [{ type: "text", text: `Found trips to ${destination}` }],
196
+ structuredContent: { results: [{ id: "1" }] },
197
+ };
198
+ });
199
+ }
21
200
  /**
22
201
  * Mock extra parameter for resource callback
23
202
  */
@@ -1 +1 @@
1
- {"version":3,"file":"utils.js","sourceRoot":"","sources":["../../../src/test/utils.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,EAAE,EAAqB,MAAM,QAAQ,CAAC;AAC/C,OAAO,EAAE,SAAS,EAAE,SAAS,IAAI,aAAa,EAAE,MAAM,qBAAqB,CAAC;AAE5E;;GAEG;AACH,MAAM,UAAU,mBAAmB;IAKjC,mCAAmC;IACnC,MAAM,MAAM,GAAG,IAAI,SAAS,CAC1B;QACE,IAAI,EAAE,kBAAkB;QACxB,OAAO,EAAE,OAAO;KACjB,EACD,EAAE,YAAY,EAAE,EAAE,EAAE,CACrB,CAAC;IAEF,6CAA6C;IAC7C,MAAM,YAAY,GAAG,EAAE,CAAC,KAAK,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;IAClD,MAAM,gBAAgB,GAAG,EAAE,CAAC,KAAK,CAAC,MAAM,EAAE,cAAc,CAAC,CAAC;IAE1D,OAAO;QACL,MAAM;QACN,YAAY;QACZ,gBAAgB;KACjB,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,eAAe,CAAC,IAAY;IAC1C,OAAO;QACL,WAAW,EAAE;YACX,OAAO,EAAE,EAAE,IAAI,EAAE;SAClB;KACF,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,UAAU,CAAC,GAA2B;IACpD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;AAClC,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,YAAY;IAC1B,OAAO,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC;AAC9B,CAAC"}
1
+ {"version":3,"file":"utils.js","sourceRoot":"","sources":["../../../src/test/utils.ts"],"names":[],"mappings":"AAAA,OAAO,EAAqB,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC/C,OAAO,KAAK,CAAC,MAAM,KAAK,CAAC;AACzB,OAAO,EAAE,SAAS,EAAE,MAAM,qBAAqB,CAAC;AAEhD;;GAEG;AACH,MAAM,UAAU,mBAAmB;IAKjC,mCAAmC;IACnC,MAAM,MAAM,GAAG,IAAI,SAAS,CAC1B;QACE,IAAI,EAAE,kBAAkB;QACxB,OAAO,EAAE,OAAO;KACjB,EACD,EAAE,YAAY,EAAE,EAAE,EAAE,CACrB,CAAC;IAEF,6CAA6C;IAC7C,MAAM,oBAAoB,GAAG,EAAE,CAAC,KAAK,CAAC,MAAM,EAAE,kBAAkB,CAAC,CAAC;IAClE,MAAM,gBAAgB,GAAG,EAAE,CAAC,KAAK,CAAC,MAAM,EAAE,cAAc,CAAC,CAAC;IAE1D,OAAO;QACL,MAAM;QACN,oBAAoB;QACpB,gBAAgB;KACjB,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,gBAAgB;IAC9B,OAAO,IAAI,SAAS,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,OAAO,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC;SAC7D,cAAc,CACb,eAAe,EACf,EAAE,EACF;QACE,WAAW,EAAE,oBAAoB;QACjC,WAAW,EAAE;YACX,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE;YACvB,aAAa,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;YACpC,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;SAChC;QACD,YAAY,EAAE;YACZ,OAAO,EAAE,CAAC,CAAC,KAAK,CACd,CAAC,CAAC,MAAM,CAAC;gBACP,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE;gBACd,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;gBAChB,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE;aAClB,CAAC,CACH;YACD,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE;SACvB;KACF,EACD,KAAK,EAAE,EAAE,WAAW,EAAE,EAAE,EAAE;QACxB,OAAO;YACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,kBAAkB,WAAW,EAAE,EAAE,CAAC;YAClE,iBAAiB,EAAE;gBACjB,OAAO,EAAE,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;gBACjD,UAAU,EAAE,CAAC;aACd;SACF,CAAC;IACJ,CAAC,CACF;SACA,cAAc,CACb,kBAAkB,EAClB,EAAE,EACF;QACE,WAAW,EAAE,kBAAkB;QAC/B,WAAW,EAAE;YACX,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE;SACnB;QACD,YAAY,EAAE;YACZ,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;YAChB,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE;YACvB,MAAM,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;SAC5B;KACF,EACD,KAAK,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE;QACnB,OAAO;YACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,eAAe,MAAM,EAAE,EAAE,CAAC;YAC1D,iBAAiB,EAAE;gBACjB,IAAI,EAAE,MAAM;gBACZ,WAAW,EAAE,cAAc;gBAC3B,MAAM,EAAE,CAAC,YAAY,CAAC;aACvB;SACF,CAAC;IACJ,CAAC,CACF;SACA,cAAc,CACb,iBAAiB,EACjB,EAAE,EACF;QACE,WAAW,EAAE,sBAAsB;QACnC,WAAW,EAAE,EAAE;QACf,YAAY,EAAE,EAAE;KACjB,EACD,KAAK,IAAI,EAAE;QACT,OAAO;YACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,iBAAiB,EAAE,CAAC;YACpD,iBAAiB,EAAE,EAAE;SACtB,CAAC;IACJ,CAAC,CACF;SACA,cAAc,CACb,wBAAwB,EACxB,EAAE,EACF;QACE,WAAW,EAAE,2CAA2C;QACxD,WAAW,EAAE;YACX,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE;SAClB;KACF,EACD,KAAK,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE;QAClB,OAAO;YACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,KAAK,EAAE,EAAE,CAAC;YACpD,iBAAiB,EAAE;gBACjB,eAAe,EAAE,CAAC,EAAE,EAAE,EAAE,YAAY,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;gBACpD,aAAa,EAAE,CAAC;aACjB;SACF,CAAC;IACJ,CAAC,CACF;SACA,YAAY,CACX,iBAAiB,EACjB;QACE,WAAW,EAAE,sBAAsB;QACnC,WAAW,EAAE;YACX,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE;YAClB,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE;SACvB;QACD,YAAY,EAAE;YACZ,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE;YACtB,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE;SACrB;KACF,EACD,KAAK,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,EAAE,EAAE;QAC/B,OAAO;YACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,aAAa,MAAM,EAAE,EAAE,CAAC;YACxD,iBAAiB,EAAE;gBACjB,UAAU,EAAE,IAAI,GAAG,UAAU;gBAC7B,QAAQ,EAAE,KAAK;aAChB;SACF,CAAC;IACJ,CAAC,CACF;SACA,YAAY,CACX,eAAe,EACf;QACE,WAAW,EAAE,yCAAyC;QACtD,WAAW,EAAE;YACX,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE;SACnB;KACF,EACD,KAAK,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE;QACnB,OAAO;YACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,SAAS,MAAM,EAAE,EAAE,CAAC;YACpD,iBAAiB,EAAE;gBACjB,WAAW,EAAE,EAAE,IAAI,EAAE,eAAe,EAAE,SAAS,EAAE,IAAI,EAAE;gBACvD,SAAS,EAAE,YAAY;aACxB;SACF,CAAC;IACJ,CAAC,CACF;SACA,cAAc,CACb,sBAAsB,EACtB,EAAE,EACF;QACE,WAAW,EAAE,uCAAuC;QACpD,WAAW,EAAE;YACX,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE;SACvB;KACF,EACD,KAAK,EAAE,EAAE,UAAU,EAAE,EAAE,EAAE;QACvB,OAAO;YACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,aAAa,UAAU,EAAE,EAAE,CAAC;YAC5D,iBAAiB,EAAE;gBACjB,IAAI,EAAE,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,EAAE,IAAI,EAAE;aACvC;YACD,KAAK,EAAE;gBACL,SAAS,EAAE,SAAS;gBACpB,SAAS,EAAE,aAAa;gBACxB,MAAM,EAAE,KAAK;aACd;SACF,CAAC;IACJ,CAAC,CACF;SACA,YAAY,CACX,oBAAoB,EACpB;QACE,WAAW,EAAE,qCAAqC;QAClD,WAAW,EAAE;YACX,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE;SAClB;KACF,EACD,KAAK,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE;QAClB,OAAO;YACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,KAAK,EAAE,EAAE,CAAC;YACpD,iBAAiB,EAAE;gBACjB,OAAO,EAAE,CAAC,KAAK,CAAC;aACjB;YACD,KAAK,EAAE;gBACL,aAAa,EAAE,GAAG;gBAClB,MAAM,EAAE,OAAO;aAChB;SACF,CAAC;IACJ,CAAC,CACF;SACA,cAAc,CACb,2BAA2B,EAC3B,EAAE,EACF;QACE,WAAW,EACT,gEAAgE;QAClE,WAAW,EAAE;YACX,aAAa,EAAE,CAAC,CAAC,OAAO,EAAE;SAC3B;KACF,EACD,KAAK,EAAE,EAAE,aAAa,EAAE,EAAE,EAAE;QAC1B,IAAI,CAAC,aAAa,EAAE,CAAC;YACnB,wBAAwB;YACxB,OAAO;gBACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,gBAAgB,EAAE,CAAC;gBACnD,iBAAiB,EAAE,EAAE,KAAK,EAAE,sBAAsB,EAAE;aACrD,CAAC;QACJ,CAAC;QACD,2BAA2B;QAC3B,OAAO;YACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;YAC5C,iBAAiB,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;YACrC,KAAK,EAAE;gBACL,WAAW,EAAE,aAAa;gBAC1B,MAAM,EAAE,WAAW;aACpB;SACF,CAAC;IACJ,CAAC,CACF,CAAC;AACN,CAAC;AAED,MAAM,UAAU,uBAAuB;IACrC,OAAO,IAAI,SAAS,CAClB,EAAE,IAAI,EAAE,UAAU,EAAE,OAAO,EAAE,OAAO,EAAE,EACtC,EAAE,CACH,CAAC,cAAc,CACd,eAAe,EACf,EAAE,EACF;QACE,WAAW,EAAE,oBAAoB;QACjC,WAAW,EAAE;YACX,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE;SACxB;QACD,YAAY,EAAE;YACZ,OAAO,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;SAC/C;KACF,EACD,KAAK,EAAE,EAAE,WAAW,EAAE,EAAE,EAAE;QACxB,OAAO;YACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,kBAAkB,WAAW,EAAE,EAAE,CAAC;YAClE,iBAAiB,EAAE,EAAE,OAAO,EAAE,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,CAAC,EAAE;SAC9C,CAAC;IACJ,CAAC,CACF,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,eAAe,CAAC,IAAY;IAC1C,OAAO;QACL,WAAW,EAAE;YACX,OAAO,EAAE,EAAE,IAAI,EAAE;SAClB;KACF,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,UAAU,CAAC,GAA2B;IACpD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;AAClC,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,YAAY;IAC1B,OAAO,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC;AAC9B,CAAC"}
@@ -1,12 +1,35 @@
1
- import { describe, it, expect, vi, beforeEach, afterEach, } from "vitest";
2
- import { McpServer } from "../server/server.js";
3
- import { createMockMcpServer, createMockExtra, setTestEnv, resetTestEnv, } from "./utils.js";
4
- describe("McpServer.widget", () => {
1
+ import { afterEach, beforeEach, describe, expect, it, vi, } from "vitest";
2
+ import { createMockExtra, createMockMcpServer, resetTestEnv, setTestEnv, } from "./utils.js";
3
+ const mockManifest = {
4
+ "src/widgets/my-widget.tsx": { file: "my-widget.js" },
5
+ "style.css": { file: "style.css" },
6
+ };
7
+ vi.mock("node:fs", async () => {
8
+ const actual = await vi.importActual("node:fs");
9
+ const readFileSyncImpl = (path, ...args) => {
10
+ if (typeof path === "string" && path.includes("manifest.json")) {
11
+ return JSON.stringify(mockManifest);
12
+ }
13
+ // Type assertion needed because readFileSync has overloads with different parameter types
14
+ // Using @ts-expect-error because the overloads are complex and we're forwarding args
15
+ // @ts-expect-error - readFileSync overloads require complex type handling
16
+ return actual.readFileSync(path, ...args);
17
+ };
18
+ const readFileSync = vi.fn(readFileSyncImpl);
19
+ return {
20
+ readFileSync,
21
+ default: {
22
+ readFileSync,
23
+ },
24
+ };
25
+ });
26
+ describe("McpServer.registerWidget", () => {
5
27
  let server;
6
- let mockResource;
28
+ let mockRegisterResource;
7
29
  let mockRegisterTool;
8
30
  beforeEach(() => {
9
- ({ server, mockResource, mockRegisterTool } = createMockMcpServer());
31
+ ({ server, mockRegisterResource, mockRegisterTool } =
32
+ createMockMcpServer());
10
33
  });
11
34
  afterEach(() => {
12
35
  vi.clearAllMocks();
@@ -15,15 +38,17 @@ describe("McpServer.widget", () => {
15
38
  it("should generate correct HTML for development mode", async () => {
16
39
  setTestEnv({ NODE_ENV: "development" });
17
40
  const mockToolCallback = vi.fn();
18
- const mockResourceConfig = { description: "Test widget" };
41
+ const mockRegisterResourceConfig = { description: "Test widget" };
19
42
  const mockToolConfig = { description: "Test tool" };
20
- server.widget("my-widget", mockResourceConfig, mockToolConfig, mockToolCallback);
43
+ server.registerWidget("my-widget", mockRegisterResourceConfig, mockToolConfig, mockToolCallback);
21
44
  // Get the resource callback function
22
- const resourceCallback = mockResource.mock.calls[0]?.[3];
45
+ const resourceCallback = mockRegisterResource.mock
46
+ .calls[0]?.[3];
23
47
  expect(resourceCallback).toBeDefined();
24
48
  const serverUrl = "http://localhost:3000";
25
49
  const mockExtra = createMockExtra("__not_used__");
26
50
  const result = await resourceCallback(new URL("ui://widgets/my-widget.html"), mockExtra);
51
+ expect(mockRegisterTool).toHaveBeenCalled();
27
52
  expect(result).toEqual({
28
53
  contents: [
29
54
  {
@@ -34,18 +59,19 @@ describe("McpServer.widget", () => {
34
59
  ],
35
60
  });
36
61
  // Check development-specific content
37
- expect(result.contents[0]?.text).toContain(serverUrl + "/@react-refresh");
38
- expect(result.contents[0]?.text).toContain(serverUrl + "/@vite/client");
39
- expect(result.contents[0]?.text).toContain(serverUrl + "/src/widgets/my-widget.tsx");
62
+ expect(result.contents[0]?.text).toContain(`${serverUrl}/@react-refresh`);
63
+ expect(result.contents[0]?.text).toContain(`${serverUrl}/@vite/client`);
64
+ expect(result.contents[0]?.text).toContain(`${serverUrl}/src/widgets/my-widget.tsx`);
40
65
  });
41
66
  it("should generate correct HTML for production mode", async () => {
42
67
  setTestEnv({ NODE_ENV: "production" });
43
68
  const mockToolCallback = vi.fn();
44
- const mockResourceConfig = { description: "Test widget" };
69
+ const mockRegisterResourceConfig = { description: "Test widget" };
45
70
  const mockToolConfig = { description: "Test tool" };
46
- server.widget("my-widget", mockResourceConfig, mockToolConfig, mockToolCallback);
71
+ server.registerWidget("my-widget", mockRegisterResourceConfig, mockToolConfig, mockToolCallback);
47
72
  // Get the resource callback function
48
- const resourceCallback = mockResource.mock.calls[0]?.[3];
73
+ const resourceCallback = mockRegisterResource.mock
74
+ .calls[0]?.[3];
49
75
  expect(resourceCallback).toBeDefined();
50
76
  const serverUrl = "https://myapp.com";
51
77
  const mockExtra = createMockExtra(serverUrl);
@@ -60,10 +86,10 @@ describe("McpServer.widget", () => {
60
86
  ],
61
87
  });
62
88
  // Check production-specific content
63
- expect(result.contents[0]?.text).not.toContain(serverUrl + "@react-refresh");
64
- expect(result.contents[0]?.text).not.toContain(serverUrl + "@vite/client");
65
- expect(result.contents[0]?.text).toContain(serverUrl + "/assets/my-widget.js");
66
- expect(result.contents[0]?.text).toContain(serverUrl + "/assets/style.css");
89
+ expect(result.contents[0]?.text).not.toContain(`${serverUrl}@react-refresh`);
90
+ expect(result.contents[0]?.text).not.toContain(`${serverUrl}@vite/client`);
91
+ expect(result.contents[0]?.text).toContain(`${serverUrl}/assets/my-widget.js`);
92
+ expect(result.contents[0]?.text).toContain(`${serverUrl}/assets/style.css`);
67
93
  });
68
94
  });
69
95
  //# sourceMappingURL=widget.test.js.map
@@ -1 +1 @@
1
- {"version":3,"file":"widget.test.js","sourceRoot":"","sources":["../../../src/test/widget.test.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,QAAQ,EACR,EAAE,EACF,MAAM,EACN,EAAE,EACF,UAAU,EACV,SAAS,GAEV,MAAM,QAAQ,CAAC;AAChB,OAAO,EAAE,SAAS,EAAE,MAAM,qBAAqB,CAAC;AAChD,OAAO,EACL,mBAAmB,EACnB,eAAe,EACf,UAAU,EACV,YAAY,GACb,MAAM,YAAY,CAAC;AAEpB,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;IAChC,IAAI,MAAiB,CAAC;IACtB,IAAI,YAAiD,CAAC;IACtD,IAAI,gBAAyD,CAAC;IAE9D,UAAU,CAAC,GAAG,EAAE;QACd,CAAC,EAAE,MAAM,EAAE,YAAY,EAAE,gBAAgB,EAAE,GAAG,mBAAmB,EAAE,CAAC,CAAC;IACvE,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,YAAY,EAAE,CAAC;IACjB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mDAAmD,EAAE,KAAK,IAAI,EAAE;QACjE,UAAU,CAAC,EAAE,QAAQ,EAAE,aAAa,EAAE,CAAC,CAAC;QAExC,MAAM,gBAAgB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACjC,MAAM,kBAAkB,GAAG,EAAE,WAAW,EAAE,aAAa,EAAE,CAAC;QAC1D,MAAM,cAAc,GAAG,EAAE,WAAW,EAAE,WAAW,EAAE,CAAC;QAEpD,MAAM,CAAC,MAAM,CACX,WAAW,EACX,kBAAkB,EAClB,cAAc,EACd,gBAAgB,CACjB,CAAC;QAEF,qCAAqC;QACrC,MAAM,gBAAgB,GAAG,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAG/C,CAAC;QACT,MAAM,CAAC,gBAAgB,CAAC,CAAC,WAAW,EAAE,CAAC;QAEvC,MAAM,SAAS,GAAG,uBAAuB,CAAC;QAC1C,MAAM,SAAS,GAAG,eAAe,CAAC,cAAc,CAAC,CAAC;QAClD,MAAM,MAAM,GAAG,MAAM,gBAAgB,CACnC,IAAI,GAAG,CAAC,6BAA6B,CAAC,EACtC,SAAS,CACV,CAAC;QAEF,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;YACrB,QAAQ,EAAE;gBACR;oBACE,GAAG,EAAE,6BAA6B;oBAClC,QAAQ,EAAE,qBAAqB;oBAC/B,IAAI,EAAE,MAAM,CAAC,gBAAgB,CAAC,uBAAuB,CAAC;iBACvD;aACF;SACF,CAAC,CAAC;QAEH,qCAAqC;QACrC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,SAAS,CAAC,SAAS,GAAG,iBAAiB,CAAC,CAAC;QAC1E,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,SAAS,CAAC,SAAS,GAAG,eAAe,CAAC,CAAC;QACxE,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,SAAS,CACxC,SAAS,GAAG,4BAA4B,CACzC,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;QAChE,UAAU,CAAC,EAAE,QAAQ,EAAE,YAAY,EAAE,CAAC,CAAC;QAEvC,MAAM,gBAAgB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACjC,MAAM,kBAAkB,GAAG,EAAE,WAAW,EAAE,aAAa,EAAE,CAAC;QAC1D,MAAM,cAAc,GAAG,EAAE,WAAW,EAAE,WAAW,EAAE,CAAC;QAEpD,MAAM,CAAC,MAAM,CACX,WAAW,EACX,kBAAkB,EAClB,cAAc,EACd,gBAAgB,CACjB,CAAC;QAEF,qCAAqC;QACrC,MAAM,gBAAgB,GAAG,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAG/C,CAAC;QACT,MAAM,CAAC,gBAAgB,CAAC,CAAC,WAAW,EAAE,CAAC;QAEvC,MAAM,SAAS,GAAG,mBAAmB,CAAC;QACtC,MAAM,SAAS,GAAG,eAAe,CAAC,SAAS,CAAC,CAAC;QAC7C,MAAM,MAAM,GAAG,MAAM,gBAAiB,CACpC,IAAI,GAAG,CAAC,6BAA6B,CAAC,EACtC,SAAS,CACV,CAAC;QAEF,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;YACrB,QAAQ,EAAE;gBACR;oBACE,GAAG,EAAE,6BAA6B;oBAClC,QAAQ,EAAE,qBAAqB;oBAC/B,IAAI,EAAE,MAAM,CAAC,gBAAgB,CAAC,uBAAuB,CAAC;iBACvD;aACF;SACF,CAAC,CAAC;QAEH,oCAAoC;QACpC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,GAAG,CAAC,SAAS,CAC5C,SAAS,GAAG,gBAAgB,CAC7B,CAAC;QACF,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,SAAS,GAAG,cAAc,CAAC,CAAC;QAC3E,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,SAAS,CACxC,SAAS,GAAG,sBAAsB,CACnC,CAAC;QACF,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,SAAS,CAAC,SAAS,GAAG,mBAAmB,CAAC,CAAC;IAC9E,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
1
+ {"version":3,"file":"widget.test.js","sourceRoot":"","sources":["../../../src/test/widget.test.ts"],"names":[],"mappings":"AAKA,OAAO,EACL,SAAS,EACT,UAAU,EACV,QAAQ,EACR,MAAM,EACN,EAAE,EAEF,EAAE,GACH,MAAM,QAAQ,CAAC;AAEhB,OAAO,EACL,eAAe,EACf,mBAAmB,EACnB,YAAY,EACZ,UAAU,GACX,MAAM,YAAY,CAAC;AAEpB,MAAM,YAAY,GAAG;IACnB,2BAA2B,EAAE,EAAE,IAAI,EAAE,cAAc,EAAE;IACrD,WAAW,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE;CACnC,CAAC;AAEF,EAAE,CAAC,IAAI,CAAC,SAAS,EAAE,KAAK,IAAI,EAAE;IAC5B,MAAM,MAAM,GAAG,MAAM,EAAE,CAAC,YAAY,CAA2B,SAAS,CAAC,CAAC;IAC1E,MAAM,gBAAgB,GAAG,CACvB,IAA+C,EAC/C,GAAG,IAAe,EACsB,EAAE;QAC1C,IAAI,OAAO,IAAI,KAAK,QAAQ,IAAI,IAAI,CAAC,QAAQ,CAAC,eAAe,CAAC,EAAE,CAAC;YAC/D,OAAO,IAAI,CAAC,SAAS,CAAC,YAAY,CAEjC,CAAC;QACJ,CAAC;QACD,0FAA0F;QAC1F,qFAAqF;QACrF,0EAA0E;QAC1E,OAAO,MAAM,CAAC,YAAY,CAAC,IAAI,EAAE,GAAG,IAAI,CAAC,CAAC;IAC5C,CAAC,CAAC;IACF,MAAM,YAAY,GAAG,EAAE,CAAC,EAAE,CAAC,gBAAgB,CAA+B,CAAC;IAE3E,OAAO;QACL,YAAY;QACZ,OAAO,EAAE;YACP,YAAY;SACb;KACF,CAAC;AACJ,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,0BAA0B,EAAE,GAAG,EAAE;IACxC,IAAI,MAAiB,CAAC;IACtB,IAAI,oBAAiE,CAAC;IACtE,IAAI,gBAAyD,CAAC;IAE9D,UAAU,CAAC,GAAG,EAAE;QACd,CAAC,EAAE,MAAM,EAAE,oBAAoB,EAAE,gBAAgB,EAAE;YACjD,mBAAmB,EAAE,CAAC,CAAC;IAC3B,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,YAAY,EAAE,CAAC;IACjB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mDAAmD,EAAE,KAAK,IAAI,EAAE;QACjE,UAAU,CAAC,EAAE,QAAQ,EAAE,aAAa,EAAE,CAAC,CAAC;QAExC,MAAM,gBAAgB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACjC,MAAM,0BAA0B,GAAG,EAAE,WAAW,EAAE,aAAa,EAAE,CAAC;QAClE,MAAM,cAAc,GAAG,EAAE,WAAW,EAAE,WAAW,EAAE,CAAC;QAEpD,MAAM,CAAC,cAAc,CACnB,WAAW,EACX,0BAA0B,EAC1B,cAAc,EACd,gBAAgB,CACjB,CAAC;QAEF,qCAAqC;QACrC,MAAM,gBAAgB,GAAG,oBAAoB,CAAC,IAAI;aAC/C,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAKb,CAAC;QACH,MAAM,CAAC,gBAAgB,CAAC,CAAC,WAAW,EAAE,CAAC;QAEvC,MAAM,SAAS,GAAG,uBAAuB,CAAC;QAC1C,MAAM,SAAS,GAAG,eAAe,CAC/B,cAAc,CACsD,CAAC;QACvE,MAAM,MAAM,GAAG,MAAM,gBAAgB,CACnC,IAAI,GAAG,CAAC,6BAA6B,CAAC,EACtC,SAAS,CACV,CAAC;QAEF,MAAM,CAAC,gBAAgB,CAAC,CAAC,gBAAgB,EAAE,CAAC;QAC5C,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;YACrB,QAAQ,EAAE;gBACR;oBACE,GAAG,EAAE,6BAA6B;oBAClC,QAAQ,EAAE,qBAAqB;oBAC/B,IAAI,EAAE,MAAM,CAAC,gBAAgB,CAAC,uBAAuB,CAAC;iBACvD;aACF;SACF,CAAC,CAAC;QAEH,qCAAqC;QACrC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,SAAS,CAAC,GAAG,SAAS,iBAAiB,CAAC,CAAC;QAC1E,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,SAAS,CAAC,GAAG,SAAS,eAAe,CAAC,CAAC;QACxE,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,SAAS,CACxC,GAAG,SAAS,4BAA4B,CACzC,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;QAChE,UAAU,CAAC,EAAE,QAAQ,EAAE,YAAY,EAAE,CAAC,CAAC;QAEvC,MAAM,gBAAgB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACjC,MAAM,0BAA0B,GAAG,EAAE,WAAW,EAAE,aAAa,EAAE,CAAC;QAClE,MAAM,cAAc,GAAG,EAAE,WAAW,EAAE,WAAW,EAAE,CAAC;QAEpD,MAAM,CAAC,cAAc,CACnB,WAAW,EACX,0BAA0B,EAC1B,cAAc,EACd,gBAAgB,CACjB,CAAC;QAEF,qCAAqC;QACrC,MAAM,gBAAgB,GAAG,oBAAoB,CAAC,IAAI;aAC/C,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAKb,CAAC;QACH,MAAM,CAAC,gBAAgB,CAAC,CAAC,WAAW,EAAE,CAAC;QAEvC,MAAM,SAAS,GAAG,mBAAmB,CAAC;QACtC,MAAM,SAAS,GAAG,eAAe,CAC/B,SAAS,CAC2D,CAAC;QACvE,MAAM,MAAM,GAAG,MAAM,gBAAgB,CACnC,IAAI,GAAG,CAAC,6BAA6B,CAAC,EACtC,SAAS,CACV,CAAC;QAEF,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;YACrB,QAAQ,EAAE;gBACR;oBACE,GAAG,EAAE,6BAA6B;oBAClC,QAAQ,EAAE,qBAAqB;oBAC/B,IAAI,EAAE,MAAM,CAAC,gBAAgB,CAAC,uBAAuB,CAAC;iBACvD;aACF;SACF,CAAC,CAAC;QAEH,oCAAoC;QACpC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,GAAG,CAAC,SAAS,CAC5C,GAAG,SAAS,gBAAgB,CAC7B,CAAC;QACF,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,GAAG,SAAS,cAAc,CAAC,CAAC;QAC3E,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,SAAS,CACxC,GAAG,SAAS,sBAAsB,CACnC,CAAC;QACF,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,SAAS,CAAC,GAAG,SAAS,mBAAmB,CAAC,CAAC;IAC9E,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,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,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"}