@imgly/codesign-mcp 0.1.1 → 0.1.2
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/dist/cli/stdio.d.ts.map +1 -1
- package/dist/cli/stdio.js +54 -21
- package/dist/cli/stdio.js.map +1 -1
- package/dist/engine/cesdk-adapter.d.ts +7 -1
- package/dist/engine/cesdk-adapter.d.ts.map +1 -1
- package/dist/engine/cesdk-adapter.js +7 -1
- package/dist/engine/cesdk-adapter.js.map +1 -1
- package/dist/http/server.d.ts.map +1 -1
- package/dist/http/server.js +0 -1
- package/dist/http/server.js.map +1 -1
- package/dist/server.d.ts +0 -2
- package/dist/server.d.ts.map +1 -1
- package/dist/server.js.map +1 -1
- package/dist/test/test/cli/redirect-console.test.d.ts +2 -0
- package/dist/test/test/cli/redirect-console.test.d.ts.map +1 -0
- package/dist/test/test/cli/redirect-console.test.js +47 -0
- package/dist/test/test/cli/redirect-console.test.js.map +1 -0
- package/dist/test/test/cli/resolve-license.test.d.ts +2 -0
- package/dist/test/test/cli/resolve-license.test.d.ts.map +1 -0
- package/dist/test/test/cli/resolve-license.test.js +20 -0
- package/dist/test/test/cli/resolve-license.test.js.map +1 -0
- package/dist/test/test/cli/stdio-license.test.d.ts +2 -0
- package/dist/test/test/cli/stdio-license.test.d.ts.map +1 -0
- package/dist/test/test/cli/stdio-license.test.js +19 -0
- package/dist/test/test/cli/stdio-license.test.js.map +1 -0
- package/dist/test/test/config.test.d.ts +2 -0
- package/dist/test/test/config.test.d.ts.map +1 -0
- package/dist/test/test/config.test.js +91 -0
- package/dist/test/test/config.test.js.map +1 -0
- package/dist/test/test/engine/asset-sources.test.d.ts +2 -0
- package/dist/test/test/engine/asset-sources.test.d.ts.map +1 -0
- package/dist/test/test/engine/asset-sources.test.js +100 -0
- package/dist/test/test/engine/asset-sources.test.js.map +1 -0
- package/dist/test/test/engine/mutex.test.d.ts +2 -0
- package/dist/test/test/engine/mutex.test.d.ts.map +1 -0
- package/dist/test/test/engine/mutex.test.js +63 -0
- package/dist/test/test/engine/mutex.test.js.map +1 -0
- package/dist/test/test/engine/session.test.js +15 -0
- package/dist/test/test/engine/session.test.js.map +1 -1
- package/dist/test/test/engine/unsplash-source.test.d.ts +2 -0
- package/dist/test/test/engine/unsplash-source.test.d.ts.map +1 -0
- package/dist/test/test/engine/unsplash-source.test.js +129 -0
- package/dist/test/test/engine/unsplash-source.test.js.map +1 -0
- package/dist/test/test/handbook-contract.test.js +8 -4
- package/dist/test/test/handbook-contract.test.js.map +1 -1
- package/dist/test/test/http/server.test.js +129 -0
- package/dist/test/test/http/server.test.js.map +1 -1
- package/dist/test/test/http/session-id-header.test.d.ts +2 -0
- package/dist/test/test/http/session-id-header.test.d.ts.map +1 -0
- package/dist/test/test/http/session-id-header.test.js +76 -0
- package/dist/test/test/http/session-id-header.test.js.map +1 -0
- package/dist/test/test/http/session-manager.test.js +14 -0
- package/dist/test/test/http/session-manager.test.js.map +1 -1
- package/dist/test/test/http/workspace-auth.test.d.ts +2 -0
- package/dist/test/test/http/workspace-auth.test.d.ts.map +1 -0
- package/dist/test/test/http/workspace-auth.test.js +41 -0
- package/dist/test/test/http/workspace-auth.test.js.map +1 -0
- package/dist/test/test/metrics/mount.test.d.ts +2 -0
- package/dist/test/test/metrics/mount.test.d.ts.map +1 -0
- package/dist/test/test/metrics/mount.test.js +59 -0
- package/dist/test/test/metrics/mount.test.js.map +1 -0
- package/dist/test/test/metrics/recorder.test.d.ts +2 -0
- package/dist/test/test/metrics/recorder.test.d.ts.map +1 -0
- package/dist/test/test/metrics/recorder.test.js +118 -0
- package/dist/test/test/metrics/recorder.test.js.map +1 -0
- package/dist/test/test/revisions/store.test.js +81 -11
- package/dist/test/test/revisions/store.test.js.map +1 -1
- package/dist/test/test/server.metrics.test.d.ts +2 -0
- package/dist/test/test/server.metrics.test.d.ts.map +1 -0
- package/dist/test/test/server.metrics.test.js +87 -0
- package/dist/test/test/server.metrics.test.js.map +1 -0
- package/dist/test/test/server.test.js +33 -4
- package/dist/test/test/server.test.js.map +1 -1
- package/dist/test/test/tools/asset-search.test.d.ts +2 -0
- package/dist/test/test/tools/asset-search.test.d.ts.map +1 -0
- package/dist/test/test/tools/asset-search.test.js +121 -0
- package/dist/test/test/tools/asset-search.test.js.map +1 -0
- package/dist/test/test/tools/edit.test.js +71 -9
- package/dist/test/test/tools/edit.test.js.map +1 -1
- package/dist/test/test/tools/export.test.js +60 -30
- package/dist/test/test/tools/export.test.js.map +1 -1
- package/dist/test/test/tools/index.test.d.ts +2 -0
- package/dist/test/test/tools/index.test.d.ts.map +1 -0
- package/dist/test/test/tools/index.test.js +55 -0
- package/dist/test/test/tools/index.test.js.map +1 -0
- package/dist/test/test/tools/metrics.test.d.ts +2 -0
- package/dist/test/test/tools/metrics.test.d.ts.map +1 -0
- package/dist/test/test/tools/metrics.test.js +29 -0
- package/dist/test/test/tools/metrics.test.js.map +1 -0
- package/dist/test/test/tools/preview.test.js +6 -4
- package/dist/test/test/tools/preview.test.js.map +1 -1
- package/dist/test/test/tools/rename.test.js +1 -2
- package/dist/test/test/tools/rename.test.js.map +1 -1
- package/dist/test/test/tools/save.test.d.ts +2 -0
- package/dist/test/test/tools/save.test.d.ts.map +1 -0
- package/dist/test/test/tools/save.test.js +137 -0
- package/dist/test/test/tools/save.test.js.map +1 -0
- package/dist/test/test/viewer/router.test.js +56 -0
- package/dist/test/test/viewer/router.test.js.map +1 -1
- package/dist/test/test/viewer/session-mount.test.d.ts +2 -0
- package/dist/test/test/viewer/session-mount.test.d.ts.map +1 -0
- package/dist/test/test/viewer/session-mount.test.js +52 -0
- package/dist/test/test/viewer/session-mount.test.js.map +1 -0
- package/dist/test/test/viewer/url-builder.test.js +76 -9
- package/dist/test/test/viewer/url-builder.test.js.map +1 -1
- package/dist/test/test/workspace/s3.test.d.ts +2 -0
- package/dist/test/test/workspace/s3.test.d.ts.map +1 -0
- package/dist/test/test/workspace/s3.test.js +153 -0
- package/dist/test/test/workspace/s3.test.js.map +1 -0
- package/dist/test/test/workspace/scoped.test.d.ts +2 -0
- package/dist/test/test/workspace/scoped.test.d.ts.map +1 -0
- package/dist/test/test/workspace/scoped.test.js +59 -0
- package/dist/test/test/workspace/scoped.test.js.map +1 -0
- package/dist/test/tsconfig.test.tsbuildinfo +1 -1
- package/package.json +1 -1
|
@@ -1,9 +1,11 @@
|
|
|
1
1
|
import { describe, expect, it, vi } from 'vitest';
|
|
2
|
+
import { Mutex } from '../../src/engine/mutex.js';
|
|
2
3
|
import { EngineSession } from '../../src/engine/session.js';
|
|
3
4
|
import { RevisionStore } from '../../src/revisions/store.js';
|
|
4
5
|
import { createExportTool } from '../../src/tools/export.js';
|
|
5
6
|
import { LocalViewerUrlBuilder } from '../../src/viewer/url-builder.js';
|
|
6
7
|
import { InMemoryWorkspaceFS } from '../../src/workspace/in-memory.js';
|
|
8
|
+
const engineLock = new Mutex();
|
|
7
9
|
const urlBuilder = new LocalViewerUrlBuilder({ base: 'http://127.0.0.1:9000' });
|
|
8
10
|
function makeDeps() {
|
|
9
11
|
const fs = new InMemoryWorkspaceFS();
|
|
@@ -29,32 +31,16 @@ function makeDeps() {
|
|
|
29
31
|
return { fs, store, session, fakeEngine };
|
|
30
32
|
}
|
|
31
33
|
describe('export tool', () => {
|
|
32
|
-
it('export target=scene writes the bytes and returns uri + httpUrl, no path', async () => {
|
|
33
|
-
const { fs, store, session } = makeDeps();
|
|
34
|
-
const root = await store.write({
|
|
35
|
-
scene: 'SCENE_ABC',
|
|
36
|
-
parent: null,
|
|
37
|
-
title: 'T'
|
|
38
|
-
});
|
|
39
|
-
const tool = createExportTool({ fs, store, session, urlBuilder });
|
|
40
|
-
const result = await tool.call({
|
|
41
|
-
target: 'scene',
|
|
42
|
-
format: 'scene',
|
|
43
|
-
revision: root.id
|
|
44
|
-
});
|
|
45
|
-
const body = JSON.parse(result.content[0].text);
|
|
46
|
-
const shard = root.id.slice(0, 2);
|
|
47
|
-
const rest = root.id.slice(2);
|
|
48
|
-
expect(body.uri).toBe(`workspace://exports/${shard}/${rest}/scene.scene`);
|
|
49
|
-
expect(body.httpUrl).toBe(`http://127.0.0.1:9000/r/exports/${shard}/${rest}/scene.scene`);
|
|
50
|
-
expect(body.path).toBeUndefined();
|
|
51
|
-
expect(await fs.readFile(`exports/${shard}/${rest}/scene.scene`, 'utf8')).toBe('SCENE_ABC');
|
|
52
|
-
expect(body.bytes).toBe(9);
|
|
53
|
-
});
|
|
54
34
|
it('export target=image renders the block and writes the file deterministically', async () => {
|
|
55
35
|
const { fs, store, session } = makeDeps();
|
|
56
36
|
const root = await store.write({ scene: 'S', parent: null, title: 'T' });
|
|
57
|
-
const tool = createExportTool({
|
|
37
|
+
const tool = createExportTool({
|
|
38
|
+
fs,
|
|
39
|
+
store,
|
|
40
|
+
session,
|
|
41
|
+
engineLock,
|
|
42
|
+
urlBuilder
|
|
43
|
+
});
|
|
58
44
|
const result = await tool.call({
|
|
59
45
|
target: 'image',
|
|
60
46
|
format: 'png',
|
|
@@ -70,7 +56,13 @@ describe('export tool', () => {
|
|
|
70
56
|
it("export target=pdf renders via block.export('application/pdf')", async () => {
|
|
71
57
|
const { fs, store, session, fakeEngine } = makeDeps();
|
|
72
58
|
const root = await store.write({ scene: 'S', parent: null, title: 'T' });
|
|
73
|
-
const tool = createExportTool({
|
|
59
|
+
const tool = createExportTool({
|
|
60
|
+
fs,
|
|
61
|
+
store,
|
|
62
|
+
session,
|
|
63
|
+
engineLock,
|
|
64
|
+
urlBuilder
|
|
65
|
+
});
|
|
74
66
|
await tool.call({
|
|
75
67
|
target: 'pdf',
|
|
76
68
|
format: 'pdf',
|
|
@@ -82,7 +74,13 @@ describe('export tool', () => {
|
|
|
82
74
|
it('returns isError when target=video (deferred to Phase 4)', async () => {
|
|
83
75
|
const { fs, store, session } = makeDeps();
|
|
84
76
|
const root = await store.write({ scene: 'S', parent: null, title: 'T' });
|
|
85
|
-
const tool = createExportTool({
|
|
77
|
+
const tool = createExportTool({
|
|
78
|
+
fs,
|
|
79
|
+
store,
|
|
80
|
+
session,
|
|
81
|
+
engineLock,
|
|
82
|
+
urlBuilder
|
|
83
|
+
});
|
|
86
84
|
const result = await tool.call({
|
|
87
85
|
target: 'video',
|
|
88
86
|
format: 'mp4',
|
|
@@ -102,18 +100,20 @@ describe('export tool', () => {
|
|
|
102
100
|
const { tmpdir } = await import('node:os');
|
|
103
101
|
const { join } = await import('node:path');
|
|
104
102
|
const dir = await mkdtemp(join(tmpdir(), 'export-out-'));
|
|
105
|
-
const outPath = join(dir, 'x.
|
|
103
|
+
const outPath = join(dir, 'x.png');
|
|
106
104
|
const tool = createExportTool({
|
|
107
105
|
fs,
|
|
108
106
|
store,
|
|
109
107
|
session,
|
|
108
|
+
engineLock,
|
|
110
109
|
urlBuilder,
|
|
111
110
|
allowOutPath: true
|
|
112
111
|
});
|
|
113
112
|
const result = await tool.call({
|
|
114
|
-
target: '
|
|
115
|
-
format: '
|
|
113
|
+
target: 'image',
|
|
114
|
+
format: 'png',
|
|
116
115
|
revision: root.id,
|
|
116
|
+
blockId: 5,
|
|
117
117
|
outPath
|
|
118
118
|
});
|
|
119
119
|
const body = JSON.parse(result.content[0].text);
|
|
@@ -128,17 +128,47 @@ describe('export tool', () => {
|
|
|
128
128
|
fs,
|
|
129
129
|
store,
|
|
130
130
|
session,
|
|
131
|
+
engineLock,
|
|
131
132
|
urlBuilder,
|
|
132
133
|
allowOutPath: false
|
|
133
134
|
});
|
|
135
|
+
const result = await tool.call({
|
|
136
|
+
target: 'image',
|
|
137
|
+
format: 'png',
|
|
138
|
+
revision: root.id,
|
|
139
|
+
blockId: 5,
|
|
140
|
+
outPath: '/tmp/x'
|
|
141
|
+
});
|
|
142
|
+
expect(result.isError).toBe(true);
|
|
143
|
+
expect(result.content[0].text).toMatch(/outPath.*not.*supported/i);
|
|
144
|
+
});
|
|
145
|
+
it('omits `outPath` from the schema when not allowed (Hosted), advertises it when allowed (Local)', () => {
|
|
146
|
+
const { fs, store, session } = makeDeps();
|
|
147
|
+
const base = { fs, store, session, engineLock, urlBuilder };
|
|
148
|
+
const hosted = createExportTool({ ...base, allowOutPath: false });
|
|
149
|
+
const local = createExportTool({ ...base, allowOutPath: true });
|
|
150
|
+
const props = (s) => s.properties;
|
|
151
|
+
expect(props(hosted.inputSchema)).not.toHaveProperty('outPath');
|
|
152
|
+
expect(props(local.inputSchema)).toHaveProperty('outPath');
|
|
153
|
+
});
|
|
154
|
+
it('rejects target=scene (moved to the save tool)', async () => {
|
|
155
|
+
const { fs, store, session } = makeDeps();
|
|
156
|
+
const root = await store.write({ scene: 'S', parent: null, title: 'T' });
|
|
157
|
+
const tool = createExportTool({
|
|
158
|
+
fs,
|
|
159
|
+
store,
|
|
160
|
+
session,
|
|
161
|
+
engineLock,
|
|
162
|
+
urlBuilder
|
|
163
|
+
});
|
|
134
164
|
const result = await tool.call({
|
|
135
165
|
target: 'scene',
|
|
136
166
|
format: 'scene',
|
|
137
167
|
revision: root.id,
|
|
138
|
-
|
|
168
|
+
blockId: 5
|
|
139
169
|
});
|
|
140
170
|
expect(result.isError).toBe(true);
|
|
141
|
-
expect(result.content[0].text).toMatch(/
|
|
171
|
+
expect(result.content[0].text).toMatch(/invalid format/i);
|
|
142
172
|
});
|
|
143
173
|
});
|
|
144
174
|
//# sourceMappingURL=export.test.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"export.test.js","sourceRoot":"","sources":["../../../../test/tools/export.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAClD,OAAO,EAAsB,aAAa,EAAE,MAAM,6BAA6B,CAAC;AAChF,OAAO,EAAE,aAAa,EAAE,MAAM,8BAA8B,CAAC;AAC7D,OAAO,EAAE,gBAAgB,EAAE,MAAM,2BAA2B,CAAC;AAC7D,OAAO,EAAE,qBAAqB,EAAE,MAAM,iCAAiC,CAAC;AACxE,OAAO,EAAE,mBAAmB,EAAE,MAAM,kCAAkC,CAAC;AAQvE,MAAM,UAAU,GAAG,IAAI,qBAAqB,CAAC,EAAE,IAAI,EAAE,uBAAuB,EAAE,CAAC,CAAC;AAEhF,SAAS,QAAQ;IACf,MAAM,EAAE,GAAG,IAAI,mBAAmB,EAAE,CAAC;IACrC,MAAM,KAAK,GAAG,IAAI,aAAa,CAAC,EAAE,CAAC,CAAC;IACpC,MAAM,UAAU,GAAe;QAC7B,MAAM,EAAE,EAAE;QACV,KAAK,EAAE;YACL,MAAM,EAAE,EAAE,CAAC,EAAE,CACX,KAAK,IAAI,EAAE,CAAC,IAAI,IAAI,CAAC,CAAC,IAAI,UAAU,CAAC,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC,CACjE;SACF;QACD,KAAK,EAAE,EAAE,YAAY,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,qBAAqB,CAAC,EAAE;KAClE,CAAC;IACF,MAAM,OAAO,GAA8B;QACzC,MAAM,EAAE,GAAG,EAAE,CAAC,UAAU;QACxB,KAAK,EAAE,KAAK,EAAE,CAAC,EAAE,EAAE;YACjB,CAAC,CAAC,MAAM,GAAG,OAAO,CAAC;QACrB,CAAC;QACD,cAAc,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE;YAC7B,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC;QACf,CAAC;QACD,YAAY,EAAE,KAAK,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM;KACpC,CAAC;IACF,MAAM,OAAO,GAAG,IAAI,aAAa,CAAa,EAAE,IAAI,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC;IACpE,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC;AAC5C,CAAC;AAED,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;IAC3B,EAAE,CAAC,yEAAyE,EAAE,KAAK,IAAI,EAAE;QACvF,MAAM,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,QAAQ,EAAE,CAAC;QAC1C,MAAM,IAAI,GAAG,MAAM,KAAK,CAAC,KAAK,CAAC;YAC7B,KAAK,EAAE,WAAW;YAClB,MAAM,EAAE,IAAI;YACZ,KAAK,EAAE,GAAG;SACX,CAAC,CAAC;QACH,MAAM,IAAI,GAAG,gBAAgB,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC,CAAC;QAClE,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC;YAC7B,MAAM,EAAE,OAAO;YACf,MAAM,EAAE,OAAO;YACf,QAAQ,EAAE,IAAI,CAAC,EAAE;SAClB,CAAC,CAAC;QACH,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAE,MAAM,CAAC,OAAO,CAAC,CAAC,CAAsB,CAAC,IAAI,CAKnE,CAAC;QACF,MAAM,KAAK,GAAG,IAAI,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QAClC,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAC9B,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,uBAAuB,KAAK,IAAI,IAAI,cAAc,CAAC,CAAC;QAC1E,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,IAAI,CACvB,mCAAmC,KAAK,IAAI,IAAI,cAAc,CAC/D,CAAC;QACF,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,aAAa,EAAE,CAAC;QAClC,MAAM,CACJ,MAAM,EAAE,CAAC,QAAQ,CAAC,WAAW,KAAK,IAAI,IAAI,cAAc,EAAE,MAAM,CAAC,CAClE,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QACpB,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAC7B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6EAA6E,EAAE,KAAK,IAAI,EAAE;QAC3F,MAAM,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,QAAQ,EAAE,CAAC;QAC1C,MAAM,IAAI,GAAG,MAAM,KAAK,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC;QACzE,MAAM,IAAI,GAAG,gBAAgB,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC,CAAC;QAClE,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC;YAC7B,MAAM,EAAE,OAAO;YACf,MAAM,EAAE,KAAK;YACb,QAAQ,EAAE,IAAI,CAAC,EAAE;YACjB,OAAO,EAAE,CAAC;SACX,CAAC,CAAC;QACH,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAE,MAAM,CAAC,OAAO,CAAC,CAAC,CAAsB,CAAC,IAAI,CAEnE,CAAC;QACF,MAAM,KAAK,GAAG,IAAI,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QAClC,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAC9B,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,uBAAuB,KAAK,IAAI,IAAI,YAAY,CAAC,CAAC;QACxE,MAAM,CAAC,MAAM,EAAE,CAAC,MAAM,CAAC,WAAW,KAAK,IAAI,IAAI,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC3E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;QAC7E,MAAM,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,UAAU,EAAE,GAAG,QAAQ,EAAE,CAAC;QACtD,MAAM,IAAI,GAAG,MAAM,KAAK,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC;QACzE,MAAM,IAAI,GAAG,gBAAgB,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC,CAAC;QAClE,MAAM,IAAI,CAAC,IAAI,CAAC;YACd,MAAM,EAAE,KAAK;YACb,MAAM,EAAE,KAAK;YACb,QAAQ,EAAE,IAAI,CAAC,EAAE;YACjB,OAAO,EAAE,CAAC;SACX,CAAC,CAAC;QACH,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,oBAAoB,CAClD,CAAC,EACD,iBAAiB,EACjB,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CACnB,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;QACvE,MAAM,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,QAAQ,EAAE,CAAC;QAC1C,MAAM,IAAI,GAAG,MAAM,KAAK,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC;QACzE,MAAM,IAAI,GAAG,gBAAgB,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC,CAAC;QAClE,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC;YAC7B,MAAM,EAAE,OAAO;YACf,MAAM,EAAE,KAAK;YACb,QAAQ,EAAE,IAAI,CAAC,EAAE;SAClB,CAAC,CAAC;QACH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAClC,MAAM,CAAE,MAAM,CAAC,OAAO,CAAC,CAAC,CAAsB,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;IAC3E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mEAAmE,EAAE,KAAK,IAAI,EAAE;QACjF,MAAM,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,QAAQ,EAAE,CAAC;QAC1C,MAAM,IAAI,GAAG,MAAM,KAAK,CAAC,KAAK,CAAC;YAC7B,KAAK,EAAE,SAAS;YAChB,MAAM,EAAE,IAAI;YACZ,KAAK,EAAE,GAAG;SACX,CAAC,CAAC;QACH,MAAM,EAAE,OAAO,EAAE,EAAE,EAAE,GAAG,MAAM,MAAM,CAAC,kBAAkB,CAAC,CAAC;QACzD,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,MAAM,CAAC,SAAS,CAAC,CAAC;QAC3C,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,MAAM,CAAC,WAAW,CAAC,CAAC;QAC3C,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,aAAa,CAAC,CAAC,CAAC;QACzD,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC;QACrC,MAAM,IAAI,GAAG,gBAAgB,CAAC;YAC5B,EAAE;YACF,KAAK;YACL,OAAO;YACP,UAAU;YACV,YAAY,EAAE,IAAI;SACnB,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC;YAC7B,MAAM,EAAE,OAAO;YACf,MAAM,EAAE,OAAO;YACf,QAAQ,EAAE,IAAI,CAAC,EAAE;YACjB,OAAO;SACR,CAAC,CAAC;QACH,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAE,MAAM,CAAC,OAAO,CAAC,CAAC,CAAsB,CAAC,IAAI,CAGnE,CAAC;QACF,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAChC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,aAAa,EAAE,CAAC;QACrC,MAAM,EAAE,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IACrC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;QACnE,MAAM,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,QAAQ,EAAE,CAAC;QAC1C,MAAM,IAAI,GAAG,MAAM,KAAK,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC;QACzE,MAAM,IAAI,GAAG,gBAAgB,CAAC;YAC5B,EAAE;YACF,KAAK;YACL,OAAO;YACP,UAAU;YACV,YAAY,EAAE,KAAK;SACpB,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC;YAC7B,MAAM,EAAE,OAAO;YACf,MAAM,EAAE,OAAO;YACf,QAAQ,EAAE,IAAI,CAAC,EAAE;YACjB,OAAO,EAAE,QAAQ;SAClB,CAAC,CAAC;QACH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAClC,MAAM,CAAE,MAAM,CAAC,OAAO,CAAC,CAAC,CAAsB,CAAC,IAAI,CAAC,CAAC,OAAO,CAC1D,wBAAwB,CACzB,CAAC;IACJ,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import { describe, expect, it, vi } from 'vitest';\nimport { type EngineAdapter, EngineSession } from '../../src/engine/session.js';\nimport { RevisionStore } from '../../src/revisions/store.js';\nimport { createExportTool } from '../../src/tools/export.js';\nimport { LocalViewerUrlBuilder } from '../../src/viewer/url-builder.js';\nimport { InMemoryWorkspaceFS } from '../../src/workspace/in-memory.js';\n\ninterface FakeEngine {\n _scene: string;\n block: { export: ReturnType<typeof vi.fn> };\n scene: { saveToString: ReturnType<typeof vi.fn> };\n}\n\nconst urlBuilder = new LocalViewerUrlBuilder({ base: 'http://127.0.0.1:9000' });\n\nfunction makeDeps() {\n const fs = new InMemoryWorkspaceFS();\n const store = new RevisionStore(fs);\n const fakeEngine: FakeEngine = {\n _scene: '',\n block: {\n export: vi.fn(\n async () => new Blob([new Uint8Array([0xff, 0xd8, 0xff, 0xe0])])\n )\n },\n scene: { saveToString: vi.fn(async () => '<scene-serialized/>') }\n };\n const adapter: EngineAdapter<FakeEngine> = {\n create: () => fakeEngine,\n reset: async (e) => {\n e._scene = 'EMPTY';\n },\n loadFromString: async (e, s) => {\n e._scene = s;\n },\n saveToString: async (e) => e._scene\n };\n const session = new EngineSession<FakeEngine>({ size: 1, adapter });\n return { fs, store, session, fakeEngine };\n}\n\ndescribe('export tool', () => {\n it('export target=scene writes the bytes and returns uri + httpUrl, no path', async () => {\n const { fs, store, session } = makeDeps();\n const root = await store.write({\n scene: 'SCENE_ABC',\n parent: null,\n title: 'T'\n });\n const tool = createExportTool({ fs, store, session, urlBuilder });\n const result = await tool.call({\n target: 'scene',\n format: 'scene',\n revision: root.id\n });\n const body = JSON.parse((result.content[0] as { text: string }).text) as {\n uri: string;\n httpUrl?: string;\n path?: string;\n bytes: number;\n };\n const shard = root.id.slice(0, 2);\n const rest = root.id.slice(2);\n expect(body.uri).toBe(`workspace://exports/${shard}/${rest}/scene.scene`);\n expect(body.httpUrl).toBe(\n `http://127.0.0.1:9000/r/exports/${shard}/${rest}/scene.scene`\n );\n expect(body.path).toBeUndefined();\n expect(\n await fs.readFile(`exports/${shard}/${rest}/scene.scene`, 'utf8')\n ).toBe('SCENE_ABC');\n expect(body.bytes).toBe(9);\n });\n\n it('export target=image renders the block and writes the file deterministically', async () => {\n const { fs, store, session } = makeDeps();\n const root = await store.write({ scene: 'S', parent: null, title: 'T' });\n const tool = createExportTool({ fs, store, session, urlBuilder });\n const result = await tool.call({\n target: 'image',\n format: 'png',\n revision: root.id,\n blockId: 5\n });\n const body = JSON.parse((result.content[0] as { text: string }).text) as {\n uri: string;\n };\n const shard = root.id.slice(0, 2);\n const rest = root.id.slice(2);\n expect(body.uri).toBe(`workspace://exports/${shard}/${rest}/image.png`);\n expect(await fs.exists(`exports/${shard}/${rest}/image.png`)).toBe(true);\n });\n\n it(\"export target=pdf renders via block.export('application/pdf')\", async () => {\n const { fs, store, session, fakeEngine } = makeDeps();\n const root = await store.write({ scene: 'S', parent: null, title: 'T' });\n const tool = createExportTool({ fs, store, session, urlBuilder });\n await tool.call({\n target: 'pdf',\n format: 'pdf',\n revision: root.id,\n blockId: 5\n });\n expect(fakeEngine.block.export).toHaveBeenCalledWith(\n 5,\n 'application/pdf',\n expect.any(Object)\n );\n });\n\n it('returns isError when target=video (deferred to Phase 4)', async () => {\n const { fs, store, session } = makeDeps();\n const root = await store.write({ scene: 'S', parent: null, title: 'T' });\n const tool = createExportTool({ fs, store, session, urlBuilder });\n const result = await tool.call({\n target: 'video',\n format: 'mp4',\n revision: root.id\n });\n expect(result.isError).toBe(true);\n expect((result.content[0] as { text: string }).text).toMatch(/phase 4/i);\n });\n\n it('respects local `outPath` escape hatch — returns path, not httpUrl', async () => {\n const { fs, store, session } = makeDeps();\n const root = await store.write({\n scene: 'SCENE_X',\n parent: null,\n title: 'T'\n });\n const { mkdtemp, rm } = await import('node:fs/promises');\n const { tmpdir } = await import('node:os');\n const { join } = await import('node:path');\n const dir = await mkdtemp(join(tmpdir(), 'export-out-'));\n const outPath = join(dir, 'x.scene');\n const tool = createExportTool({\n fs,\n store,\n session,\n urlBuilder,\n allowOutPath: true\n });\n const result = await tool.call({\n target: 'scene',\n format: 'scene',\n revision: root.id,\n outPath\n });\n const body = JSON.parse((result.content[0] as { text: string }).text) as {\n path?: string;\n httpUrl?: string;\n };\n expect(body.path).toBe(outPath);\n expect(body.httpUrl).toBeUndefined();\n await rm(dir, { recursive: true });\n });\n\n it('rejects `outPath` when not allowed (Hosted/sandbox)', async () => {\n const { fs, store, session } = makeDeps();\n const root = await store.write({ scene: 'S', parent: null, title: 'T' });\n const tool = createExportTool({\n fs,\n store,\n session,\n urlBuilder,\n allowOutPath: false\n });\n const result = await tool.call({\n target: 'scene',\n format: 'scene',\n revision: root.id,\n outPath: '/tmp/x'\n });\n expect(result.isError).toBe(true);\n expect((result.content[0] as { text: string }).text).toMatch(\n /outPath.*not.*allowed/i\n );\n });\n});\n"]}
|
|
1
|
+
{"version":3,"file":"export.test.js","sourceRoot":"","sources":["../../../../test/tools/export.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAClD,OAAO,EAAE,KAAK,EAAE,MAAM,2BAA2B,CAAC;AAClD,OAAO,EAAsB,aAAa,EAAE,MAAM,6BAA6B,CAAC;AAChF,OAAO,EAAE,aAAa,EAAE,MAAM,8BAA8B,CAAC;AAC7D,OAAO,EAAE,gBAAgB,EAAE,MAAM,2BAA2B,CAAC;AAC7D,OAAO,EAAE,qBAAqB,EAAE,MAAM,iCAAiC,CAAC;AACxE,OAAO,EAAE,mBAAmB,EAAE,MAAM,kCAAkC,CAAC;AAEvE,MAAM,UAAU,GAAG,IAAI,KAAK,EAAE,CAAC;AAQ/B,MAAM,UAAU,GAAG,IAAI,qBAAqB,CAAC,EAAE,IAAI,EAAE,uBAAuB,EAAE,CAAC,CAAC;AAEhF,SAAS,QAAQ;IACf,MAAM,EAAE,GAAG,IAAI,mBAAmB,EAAE,CAAC;IACrC,MAAM,KAAK,GAAG,IAAI,aAAa,CAAC,EAAE,CAAC,CAAC;IACpC,MAAM,UAAU,GAAe;QAC7B,MAAM,EAAE,EAAE;QACV,KAAK,EAAE;YACL,MAAM,EAAE,EAAE,CAAC,EAAE,CACX,KAAK,IAAI,EAAE,CAAC,IAAI,IAAI,CAAC,CAAC,IAAI,UAAU,CAAC,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC,CACjE;SACF;QACD,KAAK,EAAE,EAAE,YAAY,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,qBAAqB,CAAC,EAAE;KAClE,CAAC;IACF,MAAM,OAAO,GAA8B;QACzC,MAAM,EAAE,GAAG,EAAE,CAAC,UAAU;QACxB,KAAK,EAAE,KAAK,EAAE,CAAC,EAAE,EAAE;YACjB,CAAC,CAAC,MAAM,GAAG,OAAO,CAAC;QACrB,CAAC;QACD,cAAc,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE;YAC7B,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC;QACf,CAAC;QACD,YAAY,EAAE,KAAK,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM;KACpC,CAAC;IACF,MAAM,OAAO,GAAG,IAAI,aAAa,CAAa,EAAE,IAAI,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC;IACpE,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC;AAC5C,CAAC;AAED,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;IAC3B,EAAE,CAAC,6EAA6E,EAAE,KAAK,IAAI,EAAE;QAC3F,MAAM,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,QAAQ,EAAE,CAAC;QAC1C,MAAM,IAAI,GAAG,MAAM,KAAK,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC;QACzE,MAAM,IAAI,GAAG,gBAAgB,CAAC;YAC5B,EAAE;YACF,KAAK;YACL,OAAO;YACP,UAAU;YACV,UAAU;SACX,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC;YAC7B,MAAM,EAAE,OAAO;YACf,MAAM,EAAE,KAAK;YACb,QAAQ,EAAE,IAAI,CAAC,EAAE;YACjB,OAAO,EAAE,CAAC;SACX,CAAC,CAAC;QACH,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAE,MAAM,CAAC,OAAO,CAAC,CAAC,CAAsB,CAAC,IAAI,CAEnE,CAAC;QACF,MAAM,KAAK,GAAG,IAAI,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QAClC,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAC9B,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,uBAAuB,KAAK,IAAI,IAAI,YAAY,CAAC,CAAC;QACxE,MAAM,CAAC,MAAM,EAAE,CAAC,MAAM,CAAC,WAAW,KAAK,IAAI,IAAI,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC3E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;QAC7E,MAAM,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,UAAU,EAAE,GAAG,QAAQ,EAAE,CAAC;QACtD,MAAM,IAAI,GAAG,MAAM,KAAK,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC;QACzE,MAAM,IAAI,GAAG,gBAAgB,CAAC;YAC5B,EAAE;YACF,KAAK;YACL,OAAO;YACP,UAAU;YACV,UAAU;SACX,CAAC,CAAC;QACH,MAAM,IAAI,CAAC,IAAI,CAAC;YACd,MAAM,EAAE,KAAK;YACb,MAAM,EAAE,KAAK;YACb,QAAQ,EAAE,IAAI,CAAC,EAAE;YACjB,OAAO,EAAE,CAAC;SACX,CAAC,CAAC;QACH,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,oBAAoB,CAClD,CAAC,EACD,iBAAiB,EACjB,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CACnB,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;QACvE,MAAM,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,QAAQ,EAAE,CAAC;QAC1C,MAAM,IAAI,GAAG,MAAM,KAAK,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC;QACzE,MAAM,IAAI,GAAG,gBAAgB,CAAC;YAC5B,EAAE;YACF,KAAK;YACL,OAAO;YACP,UAAU;YACV,UAAU;SACX,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC;YAC7B,MAAM,EAAE,OAAO;YACf,MAAM,EAAE,KAAK;YACb,QAAQ,EAAE,IAAI,CAAC,EAAE;SAClB,CAAC,CAAC;QACH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAClC,MAAM,CAAE,MAAM,CAAC,OAAO,CAAC,CAAC,CAAsB,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;IAC3E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mEAAmE,EAAE,KAAK,IAAI,EAAE;QACjF,MAAM,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,QAAQ,EAAE,CAAC;QAC1C,MAAM,IAAI,GAAG,MAAM,KAAK,CAAC,KAAK,CAAC;YAC7B,KAAK,EAAE,SAAS;YAChB,MAAM,EAAE,IAAI;YACZ,KAAK,EAAE,GAAG;SACX,CAAC,CAAC;QACH,MAAM,EAAE,OAAO,EAAE,EAAE,EAAE,GAAG,MAAM,MAAM,CAAC,kBAAkB,CAAC,CAAC;QACzD,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,MAAM,CAAC,SAAS,CAAC,CAAC;QAC3C,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,MAAM,CAAC,WAAW,CAAC,CAAC;QAC3C,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,aAAa,CAAC,CAAC,CAAC;QACzD,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;QACnC,MAAM,IAAI,GAAG,gBAAgB,CAAC;YAC5B,EAAE;YACF,KAAK;YACL,OAAO;YACP,UAAU;YACV,UAAU;YACV,YAAY,EAAE,IAAI;SACnB,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC;YAC7B,MAAM,EAAE,OAAO;YACf,MAAM,EAAE,KAAK;YACb,QAAQ,EAAE,IAAI,CAAC,EAAE;YACjB,OAAO,EAAE,CAAC;YACV,OAAO;SACR,CAAC,CAAC;QACH,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAE,MAAM,CAAC,OAAO,CAAC,CAAC,CAAsB,CAAC,IAAI,CAGnE,CAAC;QACF,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAChC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,aAAa,EAAE,CAAC;QACrC,MAAM,EAAE,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IACrC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;QACnE,MAAM,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,QAAQ,EAAE,CAAC;QAC1C,MAAM,IAAI,GAAG,MAAM,KAAK,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC;QACzE,MAAM,IAAI,GAAG,gBAAgB,CAAC;YAC5B,EAAE;YACF,KAAK;YACL,OAAO;YACP,UAAU;YACV,UAAU;YACV,YAAY,EAAE,KAAK;SACpB,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC;YAC7B,MAAM,EAAE,OAAO;YACf,MAAM,EAAE,KAAK;YACb,QAAQ,EAAE,IAAI,CAAC,EAAE;YACjB,OAAO,EAAE,CAAC;YACV,OAAO,EAAE,QAAQ;SAClB,CAAC,CAAC;QACH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAClC,MAAM,CAAE,MAAM,CAAC,OAAO,CAAC,CAAC,CAAsB,CAAC,IAAI,CAAC,CAAC,OAAO,CAC1D,0BAA0B,CAC3B,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+FAA+F,EAAE,GAAG,EAAE;QACvG,MAAM,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,QAAQ,EAAE,CAAC;QAC1C,MAAM,IAAI,GAAG,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,CAAC;QAC5D,MAAM,MAAM,GAAG,gBAAgB,CAAC,EAAE,GAAG,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,CAAC,CAAC;QAClE,MAAM,KAAK,GAAG,gBAAgB,CAAC,EAAE,GAAG,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,CAAC,CAAC;QAChE,MAAM,KAAK,GAAG,CAAC,CAAU,EAAE,EAAE,CAC1B,CAA6C,CAAC,UAAU,CAAC;QAC5D,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,GAAG,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC;QAChE,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC;IAC7D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;QAC7D,MAAM,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,QAAQ,EAAE,CAAC;QAC1C,MAAM,IAAI,GAAG,MAAM,KAAK,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC;QACzE,MAAM,IAAI,GAAG,gBAAgB,CAAC;YAC5B,EAAE;YACF,KAAK;YACL,OAAO;YACP,UAAU;YACV,UAAU;SACX,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC;YAC7B,MAAM,EAAE,OAAO;YACf,MAAM,EAAE,OAAO;YACf,QAAQ,EAAE,IAAI,CAAC,EAAE;YACjB,OAAO,EAAE,CAAC;SACX,CAAC,CAAC;QACH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAClC,MAAM,CAAE,MAAM,CAAC,OAAO,CAAC,CAAC,CAAsB,CAAC,IAAI,CAAC,CAAC,OAAO,CAC1D,iBAAiB,CAClB,CAAC;IACJ,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import { describe, expect, it, vi } from 'vitest';\nimport { Mutex } from '../../src/engine/mutex.js';\nimport { type EngineAdapter, EngineSession } from '../../src/engine/session.js';\nimport { RevisionStore } from '../../src/revisions/store.js';\nimport { createExportTool } from '../../src/tools/export.js';\nimport { LocalViewerUrlBuilder } from '../../src/viewer/url-builder.js';\nimport { InMemoryWorkspaceFS } from '../../src/workspace/in-memory.js';\n\nconst engineLock = new Mutex();\n\ninterface FakeEngine {\n _scene: string;\n block: { export: ReturnType<typeof vi.fn> };\n scene: { saveToString: ReturnType<typeof vi.fn> };\n}\n\nconst urlBuilder = new LocalViewerUrlBuilder({ base: 'http://127.0.0.1:9000' });\n\nfunction makeDeps() {\n const fs = new InMemoryWorkspaceFS();\n const store = new RevisionStore(fs);\n const fakeEngine: FakeEngine = {\n _scene: '',\n block: {\n export: vi.fn(\n async () => new Blob([new Uint8Array([0xff, 0xd8, 0xff, 0xe0])])\n )\n },\n scene: { saveToString: vi.fn(async () => '<scene-serialized/>') }\n };\n const adapter: EngineAdapter<FakeEngine> = {\n create: () => fakeEngine,\n reset: async (e) => {\n e._scene = 'EMPTY';\n },\n loadFromString: async (e, s) => {\n e._scene = s;\n },\n saveToString: async (e) => e._scene\n };\n const session = new EngineSession<FakeEngine>({ size: 1, adapter });\n return { fs, store, session, fakeEngine };\n}\n\ndescribe('export tool', () => {\n it('export target=image renders the block and writes the file deterministically', async () => {\n const { fs, store, session } = makeDeps();\n const root = await store.write({ scene: 'S', parent: null, title: 'T' });\n const tool = createExportTool({\n fs,\n store,\n session,\n engineLock,\n urlBuilder\n });\n const result = await tool.call({\n target: 'image',\n format: 'png',\n revision: root.id,\n blockId: 5\n });\n const body = JSON.parse((result.content[0] as { text: string }).text) as {\n uri: string;\n };\n const shard = root.id.slice(0, 2);\n const rest = root.id.slice(2);\n expect(body.uri).toBe(`workspace://exports/${shard}/${rest}/image.png`);\n expect(await fs.exists(`exports/${shard}/${rest}/image.png`)).toBe(true);\n });\n\n it(\"export target=pdf renders via block.export('application/pdf')\", async () => {\n const { fs, store, session, fakeEngine } = makeDeps();\n const root = await store.write({ scene: 'S', parent: null, title: 'T' });\n const tool = createExportTool({\n fs,\n store,\n session,\n engineLock,\n urlBuilder\n });\n await tool.call({\n target: 'pdf',\n format: 'pdf',\n revision: root.id,\n blockId: 5\n });\n expect(fakeEngine.block.export).toHaveBeenCalledWith(\n 5,\n 'application/pdf',\n expect.any(Object)\n );\n });\n\n it('returns isError when target=video (deferred to Phase 4)', async () => {\n const { fs, store, session } = makeDeps();\n const root = await store.write({ scene: 'S', parent: null, title: 'T' });\n const tool = createExportTool({\n fs,\n store,\n session,\n engineLock,\n urlBuilder\n });\n const result = await tool.call({\n target: 'video',\n format: 'mp4',\n revision: root.id\n });\n expect(result.isError).toBe(true);\n expect((result.content[0] as { text: string }).text).toMatch(/phase 4/i);\n });\n\n it('respects local `outPath` escape hatch — returns path, not httpUrl', async () => {\n const { fs, store, session } = makeDeps();\n const root = await store.write({\n scene: 'SCENE_X',\n parent: null,\n title: 'T'\n });\n const { mkdtemp, rm } = await import('node:fs/promises');\n const { tmpdir } = await import('node:os');\n const { join } = await import('node:path');\n const dir = await mkdtemp(join(tmpdir(), 'export-out-'));\n const outPath = join(dir, 'x.png');\n const tool = createExportTool({\n fs,\n store,\n session,\n engineLock,\n urlBuilder,\n allowOutPath: true\n });\n const result = await tool.call({\n target: 'image',\n format: 'png',\n revision: root.id,\n blockId: 5,\n outPath\n });\n const body = JSON.parse((result.content[0] as { text: string }).text) as {\n path?: string;\n httpUrl?: string;\n };\n expect(body.path).toBe(outPath);\n expect(body.httpUrl).toBeUndefined();\n await rm(dir, { recursive: true });\n });\n\n it('rejects `outPath` when not allowed (Hosted/sandbox)', async () => {\n const { fs, store, session } = makeDeps();\n const root = await store.write({ scene: 'S', parent: null, title: 'T' });\n const tool = createExportTool({\n fs,\n store,\n session,\n engineLock,\n urlBuilder,\n allowOutPath: false\n });\n const result = await tool.call({\n target: 'image',\n format: 'png',\n revision: root.id,\n blockId: 5,\n outPath: '/tmp/x'\n });\n expect(result.isError).toBe(true);\n expect((result.content[0] as { text: string }).text).toMatch(\n /outPath.*not.*supported/i\n );\n });\n\n it('omits `outPath` from the schema when not allowed (Hosted), advertises it when allowed (Local)', () => {\n const { fs, store, session } = makeDeps();\n const base = { fs, store, session, engineLock, urlBuilder };\n const hosted = createExportTool({ ...base, allowOutPath: false });\n const local = createExportTool({ ...base, allowOutPath: true });\n const props = (s: unknown) =>\n (s as { properties: Record<string, unknown> }).properties;\n expect(props(hosted.inputSchema)).not.toHaveProperty('outPath');\n expect(props(local.inputSchema)).toHaveProperty('outPath');\n });\n\n it('rejects target=scene (moved to the save tool)', async () => {\n const { fs, store, session } = makeDeps();\n const root = await store.write({ scene: 'S', parent: null, title: 'T' });\n const tool = createExportTool({\n fs,\n store,\n session,\n engineLock,\n urlBuilder\n });\n const result = await tool.call({\n target: 'scene',\n format: 'scene',\n revision: root.id,\n blockId: 5\n });\n expect(result.isError).toBe(true);\n expect((result.content[0] as { text: string }).text).toMatch(\n /invalid format/i\n );\n });\n});\n"]}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.test.d.ts","sourceRoot":"","sources":["../../../../test/tools/index.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,55 @@
|
|
|
1
|
+
import { describe, expect, it } from 'vitest';
|
|
2
|
+
import { buildDesignToolRegistrations } from '../../src/tools/index.js';
|
|
3
|
+
// The tool factories only close over their deps (they touch them on `call`,
|
|
4
|
+
// not at construction), so stub primitives are enough to assert which tools
|
|
5
|
+
// get registered.
|
|
6
|
+
const stub = {};
|
|
7
|
+
function gateway(available) {
|
|
8
|
+
return {
|
|
9
|
+
available,
|
|
10
|
+
generateImage: async () => ({
|
|
11
|
+
bytes: new Uint8Array(),
|
|
12
|
+
mimeType: 'image/png'
|
|
13
|
+
})
|
|
14
|
+
};
|
|
15
|
+
}
|
|
16
|
+
function toolNames(available) {
|
|
17
|
+
return buildDesignToolRegistrations({
|
|
18
|
+
fs: stub,
|
|
19
|
+
store: stub,
|
|
20
|
+
session: stub,
|
|
21
|
+
engineLock: stub,
|
|
22
|
+
gateway: gateway(available),
|
|
23
|
+
urlBuilder: stub,
|
|
24
|
+
allowOutPath: true
|
|
25
|
+
}).map((t) => t.name);
|
|
26
|
+
}
|
|
27
|
+
// asset_generate is hidden for the beta via the ASSET_GENERATE_ENABLED
|
|
28
|
+
// kill-switch in src/tools/index.ts, so it is omitted regardless of gateway
|
|
29
|
+
// availability. When the feature ships and the flag flips back to true, the
|
|
30
|
+
// "available" case should assert it IS registered again.
|
|
31
|
+
describe('buildDesignToolRegistrations — asset_generate gating (beta: hidden)', () => {
|
|
32
|
+
const alwaysOn = [
|
|
33
|
+
'edit',
|
|
34
|
+
'preview',
|
|
35
|
+
'export',
|
|
36
|
+
'save',
|
|
37
|
+
'list',
|
|
38
|
+
'history',
|
|
39
|
+
'inspect',
|
|
40
|
+
'rename',
|
|
41
|
+
'open_editor',
|
|
42
|
+
'asset_search'
|
|
43
|
+
];
|
|
44
|
+
it('omits asset_generate when the gateway is unavailable (no API_KEY)', () => {
|
|
45
|
+
const names = toolNames(false);
|
|
46
|
+
expect(names).not.toContain('asset_generate');
|
|
47
|
+
expect(names).toEqual(alwaysOn);
|
|
48
|
+
});
|
|
49
|
+
it('stays hidden even when the gateway is available (beta kill-switch)', () => {
|
|
50
|
+
const names = toolNames(true);
|
|
51
|
+
expect(names).not.toContain('asset_generate');
|
|
52
|
+
expect(names).toEqual(alwaysOn);
|
|
53
|
+
});
|
|
54
|
+
});
|
|
55
|
+
//# sourceMappingURL=index.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.test.js","sourceRoot":"","sources":["../../../../test/tools/index.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAE9C,OAAO,EAAE,4BAA4B,EAAE,MAAM,0BAA0B,CAAC;AAExE,4EAA4E;AAC5E,4EAA4E;AAC5E,kBAAkB;AAClB,MAAM,IAAI,GAAG,EAAW,CAAC;AAEzB,SAAS,OAAO,CAAC,SAAkB;IACjC,OAAO;QACL,SAAS;QACT,aAAa,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC;YAC1B,KAAK,EAAE,IAAI,UAAU,EAAE;YACvB,QAAQ,EAAE,WAAW;SACtB,CAAC;KACH,CAAC;AACJ,CAAC;AAED,SAAS,SAAS,CAAC,SAAkB;IACnC,OAAO,4BAA4B,CAAC;QAClC,EAAE,EAAE,IAAI;QACR,KAAK,EAAE,IAAI;QACX,OAAO,EAAE,IAAI;QACb,UAAU,EAAE,IAAI;QAChB,OAAO,EAAE,OAAO,CAAC,SAAS,CAAC;QAC3B,UAAU,EAAE,IAAI;QAChB,YAAY,EAAE,IAAI;KACnB,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;AACxB,CAAC;AAED,uEAAuE;AACvE,4EAA4E;AAC5E,4EAA4E;AAC5E,yDAAyD;AACzD,QAAQ,CAAC,qEAAqE,EAAE,GAAG,EAAE;IACnF,MAAM,QAAQ,GAAG;QACf,MAAM;QACN,SAAS;QACT,QAAQ;QACR,MAAM;QACN,MAAM;QACN,SAAS;QACT,SAAS;QACT,QAAQ;QACR,aAAa;QACb,cAAc;KACf,CAAC;IAEF,EAAE,CAAC,mEAAmE,EAAE,GAAG,EAAE;QAC3E,MAAM,KAAK,GAAG,SAAS,CAAC,KAAK,CAAC,CAAC;QAC/B,MAAM,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;QAC9C,MAAM,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IAClC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oEAAoE,EAAE,GAAG,EAAE;QAC5E,MAAM,KAAK,GAAG,SAAS,CAAC,IAAI,CAAC,CAAC;QAC9B,MAAM,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;QAC9C,MAAM,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IAClC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import { describe, expect, it } from 'vitest';\nimport type { GatewayClient } from '../../src/gateway/types.js';\nimport { buildDesignToolRegistrations } from '../../src/tools/index.js';\n\n// The tool factories only close over their deps (they touch them on `call`,\n// not at construction), so stub primitives are enough to assert which tools\n// get registered.\nconst stub = {} as never;\n\nfunction gateway(available: boolean): GatewayClient {\n return {\n available,\n generateImage: async () => ({\n bytes: new Uint8Array(),\n mimeType: 'image/png'\n })\n };\n}\n\nfunction toolNames(available: boolean): string[] {\n return buildDesignToolRegistrations({\n fs: stub,\n store: stub,\n session: stub,\n engineLock: stub,\n gateway: gateway(available),\n urlBuilder: stub,\n allowOutPath: true\n }).map((t) => t.name);\n}\n\n// asset_generate is hidden for the beta via the ASSET_GENERATE_ENABLED\n// kill-switch in src/tools/index.ts, so it is omitted regardless of gateway\n// availability. When the feature ships and the flag flips back to true, the\n// \"available\" case should assert it IS registered again.\ndescribe('buildDesignToolRegistrations — asset_generate gating (beta: hidden)', () => {\n const alwaysOn = [\n 'edit',\n 'preview',\n 'export',\n 'save',\n 'list',\n 'history',\n 'inspect',\n 'rename',\n 'open_editor',\n 'asset_search'\n ];\n\n it('omits asset_generate when the gateway is unavailable (no API_KEY)', () => {\n const names = toolNames(false);\n expect(names).not.toContain('asset_generate');\n expect(names).toEqual(alwaysOn);\n });\n\n it('stays hidden even when the gateway is available (beta kill-switch)', () => {\n const names = toolNames(true);\n expect(names).not.toContain('asset_generate');\n expect(names).toEqual(alwaysOn);\n });\n});\n"]}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"metrics.test.d.ts","sourceRoot":"","sources":["../../../../test/tools/metrics.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,29 @@
|
|
|
1
|
+
import { describe, expect, it } from 'vitest';
|
|
2
|
+
import { MetricsRecorder } from '../../src/metrics/recorder.js';
|
|
3
|
+
import { createMetricsTool } from '../../src/tools/metrics.js';
|
|
4
|
+
describe('metrics tool', () => {
|
|
5
|
+
it('returns the recorder snapshot plus live session facts', async () => {
|
|
6
|
+
const recorder = new MetricsRecorder();
|
|
7
|
+
recorder.record({ tool: 'edit', durationMs: 120, ok: true });
|
|
8
|
+
recorder.record({ tool: 'edit', durationMs: 80, ok: false });
|
|
9
|
+
const tool = createMetricsTool({
|
|
10
|
+
recorder,
|
|
11
|
+
activeSessions: () => 3,
|
|
12
|
+
maxSessions: 5
|
|
13
|
+
});
|
|
14
|
+
expect(tool.name).toBe('metrics');
|
|
15
|
+
const result = await tool.call({});
|
|
16
|
+
const first = result.content[0];
|
|
17
|
+
if (first.type !== 'text')
|
|
18
|
+
throw new Error('expected text content');
|
|
19
|
+
const body = JSON.parse(first.text);
|
|
20
|
+
const edit = body.tools.find((t) => t.tool === 'edit');
|
|
21
|
+
expect(edit?.count).toBe(2);
|
|
22
|
+
expect(edit?.errorCount).toBe(1);
|
|
23
|
+
expect(body.recent).toHaveLength(2);
|
|
24
|
+
expect(body.activeSessions).toBe(3);
|
|
25
|
+
expect(body.maxSessions).toBe(5);
|
|
26
|
+
expect(body.uptimeMs).toBeGreaterThan(0);
|
|
27
|
+
});
|
|
28
|
+
});
|
|
29
|
+
//# sourceMappingURL=metrics.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"metrics.test.js","sourceRoot":"","sources":["../../../../test/tools/metrics.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,eAAe,EAAE,MAAM,+BAA+B,CAAC;AAChE,OAAO,EAAE,iBAAiB,EAAE,MAAM,4BAA4B,CAAC;AAE/D,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;IAC5B,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;QACrE,MAAM,QAAQ,GAAG,IAAI,eAAe,EAAE,CAAC;QACvC,QAAQ,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,UAAU,EAAE,GAAG,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC;QAC7D,QAAQ,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,UAAU,EAAE,EAAE,EAAE,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;QAC7D,MAAM,IAAI,GAAG,iBAAiB,CAAC;YAC7B,QAAQ;YACR,cAAc,EAAE,GAAG,EAAE,CAAC,CAAC;YACvB,WAAW,EAAE,CAAC;SACf,CAAC,CAAC;QACH,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAElC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACnC,MAAM,KAAK,GAAG,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;QAChC,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM;YAAE,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;QACpE,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAMjC,CAAC;QACF,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC;QACvD,MAAM,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC5B,MAAM,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACjC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACpC,MAAM,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACpC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACjC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;IAC3C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import { describe, expect, it } from 'vitest';\nimport { MetricsRecorder } from '../../src/metrics/recorder.js';\nimport { createMetricsTool } from '../../src/tools/metrics.js';\n\ndescribe('metrics tool', () => {\n it('returns the recorder snapshot plus live session facts', async () => {\n const recorder = new MetricsRecorder();\n recorder.record({ tool: 'edit', durationMs: 120, ok: true });\n recorder.record({ tool: 'edit', durationMs: 80, ok: false });\n const tool = createMetricsTool({\n recorder,\n activeSessions: () => 3,\n maxSessions: 5\n });\n expect(tool.name).toBe('metrics');\n\n const result = await tool.call({});\n const first = result.content[0];\n if (first.type !== 'text') throw new Error('expected text content');\n const body = JSON.parse(first.text) as {\n tools: Array<{ tool: string; count: number; errorCount: number }>;\n recent: unknown[];\n activeSessions: number;\n maxSessions: number;\n uptimeMs: number;\n };\n const edit = body.tools.find((t) => t.tool === 'edit');\n expect(edit?.count).toBe(2);\n expect(edit?.errorCount).toBe(1);\n expect(body.recent).toHaveLength(2);\n expect(body.activeSessions).toBe(3);\n expect(body.maxSessions).toBe(5);\n expect(body.uptimeMs).toBeGreaterThan(0);\n });\n});\n"]}
|
|
@@ -1,8 +1,10 @@
|
|
|
1
1
|
import { describe, expect, it, vi } from 'vitest';
|
|
2
|
+
import { Mutex } from '../../src/engine/mutex.js';
|
|
2
3
|
import { EngineSession } from '../../src/engine/session.js';
|
|
3
4
|
import { RevisionStore } from '../../src/revisions/store.js';
|
|
4
5
|
import { createPreviewTool } from '../../src/tools/preview.js';
|
|
5
6
|
import { InMemoryWorkspaceFS } from '../../src/workspace/in-memory.js';
|
|
7
|
+
const engineLock = new Mutex();
|
|
6
8
|
function makeDeps() {
|
|
7
9
|
const fs = new InMemoryWorkspaceFS();
|
|
8
10
|
const store = new RevisionStore(fs);
|
|
@@ -31,7 +33,7 @@ describe('preview tool', () => {
|
|
|
31
33
|
const slot = await session.acquire(null);
|
|
32
34
|
session.commit(slot, 'rev_warm');
|
|
33
35
|
fakeEngine._scene = 'WARM';
|
|
34
|
-
const tool = createPreviewTool({ store, session });
|
|
36
|
+
const tool = createPreviewTool({ store, session, engineLock });
|
|
35
37
|
const result = await tool.call({ blockId: 42 });
|
|
36
38
|
expect(result.isError).toBeFalsy();
|
|
37
39
|
expect(result.content[0].type).toBe('image');
|
|
@@ -41,7 +43,7 @@ describe('preview tool', () => {
|
|
|
41
43
|
});
|
|
42
44
|
it('returns isError when no revision supplied AND no warm slot exists', async () => {
|
|
43
45
|
const { store, session } = makeDeps();
|
|
44
|
-
const tool = createPreviewTool({ store, session });
|
|
46
|
+
const tool = createPreviewTool({ store, session, engineLock });
|
|
45
47
|
const result = await tool.call({ blockId: 42 });
|
|
46
48
|
expect(result.isError).toBe(true);
|
|
47
49
|
expect(result.content[0].text).toMatch(/no.*engine.*state/i);
|
|
@@ -53,7 +55,7 @@ describe('preview tool', () => {
|
|
|
53
55
|
parent: null,
|
|
54
56
|
title: 'T'
|
|
55
57
|
});
|
|
56
|
-
const tool = createPreviewTool({ store, session });
|
|
58
|
+
const tool = createPreviewTool({ store, session, engineLock });
|
|
57
59
|
// adapter.loadFromString is set as a plain async fn (not a vi.fn) above, so wrap it
|
|
58
60
|
const loadSpy = vi.spyOn(adapter, 'loadFromString');
|
|
59
61
|
await tool.call({ blockId: 42, revision: root.id });
|
|
@@ -63,7 +65,7 @@ describe('preview tool', () => {
|
|
|
63
65
|
const { store, session, fakeEngine } = makeDeps();
|
|
64
66
|
const slot = await session.acquire(null);
|
|
65
67
|
session.commit(slot, 'rev_warm');
|
|
66
|
-
const tool = createPreviewTool({ store, session });
|
|
68
|
+
const tool = createPreviewTool({ store, session, engineLock });
|
|
67
69
|
await tool.call({ blockId: 42, width: 9999, height: 9999 });
|
|
68
70
|
const call = fakeEngine.block.export.mock
|
|
69
71
|
.calls[0];
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"preview.test.js","sourceRoot":"","sources":["../../../../test/tools/preview.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAClD,OAAO,EAAsB,aAAa,EAAE,MAAM,6BAA6B,CAAC;AAChF,OAAO,EAAE,aAAa,EAAE,MAAM,8BAA8B,CAAC;AAC7D,OAAO,EAAE,iBAAiB,EAAE,MAAM,4BAA4B,CAAC;AAC/D,OAAO,EAAE,mBAAmB,EAAE,MAAM,kCAAkC,CAAC;
|
|
1
|
+
{"version":3,"file":"preview.test.js","sourceRoot":"","sources":["../../../../test/tools/preview.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAClD,OAAO,EAAE,KAAK,EAAE,MAAM,2BAA2B,CAAC;AAClD,OAAO,EAAsB,aAAa,EAAE,MAAM,6BAA6B,CAAC;AAChF,OAAO,EAAE,aAAa,EAAE,MAAM,8BAA8B,CAAC;AAC7D,OAAO,EAAE,iBAAiB,EAAE,MAAM,4BAA4B,CAAC;AAC/D,OAAO,EAAE,mBAAmB,EAAE,MAAM,kCAAkC,CAAC;AAEvE,MAAM,UAAU,GAAG,IAAI,KAAK,EAAE,CAAC;AAO/B,SAAS,QAAQ;IACf,MAAM,EAAE,GAAG,IAAI,mBAAmB,EAAE,CAAC;IACrC,MAAM,KAAK,GAAG,IAAI,aAAa,CAAC,EAAE,CAAC,CAAC;IACpC,MAAM,UAAU,GAAe;QAC7B,MAAM,EAAE,EAAE;QACV,KAAK,EAAE;YACL,MAAM,EAAE,EAAE,CAAC,EAAE,CACX,KAAK,EAAE,GAAW,EAAE,KAAa,EAAE,EAAE,CACnC,IAAI,IAAI,CAAC,CAAC,IAAI,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAC3C;SACF;KACF,CAAC;IACF,MAAM,OAAO,GAA8B;QACzC,MAAM,EAAE,GAAG,EAAE,CAAC,UAAU;QACxB,KAAK,EAAE,KAAK,EAAE,CAAC,EAAE,EAAE;YACjB,CAAC,CAAC,MAAM,GAAG,OAAO,CAAC;QACrB,CAAC;QACD,cAAc,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE;YAC7B,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC;QACf,CAAC;QACD,YAAY,EAAE,KAAK,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM;KACpC,CAAC;IACF,MAAM,OAAO,GAAG,IAAI,aAAa,CAAa,EAAE,IAAI,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC;IACpE,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,UAAU,EAAE,OAAO,EAAE,CAAC;AACrD,CAAC;AAED,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;IAC5B,EAAE,CAAC,8DAA8D,EAAE,KAAK,IAAI,EAAE;QAC5E,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,UAAU,EAAE,GAAG,QAAQ,EAAE,CAAC;QAClD,MAAM,IAAI,GAAG,MAAM,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QACzC,OAAO,CAAC,MAAM,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;QACjC,UAAU,CAAC,MAAM,GAAG,MAAM,CAAC;QAC3B,MAAM,IAAI,GAAG,iBAAiB,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC,CAAC;QAC/D,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC;QAChD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,CAAC;QACnC,MAAM,CAAE,MAAM,CAAC,OAAO,CAAC,CAAC,CAAsB,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACnE,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,oBAAoB,CAClD,EAAE,EACF,YAAY,EACZ,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CACnB,CAAC;QACF,sEAAsE;QACtE,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mEAAmE,EAAE,KAAK,IAAI,EAAE;QACjF,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,QAAQ,EAAE,CAAC;QACtC,MAAM,IAAI,GAAG,iBAAiB,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC,CAAC;QAC/D,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC;QAChD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAClC,MAAM,CAAE,MAAM,CAAC,OAAO,CAAC,CAAC,CAAsB,CAAC,IAAI,CAAC,CAAC,OAAO,CAC1D,oBAAoB,CACrB,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8DAA8D,EAAE,KAAK,IAAI,EAAE;QAC5E,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,OAAO,EAAE,GAAG,QAAQ,EAAE,CAAC;QAC/C,MAAM,IAAI,GAAG,MAAM,KAAK,CAAC,KAAK,CAAC;YAC7B,KAAK,EAAE,mBAAmB;YAC1B,MAAM,EAAE,IAAI;YACZ,KAAK,EAAE,GAAG;SACX,CAAC,CAAC;QACH,MAAM,IAAI,GAAG,iBAAiB,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC,CAAC;QAC/D,oFAAoF;QACpF,MAAM,OAAO,GAAG,EAAE,CAAC,KAAK,CAAC,OAAO,EAAE,gBAAgB,CAAC,CAAC;QACpD,MAAM,IAAI,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,CAAC;QACpD,MAAM,CAAC,OAAO,CAAC,CAAC,oBAAoB,CAClC,MAAM,CAAC,QAAQ,EAAE,EACjB,mBAAmB,CACpB,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;QAC9C,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,UAAU,EAAE,GAAG,QAAQ,EAAE,CAAC;QAClD,MAAM,IAAI,GAAG,MAAM,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QACzC,OAAO,CAAC,MAAM,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;QACjC,MAAM,IAAI,GAAG,iBAAiB,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC,CAAC;QAC/D,MAAM,IAAI,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;QAC5D,MAAM,IAAI,GAAI,UAAU,CAAC,KAAK,CAAC,MAAmC,CAAC,IAAI;aACpE,KAAK,CAAC,CAAC,CAAC,CAAC;QACZ,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,EAAE,WAAW,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,CAAC,CAAC;IAC3E,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import { describe, expect, it, vi } from 'vitest';\nimport { Mutex } from '../../src/engine/mutex.js';\nimport { type EngineAdapter, EngineSession } from '../../src/engine/session.js';\nimport { RevisionStore } from '../../src/revisions/store.js';\nimport { createPreviewTool } from '../../src/tools/preview.js';\nimport { InMemoryWorkspaceFS } from '../../src/workspace/in-memory.js';\n\nconst engineLock = new Mutex();\n\ninterface FakeEngine {\n _scene: string;\n block: { export: ReturnType<typeof vi.fn> };\n}\n\nfunction makeDeps() {\n const fs = new InMemoryWorkspaceFS();\n const store = new RevisionStore(fs);\n const fakeEngine: FakeEngine = {\n _scene: '',\n block: {\n export: vi.fn(\n async (_id: number, _mime: string) =>\n new Blob([new Uint8Array([1, 2, 3, 4])])\n )\n }\n };\n const adapter: EngineAdapter<FakeEngine> = {\n create: () => fakeEngine,\n reset: async (e) => {\n e._scene = 'EMPTY';\n },\n loadFromString: async (e, s) => {\n e._scene = s;\n },\n saveToString: async (e) => e._scene\n };\n const session = new EngineSession<FakeEngine>({ size: 1, adapter });\n return { fs, store, session, fakeEngine, adapter };\n}\n\ndescribe('preview tool', () => {\n it('renders the active engine state when no revision is supplied', async () => {\n const { store, session, fakeEngine } = makeDeps();\n const slot = await session.acquire(null);\n session.commit(slot, 'rev_warm');\n fakeEngine._scene = 'WARM';\n const tool = createPreviewTool({ store, session, engineLock });\n const result = await tool.call({ blockId: 42 });\n expect(result.isError).toBeFalsy();\n expect((result.content[0] as { type: string }).type).toBe('image');\n expect(fakeEngine.block.export).toHaveBeenCalledWith(\n 42,\n 'image/webp',\n expect.any(Object)\n );\n // Engine state is preserved (NOT reset) when no revision is supplied.\n expect(fakeEngine._scene).toBe('WARM');\n });\n\n it('returns isError when no revision supplied AND no warm slot exists', async () => {\n const { store, session } = makeDeps();\n const tool = createPreviewTool({ store, session, engineLock });\n const result = await tool.call({ blockId: 42 });\n expect(result.isError).toBe(true);\n expect((result.content[0] as { text: string }).text).toMatch(\n /no.*engine.*state/i\n );\n });\n\n it('loads a revision into the engine when `revision` is supplied', async () => {\n const { store, session, adapter } = makeDeps();\n const root = await store.write({\n scene: 'SCENE_FOR_PREVIEW',\n parent: null,\n title: 'T'\n });\n const tool = createPreviewTool({ store, session, engineLock });\n // adapter.loadFromString is set as a plain async fn (not a vi.fn) above, so wrap it\n const loadSpy = vi.spyOn(adapter, 'loadFromString');\n await tool.call({ blockId: 42, revision: root.id });\n expect(loadSpy).toHaveBeenCalledWith(\n expect.anything(),\n 'SCENE_FOR_PREVIEW'\n );\n });\n\n it('clamps width/height to 1800 px', async () => {\n const { store, session, fakeEngine } = makeDeps();\n const slot = await session.acquire(null);\n session.commit(slot, 'rev_warm');\n const tool = createPreviewTool({ store, session, engineLock });\n await tool.call({ blockId: 42, width: 9999, height: 9999 });\n const call = (fakeEngine.block.export as ReturnType<typeof vi.fn>).mock\n .calls[0];\n expect(call[2]).toMatchObject({ targetWidth: 1800, targetHeight: 1800 });\n });\n});\n"]}
|
|
@@ -10,8 +10,7 @@ describe('rename tool', () => {
|
|
|
10
10
|
const tool = createRenameTool({ store });
|
|
11
11
|
const result = await tool.call({ revision: r2.id, title: 'New' });
|
|
12
12
|
expect(result.content[0].text).toContain('ok');
|
|
13
|
-
|
|
14
|
-
expect(root.meta.title).toBe('New');
|
|
13
|
+
expect(await store.currentTitle(r1.id)).toBe('New');
|
|
15
14
|
});
|
|
16
15
|
});
|
|
17
16
|
//# sourceMappingURL=rename.test.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"rename.test.js","sourceRoot":"","sources":["../../../../test/tools/rename.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,aAAa,EAAE,MAAM,8BAA8B,CAAC;AAC7D,OAAO,EAAE,gBAAgB,EAAE,MAAM,2BAA2B,CAAC;AAC7D,OAAO,EAAE,mBAAmB,EAAE,MAAM,kCAAkC,CAAC;AAEvE,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;IAC3B,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;QAClE,MAAM,KAAK,GAAG,IAAI,aAAa,CAAC,IAAI,mBAAmB,EAAE,CAAC,CAAC;QAC3D,MAAM,EAAE,GAAG,MAAM,KAAK,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC;QACzE,MAAM,EAAE,GAAG,MAAM,KAAK,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;QAC5D,MAAM,IAAI,GAAG,gBAAgB,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC;QACzC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,EAAE,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC;QAClE,MAAM,CAAE,MAAM,CAAC,OAAO,CAAC,CAAC,CAAsB,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;QACrE,MAAM,
|
|
1
|
+
{"version":3,"file":"rename.test.js","sourceRoot":"","sources":["../../../../test/tools/rename.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,aAAa,EAAE,MAAM,8BAA8B,CAAC;AAC7D,OAAO,EAAE,gBAAgB,EAAE,MAAM,2BAA2B,CAAC;AAC7D,OAAO,EAAE,mBAAmB,EAAE,MAAM,kCAAkC,CAAC;AAEvE,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;IAC3B,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;QAClE,MAAM,KAAK,GAAG,IAAI,aAAa,CAAC,IAAI,mBAAmB,EAAE,CAAC,CAAC;QAC3D,MAAM,EAAE,GAAG,MAAM,KAAK,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC;QACzE,MAAM,EAAE,GAAG,MAAM,KAAK,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;QAC5D,MAAM,IAAI,GAAG,gBAAgB,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC;QACzC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,EAAE,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC;QAClE,MAAM,CAAE,MAAM,CAAC,OAAO,CAAC,CAAC,CAAsB,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;QACrE,MAAM,CAAC,MAAM,KAAK,CAAC,YAAY,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACtD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import { describe, expect, it } from 'vitest';\nimport { RevisionStore } from '../../src/revisions/store.js';\nimport { createRenameTool } from '../../src/tools/rename.js';\nimport { InMemoryWorkspaceFS } from '../../src/workspace/in-memory.js';\n\ndescribe('rename tool', () => {\n it('renames the root reachable from the given revision', async () => {\n const store = new RevisionStore(new InMemoryWorkspaceFS());\n const r1 = await store.write({ scene: 'a', parent: null, title: 'Old' });\n const r2 = await store.write({ scene: 'b', parent: r1.id });\n const tool = createRenameTool({ store });\n const result = await tool.call({ revision: r2.id, title: 'New' });\n expect((result.content[0] as { text: string }).text).toContain('ok');\n expect(await store.currentTitle(r1.id)).toBe('New');\n });\n});\n"]}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"save.test.d.ts","sourceRoot":"","sources":["../../../../test/tools/save.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,137 @@
|
|
|
1
|
+
import { describe, expect, it, vi } from 'vitest';
|
|
2
|
+
import { Mutex } from '../../src/engine/mutex.js';
|
|
3
|
+
import { EngineSession } from '../../src/engine/session.js';
|
|
4
|
+
import { RevisionStore } from '../../src/revisions/store.js';
|
|
5
|
+
import { createSaveTool } from '../../src/tools/save.js';
|
|
6
|
+
import { LocalViewerUrlBuilder } from '../../src/viewer/url-builder.js';
|
|
7
|
+
import { InMemoryWorkspaceFS } from '../../src/workspace/in-memory.js';
|
|
8
|
+
const engineLock = new Mutex();
|
|
9
|
+
const urlBuilder = new LocalViewerUrlBuilder({ base: 'http://127.0.0.1:9000' });
|
|
10
|
+
function makeDeps() {
|
|
11
|
+
const fs = new InMemoryWorkspaceFS();
|
|
12
|
+
const store = new RevisionStore(fs);
|
|
13
|
+
const fakeEngine = {
|
|
14
|
+
_scene: '',
|
|
15
|
+
_resourcesLoaded: false,
|
|
16
|
+
scene: {
|
|
17
|
+
getPages: vi.fn(() => [10]),
|
|
18
|
+
// The archive bundles the asset marker ONLY if resources were loaded
|
|
19
|
+
// first — mirrors saveToArchive's "assets accessible when called" contract.
|
|
20
|
+
saveToArchive: vi.fn(async () => {
|
|
21
|
+
const body = fakeEngine._resourcesLoaded ? 'ZIP+ASSET' : 'ZIP';
|
|
22
|
+
return new Blob([new TextEncoder().encode(body)]);
|
|
23
|
+
})
|
|
24
|
+
},
|
|
25
|
+
block: {
|
|
26
|
+
forceLoadResources: vi.fn(async () => {
|
|
27
|
+
fakeEngine._resourcesLoaded = true;
|
|
28
|
+
})
|
|
29
|
+
}
|
|
30
|
+
};
|
|
31
|
+
const adapter = {
|
|
32
|
+
create: () => fakeEngine,
|
|
33
|
+
reset: async (e) => {
|
|
34
|
+
e._scene = 'EMPTY';
|
|
35
|
+
},
|
|
36
|
+
loadFromString: async (e, s) => {
|
|
37
|
+
e._scene = s;
|
|
38
|
+
},
|
|
39
|
+
saveToString: async (e) => e._scene
|
|
40
|
+
};
|
|
41
|
+
const session = new EngineSession({ size: 1, adapter });
|
|
42
|
+
return { fs, store, session, fakeEngine };
|
|
43
|
+
}
|
|
44
|
+
describe('save tool', () => {
|
|
45
|
+
it('writes archive.zip and returns uri + httpUrl, no path', async () => {
|
|
46
|
+
const { fs, store, session } = makeDeps();
|
|
47
|
+
const root = await store.write({
|
|
48
|
+
scene: 'SCENE_ABC',
|
|
49
|
+
parent: null,
|
|
50
|
+
title: 'T'
|
|
51
|
+
});
|
|
52
|
+
const tool = createSaveTool({ fs, store, session, engineLock, urlBuilder });
|
|
53
|
+
const result = await tool.call({ revision: root.id });
|
|
54
|
+
const body = JSON.parse(result.content[0].text);
|
|
55
|
+
const shard = root.id.slice(0, 2);
|
|
56
|
+
const rest = root.id.slice(2);
|
|
57
|
+
expect(body.uri).toBe(`workspace://exports/${shard}/${rest}/archive.zip`);
|
|
58
|
+
expect(body.httpUrl).toBe(`http://127.0.0.1:9000/r/exports/${shard}/${rest}/archive.zip`);
|
|
59
|
+
expect(body.path).toBeUndefined();
|
|
60
|
+
expect(body.format).toBe('archive');
|
|
61
|
+
expect(body.revision).toBe(root.id);
|
|
62
|
+
expect(await fs.exists(`exports/${shard}/${rest}/archive.zip`)).toBe(true);
|
|
63
|
+
});
|
|
64
|
+
it('respects local `outPath` escape hatch — returns path, not httpUrl', async () => {
|
|
65
|
+
const { fs, store, session } = makeDeps();
|
|
66
|
+
const root = await store.write({ scene: 'S', parent: null, title: 'T' });
|
|
67
|
+
const { mkdtemp, rm } = await import('node:fs/promises');
|
|
68
|
+
const { tmpdir } = await import('node:os');
|
|
69
|
+
const { join } = await import('node:path');
|
|
70
|
+
const dir = await mkdtemp(join(tmpdir(), 'save-out-'));
|
|
71
|
+
const outPath = join(dir, 'design.zip');
|
|
72
|
+
const tool = createSaveTool({
|
|
73
|
+
fs,
|
|
74
|
+
store,
|
|
75
|
+
session,
|
|
76
|
+
engineLock,
|
|
77
|
+
urlBuilder,
|
|
78
|
+
allowOutPath: true
|
|
79
|
+
});
|
|
80
|
+
const result = await tool.call({ revision: root.id, outPath });
|
|
81
|
+
const body = JSON.parse(result.content[0].text);
|
|
82
|
+
expect(body.path).toBe(outPath);
|
|
83
|
+
expect(body.httpUrl).toBeUndefined();
|
|
84
|
+
await rm(dir, { recursive: true });
|
|
85
|
+
});
|
|
86
|
+
it('rejects `outPath` when not allowed (Hosted/sandbox)', async () => {
|
|
87
|
+
const { fs, store, session } = makeDeps();
|
|
88
|
+
const root = await store.write({ scene: 'S', parent: null, title: 'T' });
|
|
89
|
+
const tool = createSaveTool({
|
|
90
|
+
fs,
|
|
91
|
+
store,
|
|
92
|
+
session,
|
|
93
|
+
engineLock,
|
|
94
|
+
urlBuilder,
|
|
95
|
+
allowOutPath: false
|
|
96
|
+
});
|
|
97
|
+
const result = await tool.call({
|
|
98
|
+
revision: root.id,
|
|
99
|
+
outPath: '/tmp/x.zip'
|
|
100
|
+
});
|
|
101
|
+
expect(result.isError).toBe(true);
|
|
102
|
+
expect(result.content[0].text).toMatch(/outPath.*not.*supported/i);
|
|
103
|
+
});
|
|
104
|
+
it('omits `outPath` from the schema when not allowed (Hosted), advertises it when allowed (Local)', () => {
|
|
105
|
+
const { fs, store, session } = makeDeps();
|
|
106
|
+
const base = { fs, store, session, engineLock, urlBuilder };
|
|
107
|
+
const hosted = createSaveTool({ ...base, allowOutPath: false });
|
|
108
|
+
const local = createSaveTool({ ...base, allowOutPath: true });
|
|
109
|
+
const props = (s) => s.properties;
|
|
110
|
+
expect(props(hosted.inputSchema)).not.toHaveProperty('outPath');
|
|
111
|
+
expect(props(local.inputSchema)).toHaveProperty('outPath');
|
|
112
|
+
});
|
|
113
|
+
it('errors when revision is missing', async () => {
|
|
114
|
+
const { fs, store, session } = makeDeps();
|
|
115
|
+
const tool = createSaveTool({ fs, store, session, engineLock, urlBuilder });
|
|
116
|
+
const result = await tool.call({});
|
|
117
|
+
expect(result.isError).toBe(true);
|
|
118
|
+
expect(result.content[0].text).toMatch(/revision/i);
|
|
119
|
+
});
|
|
120
|
+
it('force-loads resources before archiving so assets are bundled', async () => {
|
|
121
|
+
const { fs, store, session, fakeEngine } = makeDeps();
|
|
122
|
+
const root = await store.write({ scene: 'S', parent: null, title: 'T' });
|
|
123
|
+
const tool = createSaveTool({ fs, store, session, engineLock, urlBuilder });
|
|
124
|
+
await tool.call({ revision: root.id });
|
|
125
|
+
// forceLoadResources must run with the scene's pages...
|
|
126
|
+
expect(fakeEngine.block.forceLoadResources).toHaveBeenCalledWith([10]);
|
|
127
|
+
// ...and BEFORE saveToArchive.
|
|
128
|
+
const callOrder = (m) => m.mock.invocationCallOrder[0];
|
|
129
|
+
expect(callOrder(fakeEngine.block.forceLoadResources)).toBeLessThan(callOrder(fakeEngine.scene.saveToArchive));
|
|
130
|
+
// ...so the produced archive actually contains the asset bytes.
|
|
131
|
+
const shard = root.id.slice(0, 2);
|
|
132
|
+
const rest = root.id.slice(2);
|
|
133
|
+
const zip = await fs.readFile(`exports/${shard}/${rest}/archive.zip`, 'utf8');
|
|
134
|
+
expect(zip).toContain('ASSET');
|
|
135
|
+
});
|
|
136
|
+
});
|
|
137
|
+
//# sourceMappingURL=save.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"save.test.js","sourceRoot":"","sources":["../../../../test/tools/save.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAClD,OAAO,EAAE,KAAK,EAAE,MAAM,2BAA2B,CAAC;AAClD,OAAO,EAAsB,aAAa,EAAE,MAAM,6BAA6B,CAAC;AAChF,OAAO,EAAE,aAAa,EAAE,MAAM,8BAA8B,CAAC;AAC7D,OAAO,EAAE,cAAc,EAAE,MAAM,yBAAyB,CAAC;AACzD,OAAO,EAAE,qBAAqB,EAAE,MAAM,iCAAiC,CAAC;AACxE,OAAO,EAAE,mBAAmB,EAAE,MAAM,kCAAkC,CAAC;AAEvE,MAAM,UAAU,GAAG,IAAI,KAAK,EAAE,CAAC;AAC/B,MAAM,UAAU,GAAG,IAAI,qBAAqB,CAAC,EAAE,IAAI,EAAE,uBAAuB,EAAE,CAAC,CAAC;AAYhF,SAAS,QAAQ;IACf,MAAM,EAAE,GAAG,IAAI,mBAAmB,EAAE,CAAC;IACrC,MAAM,KAAK,GAAG,IAAI,aAAa,CAAC,EAAE,CAAC,CAAC;IACpC,MAAM,UAAU,GAAe;QAC7B,MAAM,EAAE,EAAE;QACV,gBAAgB,EAAE,KAAK;QACvB,KAAK,EAAE;YACL,QAAQ,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;YAC3B,qEAAqE;YACrE,4EAA4E;YAC5E,aAAa,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE;gBAC9B,MAAM,IAAI,GAAG,UAAU,CAAC,gBAAgB,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,KAAK,CAAC;gBAC/D,OAAO,IAAI,IAAI,CAAC,CAAC,IAAI,WAAW,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YACpD,CAAC,CAAC;SACH;QACD,KAAK,EAAE;YACL,kBAAkB,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE;gBACnC,UAAU,CAAC,gBAAgB,GAAG,IAAI,CAAC;YACrC,CAAC,CAAC;SACH;KACF,CAAC;IACF,MAAM,OAAO,GAA8B;QACzC,MAAM,EAAE,GAAG,EAAE,CAAC,UAAU;QACxB,KAAK,EAAE,KAAK,EAAE,CAAC,EAAE,EAAE;YACjB,CAAC,CAAC,MAAM,GAAG,OAAO,CAAC;QACrB,CAAC;QACD,cAAc,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE;YAC7B,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC;QACf,CAAC;QACD,YAAY,EAAE,KAAK,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM;KACpC,CAAC;IACF,MAAM,OAAO,GAAG,IAAI,aAAa,CAAa,EAAE,IAAI,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC;IACpE,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC;AAC5C,CAAC;AAED,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;IACzB,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;QACrE,MAAM,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,QAAQ,EAAE,CAAC;QAC1C,MAAM,IAAI,GAAG,MAAM,KAAK,CAAC,KAAK,CAAC;YAC7B,KAAK,EAAE,WAAW;YAClB,MAAM,EAAE,IAAI;YACZ,KAAK,EAAE,GAAG;SACX,CAAC,CAAC;QACH,MAAM,IAAI,GAAG,cAAc,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,CAAC,CAAC;QAC5E,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,EAAE,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,CAAC;QACtD,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAE,MAAM,CAAC,OAAO,CAAC,CAAC,CAAsB,CAAC,IAAI,CAOnE,CAAC;QACF,MAAM,KAAK,GAAG,IAAI,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QAClC,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAC9B,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,uBAAuB,KAAK,IAAI,IAAI,cAAc,CAAC,CAAC;QAC1E,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,IAAI,CACvB,mCAAmC,KAAK,IAAI,IAAI,cAAc,CAC/D,CAAC;QACF,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,aAAa,EAAE,CAAC;QAClC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACpC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACpC,MAAM,CAAC,MAAM,EAAE,CAAC,MAAM,CAAC,WAAW,KAAK,IAAI,IAAI,cAAc,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC7E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mEAAmE,EAAE,KAAK,IAAI,EAAE;QACjF,MAAM,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,QAAQ,EAAE,CAAC;QAC1C,MAAM,IAAI,GAAG,MAAM,KAAK,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC;QACzE,MAAM,EAAE,OAAO,EAAE,EAAE,EAAE,GAAG,MAAM,MAAM,CAAC,kBAAkB,CAAC,CAAC;QACzD,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,MAAM,CAAC,SAAS,CAAC,CAAC;QAC3C,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,MAAM,CAAC,WAAW,CAAC,CAAC;QAC3C,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,WAAW,CAAC,CAAC,CAAC;QACvD,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,YAAY,CAAC,CAAC;QACxC,MAAM,IAAI,GAAG,cAAc,CAAC;YAC1B,EAAE;YACF,KAAK;YACL,OAAO;YACP,UAAU;YACV,UAAU;YACV,YAAY,EAAE,IAAI;SACnB,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,EAAE,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;QAC/D,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAE,MAAM,CAAC,OAAO,CAAC,CAAC,CAAsB,CAAC,IAAI,CAGnE,CAAC;QACF,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAChC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,aAAa,EAAE,CAAC;QACrC,MAAM,EAAE,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IACrC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;QACnE,MAAM,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,QAAQ,EAAE,CAAC;QAC1C,MAAM,IAAI,GAAG,MAAM,KAAK,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC;QACzE,MAAM,IAAI,GAAG,cAAc,CAAC;YAC1B,EAAE;YACF,KAAK;YACL,OAAO;YACP,UAAU;YACV,UAAU;YACV,YAAY,EAAE,KAAK;SACpB,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC;YAC7B,QAAQ,EAAE,IAAI,CAAC,EAAE;YACjB,OAAO,EAAE,YAAY;SACtB,CAAC,CAAC;QACH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAClC,MAAM,CAAE,MAAM,CAAC,OAAO,CAAC,CAAC,CAAsB,CAAC,IAAI,CAAC,CAAC,OAAO,CAC1D,0BAA0B,CAC3B,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+FAA+F,EAAE,GAAG,EAAE;QACvG,MAAM,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,QAAQ,EAAE,CAAC;QAC1C,MAAM,IAAI,GAAG,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,CAAC;QAC5D,MAAM,MAAM,GAAG,cAAc,CAAC,EAAE,GAAG,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,CAAC,CAAC;QAChE,MAAM,KAAK,GAAG,cAAc,CAAC,EAAE,GAAG,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,CAAC,CAAC;QAC9D,MAAM,KAAK,GAAG,CAAC,CAAU,EAAE,EAAE,CAC1B,CAA6C,CAAC,UAAU,CAAC;QAC5D,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,GAAG,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC;QAChE,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC;IAC7D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iCAAiC,EAAE,KAAK,IAAI,EAAE;QAC/C,MAAM,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,QAAQ,EAAE,CAAC;QAC1C,MAAM,IAAI,GAAG,cAAc,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,CAAC,CAAC;QAC5E,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACnC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAClC,MAAM,CAAE,MAAM,CAAC,OAAO,CAAC,CAAC,CAAsB,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;IAC5E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8DAA8D,EAAE,KAAK,IAAI,EAAE;QAC5E,MAAM,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,UAAU,EAAE,GAAG,QAAQ,EAAE,CAAC;QACtD,MAAM,IAAI,GAAG,MAAM,KAAK,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC;QACzE,MAAM,IAAI,GAAG,cAAc,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,CAAC,CAAC;QAC5E,MAAM,IAAI,CAAC,IAAI,CAAC,EAAE,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,CAAC;QAEvC,wDAAwD;QACxD,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,kBAAkB,CAAC,CAAC,oBAAoB,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QACvE,+BAA+B;QAC/B,MAAM,SAAS,GAAG,CAAC,CAA2B,EAAU,EAAE,CACxD,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC,CAAC,CAAC;QAChC,MAAM,CAAC,SAAS,CAAC,UAAU,CAAC,KAAK,CAAC,kBAAkB,CAAC,CAAC,CAAC,YAAY,CACjE,SAAS,CAAC,UAAU,CAAC,KAAK,CAAC,aAAa,CAAC,CAC1C,CAAC;QAEF,gEAAgE;QAChE,MAAM,KAAK,GAAG,IAAI,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QAClC,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAC9B,MAAM,GAAG,GAAG,MAAM,EAAE,CAAC,QAAQ,CAC3B,WAAW,KAAK,IAAI,IAAI,cAAc,EACtC,MAAM,CACP,CAAC;QACF,MAAM,CAAC,GAAG,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;IACjC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import { describe, expect, it, vi } from 'vitest';\nimport { Mutex } from '../../src/engine/mutex.js';\nimport { type EngineAdapter, EngineSession } from '../../src/engine/session.js';\nimport { RevisionStore } from '../../src/revisions/store.js';\nimport { createSaveTool } from '../../src/tools/save.js';\nimport { LocalViewerUrlBuilder } from '../../src/viewer/url-builder.js';\nimport { InMemoryWorkspaceFS } from '../../src/workspace/in-memory.js';\n\nconst engineLock = new Mutex();\nconst urlBuilder = new LocalViewerUrlBuilder({ base: 'http://127.0.0.1:9000' });\n\ninterface FakeEngine {\n _scene: string;\n _resourcesLoaded: boolean;\n scene: {\n getPages: ReturnType<typeof vi.fn>;\n saveToArchive: ReturnType<typeof vi.fn>;\n };\n block: { forceLoadResources: ReturnType<typeof vi.fn> };\n}\n\nfunction makeDeps() {\n const fs = new InMemoryWorkspaceFS();\n const store = new RevisionStore(fs);\n const fakeEngine: FakeEngine = {\n _scene: '',\n _resourcesLoaded: false,\n scene: {\n getPages: vi.fn(() => [10]),\n // The archive bundles the asset marker ONLY if resources were loaded\n // first — mirrors saveToArchive's \"assets accessible when called\" contract.\n saveToArchive: vi.fn(async () => {\n const body = fakeEngine._resourcesLoaded ? 'ZIP+ASSET' : 'ZIP';\n return new Blob([new TextEncoder().encode(body)]);\n })\n },\n block: {\n forceLoadResources: vi.fn(async () => {\n fakeEngine._resourcesLoaded = true;\n })\n }\n };\n const adapter: EngineAdapter<FakeEngine> = {\n create: () => fakeEngine,\n reset: async (e) => {\n e._scene = 'EMPTY';\n },\n loadFromString: async (e, s) => {\n e._scene = s;\n },\n saveToString: async (e) => e._scene\n };\n const session = new EngineSession<FakeEngine>({ size: 1, adapter });\n return { fs, store, session, fakeEngine };\n}\n\ndescribe('save tool', () => {\n it('writes archive.zip and returns uri + httpUrl, no path', async () => {\n const { fs, store, session } = makeDeps();\n const root = await store.write({\n scene: 'SCENE_ABC',\n parent: null,\n title: 'T'\n });\n const tool = createSaveTool({ fs, store, session, engineLock, urlBuilder });\n const result = await tool.call({ revision: root.id });\n const body = JSON.parse((result.content[0] as { text: string }).text) as {\n uri: string;\n httpUrl?: string;\n path?: string;\n bytes: number;\n format: string;\n revision: string;\n };\n const shard = root.id.slice(0, 2);\n const rest = root.id.slice(2);\n expect(body.uri).toBe(`workspace://exports/${shard}/${rest}/archive.zip`);\n expect(body.httpUrl).toBe(\n `http://127.0.0.1:9000/r/exports/${shard}/${rest}/archive.zip`\n );\n expect(body.path).toBeUndefined();\n expect(body.format).toBe('archive');\n expect(body.revision).toBe(root.id);\n expect(await fs.exists(`exports/${shard}/${rest}/archive.zip`)).toBe(true);\n });\n\n it('respects local `outPath` escape hatch — returns path, not httpUrl', async () => {\n const { fs, store, session } = makeDeps();\n const root = await store.write({ scene: 'S', parent: null, title: 'T' });\n const { mkdtemp, rm } = await import('node:fs/promises');\n const { tmpdir } = await import('node:os');\n const { join } = await import('node:path');\n const dir = await mkdtemp(join(tmpdir(), 'save-out-'));\n const outPath = join(dir, 'design.zip');\n const tool = createSaveTool({\n fs,\n store,\n session,\n engineLock,\n urlBuilder,\n allowOutPath: true\n });\n const result = await tool.call({ revision: root.id, outPath });\n const body = JSON.parse((result.content[0] as { text: string }).text) as {\n path?: string;\n httpUrl?: string;\n };\n expect(body.path).toBe(outPath);\n expect(body.httpUrl).toBeUndefined();\n await rm(dir, { recursive: true });\n });\n\n it('rejects `outPath` when not allowed (Hosted/sandbox)', async () => {\n const { fs, store, session } = makeDeps();\n const root = await store.write({ scene: 'S', parent: null, title: 'T' });\n const tool = createSaveTool({\n fs,\n store,\n session,\n engineLock,\n urlBuilder,\n allowOutPath: false\n });\n const result = await tool.call({\n revision: root.id,\n outPath: '/tmp/x.zip'\n });\n expect(result.isError).toBe(true);\n expect((result.content[0] as { text: string }).text).toMatch(\n /outPath.*not.*supported/i\n );\n });\n\n it('omits `outPath` from the schema when not allowed (Hosted), advertises it when allowed (Local)', () => {\n const { fs, store, session } = makeDeps();\n const base = { fs, store, session, engineLock, urlBuilder };\n const hosted = createSaveTool({ ...base, allowOutPath: false });\n const local = createSaveTool({ ...base, allowOutPath: true });\n const props = (s: unknown) =>\n (s as { properties: Record<string, unknown> }).properties;\n expect(props(hosted.inputSchema)).not.toHaveProperty('outPath');\n expect(props(local.inputSchema)).toHaveProperty('outPath');\n });\n\n it('errors when revision is missing', async () => {\n const { fs, store, session } = makeDeps();\n const tool = createSaveTool({ fs, store, session, engineLock, urlBuilder });\n const result = await tool.call({});\n expect(result.isError).toBe(true);\n expect((result.content[0] as { text: string }).text).toMatch(/revision/i);\n });\n\n it('force-loads resources before archiving so assets are bundled', async () => {\n const { fs, store, session, fakeEngine } = makeDeps();\n const root = await store.write({ scene: 'S', parent: null, title: 'T' });\n const tool = createSaveTool({ fs, store, session, engineLock, urlBuilder });\n await tool.call({ revision: root.id });\n\n // forceLoadResources must run with the scene's pages...\n expect(fakeEngine.block.forceLoadResources).toHaveBeenCalledWith([10]);\n // ...and BEFORE saveToArchive.\n const callOrder = (m: ReturnType<typeof vi.fn>): number =>\n m.mock.invocationCallOrder[0];\n expect(callOrder(fakeEngine.block.forceLoadResources)).toBeLessThan(\n callOrder(fakeEngine.scene.saveToArchive)\n );\n\n // ...so the produced archive actually contains the asset bytes.\n const shard = root.id.slice(0, 2);\n const rest = root.id.slice(2);\n const zip = await fs.readFile(\n `exports/${shard}/${rest}/archive.zip`,\n 'utf8'\n );\n expect(zip).toContain('ASSET');\n });\n});\n"]}
|