@imgly/codesign-mcp 0.1.1 → 0.1.3
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 +82 -20
- 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/dist/tools/edit.d.ts.map +1 -1
- package/dist/tools/edit.js +6 -13
- package/dist/tools/edit.js.map +1 -1
- package/package.json +1 -1
|
@@ -6,8 +6,8 @@ const FIXTURES = {
|
|
|
6
6
|
'docs/alpha/SKILL.md': '---\nname: alpha\ndescription: First skill.\n---\n# Alpha\nSee [more](more.md).',
|
|
7
7
|
'docs/alpha/more.md': '---\nname: more\ndescription: details\n---\nMore details.'
|
|
8
8
|
};
|
|
9
|
-
async function connect(assets = FIXTURES) {
|
|
10
|
-
const server = createCodesignMcpServer({
|
|
9
|
+
async function connect(assets = FIXTURES, extra = {}) {
|
|
10
|
+
const server = createCodesignMcpServer({ assets, ...extra });
|
|
11
11
|
const [clientT, serverT] = InMemoryTransport.createLinkedPair();
|
|
12
12
|
await server.connect(serverT);
|
|
13
13
|
const client = new Client({ name: 'test', version: '0' });
|
|
@@ -15,6 +15,14 @@ async function connect(assets = FIXTURES) {
|
|
|
15
15
|
return { client, server };
|
|
16
16
|
}
|
|
17
17
|
describe('createCodesignMcpServer (initialize + resources)', () => {
|
|
18
|
+
it('reports buildId as serverInfo.version (defaulting to "dev")', async () => {
|
|
19
|
+
const def = await connect();
|
|
20
|
+
expect(def.client.getServerVersion()?.version).toBe('dev');
|
|
21
|
+
// The deploy injects a git SHA; the smoketest asserts it to detect a
|
|
22
|
+
// stale, not-yet-rolled-out container.
|
|
23
|
+
const stamped = await connect(FIXTURES, { buildId: 'abc1234' });
|
|
24
|
+
expect(stamped.client.getServerVersion()?.version).toBe('abc1234');
|
|
25
|
+
});
|
|
18
26
|
it('instructions advertise the codesign tools + force-direct the model to read codesign-system first', async () => {
|
|
19
27
|
const { client } = await connect();
|
|
20
28
|
const instructions = client.getInstructions() ?? '';
|
|
@@ -28,7 +36,9 @@ describe('createCodesignMcpServer (initialize + resources)', () => {
|
|
|
28
36
|
expect(instructions).toContain('inspect');
|
|
29
37
|
expect(instructions).toContain('rename');
|
|
30
38
|
expect(instructions).toContain('open_editor');
|
|
31
|
-
|
|
39
|
+
// asset_generate is opt-in: with no tools registered (no gateway), the
|
|
40
|
+
// hint is omitted so we never advertise a hidden tool.
|
|
41
|
+
expect(instructions).not.toContain('asset_generate');
|
|
32
42
|
expect(instructions).toContain('parent');
|
|
33
43
|
// Hard direction: read handbook before any design call.
|
|
34
44
|
expect(instructions).toContain('handbook');
|
|
@@ -39,6 +49,26 @@ describe('createCodesignMcpServer (initialize + resources)', () => {
|
|
|
39
49
|
// truncation (target: well under 4 KB).
|
|
40
50
|
expect(instructions.length).toBeLessThan(4096);
|
|
41
51
|
});
|
|
52
|
+
it('includes the asset-generation hint only when an asset_generate tool is registered', async () => {
|
|
53
|
+
const server = createCodesignMcpServer({
|
|
54
|
+
assets: FIXTURES,
|
|
55
|
+
tools: [
|
|
56
|
+
{
|
|
57
|
+
name: 'asset_generate',
|
|
58
|
+
description: 'gen',
|
|
59
|
+
inputSchema: { type: 'object', properties: {} },
|
|
60
|
+
call: async () => ({ content: [] })
|
|
61
|
+
}
|
|
62
|
+
]
|
|
63
|
+
});
|
|
64
|
+
const [clientT, serverT] = InMemoryTransport.createLinkedPair();
|
|
65
|
+
await server.connect(serverT);
|
|
66
|
+
const client = new Client({ name: 'test', version: '0' });
|
|
67
|
+
await client.connect(clientT);
|
|
68
|
+
const instructions = client.getInstructions() ?? '';
|
|
69
|
+
expect(instructions).toContain('Asset generation');
|
|
70
|
+
expect(instructions).toContain('asset_generate');
|
|
71
|
+
});
|
|
42
72
|
it('resources/list returns SKILL.md entry points (sub-files reachable via the read tools)', async () => {
|
|
43
73
|
const { client } = await connect();
|
|
44
74
|
const list = await client.listResources();
|
|
@@ -109,7 +139,6 @@ describe('createCodesignMcpServer (tools)', () => {
|
|
|
109
139
|
const { Client } = await import('@modelcontextprotocol/sdk/client/index.js');
|
|
110
140
|
const { InMemoryTransport } = await import('@modelcontextprotocol/sdk/inMemory.js');
|
|
111
141
|
const server = createCodesignMcpServer({
|
|
112
|
-
engine: {},
|
|
113
142
|
assets: { 'agent/SYSTEM.md': 'sys' },
|
|
114
143
|
tools: [
|
|
115
144
|
{
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"server.test.js","sourceRoot":"","sources":["../../../test/server.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAC;AACnE,OAAO,EAAE,iBAAiB,EAAE,MAAM,uCAAuC,CAAC;AAC1E,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,uBAAuB,EAAE,MAAM,kBAAkB,CAAC;AAE3D,MAAM,QAAQ,GAAG;IACf,qBAAqB,EACnB,iFAAiF;IACnF,oBAAoB,EAClB,2DAA2D;CAC9D,CAAC;AAEF,KAAK,UAAU,OAAO,CAAC,SAAiC,QAAQ;IAC9D,MAAM,MAAM,GAAG,uBAAuB,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;IAC/D,MAAM,CAAC,OAAO,EAAE,OAAO,CAAC,GAAG,iBAAiB,CAAC,gBAAgB,EAAE,CAAC;IAChE,MAAM,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;IAC9B,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,EAAE,CAAC,CAAC;IAC1D,MAAM,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;IAC9B,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC;AAC5B,CAAC;AAED,QAAQ,CAAC,kDAAkD,EAAE,GAAG,EAAE;IAChE,EAAE,CAAC,kGAAkG,EAAE,KAAK,IAAI,EAAE;QAChH,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,OAAO,EAAE,CAAC;QACnC,MAAM,YAAY,GAAG,MAAM,CAAC,eAAe,EAAE,IAAI,EAAE,CAAC;QACpD,4DAA4D;QAC5D,MAAM,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,qCAAqC,CAAC,CAAC;QACtE,MAAM,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QACvC,MAAM,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;QAC1C,MAAM,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;QACzC,MAAM,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QACvC,MAAM,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;QAC1C,MAAM,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;QAC1C,MAAM,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;QACzC,MAAM,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC;QAC9C,MAAM,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;QACjD,MAAM,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;QACzC,wDAAwD;QACxD,MAAM,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;QAC3C,MAAM,CAAC,YAAY,CAAC,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC;QAChD,wEAAwE;QACxE,MAAM,CAAC,YAAY,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAC;QACxD,sEAAsE;QACtE,wCAAwC;QACxC,MAAM,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;IACjD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uFAAuF,EAAE,KAAK,IAAI,EAAE;QACrG,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,OAAO,EAAE,CAAC;QACnC,MAAM,IAAI,GAAG,MAAM,MAAM,CAAC,aAAa,EAAE,CAAC;QAC1C,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC;QACrD,sEAAsE;QACtE,sEAAsE;QACtE,MAAM,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,gCAAgC,CAAC,CAAC,CAAC;IAC3D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;QAChE,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,OAAO,EAAE,CAAC;QACnC,MAAM,GAAG,GAAG,MAAM,MAAM,CAAC,YAAY,CAAC;YACpC,GAAG,EAAE,gCAAgC;SACtC,CAAC,CAAC;QACH,MAAM,IAAI,GAAI,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAuB,CAAC,IAAI,CAAC;QACzD,MAAM,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,uCAAuC,CAAC,CAAC;QAChE,MAAM,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;IACpC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;QAChE,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,OAAO,EAAE,CAAC;QACnC,MAAM,MAAM,CACV,MAAM,CAAC,YAAY,CAAC,EAAE,GAAG,EAAE,iCAAiC,EAAE,CAAC,CAChE,CAAC,OAAO,CAAC,aAAa,CAAC;YACtB,IAAI,EAAE,CAAC,KAAK;SACb,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,MAAM,eAAe,GAAG;IACtB,iBAAiB,EAAE,KAAK;IACxB,oBAAoB,EAClB,yFAAyF;IAC3F,mBAAmB,EACjB,yEAAyE;CAC5E,CAAC;AAEF,QAAQ,CAAC,mCAAmC,EAAE,GAAG,EAAE;IACjD,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;QACxD,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,OAAO,CAAC,eAAe,CAAC,CAAC;QAClD,MAAM,IAAI,GAAG,MAAM,MAAM,CAAC,WAAW,EAAE,CAAC;QACxC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC;YACrD,SAAS;YACT,QAAQ;SACT,CAAC,CAAC;QACH,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC;QAC/D,MAAM,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;IACvD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;QAC5D,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,OAAO,CAAC,eAAe,CAAC,CAAC;QAClD,MAAM,GAAG,GAAG,MAAM,MAAM,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,CAAC;QACxD,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACrC,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC1C,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,aAAa,CAAC;YAC5C,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,MAAM,CAAC,gBAAgB,CAAC,gBAAgB,CAAC;SAChD,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6BAA6B,EAAE,KAAK,IAAI,EAAE;QAC3C,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,OAAO,CAAC,eAAe,CAAC,CAAC;QAClD,MAAM,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC;YACrE,IAAI,EAAE,CAAC,KAAK;SACb,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;QACpE,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,OAAO,CAAC;YAC/B,qBAAqB,EAAE,WAAW;SACnC,CAAC,CAAC;QACH,MAAM,IAAI,GAAG,MAAM,MAAM,CAAC,WAAW,EAAE,CAAC;QACxC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC;IAChE,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,iCAAiC,EAAE,GAAG,EAAE;IAC/C,KAAK,UAAU,KAAK;QAClB,MAAM,KAAK,GAAsC,EAAE,CAAC;QACpD,MAAM,EAAE,MAAM,EAAE,GACd,MAAM,MAAM,CAAC,2CAA2C,CAAC,CAAC;QAC5D,MAAM,EAAE,iBAAiB,EAAE,GACzB,MAAM,MAAM,CAAC,uCAAuC,CAAC,CAAC;QACxD,MAAM,MAAM,GAAG,uBAAuB,CAAC;YACrC,MAAM,EAAE,EAAE;YACV,MAAM,EAAE,EAAE,iBAAiB,EAAE,KAAK,EAAE;YACpC,KAAK,EAAE;gBACL;oBACE,IAAI,EAAE,MAAM;oBACZ,WAAW,EAAE,uBAAuB;oBACpC,WAAW,EAAE;wBACX,IAAI,EAAE,QAAQ;wBACd,UAAU,EAAE,EAAE,GAAG,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE;wBACvC,QAAQ,EAAE,CAAC,KAAK,CAAC;qBAClB;oBACD,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE;wBACnB,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;wBACnC,MAAM,GAAG,GAAI,IAAwB,CAAC,GAAG,CAAC;wBAC1C,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,EAAE,CAAC,EAAE,CAAC;oBACpD,CAAC;iBACF;aACF;SACF,CAAC,CAAC;QACH,MAAM,CAAC,OAAO,EAAE,OAAO,CAAC,GAAG,iBAAiB,CAAC,gBAAgB,EAAE,CAAC;QAChE,MAAM,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QAC9B,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,EAAE,CAAC,CAAC;QAC1D,MAAM,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QAC9B,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC;IAC3B,CAAC;IAED,EAAE,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;QAC9C,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,KAAK,EAAE,CAAC;QACjC,MAAM,IAAI,GAAG,MAAM,MAAM,CAAC,SAAS,EAAE,CAAC;QACtC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC;QACxD,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC;IAClE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;QAChE,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,KAAK,EAAE,CAAC;QACxC,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,QAAQ,CAAC;YACnC,IAAI,EAAE,MAAM;YACZ,SAAS,EAAE,EAAE,GAAG,EAAE,IAAI,EAAE;SACzB,CAAC,CAAC;QACH,MAAM,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE,CAAC,CAAC,CAAC;QAC/D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;IACjE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2BAA2B,EAAE,KAAK,IAAI,EAAE;QACzC,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,KAAK,EAAE,CAAC;QACjC,MAAM,MAAM,CACV,MAAM,CAAC,QAAQ,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,SAAS,EAAE,EAAE,EAAE,CAAC,CACjD,CAAC,OAAO,CAAC,aAAa,CAAC,EAAE,IAAI,EAAE,CAAC,KAAK,EAAE,CAAC,CAAC;IAC5C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import { Client } from '@modelcontextprotocol/sdk/client/index.js';\nimport { InMemoryTransport } from '@modelcontextprotocol/sdk/inMemory.js';\nimport { describe, expect, it } from 'vitest';\nimport { createCodesignMcpServer } from '../src/server.js';\n\nconst FIXTURES = {\n 'docs/alpha/SKILL.md':\n '---\\nname: alpha\\ndescription: First skill.\\n---\\n# Alpha\\nSee [more](more.md).',\n 'docs/alpha/more.md':\n '---\\nname: more\\ndescription: details\\n---\\nMore details.'\n};\n\nasync function connect(assets: Record<string, string> = FIXTURES) {\n const server = createCodesignMcpServer({ engine: {}, assets });\n const [clientT, serverT] = InMemoryTransport.createLinkedPair();\n await server.connect(serverT);\n const client = new Client({ name: 'test', version: '0' });\n await client.connect(clientT);\n return { client, server };\n}\n\ndescribe('createCodesignMcpServer (initialize + resources)', () => {\n it('instructions advertise the codesign tools + force-direct the model to read codesign-system first', async () => {\n const { client } = await connect();\n const instructions = client.getInstructions() ?? '';\n // Short pointer-style instructions (no inlined SYSTEM body)\n expect(instructions).toContain('This is the **codesign** MCP server');\n expect(instructions).toContain('edit');\n expect(instructions).toContain('preview');\n expect(instructions).toContain('export');\n expect(instructions).toContain('list');\n expect(instructions).toContain('history');\n expect(instructions).toContain('inspect');\n expect(instructions).toContain('rename');\n expect(instructions).toContain('open_editor');\n expect(instructions).toContain('asset_generate');\n expect(instructions).toContain('parent');\n // Hard direction: read handbook before any design call.\n expect(instructions).toContain('handbook');\n expect(instructions).toMatch(/[Bb]efore.*edit/);\n // No more <codesign_skills> XML — discovery now happens via tools/list.\n expect(instructions).not.toContain('<codesign_skills>');\n // Cap on size — keep instructions short enough to survive Claude Code\n // truncation (target: well under 4 KB).\n expect(instructions.length).toBeLessThan(4096);\n });\n\n it('resources/list returns SKILL.md entry points (sub-files reachable via the read tools)', async () => {\n const { client } = await connect();\n const list = await client.listResources();\n const uris = list.resources.map((r) => r.uri).sort();\n // Only the SKILL.md is advertised — the nested `more.md` is reachable\n // via `resources/read` or via codesign_read_alpha but not enumerated.\n expect(uris).toEqual(['codesign://docs/alpha/SKILL.md']);\n });\n\n it('returns rewritten cross-refs in resource content', async () => {\n const { client } = await connect();\n const got = await client.readResource({\n uri: 'codesign://docs/alpha/SKILL.md'\n });\n const text = (got.contents[0] as { text?: string }).text;\n expect(text).toContain('[more](codesign://docs/alpha/more.md)');\n expect(text).toContain('# Alpha');\n });\n\n it('rejects unknown resource URIs with InvalidParams', async () => {\n const { client } = await connect();\n await expect(\n client.readResource({ uri: 'codesign://docs/missing/file.md' })\n ).rejects.toMatchObject({\n code: -32602\n });\n });\n});\n\nconst PROMPT_FIXTURES = {\n 'agent/SYSTEM.md': 'sys',\n 'prompts/explain.md':\n \"---\\nname: explain\\ndescription: Explain things.\\n---\\nPlease explain what's happening.\",\n 'prompts/review.md':\n '---\\nname: review\\ndescription: Review the code.\\n---\\nReview the diff.'\n};\n\ndescribe('createCodesignMcpServer (prompts)', () => {\n it('lists every prompt with name+description', async () => {\n const { client } = await connect(PROMPT_FIXTURES);\n const list = await client.listPrompts();\n expect(list.prompts.map((p) => p.name).sort()).toEqual([\n 'explain',\n 'review'\n ]);\n const explain = list.prompts.find((p) => p.name === 'explain');\n expect(explain?.description).toBe('Explain things.');\n });\n\n it('returns prompt body as a single user message', async () => {\n const { client } = await connect(PROMPT_FIXTURES);\n const got = await client.getPrompt({ name: 'explain' });\n expect(got.messages).toHaveLength(1);\n expect(got.messages[0].role).toBe('user');\n expect(got.messages[0].content).toMatchObject({\n type: 'text',\n text: expect.stringContaining('Please explain')\n });\n });\n\n it('rejects unknown prompt name', async () => {\n const { client } = await connect(PROMPT_FIXTURES);\n await expect(client.getPrompt({ name: 'nope' })).rejects.toMatchObject({\n code: -32602\n });\n });\n\n it('falls back to filename when frontmatter omits `name`', async () => {\n const { client } = await connect({\n 'prompts/no-front.md': 'just text'\n });\n const list = await client.listPrompts();\n expect(list.prompts.map((p) => p.name)).toEqual(['no-front']);\n });\n});\n\ndescribe('createCodesignMcpServer (tools)', () => {\n async function setup() {\n const calls: { name: string; args: unknown }[] = [];\n const { Client } =\n await import('@modelcontextprotocol/sdk/client/index.js');\n const { InMemoryTransport } =\n await import('@modelcontextprotocol/sdk/inMemory.js');\n const server = createCodesignMcpServer({\n engine: {},\n assets: { 'agent/SYSTEM.md': 'sys' },\n tools: [\n {\n name: 'echo',\n description: 'Echo arg back as text',\n inputSchema: {\n type: 'object',\n properties: { msg: { type: 'string' } },\n required: ['msg']\n },\n call: async (args) => {\n calls.push({ name: 'echo', args });\n const msg = (args as { msg: string }).msg;\n return { content: [{ type: 'text', text: msg }] };\n }\n }\n ]\n });\n const [clientT, serverT] = InMemoryTransport.createLinkedPair();\n await server.connect(serverT);\n const client = new Client({ name: 'test', version: '0' });\n await client.connect(clientT);\n return { client, calls };\n }\n\n it('lists tools with their schemas', async () => {\n const { client } = await setup();\n const list = await client.listTools();\n expect(list.tools.map((t) => t.name)).toEqual(['echo']);\n expect(list.tools[0].description).toBe('Echo arg back as text');\n });\n\n it('forwards arguments to call() and returns content', async () => {\n const { client, calls } = await setup();\n const result = await client.callTool({\n name: 'echo',\n arguments: { msg: 'hi' }\n });\n expect(calls).toEqual([{ name: 'echo', args: { msg: 'hi' } }]);\n expect(result.content).toEqual([{ type: 'text', text: 'hi' }]);\n });\n\n it('rejects unknown tool name', async () => {\n const { client } = await setup();\n await expect(\n client.callTool({ name: 'nope', arguments: {} })\n ).rejects.toMatchObject({ code: -32602 });\n });\n});\n"]}
|
|
1
|
+
{"version":3,"file":"server.test.js","sourceRoot":"","sources":["../../../test/server.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAC;AACnE,OAAO,EAAE,iBAAiB,EAAE,MAAM,uCAAuC,CAAC;AAC1E,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,uBAAuB,EAAE,MAAM,kBAAkB,CAAC;AAE3D,MAAM,QAAQ,GAAG;IACf,qBAAqB,EACnB,iFAAiF;IACnF,oBAAoB,EAClB,2DAA2D;CAC9D,CAAC;AAEF,KAAK,UAAU,OAAO,CACpB,SAAiC,QAAQ,EACzC,QAA8B,EAAE;IAEhC,MAAM,MAAM,GAAG,uBAAuB,CAAC,EAAE,MAAM,EAAE,GAAG,KAAK,EAAE,CAAC,CAAC;IAC7D,MAAM,CAAC,OAAO,EAAE,OAAO,CAAC,GAAG,iBAAiB,CAAC,gBAAgB,EAAE,CAAC;IAChE,MAAM,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;IAC9B,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,EAAE,CAAC,CAAC;IAC1D,MAAM,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;IAC9B,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC;AAC5B,CAAC;AAED,QAAQ,CAAC,kDAAkD,EAAE,GAAG,EAAE;IAChE,EAAE,CAAC,6DAA6D,EAAE,KAAK,IAAI,EAAE;QAC3E,MAAM,GAAG,GAAG,MAAM,OAAO,EAAE,CAAC;QAC5B,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,gBAAgB,EAAE,EAAE,OAAO,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC3D,qEAAqE;QACrE,uCAAuC;QACvC,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,QAAQ,EAAE,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC,CAAC;QAChE,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,gBAAgB,EAAE,EAAE,OAAO,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IACrE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kGAAkG,EAAE,KAAK,IAAI,EAAE;QAChH,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,OAAO,EAAE,CAAC;QACnC,MAAM,YAAY,GAAG,MAAM,CAAC,eAAe,EAAE,IAAI,EAAE,CAAC;QACpD,4DAA4D;QAC5D,MAAM,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,qCAAqC,CAAC,CAAC;QACtE,MAAM,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QACvC,MAAM,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;QAC1C,MAAM,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;QACzC,MAAM,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QACvC,MAAM,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;QAC1C,MAAM,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;QAC1C,MAAM,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;QACzC,MAAM,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC;QAC9C,uEAAuE;QACvE,uDAAuD;QACvD,MAAM,CAAC,YAAY,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;QACrD,MAAM,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;QACzC,wDAAwD;QACxD,MAAM,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;QAC3C,MAAM,CAAC,YAAY,CAAC,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC;QAChD,wEAAwE;QACxE,MAAM,CAAC,YAAY,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAC;QACxD,sEAAsE;QACtE,wCAAwC;QACxC,MAAM,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;IACjD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mFAAmF,EAAE,KAAK,IAAI,EAAE;QACjG,MAAM,MAAM,GAAG,uBAAuB,CAAC;YACrC,MAAM,EAAE,QAAQ;YAChB,KAAK,EAAE;gBACL;oBACE,IAAI,EAAE,gBAAgB;oBACtB,WAAW,EAAE,KAAK;oBAClB,WAAW,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,EAAE,EAAE;oBAC/C,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC;iBACpC;aACF;SACF,CAAC,CAAC;QACH,MAAM,CAAC,OAAO,EAAE,OAAO,CAAC,GAAG,iBAAiB,CAAC,gBAAgB,EAAE,CAAC;QAChE,MAAM,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QAC9B,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,EAAE,CAAC,CAAC;QAC1D,MAAM,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QAC9B,MAAM,YAAY,GAAG,MAAM,CAAC,eAAe,EAAE,IAAI,EAAE,CAAC;QACpD,MAAM,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,kBAAkB,CAAC,CAAC;QACnD,MAAM,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;IACnD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uFAAuF,EAAE,KAAK,IAAI,EAAE;QACrG,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,OAAO,EAAE,CAAC;QACnC,MAAM,IAAI,GAAG,MAAM,MAAM,CAAC,aAAa,EAAE,CAAC;QAC1C,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC;QACrD,sEAAsE;QACtE,sEAAsE;QACtE,MAAM,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,gCAAgC,CAAC,CAAC,CAAC;IAC3D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;QAChE,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,OAAO,EAAE,CAAC;QACnC,MAAM,GAAG,GAAG,MAAM,MAAM,CAAC,YAAY,CAAC;YACpC,GAAG,EAAE,gCAAgC;SACtC,CAAC,CAAC;QACH,MAAM,IAAI,GAAI,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAuB,CAAC,IAAI,CAAC;QACzD,MAAM,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,uCAAuC,CAAC,CAAC;QAChE,MAAM,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;IACpC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;QAChE,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,OAAO,EAAE,CAAC;QACnC,MAAM,MAAM,CACV,MAAM,CAAC,YAAY,CAAC,EAAE,GAAG,EAAE,iCAAiC,EAAE,CAAC,CAChE,CAAC,OAAO,CAAC,aAAa,CAAC;YACtB,IAAI,EAAE,CAAC,KAAK;SACb,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,MAAM,eAAe,GAAG;IACtB,iBAAiB,EAAE,KAAK;IACxB,oBAAoB,EAClB,yFAAyF;IAC3F,mBAAmB,EACjB,yEAAyE;CAC5E,CAAC;AAEF,QAAQ,CAAC,mCAAmC,EAAE,GAAG,EAAE;IACjD,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;QACxD,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,OAAO,CAAC,eAAe,CAAC,CAAC;QAClD,MAAM,IAAI,GAAG,MAAM,MAAM,CAAC,WAAW,EAAE,CAAC;QACxC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC;YACrD,SAAS;YACT,QAAQ;SACT,CAAC,CAAC;QACH,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC;QAC/D,MAAM,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;IACvD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;QAC5D,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,OAAO,CAAC,eAAe,CAAC,CAAC;QAClD,MAAM,GAAG,GAAG,MAAM,MAAM,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,CAAC;QACxD,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACrC,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC1C,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,aAAa,CAAC;YAC5C,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,MAAM,CAAC,gBAAgB,CAAC,gBAAgB,CAAC;SAChD,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6BAA6B,EAAE,KAAK,IAAI,EAAE;QAC3C,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,OAAO,CAAC,eAAe,CAAC,CAAC;QAClD,MAAM,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC;YACrE,IAAI,EAAE,CAAC,KAAK;SACb,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;QACpE,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,OAAO,CAAC;YAC/B,qBAAqB,EAAE,WAAW;SACnC,CAAC,CAAC;QACH,MAAM,IAAI,GAAG,MAAM,MAAM,CAAC,WAAW,EAAE,CAAC;QACxC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC;IAChE,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,iCAAiC,EAAE,GAAG,EAAE;IAC/C,KAAK,UAAU,KAAK;QAClB,MAAM,KAAK,GAAsC,EAAE,CAAC;QACpD,MAAM,EAAE,MAAM,EAAE,GACd,MAAM,MAAM,CAAC,2CAA2C,CAAC,CAAC;QAC5D,MAAM,EAAE,iBAAiB,EAAE,GACzB,MAAM,MAAM,CAAC,uCAAuC,CAAC,CAAC;QACxD,MAAM,MAAM,GAAG,uBAAuB,CAAC;YACrC,MAAM,EAAE,EAAE,iBAAiB,EAAE,KAAK,EAAE;YACpC,KAAK,EAAE;gBACL;oBACE,IAAI,EAAE,MAAM;oBACZ,WAAW,EAAE,uBAAuB;oBACpC,WAAW,EAAE;wBACX,IAAI,EAAE,QAAQ;wBACd,UAAU,EAAE,EAAE,GAAG,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE;wBACvC,QAAQ,EAAE,CAAC,KAAK,CAAC;qBAClB;oBACD,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE;wBACnB,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;wBACnC,MAAM,GAAG,GAAI,IAAwB,CAAC,GAAG,CAAC;wBAC1C,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,EAAE,CAAC,EAAE,CAAC;oBACpD,CAAC;iBACF;aACF;SACF,CAAC,CAAC;QACH,MAAM,CAAC,OAAO,EAAE,OAAO,CAAC,GAAG,iBAAiB,CAAC,gBAAgB,EAAE,CAAC;QAChE,MAAM,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QAC9B,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,EAAE,CAAC,CAAC;QAC1D,MAAM,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QAC9B,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC;IAC3B,CAAC;IAED,EAAE,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;QAC9C,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,KAAK,EAAE,CAAC;QACjC,MAAM,IAAI,GAAG,MAAM,MAAM,CAAC,SAAS,EAAE,CAAC;QACtC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC;QACxD,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC;IAClE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;QAChE,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,KAAK,EAAE,CAAC;QACxC,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,QAAQ,CAAC;YACnC,IAAI,EAAE,MAAM;YACZ,SAAS,EAAE,EAAE,GAAG,EAAE,IAAI,EAAE;SACzB,CAAC,CAAC;QACH,MAAM,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE,CAAC,CAAC,CAAC;QAC/D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;IACjE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2BAA2B,EAAE,KAAK,IAAI,EAAE;QACzC,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,KAAK,EAAE,CAAC;QACjC,MAAM,MAAM,CACV,MAAM,CAAC,QAAQ,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,SAAS,EAAE,EAAE,EAAE,CAAC,CACjD,CAAC,OAAO,CAAC,aAAa,CAAC,EAAE,IAAI,EAAE,CAAC,KAAK,EAAE,CAAC,CAAC;IAC5C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import { Client } from '@modelcontextprotocol/sdk/client/index.js';\nimport { InMemoryTransport } from '@modelcontextprotocol/sdk/inMemory.js';\nimport { describe, expect, it } from 'vitest';\nimport { createCodesignMcpServer } from '../src/server.js';\n\nconst FIXTURES = {\n 'docs/alpha/SKILL.md':\n '---\\nname: alpha\\ndescription: First skill.\\n---\\n# Alpha\\nSee [more](more.md).',\n 'docs/alpha/more.md':\n '---\\nname: more\\ndescription: details\\n---\\nMore details.'\n};\n\nasync function connect(\n assets: Record<string, string> = FIXTURES,\n extra: { buildId?: string } = {}\n) {\n const server = createCodesignMcpServer({ assets, ...extra });\n const [clientT, serverT] = InMemoryTransport.createLinkedPair();\n await server.connect(serverT);\n const client = new Client({ name: 'test', version: '0' });\n await client.connect(clientT);\n return { client, server };\n}\n\ndescribe('createCodesignMcpServer (initialize + resources)', () => {\n it('reports buildId as serverInfo.version (defaulting to \"dev\")', async () => {\n const def = await connect();\n expect(def.client.getServerVersion()?.version).toBe('dev');\n // The deploy injects a git SHA; the smoketest asserts it to detect a\n // stale, not-yet-rolled-out container.\n const stamped = await connect(FIXTURES, { buildId: 'abc1234' });\n expect(stamped.client.getServerVersion()?.version).toBe('abc1234');\n });\n\n it('instructions advertise the codesign tools + force-direct the model to read codesign-system first', async () => {\n const { client } = await connect();\n const instructions = client.getInstructions() ?? '';\n // Short pointer-style instructions (no inlined SYSTEM body)\n expect(instructions).toContain('This is the **codesign** MCP server');\n expect(instructions).toContain('edit');\n expect(instructions).toContain('preview');\n expect(instructions).toContain('export');\n expect(instructions).toContain('list');\n expect(instructions).toContain('history');\n expect(instructions).toContain('inspect');\n expect(instructions).toContain('rename');\n expect(instructions).toContain('open_editor');\n // asset_generate is opt-in: with no tools registered (no gateway), the\n // hint is omitted so we never advertise a hidden tool.\n expect(instructions).not.toContain('asset_generate');\n expect(instructions).toContain('parent');\n // Hard direction: read handbook before any design call.\n expect(instructions).toContain('handbook');\n expect(instructions).toMatch(/[Bb]efore.*edit/);\n // No more <codesign_skills> XML — discovery now happens via tools/list.\n expect(instructions).not.toContain('<codesign_skills>');\n // Cap on size — keep instructions short enough to survive Claude Code\n // truncation (target: well under 4 KB).\n expect(instructions.length).toBeLessThan(4096);\n });\n\n it('includes the asset-generation hint only when an asset_generate tool is registered', async () => {\n const server = createCodesignMcpServer({\n assets: FIXTURES,\n tools: [\n {\n name: 'asset_generate',\n description: 'gen',\n inputSchema: { type: 'object', properties: {} },\n call: async () => ({ content: [] })\n }\n ]\n });\n const [clientT, serverT] = InMemoryTransport.createLinkedPair();\n await server.connect(serverT);\n const client = new Client({ name: 'test', version: '0' });\n await client.connect(clientT);\n const instructions = client.getInstructions() ?? '';\n expect(instructions).toContain('Asset generation');\n expect(instructions).toContain('asset_generate');\n });\n\n it('resources/list returns SKILL.md entry points (sub-files reachable via the read tools)', async () => {\n const { client } = await connect();\n const list = await client.listResources();\n const uris = list.resources.map((r) => r.uri).sort();\n // Only the SKILL.md is advertised — the nested `more.md` is reachable\n // via `resources/read` or via codesign_read_alpha but not enumerated.\n expect(uris).toEqual(['codesign://docs/alpha/SKILL.md']);\n });\n\n it('returns rewritten cross-refs in resource content', async () => {\n const { client } = await connect();\n const got = await client.readResource({\n uri: 'codesign://docs/alpha/SKILL.md'\n });\n const text = (got.contents[0] as { text?: string }).text;\n expect(text).toContain('[more](codesign://docs/alpha/more.md)');\n expect(text).toContain('# Alpha');\n });\n\n it('rejects unknown resource URIs with InvalidParams', async () => {\n const { client } = await connect();\n await expect(\n client.readResource({ uri: 'codesign://docs/missing/file.md' })\n ).rejects.toMatchObject({\n code: -32602\n });\n });\n});\n\nconst PROMPT_FIXTURES = {\n 'agent/SYSTEM.md': 'sys',\n 'prompts/explain.md':\n \"---\\nname: explain\\ndescription: Explain things.\\n---\\nPlease explain what's happening.\",\n 'prompts/review.md':\n '---\\nname: review\\ndescription: Review the code.\\n---\\nReview the diff.'\n};\n\ndescribe('createCodesignMcpServer (prompts)', () => {\n it('lists every prompt with name+description', async () => {\n const { client } = await connect(PROMPT_FIXTURES);\n const list = await client.listPrompts();\n expect(list.prompts.map((p) => p.name).sort()).toEqual([\n 'explain',\n 'review'\n ]);\n const explain = list.prompts.find((p) => p.name === 'explain');\n expect(explain?.description).toBe('Explain things.');\n });\n\n it('returns prompt body as a single user message', async () => {\n const { client } = await connect(PROMPT_FIXTURES);\n const got = await client.getPrompt({ name: 'explain' });\n expect(got.messages).toHaveLength(1);\n expect(got.messages[0].role).toBe('user');\n expect(got.messages[0].content).toMatchObject({\n type: 'text',\n text: expect.stringContaining('Please explain')\n });\n });\n\n it('rejects unknown prompt name', async () => {\n const { client } = await connect(PROMPT_FIXTURES);\n await expect(client.getPrompt({ name: 'nope' })).rejects.toMatchObject({\n code: -32602\n });\n });\n\n it('falls back to filename when frontmatter omits `name`', async () => {\n const { client } = await connect({\n 'prompts/no-front.md': 'just text'\n });\n const list = await client.listPrompts();\n expect(list.prompts.map((p) => p.name)).toEqual(['no-front']);\n });\n});\n\ndescribe('createCodesignMcpServer (tools)', () => {\n async function setup() {\n const calls: { name: string; args: unknown }[] = [];\n const { Client } =\n await import('@modelcontextprotocol/sdk/client/index.js');\n const { InMemoryTransport } =\n await import('@modelcontextprotocol/sdk/inMemory.js');\n const server = createCodesignMcpServer({\n assets: { 'agent/SYSTEM.md': 'sys' },\n tools: [\n {\n name: 'echo',\n description: 'Echo arg back as text',\n inputSchema: {\n type: 'object',\n properties: { msg: { type: 'string' } },\n required: ['msg']\n },\n call: async (args) => {\n calls.push({ name: 'echo', args });\n const msg = (args as { msg: string }).msg;\n return { content: [{ type: 'text', text: msg }] };\n }\n }\n ]\n });\n const [clientT, serverT] = InMemoryTransport.createLinkedPair();\n await server.connect(serverT);\n const client = new Client({ name: 'test', version: '0' });\n await client.connect(clientT);\n return { client, calls };\n }\n\n it('lists tools with their schemas', async () => {\n const { client } = await setup();\n const list = await client.listTools();\n expect(list.tools.map((t) => t.name)).toEqual(['echo']);\n expect(list.tools[0].description).toBe('Echo arg back as text');\n });\n\n it('forwards arguments to call() and returns content', async () => {\n const { client, calls } = await setup();\n const result = await client.callTool({\n name: 'echo',\n arguments: { msg: 'hi' }\n });\n expect(calls).toEqual([{ name: 'echo', args: { msg: 'hi' } }]);\n expect(result.content).toEqual([{ type: 'text', text: 'hi' }]);\n });\n\n it('rejects unknown tool name', async () => {\n const { client } = await setup();\n await expect(\n client.callTool({ name: 'nope', arguments: {} })\n ).rejects.toMatchObject({ code: -32602 });\n });\n});\n"]}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"asset-search.test.d.ts","sourceRoot":"","sources":["../../../../test/tools/asset-search.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,121 @@
|
|
|
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 { createAssetSearchTool } from '../../src/tools/asset-search.js';
|
|
5
|
+
function makeDeps() {
|
|
6
|
+
const fakeEngine = {
|
|
7
|
+
asset: {
|
|
8
|
+
findAllSources: vi.fn(() => ['ly.img.image', 'ly.img.sticker']),
|
|
9
|
+
getSupportedMimeTypes: vi.fn((id) => id === 'ly.img.image' ? ['image/jpeg', 'image/png'] : []),
|
|
10
|
+
findAssets: vi.fn(async () => ({
|
|
11
|
+
assets: [
|
|
12
|
+
{
|
|
13
|
+
id: 'beach-1',
|
|
14
|
+
label: 'Beach',
|
|
15
|
+
groups: ['nature'],
|
|
16
|
+
meta: {
|
|
17
|
+
uri: 'https://cdn.example/beach.jpg',
|
|
18
|
+
thumbUri: 'https://cdn.example/beach-thumb.jpg',
|
|
19
|
+
width: 1200,
|
|
20
|
+
height: 800,
|
|
21
|
+
blockType: '//ly.img.ubq/graphic'
|
|
22
|
+
}
|
|
23
|
+
}
|
|
24
|
+
],
|
|
25
|
+
currentPage: 0,
|
|
26
|
+
nextPage: 1,
|
|
27
|
+
total: 42
|
|
28
|
+
}))
|
|
29
|
+
}
|
|
30
|
+
};
|
|
31
|
+
const reset = vi.fn(async () => { });
|
|
32
|
+
const adapter = {
|
|
33
|
+
create: () => fakeEngine,
|
|
34
|
+
reset,
|
|
35
|
+
loadFromString: async () => { },
|
|
36
|
+
saveToString: async () => ''
|
|
37
|
+
};
|
|
38
|
+
const session = new EngineSession({ size: 1, adapter });
|
|
39
|
+
const tool = createAssetSearchTool({ session, engineLock: new Mutex() });
|
|
40
|
+
return { fakeEngine, session, tool, reset };
|
|
41
|
+
}
|
|
42
|
+
function bodyOf(result) {
|
|
43
|
+
return JSON.parse(result.content[0].text);
|
|
44
|
+
}
|
|
45
|
+
describe('asset_search tool', () => {
|
|
46
|
+
it('lists all sources with supported MIME types when sourceId is omitted', async () => {
|
|
47
|
+
const { tool } = makeDeps();
|
|
48
|
+
const result = await tool.call({});
|
|
49
|
+
expect(result.isError).toBeUndefined();
|
|
50
|
+
expect(bodyOf(result)).toEqual({
|
|
51
|
+
sources: [
|
|
52
|
+
{ id: 'ly.img.image', supportedMimeTypes: ['image/jpeg', 'image/png'] },
|
|
53
|
+
{ id: 'ly.img.sticker', supportedMimeTypes: [] }
|
|
54
|
+
]
|
|
55
|
+
});
|
|
56
|
+
});
|
|
57
|
+
it('searches a source and returns compact asset JSON', async () => {
|
|
58
|
+
const { tool, fakeEngine } = makeDeps();
|
|
59
|
+
const result = await tool.call({
|
|
60
|
+
sourceId: 'ly.img.image',
|
|
61
|
+
query: 'beach',
|
|
62
|
+
page: 2
|
|
63
|
+
});
|
|
64
|
+
expect(result.isError).toBeUndefined();
|
|
65
|
+
expect(fakeEngine.asset.findAssets).toHaveBeenCalledWith('ly.img.image', {
|
|
66
|
+
query: 'beach',
|
|
67
|
+
page: 2,
|
|
68
|
+
perPage: 20
|
|
69
|
+
});
|
|
70
|
+
expect(bodyOf(result)).toEqual({
|
|
71
|
+
total: 42,
|
|
72
|
+
currentPage: 0,
|
|
73
|
+
nextPage: 1,
|
|
74
|
+
assets: [
|
|
75
|
+
{
|
|
76
|
+
id: 'beach-1',
|
|
77
|
+
label: 'Beach',
|
|
78
|
+
groups: ['nature'],
|
|
79
|
+
uri: 'https://cdn.example/beach.jpg',
|
|
80
|
+
thumbUri: 'https://cdn.example/beach-thumb.jpg',
|
|
81
|
+
width: 1200,
|
|
82
|
+
height: 800
|
|
83
|
+
}
|
|
84
|
+
]
|
|
85
|
+
});
|
|
86
|
+
});
|
|
87
|
+
it('clamps perPage to 50', async () => {
|
|
88
|
+
const { tool, fakeEngine } = makeDeps();
|
|
89
|
+
await tool.call({ sourceId: 'ly.img.image', perPage: 500 });
|
|
90
|
+
expect(fakeEngine.asset.findAssets).toHaveBeenCalledWith('ly.img.image', {
|
|
91
|
+
page: 0,
|
|
92
|
+
perPage: 50
|
|
93
|
+
});
|
|
94
|
+
});
|
|
95
|
+
it('errors on an unknown sourceId and lists the valid ones', async () => {
|
|
96
|
+
const { tool } = makeDeps();
|
|
97
|
+
const result = await tool.call({ sourceId: 'nope' });
|
|
98
|
+
expect(result.isError).toBe(true);
|
|
99
|
+
const text = result.content[0].text;
|
|
100
|
+
expect(text).toContain('unknown sourceId "nope"');
|
|
101
|
+
expect(text).toContain('ly.img.image');
|
|
102
|
+
expect(text).toContain('ly.img.sticker');
|
|
103
|
+
});
|
|
104
|
+
it('returns isError when findAssets rejects (scene untouched)', async () => {
|
|
105
|
+
const { tool, fakeEngine } = makeDeps();
|
|
106
|
+
fakeEngine.asset.findAssets.mockRejectedValueOnce(new Error('CDN unreachable'));
|
|
107
|
+
const result = await tool.call({ sourceId: 'ly.img.image' });
|
|
108
|
+
expect(result.isError).toBe(true);
|
|
109
|
+
expect(result.content[0].text).toContain('CDN unreachable');
|
|
110
|
+
});
|
|
111
|
+
it('reuses a warm slot without resetting the engine', async () => {
|
|
112
|
+
const { tool, session, fakeEngine, reset } = makeDeps();
|
|
113
|
+
const slot = await session.acquire(null);
|
|
114
|
+
session.commit(slot, 'rev-1');
|
|
115
|
+
reset.mockClear(); // acquire(null) above legitimately reset once
|
|
116
|
+
await tool.call({ sourceId: 'ly.img.image' });
|
|
117
|
+
expect(fakeEngine.asset.findAssets).toHaveBeenCalled();
|
|
118
|
+
expect(reset).not.toHaveBeenCalled();
|
|
119
|
+
});
|
|
120
|
+
});
|
|
121
|
+
//# sourceMappingURL=asset-search.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"asset-search.test.js","sourceRoot":"","sources":["../../../../test/tools/asset-search.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,qBAAqB,EAAE,MAAM,iCAAiC,CAAC;AAUxE,SAAS,QAAQ;IACf,MAAM,UAAU,GAAe;QAC7B,KAAK,EAAE;YACL,cAAc,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC,cAAc,EAAE,gBAAgB,CAAC,CAAC;YAC/D,qBAAqB,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,EAAU,EAAE,EAAE,CAC1C,EAAE,KAAK,cAAc,CAAC,CAAC,CAAC,CAAC,YAAY,EAAE,WAAW,CAAC,CAAC,CAAC,CAAC,EAAE,CACzD;YACD,UAAU,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;gBAC7B,MAAM,EAAE;oBACN;wBACE,EAAE,EAAE,SAAS;wBACb,KAAK,EAAE,OAAO;wBACd,MAAM,EAAE,CAAC,QAAQ,CAAC;wBAClB,IAAI,EAAE;4BACJ,GAAG,EAAE,+BAA+B;4BACpC,QAAQ,EAAE,qCAAqC;4BAC/C,KAAK,EAAE,IAAI;4BACX,MAAM,EAAE,GAAG;4BACX,SAAS,EAAE,sBAAsB;yBAClC;qBACF;iBACF;gBACD,WAAW,EAAE,CAAC;gBACd,QAAQ,EAAE,CAAC;gBACX,KAAK,EAAE,EAAE;aACV,CAAC,CAAC;SACJ;KACF,CAAC;IACF,MAAM,KAAK,GAAG,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,GAAE,CAAC,CAAC,CAAC;IACpC,MAAM,OAAO,GAA8B;QACzC,MAAM,EAAE,GAAG,EAAE,CAAC,UAAU;QACxB,KAAK;QACL,cAAc,EAAE,KAAK,IAAI,EAAE,GAAE,CAAC;QAC9B,YAAY,EAAE,KAAK,IAAI,EAAE,CAAC,EAAE;KAC7B,CAAC;IACF,MAAM,OAAO,GAAG,IAAI,aAAa,CAAa,EAAE,IAAI,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC;IACpE,MAAM,IAAI,GAAG,qBAAqB,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,IAAI,KAAK,EAAE,EAAE,CAAC,CAAC;IACzE,OAAO,EAAE,UAAU,EAAE,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC;AAC9C,CAAC;AAED,SAAS,MAAM,CAAC,MAEf;IACC,OAAO,IAAI,CAAC,KAAK,CAAE,MAAM,CAAC,OAAO,CAAC,CAAC,CAAsB,CAAC,IAAI,CAAC,CAAC;AAClE,CAAC;AAED,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;IACjC,EAAE,CAAC,sEAAsE,EAAE,KAAK,IAAI,EAAE;QACpF,MAAM,EAAE,IAAI,EAAE,GAAG,QAAQ,EAAE,CAAC;QAC5B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACnC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,aAAa,EAAE,CAAC;QACvC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC;YAC7B,OAAO,EAAE;gBACP,EAAE,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE,CAAC,YAAY,EAAE,WAAW,CAAC,EAAE;gBACvE,EAAE,EAAE,EAAE,gBAAgB,EAAE,kBAAkB,EAAE,EAAE,EAAE;aACjD;SACF,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;QAChE,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,GAAG,QAAQ,EAAE,CAAC;QACxC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC;YAC7B,QAAQ,EAAE,cAAc;YACxB,KAAK,EAAE,OAAO;YACd,IAAI,EAAE,CAAC;SACR,CAAC,CAAC;QACH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,aAAa,EAAE,CAAC;QACvC,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,oBAAoB,CAAC,cAAc,EAAE;YACvE,KAAK,EAAE,OAAO;YACd,IAAI,EAAE,CAAC;YACP,OAAO,EAAE,EAAE;SACZ,CAAC,CAAC;QACH,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC;YAC7B,KAAK,EAAE,EAAE;YACT,WAAW,EAAE,CAAC;YACd,QAAQ,EAAE,CAAC;YACX,MAAM,EAAE;gBACN;oBACE,EAAE,EAAE,SAAS;oBACb,KAAK,EAAE,OAAO;oBACd,MAAM,EAAE,CAAC,QAAQ,CAAC;oBAClB,GAAG,EAAE,+BAA+B;oBACpC,QAAQ,EAAE,qCAAqC;oBAC/C,KAAK,EAAE,IAAI;oBACX,MAAM,EAAE,GAAG;iBACZ;aACF;SACF,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sBAAsB,EAAE,KAAK,IAAI,EAAE;QACpC,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,GAAG,QAAQ,EAAE,CAAC;QACxC,MAAM,IAAI,CAAC,IAAI,CAAC,EAAE,QAAQ,EAAE,cAAc,EAAE,OAAO,EAAE,GAAG,EAAE,CAAC,CAAC;QAC5D,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,oBAAoB,CAAC,cAAc,EAAE;YACvE,IAAI,EAAE,CAAC;YACP,OAAO,EAAE,EAAE;SACZ,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;QACtE,MAAM,EAAE,IAAI,EAAE,GAAG,QAAQ,EAAE,CAAC;QAC5B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC,CAAC;QACrD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAClC,MAAM,IAAI,GAAI,MAAM,CAAC,OAAO,CAAC,CAAC,CAAsB,CAAC,IAAI,CAAC;QAC1D,MAAM,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,yBAAyB,CAAC,CAAC;QAClD,MAAM,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;QACvC,MAAM,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;IAC3C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;QACzE,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,GAAG,QAAQ,EAAE,CAAC;QACxC,UAAU,CAAC,KAAK,CAAC,UAAU,CAAC,qBAAqB,CAC/C,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAC7B,CAAC;QACF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,EAAE,QAAQ,EAAE,cAAc,EAAE,CAAC,CAAC;QAC7D,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,SAAS,CAC5D,iBAAiB,CAClB,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;QAC/D,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,UAAU,EAAE,KAAK,EAAE,GAAG,QAAQ,EAAE,CAAC;QACxD,MAAM,IAAI,GAAG,MAAM,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QACzC,OAAO,CAAC,MAAM,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;QAC9B,KAAK,CAAC,SAAS,EAAE,CAAC,CAAC,8CAA8C;QACjE,MAAM,IAAI,CAAC,IAAI,CAAC,EAAE,QAAQ,EAAE,cAAc,EAAE,CAAC,CAAC;QAC9C,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACvD,MAAM,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IACvC,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 { createAssetSearchTool } from '../../src/tools/asset-search.js';\n\ninterface FakeEngine {\n asset: {\n findAllSources: ReturnType<typeof vi.fn>;\n getSupportedMimeTypes: ReturnType<typeof vi.fn>;\n findAssets: ReturnType<typeof vi.fn>;\n };\n}\n\nfunction makeDeps() {\n const fakeEngine: FakeEngine = {\n asset: {\n findAllSources: vi.fn(() => ['ly.img.image', 'ly.img.sticker']),\n getSupportedMimeTypes: vi.fn((id: string) =>\n id === 'ly.img.image' ? ['image/jpeg', 'image/png'] : []\n ),\n findAssets: vi.fn(async () => ({\n assets: [\n {\n id: 'beach-1',\n label: 'Beach',\n groups: ['nature'],\n meta: {\n uri: 'https://cdn.example/beach.jpg',\n thumbUri: 'https://cdn.example/beach-thumb.jpg',\n width: 1200,\n height: 800,\n blockType: '//ly.img.ubq/graphic'\n }\n }\n ],\n currentPage: 0,\n nextPage: 1,\n total: 42\n }))\n }\n };\n const reset = vi.fn(async () => {});\n const adapter: EngineAdapter<FakeEngine> = {\n create: () => fakeEngine,\n reset,\n loadFromString: async () => {},\n saveToString: async () => ''\n };\n const session = new EngineSession<FakeEngine>({ size: 1, adapter });\n const tool = createAssetSearchTool({ session, engineLock: new Mutex() });\n return { fakeEngine, session, tool, reset };\n}\n\nfunction bodyOf(result: {\n content: Array<{ type: string; text?: string }>;\n}): unknown {\n return JSON.parse((result.content[0] as { text: string }).text);\n}\n\ndescribe('asset_search tool', () => {\n it('lists all sources with supported MIME types when sourceId is omitted', async () => {\n const { tool } = makeDeps();\n const result = await tool.call({});\n expect(result.isError).toBeUndefined();\n expect(bodyOf(result)).toEqual({\n sources: [\n { id: 'ly.img.image', supportedMimeTypes: ['image/jpeg', 'image/png'] },\n { id: 'ly.img.sticker', supportedMimeTypes: [] }\n ]\n });\n });\n\n it('searches a source and returns compact asset JSON', async () => {\n const { tool, fakeEngine } = makeDeps();\n const result = await tool.call({\n sourceId: 'ly.img.image',\n query: 'beach',\n page: 2\n });\n expect(result.isError).toBeUndefined();\n expect(fakeEngine.asset.findAssets).toHaveBeenCalledWith('ly.img.image', {\n query: 'beach',\n page: 2,\n perPage: 20\n });\n expect(bodyOf(result)).toEqual({\n total: 42,\n currentPage: 0,\n nextPage: 1,\n assets: [\n {\n id: 'beach-1',\n label: 'Beach',\n groups: ['nature'],\n uri: 'https://cdn.example/beach.jpg',\n thumbUri: 'https://cdn.example/beach-thumb.jpg',\n width: 1200,\n height: 800\n }\n ]\n });\n });\n\n it('clamps perPage to 50', async () => {\n const { tool, fakeEngine } = makeDeps();\n await tool.call({ sourceId: 'ly.img.image', perPage: 500 });\n expect(fakeEngine.asset.findAssets).toHaveBeenCalledWith('ly.img.image', {\n page: 0,\n perPage: 50\n });\n });\n\n it('errors on an unknown sourceId and lists the valid ones', async () => {\n const { tool } = makeDeps();\n const result = await tool.call({ sourceId: 'nope' });\n expect(result.isError).toBe(true);\n const text = (result.content[0] as { text: string }).text;\n expect(text).toContain('unknown sourceId \"nope\"');\n expect(text).toContain('ly.img.image');\n expect(text).toContain('ly.img.sticker');\n });\n\n it('returns isError when findAssets rejects (scene untouched)', async () => {\n const { tool, fakeEngine } = makeDeps();\n fakeEngine.asset.findAssets.mockRejectedValueOnce(\n new Error('CDN unreachable')\n );\n const result = await tool.call({ sourceId: 'ly.img.image' });\n expect(result.isError).toBe(true);\n expect((result.content[0] as { text: string }).text).toContain(\n 'CDN unreachable'\n );\n });\n\n it('reuses a warm slot without resetting the engine', async () => {\n const { tool, session, fakeEngine, reset } = makeDeps();\n const slot = await session.acquire(null);\n session.commit(slot, 'rev-1');\n reset.mockClear(); // acquire(null) above legitimately reset once\n await tool.call({ sourceId: 'ly.img.image' });\n expect(fakeEngine.asset.findAssets).toHaveBeenCalled();\n expect(reset).not.toHaveBeenCalled();\n });\n});\n"]}
|
|
@@ -1,9 +1,11 @@
|
|
|
1
1
|
// packages/codesign-mcp/test/tools/edit.test.ts
|
|
2
2
|
import { describe, expect, it, vi } from 'vitest';
|
|
3
|
+
import { Mutex } from '../../src/engine/mutex.js';
|
|
3
4
|
import { EngineSession } from '../../src/engine/session.js';
|
|
4
5
|
import { RevisionStore } from '../../src/revisions/store.js';
|
|
5
6
|
import { createEditTool } from '../../src/tools/edit.js';
|
|
6
7
|
import { InMemoryWorkspaceFS } from '../../src/workspace/in-memory.js';
|
|
8
|
+
const engineLock = new Mutex();
|
|
7
9
|
function makeDeps() {
|
|
8
10
|
const fs = new InMemoryWorkspaceFS();
|
|
9
11
|
const store = new RevisionStore(fs);
|
|
@@ -25,7 +27,7 @@ function makeDeps() {
|
|
|
25
27
|
describe('edit tool — happy path', () => {
|
|
26
28
|
it('parent=null resets engine, writes a root revision, returns {revision, parent}', async () => {
|
|
27
29
|
const { store, session, adapter, onLeafWritten } = makeDeps();
|
|
28
|
-
const tool = createEditTool({ store, session, onLeafWritten });
|
|
30
|
+
const tool = createEditTool({ store, session, engineLock, onLeafWritten });
|
|
29
31
|
// The user `code` is what mutates the engine — it runs AFTER reset/loadFromString
|
|
30
32
|
// resets the slot. So we set _scene from inside the code string.
|
|
31
33
|
const result = await tool.call({
|
|
@@ -52,7 +54,7 @@ describe('edit tool — happy path', () => {
|
|
|
52
54
|
});
|
|
53
55
|
it('parent=existing revision is a linear edit (no reload on warm slot)', async () => {
|
|
54
56
|
const { store, session, adapter, onLeafWritten } = makeDeps();
|
|
55
|
-
const tool = createEditTool({ store, session, onLeafWritten });
|
|
57
|
+
const tool = createEditTool({ store, session, engineLock, onLeafWritten });
|
|
56
58
|
const first = await tool.call({
|
|
57
59
|
parent: null,
|
|
58
60
|
code: "engine._scene = 'SCENE_V1'",
|
|
@@ -73,7 +75,7 @@ describe('edit tool — happy path', () => {
|
|
|
73
75
|
});
|
|
74
76
|
it('parent=older revision reloads engine and creates a fork', async () => {
|
|
75
77
|
const { store, session, adapter, onLeafWritten } = makeDeps();
|
|
76
|
-
const tool = createEditTool({ store, session, onLeafWritten });
|
|
78
|
+
const tool = createEditTool({ store, session, engineLock, onLeafWritten });
|
|
77
79
|
const r1 = await tool.call({
|
|
78
80
|
parent: null,
|
|
79
81
|
code: "engine._scene = 'SCENE_V1'",
|
|
@@ -100,7 +102,7 @@ describe('edit tool — happy path', () => {
|
|
|
100
102
|
describe('edit tool — safety + validation', () => {
|
|
101
103
|
it('marks slot dirty when code throws; next edit pays one reload', async () => {
|
|
102
104
|
const { store, session, adapter, onLeafWritten } = makeDeps();
|
|
103
|
-
const tool = createEditTool({ store, session, onLeafWritten });
|
|
105
|
+
const tool = createEditTool({ store, session, engineLock, onLeafWritten });
|
|
104
106
|
const r1 = await tool.call({
|
|
105
107
|
parent: null,
|
|
106
108
|
code: "engine._scene = 'SCENE_V1'",
|
|
@@ -122,7 +124,7 @@ describe('edit tool — safety + validation', () => {
|
|
|
122
124
|
});
|
|
123
125
|
it('returns isError when parent property is missing entirely', async () => {
|
|
124
126
|
const { store, session } = makeDeps();
|
|
125
|
-
const tool = createEditTool({ store, session });
|
|
127
|
+
const tool = createEditTool({ store, session, engineLock });
|
|
126
128
|
const result = await tool.call({
|
|
127
129
|
code: '// no parent'
|
|
128
130
|
});
|
|
@@ -131,27 +133,27 @@ describe('edit tool — safety + validation', () => {
|
|
|
131
133
|
});
|
|
132
134
|
it('returns isError when code is missing', async () => {
|
|
133
135
|
const { store, session } = makeDeps();
|
|
134
|
-
const tool = createEditTool({ store, session });
|
|
136
|
+
const tool = createEditTool({ store, session, engineLock });
|
|
135
137
|
const result = await tool.call({ parent: null });
|
|
136
138
|
expect(result.isError).toBe(true);
|
|
137
139
|
expect(result.content[0].text).toMatch(/code.*required/i);
|
|
138
140
|
});
|
|
139
|
-
it('
|
|
140
|
-
// LLMs
|
|
141
|
-
//
|
|
142
|
-
//
|
|
143
|
-
//
|
|
144
|
-
|
|
145
|
-
const { store, session }
|
|
146
|
-
const tool = createEditTool({ store, session });
|
|
141
|
+
it('coerces the literal string "null" to a real null parent (new design root)', async () => {
|
|
142
|
+
// Some LLMs / MCP clients emit the four-character string "null" instead of
|
|
143
|
+
// the bare JSON null literal — a quoting slip, not a real revision id. The
|
|
144
|
+
// tool coerces "null" → null rather than rejecting, so the slip never
|
|
145
|
+
// blocks creating a new design root. (Stochastic, model-dependent.)
|
|
146
|
+
const { store, session, adapter } = makeDeps();
|
|
147
|
+
const tool = createEditTool({ store, session, engineLock });
|
|
147
148
|
const result = await tool.call({
|
|
148
149
|
parent: 'null',
|
|
149
150
|
code: "engine._scene = 'X'"
|
|
150
151
|
});
|
|
151
|
-
expect(result.isError).
|
|
152
|
-
|
|
153
|
-
|
|
154
|
-
expect(
|
|
152
|
+
expect(result.isError).toBeFalsy();
|
|
153
|
+
expect(adapter.reset).toHaveBeenCalledOnce(); // treated as parent=null
|
|
154
|
+
const envelope = JSON.parse(result.content[0].text);
|
|
155
|
+
expect(envelope.parent).toBeNull();
|
|
156
|
+
expect(envelope.revision).toMatch(/^[0-9a-f]{12}$/);
|
|
155
157
|
});
|
|
156
158
|
});
|
|
157
159
|
describe('edit tool — per-leaf thumbnail', () => {
|
|
@@ -167,7 +169,13 @@ describe('edit tool — per-leaf thumbnail', () => {
|
|
|
167
169
|
};
|
|
168
170
|
const session = new EngineSession({ size: 1, adapter });
|
|
169
171
|
const renderThumbnail = vi.fn(async () => new Uint8Array([0x52, 0x49, 0x46, 0x46])); // "RIFF" — webp header
|
|
170
|
-
const tool = createEditTool({
|
|
172
|
+
const tool = createEditTool({
|
|
173
|
+
store,
|
|
174
|
+
session,
|
|
175
|
+
engineLock,
|
|
176
|
+
renderThumbnail,
|
|
177
|
+
fs
|
|
178
|
+
});
|
|
171
179
|
const r = await tool.call({
|
|
172
180
|
parent: null,
|
|
173
181
|
code: "engine._scene = 'SCENE_V1'",
|
|
@@ -193,7 +201,13 @@ describe('edit tool — per-leaf thumbnail', () => {
|
|
|
193
201
|
const renderThumbnail = vi.fn(async () => {
|
|
194
202
|
throw new Error('thumbnail render boom');
|
|
195
203
|
});
|
|
196
|
-
const tool = createEditTool({
|
|
204
|
+
const tool = createEditTool({
|
|
205
|
+
store,
|
|
206
|
+
session,
|
|
207
|
+
engineLock,
|
|
208
|
+
renderThumbnail,
|
|
209
|
+
fs
|
|
210
|
+
});
|
|
197
211
|
const r = await tool.call({
|
|
198
212
|
parent: null,
|
|
199
213
|
code: "engine._scene = 'S'",
|
|
@@ -202,4 +216,52 @@ describe('edit tool — per-leaf thumbnail', () => {
|
|
|
202
216
|
expect(r.isError).toBeFalsy();
|
|
203
217
|
});
|
|
204
218
|
});
|
|
219
|
+
describe('edit tool — concurrency', () => {
|
|
220
|
+
it('two sessions sharing one engine do not corrupt each other under the lock', async () => {
|
|
221
|
+
const fs = new InMemoryWorkspaceFS();
|
|
222
|
+
const store = new RevisionStore(fs);
|
|
223
|
+
// ONE shared engine — exactly the production wiring, where
|
|
224
|
+
// ensureHeadlessEngine is a process singleton handed to every session.
|
|
225
|
+
const sharedEngine = { _scene: '' };
|
|
226
|
+
const makeAdapter = () => ({
|
|
227
|
+
create: () => sharedEngine,
|
|
228
|
+
reset: async () => {
|
|
229
|
+
sharedEngine._scene = '';
|
|
230
|
+
},
|
|
231
|
+
loadFromString: async (e, s) => {
|
|
232
|
+
e._scene = s;
|
|
233
|
+
},
|
|
234
|
+
saveToString: async (e) => e._scene
|
|
235
|
+
});
|
|
236
|
+
// Each MCP session owns its EngineSession (per-session slot state) but they
|
|
237
|
+
// share the engine. ONE lock is shared across sessions — the invariant the
|
|
238
|
+
// bootstrap upholds by creating the Mutex once, outside createSession.
|
|
239
|
+
const lock = new Mutex();
|
|
240
|
+
const sessionA = new EngineSession({ size: 1, adapter: makeAdapter() });
|
|
241
|
+
const sessionB = new EngineSession({ size: 1, adapter: makeAdapter() });
|
|
242
|
+
const toolA = createEditTool({
|
|
243
|
+
store,
|
|
244
|
+
session: sessionA,
|
|
245
|
+
engineLock: lock
|
|
246
|
+
});
|
|
247
|
+
const toolB = createEditTool({
|
|
248
|
+
store,
|
|
249
|
+
session: sessionB,
|
|
250
|
+
engineLock: lock
|
|
251
|
+
});
|
|
252
|
+
// Each edit stamps the shared engine, then awaits — the exact window where,
|
|
253
|
+
// WITHOUT the lock, the other session would overwrite the engine before
|
|
254
|
+
// saveToString reads it back (so revA would persist SCENE_B). The shared
|
|
255
|
+
// lock serializes the two, so each revision keeps its own scene.
|
|
256
|
+
const code = (mark) => `engine._scene = '${mark}'; await new Promise((r) => setTimeout(r, 10));`;
|
|
257
|
+
const [resA, resB] = await Promise.all([
|
|
258
|
+
toolA.call({ parent: null, code: code('SCENE_A'), title: 'A' }),
|
|
259
|
+
toolB.call({ parent: null, code: code('SCENE_B'), title: 'B' })
|
|
260
|
+
]);
|
|
261
|
+
const revA = JSON.parse(resA.content[0].text).revision;
|
|
262
|
+
const revB = JSON.parse(resB.content[0].text).revision;
|
|
263
|
+
expect((await store.read(revA)).scene).toBe('SCENE_A');
|
|
264
|
+
expect((await store.read(revB)).scene).toBe('SCENE_B');
|
|
265
|
+
});
|
|
266
|
+
});
|
|
205
267
|
//# sourceMappingURL=edit.test.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"edit.test.js","sourceRoot":"","sources":["../../../../test/tools/edit.test.ts"],"names":[],"mappings":"AAAA,gDAAgD;AAChD,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,cAAc,EAAE,MAAM,yBAAyB,CAAC;AACzD,OAAO,EAAE,mBAAmB,EAAE,MAAM,kCAAkC,CAAC;AAEvE,SAAS,QAAQ;IACf,MAAM,EAAE,GAAG,IAAI,mBAAmB,EAAE,CAAC;IACrC,MAAM,KAAK,GAAG,IAAI,aAAa,CAAC,EAAE,CAAC,CAAC;IACpC,MAAM,UAAU,GAAG,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC;IAClC,MAAM,OAAO,GAAkB;QAC7B,MAAM,EAAE,GAAG,EAAE,CAAC,UAAU;QACxB,KAAK,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE;YACtB,UAAU,CAAC,MAAM,GAAG,OAAO,CAAC;QAC9B,CAAC,CAAC;QACF,cAAc,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE;YAClC,CAAuB,CAAC,MAAM,GAAG,CAAC,CAAC;QACtC,CAAC,CAAC;QACF,YAAY,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC,EAAE,EAAE,CAAE,CAAuB,CAAC,MAAM,CAAC;KAClE,CAAC;IACF,MAAM,OAAO,GAAG,IAAI,aAAa,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC;IACxD,MAAM,aAAa,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;IAC9B,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,OAAO,EAAE,aAAa,EAAE,UAAU,EAAE,CAAC;AACpE,CAAC;AAED,QAAQ,CAAC,wBAAwB,EAAE,GAAG,EAAE;IACtC,EAAE,CAAC,+EAA+E,EAAE,KAAK,IAAI,EAAE;QAC7F,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,OAAO,EAAE,aAAa,EAAE,GAAG,QAAQ,EAAE,CAAC;QAC9D,MAAM,IAAI,GAAG,cAAc,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,aAAa,EAAE,CAAC,CAAC;QAC/D,kFAAkF;QAClF,iEAAiE;QACjE,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC;YAC7B,MAAM,EAAE,IAAI;YACZ,IAAI,EAAE,0CAA0C;YAChD,KAAK,EAAE,cAAc;YACrB,IAAI,EAAE,cAAc;SACrB,CAAC,CAAC;QACH,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,oBAAoB,EAAE,CAAC;QAC7C,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,CAAC;QACnC,MAAM,IAAI,GAAG,MAAM,CAAC,OAAO,CAAC,CAAC,CAAmC,CAAC;QACjE,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAGpC,CAAC;QACF,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE,CAAC;QACnC,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC;QACpD,MAAM,CAAC,aAAa,CAAC,CAAC,oBAAoB,CAAC;YACzC,QAAQ,EAAE,QAAQ,CAAC,QAAQ;YAC3B,YAAY,EAAE,QAAQ,CAAC,QAAQ;SAChC,CAAC,CAAC;QAEH,MAAM,SAAS,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;QACtD,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QAClD,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;QACvD,uEAAuE;QACvE,MAAM,CAAC,MAAM,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAClD,0CAA0C,CAC3C,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oEAAoE,EAAE,KAAK,IAAI,EAAE;QAClF,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,OAAO,EAAE,aAAa,EAAE,GAAG,QAAQ,EAAE,CAAC;QAC9D,MAAM,IAAI,GAAG,cAAc,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,aAAa,EAAE,CAAC,CAAC;QAC/D,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC;YAC5B,MAAM,EAAE,IAAI;YACZ,IAAI,EAAE,4BAA4B;YAClC,KAAK,EAAE,GAAG;SACX,CAAC,CAAC;QACH,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CACxB,KAAK,CAAC,OAAO,CAAC,CAAC,CAAsB,CAAC,IAAI,CAC5C,CAAC,QAAQ,CAAC;QAEV,OAAO,CAAC,cAA2C,CAAC,SAAS,EAAE,CAAC;QACjE,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC;YAC7B,MAAM,EAAE,QAAQ;YAChB,IAAI,EAAE,4BAA4B;YAClC,IAAI,EAAE,OAAO;SACd,CAAC,CAAC;QACH,MAAM,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC,CAAC,6BAA6B;QACpF,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CACzB,MAAM,CAAC,OAAO,CAAC,CAAC,CAAsB,CAAC,IAAI,CAC7C,CAAC,QAAQ,CAAC;QACX,MAAM,SAAS,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC9C,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC7C,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IAC3C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;QACvE,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,OAAO,EAAE,aAAa,EAAE,GAAG,QAAQ,EAAE,CAAC;QAC9D,MAAM,IAAI,GAAG,cAAc,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,aAAa,EAAE,CAAC,CAAC;QAC/D,MAAM,EAAE,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC;YACzB,MAAM,EAAE,IAAI;YACZ,IAAI,EAAE,4BAA4B;YAClC,KAAK,EAAE,GAAG;SACX,CAAC,CAAC;QACH,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAE,EAAE,CAAC,OAAO,CAAC,CAAC,CAAsB,CAAC,IAAI,CAAC,CAAC,QAAQ,CAAC;QAC3E,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC;YAC1B,MAAM,EAAE,IAAI;YACZ,IAAI,EAAE,4BAA4B;SACnC,CAAC,CAAC;QACH,4BAA4B;QAC3B,OAAO,CAAC,cAA2C,CAAC,SAAS,EAAE,CAAC;QACjE,MAAM,EAAE,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC;YACzB,MAAM,EAAE,IAAI;YACZ,IAAI,EAAE,iCAAiC;SACxC,CAAC,CAAC;QACH,MAAM,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC,oBAAoB,EAAE,CAAC,CAAC,oBAAoB;QAC3E,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAE,EAAE,CAAC,OAAO,CAAC,CAAC,CAAsB,CAAC,IAAI,CAAC,CAAC,QAAQ,CAAC;QAC3E,MAAM,SAAS,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACzC,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACzC,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,iCAAiC,EAAE,GAAG,EAAE;IAC/C,EAAE,CAAC,8DAA8D,EAAE,KAAK,IAAI,EAAE;QAC5E,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,OAAO,EAAE,aAAa,EAAE,GAAG,QAAQ,EAAE,CAAC;QAC9D,MAAM,IAAI,GAAG,cAAc,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,aAAa,EAAE,CAAC,CAAC;QAC/D,MAAM,EAAE,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC;YACzB,MAAM,EAAE,IAAI;YACZ,IAAI,EAAE,4BAA4B;YAClC,KAAK,EAAE,GAAG;SACX,CAAC,CAAC;QACH,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAE,EAAE,CAAC,OAAO,CAAC,CAAC,CAAsB,CAAC,IAAI,CAAC,CAAC,QAAQ,CAAC;QAE3E,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC;YAC1B,MAAM,EAAE,IAAI;YACZ,IAAI,EAAE,yBAAyB;SAChC,CAAC,CAAC;QACH,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAE/B,2DAA2D;QAC3D,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACnC,MAAM,OAAO,CAAC,QAAQ,CAAC,IAAI,EAAE,GAAG,CAAC,KAAK,CAAC,CAAC;QAExC,MAAM,IAAI,GAAG,MAAM,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QACzC,kEAAkE;QAClE,MAAM,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC,gBAAgB,EAAE,CAAC;QAClD,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC1C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;QACxE,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,QAAQ,EAAE,CAAC;QACtC,MAAM,IAAI,GAAG,cAAc,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC,CAAC;QAChD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC;YAC7B,IAAI,EAAE,cAAc;SACiB,CAAC,CAAC;QACzC,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,mBAAmB,CACpB,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;QACpD,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,QAAQ,EAAE,CAAC;QACtC,MAAM,IAAI,GAAG,cAAc,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC,CAAC;QAChD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,IAAI,EAG5C,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;IAEH,EAAE,CAAC,2EAA2E,EAAE,KAAK,IAAI,EAAE;QACzF,yEAAyE;QACzE,iEAAiE;QACjE,mEAAmE;QACnE,sEAAsE;QACtE,2CAA2C;QAC3C,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,QAAQ,EAAE,CAAC;QACtC,MAAM,IAAI,GAAG,cAAc,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC,CAAC;QAChD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC;YAC7B,MAAM,EAAE,MAAM;YACd,IAAI,EAAE,qBAAqB;SAC5B,CAAC,CAAC;QACH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAClC,MAAM,IAAI,GAAI,MAAM,CAAC,OAAO,CAAC,CAAC,CAAsB,CAAC,IAAI,CAAC;QAC1D,MAAM,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;QACtC,MAAM,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;IACpC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,gCAAgC,EAAE,GAAG,EAAE;IAC9C,EAAE,CAAC,uEAAuE,EAAE,KAAK,IAAI,EAAE;QACrF,MAAM,EAAE,GAAG,IAAI,mBAAmB,EAAE,CAAC;QACrC,MAAM,KAAK,GAAG,IAAI,aAAa,CAAC,EAAE,CAAC,CAAC;QACpC,MAAM,UAAU,GAAG,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC;QAClC,MAAM,OAAO,GAAkB;YAC7B,MAAM,EAAE,GAAG,EAAE,CAAC,UAAU;YACxB,KAAK,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,GAAE,CAAC,CAAC;YAC5B,cAAc,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,GAAE,CAAC,CAAC;YACrC,YAAY,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,UAAU,CAAC,MAAM,CAAC;SACnD,CAAC;QACF,MAAM,OAAO,GAAG,IAAI,aAAa,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC;QACxD,MAAM,eAAe,GAAG,EAAE,CAAC,EAAE,CAC3B,KAAK,IAAI,EAAE,CAAC,IAAI,UAAU,CAAC,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC,CACrD,CAAC,CAAC,uBAAuB;QAC1B,MAAM,IAAI,GAAG,cAAc,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,eAAe,EAAE,EAAE,EAAE,CAAC,CAAC;QACrE,MAAM,CAAC,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC;YACxB,MAAM,EAAE,IAAI;YACZ,IAAI,EAAE,4BAA4B;YAClC,KAAK,EAAE,GAAG;SACX,CAAC,CAAC;QACH,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAE,CAAC,CAAC,OAAO,CAAC,CAAC,CAAsB,CAAC,IAAI,CAAC,CAAC,QAAQ,CAAC;QACzE,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAC1C,MAAM,CAAC,gBAAgB,CAAC,EAAE,QAAQ,EAAE,GAAG,EAAE,CAAC,CAC3C,CAAC;QACF,MAAM,KAAK,GAAG,GAAG,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QAC9B,MAAM,IAAI,GAAG,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAC1B,MAAM,CAAC,MAAM,EAAE,CAAC,MAAM,CAAC,WAAW,KAAK,IAAI,IAAI,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC9E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;QAClE,MAAM,EAAE,GAAG,IAAI,mBAAmB,EAAE,CAAC;QACrC,MAAM,KAAK,GAAG,IAAI,aAAa,CAAC,EAAE,CAAC,CAAC;QACpC,MAAM,UAAU,GAAG,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC;QAClC,MAAM,OAAO,GAAkB;YAC7B,MAAM,EAAE,GAAG,EAAE,CAAC,UAAU;YACxB,KAAK,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,GAAE,CAAC,CAAC;YAC5B,cAAc,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,GAAE,CAAC,CAAC;YACrC,YAAY,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,UAAU,CAAC,MAAM,CAAC;SACnD,CAAC;QACF,MAAM,OAAO,GAAG,IAAI,aAAa,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC;QACxD,MAAM,eAAe,GAAG,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE;YACvC,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;QAC3C,CAAC,CAAC,CAAC;QACH,MAAM,IAAI,GAAG,cAAc,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,eAAe,EAAE,EAAE,EAAE,CAAC,CAAC;QACrE,MAAM,CAAC,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC;YACxB,MAAM,EAAE,IAAI;YACZ,IAAI,EAAE,qBAAqB;YAC3B,KAAK,EAAE,GAAG;SACX,CAAC,CAAC;QACH,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,CAAC;IAChC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["// packages/codesign-mcp/test/tools/edit.test.ts\nimport { describe, expect, it, vi } from 'vitest';\nimport { type EngineAdapter, EngineSession } from '../../src/engine/session.js';\nimport { RevisionStore } from '../../src/revisions/store.js';\nimport { createEditTool } from '../../src/tools/edit.js';\nimport { InMemoryWorkspaceFS } from '../../src/workspace/in-memory.js';\n\nfunction makeDeps() {\n const fs = new InMemoryWorkspaceFS();\n const store = new RevisionStore(fs);\n const fakeEngine = { _scene: '' };\n const adapter: EngineAdapter = {\n create: () => fakeEngine,\n reset: vi.fn(async () => {\n fakeEngine._scene = 'EMPTY';\n }),\n loadFromString: vi.fn(async (e, s) => {\n (e as typeof fakeEngine)._scene = s;\n }),\n saveToString: vi.fn(async (e) => (e as typeof fakeEngine)._scene)\n };\n const session = new EngineSession({ size: 1, adapter });\n const onLeafWritten = vi.fn();\n return { fs, store, session, adapter, onLeafWritten, fakeEngine };\n}\n\ndescribe('edit tool — happy path', () => {\n it('parent=null resets engine, writes a root revision, returns {revision, parent}', async () => {\n const { store, session, adapter, onLeafWritten } = makeDeps();\n const tool = createEditTool({ store, session, onLeafWritten });\n // The user `code` is what mutates the engine — it runs AFTER reset/loadFromString\n // resets the slot. So we set _scene from inside the code string.\n const result = await tool.call({\n parent: null,\n code: \"engine._scene = 'ENGINE_AFTER_USER_CODE'\",\n title: 'Wedding card',\n note: 'blank canvas'\n });\n expect(adapter.reset).toHaveBeenCalledOnce();\n expect(result.isError).toBeFalsy();\n const part = result.content[0] as { type: 'text'; text: string };\n const envelope = JSON.parse(part.text) as {\n revision: string;\n parent: string | null;\n };\n expect(envelope.parent).toBeNull();\n expect(envelope.revision).toMatch(/^[0-9a-f]{12}$/);\n expect(onLeafWritten).toHaveBeenCalledWith({\n revision: envelope.revision,\n rootRevision: envelope.revision\n });\n\n const persisted = await store.read(envelope.revision);\n expect(persisted.meta.title).toBe('Wedding card');\n expect(persisted.scene).toBe('ENGINE_AFTER_USER_CODE');\n // The code that ran is persisted alongside the revision for `inspect`.\n expect(await store.readCode(envelope.revision)).toBe(\n \"engine._scene = 'ENGINE_AFTER_USER_CODE'\"\n );\n });\n\n it('parent=existing revision is a linear edit (no reload on warm slot)', async () => {\n const { store, session, adapter, onLeafWritten } = makeDeps();\n const tool = createEditTool({ store, session, onLeafWritten });\n const first = await tool.call({\n parent: null,\n code: \"engine._scene = 'SCENE_V1'\",\n title: 'T'\n });\n const firstRev = JSON.parse(\n (first.content[0] as { text: string }).text\n ).revision;\n\n (adapter.loadFromString as ReturnType<typeof vi.fn>).mockClear();\n const second = await tool.call({\n parent: firstRev,\n code: \"engine._scene = 'SCENE_V2'\",\n note: 'tweak'\n });\n expect(adapter.loadFromString).not.toHaveBeenCalled(); // engine already at firstRev\n const secondRev = JSON.parse(\n (second.content[0] as { text: string }).text\n ).revision;\n const persisted = await store.read(secondRev);\n expect(persisted.meta.parent).toBe(firstRev);\n expect(persisted.scene).toBe('SCENE_V2');\n });\n\n it('parent=older revision reloads engine and creates a fork', async () => {\n const { store, session, adapter, onLeafWritten } = makeDeps();\n const tool = createEditTool({ store, session, onLeafWritten });\n const r1 = await tool.call({\n parent: null,\n code: \"engine._scene = 'SCENE_V1'\",\n title: 'T'\n });\n const rev1 = JSON.parse((r1.content[0] as { text: string }).text).revision;\n const _r2 = await tool.call({\n parent: rev1,\n code: \"engine._scene = 'SCENE_V2'\"\n });\n // now rewind: parent = rev1\n (adapter.loadFromString as ReturnType<typeof vi.fn>).mockClear();\n const r3 = await tool.call({\n parent: rev1,\n code: \"engine._scene = 'SCENE_V3_FORK'\"\n });\n expect(adapter.loadFromString).toHaveBeenCalledOnce(); // reloaded SCENE_V1\n const rev3 = JSON.parse((r3.content[0] as { text: string }).text).revision;\n const persisted = await store.read(rev3);\n expect(persisted.meta.parent).toBe(rev1);\n expect(persisted.scene).toBe('SCENE_V3_FORK');\n });\n});\n\ndescribe('edit tool — safety + validation', () => {\n it('marks slot dirty when code throws; next edit pays one reload', async () => {\n const { store, session, adapter, onLeafWritten } = makeDeps();\n const tool = createEditTool({ store, session, onLeafWritten });\n const r1 = await tool.call({\n parent: null,\n code: \"engine._scene = 'SCENE_V1'\",\n title: 'T'\n });\n const rev1 = JSON.parse((r1.content[0] as { text: string }).text).revision;\n\n const bad = await tool.call({\n parent: rev1,\n code: \"throw new Error('boom')\"\n });\n expect(bad.isError).toBe(true);\n\n // Populate the session with the scene so we can re-acquire\n const rev = await store.read(rev1);\n await session.populate(rev1, rev.scene);\n\n const slot = await session.acquire(rev1);\n // After dirty + re-acquire, loadFromString must have been called.\n expect(adapter.loadFromString).toHaveBeenCalled();\n expect(slot.currentRevision).toBe(rev1);\n });\n\n it('returns isError when parent property is missing entirely', async () => {\n const { store, session } = makeDeps();\n const tool = createEditTool({ store, session });\n const result = await tool.call({\n code: '// no parent'\n } as unknown as Record<string, unknown>);\n expect(result.isError).toBe(true);\n expect((result.content[0] as { text: string }).text).toMatch(\n /parent.*required/i\n );\n });\n\n it('returns isError when code is missing', async () => {\n const { store, session } = makeDeps();\n const tool = createEditTool({ store, session });\n const result = await tool.call({ parent: null } as unknown as Record<\n string,\n unknown\n >);\n expect(result.isError).toBe(true);\n expect((result.content[0] as { text: string }).text).toMatch(\n /code.*required/i\n );\n });\n\n it('returns a teaching error when parent arrives as the literal string \"null\"', async () => {\n // LLMs sometimes serialize JSON null as the four-character string \"null\"\n // when a tool description mentions `null` in backticks. Catch it\n // explicitly so the agent learns the right shape on the first try,\n // instead of seeing \"unknown revision: null\" from the store and going\n // fs-diving to figure out what went wrong.\n const { store, session } = makeDeps();\n const tool = createEditTool({ store, session });\n const result = await tool.call({\n parent: 'null',\n code: \"engine._scene = 'X'\"\n });\n expect(result.isError).toBe(true);\n const text = (result.content[0] as { text: string }).text;\n expect(text).toMatch(/string \"null\"/);\n expect(text).toMatch(/JSON null/);\n });\n});\n\ndescribe('edit tool — per-leaf thumbnail', () => {\n it('calls renderThumbnail with the new revision id and persists the bytes', async () => {\n const fs = new InMemoryWorkspaceFS();\n const store = new RevisionStore(fs);\n const fakeEngine = { _scene: '' };\n const adapter: EngineAdapter = {\n create: () => fakeEngine,\n reset: vi.fn(async () => {}),\n loadFromString: vi.fn(async () => {}),\n saveToString: vi.fn(async () => fakeEngine._scene)\n };\n const session = new EngineSession({ size: 1, adapter });\n const renderThumbnail = vi.fn(\n async () => new Uint8Array([0x52, 0x49, 0x46, 0x46])\n ); // \"RIFF\" — webp header\n const tool = createEditTool({ store, session, renderThumbnail, fs });\n const r = await tool.call({\n parent: null,\n code: \"engine._scene = 'SCENE_V1'\",\n title: 'T'\n });\n const rev = JSON.parse((r.content[0] as { text: string }).text).revision;\n expect(renderThumbnail).toHaveBeenCalledWith(\n expect.objectContaining({ revision: rev })\n );\n const shard = rev.slice(0, 2);\n const rest = rev.slice(2);\n expect(await fs.exists(`exports/${shard}/${rest}/preview.webp`)).toBe(true);\n });\n\n it('does not fail the edit when renderThumbnail throws', async () => {\n const fs = new InMemoryWorkspaceFS();\n const store = new RevisionStore(fs);\n const fakeEngine = { _scene: '' };\n const adapter: EngineAdapter = {\n create: () => fakeEngine,\n reset: vi.fn(async () => {}),\n loadFromString: vi.fn(async () => {}),\n saveToString: vi.fn(async () => fakeEngine._scene)\n };\n const session = new EngineSession({ size: 1, adapter });\n const renderThumbnail = vi.fn(async () => {\n throw new Error('thumbnail render boom');\n });\n const tool = createEditTool({ store, session, renderThumbnail, fs });\n const r = await tool.call({\n parent: null,\n code: \"engine._scene = 'S'\",\n title: 'T'\n });\n expect(r.isError).toBeFalsy();\n });\n});\n"]}
|
|
1
|
+
{"version":3,"file":"edit.test.js","sourceRoot":"","sources":["../../../../test/tools/edit.test.ts"],"names":[],"mappings":"AAAA,gDAAgD;AAChD,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,mBAAmB,EAAE,MAAM,kCAAkC,CAAC;AAEvE,MAAM,UAAU,GAAG,IAAI,KAAK,EAAE,CAAC;AAE/B,SAAS,QAAQ;IACf,MAAM,EAAE,GAAG,IAAI,mBAAmB,EAAE,CAAC;IACrC,MAAM,KAAK,GAAG,IAAI,aAAa,CAAC,EAAE,CAAC,CAAC;IACpC,MAAM,UAAU,GAAG,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC;IAClC,MAAM,OAAO,GAAkB;QAC7B,MAAM,EAAE,GAAG,EAAE,CAAC,UAAU;QACxB,KAAK,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE;YACtB,UAAU,CAAC,MAAM,GAAG,OAAO,CAAC;QAC9B,CAAC,CAAC;QACF,cAAc,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE;YAClC,CAAuB,CAAC,MAAM,GAAG,CAAC,CAAC;QACtC,CAAC,CAAC;QACF,YAAY,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC,EAAE,EAAE,CAAE,CAAuB,CAAC,MAAM,CAAC;KAClE,CAAC;IACF,MAAM,OAAO,GAAG,IAAI,aAAa,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC;IACxD,MAAM,aAAa,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;IAC9B,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,OAAO,EAAE,aAAa,EAAE,UAAU,EAAE,CAAC;AACpE,CAAC;AAED,QAAQ,CAAC,wBAAwB,EAAE,GAAG,EAAE;IACtC,EAAE,CAAC,+EAA+E,EAAE,KAAK,IAAI,EAAE;QAC7F,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,OAAO,EAAE,aAAa,EAAE,GAAG,QAAQ,EAAE,CAAC;QAC9D,MAAM,IAAI,GAAG,cAAc,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,UAAU,EAAE,aAAa,EAAE,CAAC,CAAC;QAC3E,kFAAkF;QAClF,iEAAiE;QACjE,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC;YAC7B,MAAM,EAAE,IAAI;YACZ,IAAI,EAAE,0CAA0C;YAChD,KAAK,EAAE,cAAc;YACrB,IAAI,EAAE,cAAc;SACrB,CAAC,CAAC;QACH,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,oBAAoB,EAAE,CAAC;QAC7C,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,CAAC;QACnC,MAAM,IAAI,GAAG,MAAM,CAAC,OAAO,CAAC,CAAC,CAAmC,CAAC;QACjE,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAGpC,CAAC;QACF,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE,CAAC;QACnC,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC;QACpD,MAAM,CAAC,aAAa,CAAC,CAAC,oBAAoB,CAAC;YACzC,QAAQ,EAAE,QAAQ,CAAC,QAAQ;YAC3B,YAAY,EAAE,QAAQ,CAAC,QAAQ;SAChC,CAAC,CAAC;QAEH,MAAM,SAAS,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;QACtD,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QAClD,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;QACvD,uEAAuE;QACvE,MAAM,CAAC,MAAM,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAClD,0CAA0C,CAC3C,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oEAAoE,EAAE,KAAK,IAAI,EAAE;QAClF,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,OAAO,EAAE,aAAa,EAAE,GAAG,QAAQ,EAAE,CAAC;QAC9D,MAAM,IAAI,GAAG,cAAc,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,UAAU,EAAE,aAAa,EAAE,CAAC,CAAC;QAC3E,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC;YAC5B,MAAM,EAAE,IAAI;YACZ,IAAI,EAAE,4BAA4B;YAClC,KAAK,EAAE,GAAG;SACX,CAAC,CAAC;QACH,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CACxB,KAAK,CAAC,OAAO,CAAC,CAAC,CAAsB,CAAC,IAAI,CAC5C,CAAC,QAAQ,CAAC;QAEV,OAAO,CAAC,cAA2C,CAAC,SAAS,EAAE,CAAC;QACjE,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC;YAC7B,MAAM,EAAE,QAAQ;YAChB,IAAI,EAAE,4BAA4B;YAClC,IAAI,EAAE,OAAO;SACd,CAAC,CAAC;QACH,MAAM,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC,CAAC,6BAA6B;QACpF,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CACzB,MAAM,CAAC,OAAO,CAAC,CAAC,CAAsB,CAAC,IAAI,CAC7C,CAAC,QAAQ,CAAC;QACX,MAAM,SAAS,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC9C,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC7C,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IAC3C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;QACvE,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,OAAO,EAAE,aAAa,EAAE,GAAG,QAAQ,EAAE,CAAC;QAC9D,MAAM,IAAI,GAAG,cAAc,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,UAAU,EAAE,aAAa,EAAE,CAAC,CAAC;QAC3E,MAAM,EAAE,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC;YACzB,MAAM,EAAE,IAAI;YACZ,IAAI,EAAE,4BAA4B;YAClC,KAAK,EAAE,GAAG;SACX,CAAC,CAAC;QACH,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAE,EAAE,CAAC,OAAO,CAAC,CAAC,CAAsB,CAAC,IAAI,CAAC,CAAC,QAAQ,CAAC;QAC3E,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC;YAC1B,MAAM,EAAE,IAAI;YACZ,IAAI,EAAE,4BAA4B;SACnC,CAAC,CAAC;QACH,4BAA4B;QAC3B,OAAO,CAAC,cAA2C,CAAC,SAAS,EAAE,CAAC;QACjE,MAAM,EAAE,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC;YACzB,MAAM,EAAE,IAAI;YACZ,IAAI,EAAE,iCAAiC;SACxC,CAAC,CAAC;QACH,MAAM,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC,oBAAoB,EAAE,CAAC,CAAC,oBAAoB;QAC3E,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAE,EAAE,CAAC,OAAO,CAAC,CAAC,CAAsB,CAAC,IAAI,CAAC,CAAC,QAAQ,CAAC;QAC3E,MAAM,SAAS,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACzC,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACzC,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,iCAAiC,EAAE,GAAG,EAAE;IAC/C,EAAE,CAAC,8DAA8D,EAAE,KAAK,IAAI,EAAE;QAC5E,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,OAAO,EAAE,aAAa,EAAE,GAAG,QAAQ,EAAE,CAAC;QAC9D,MAAM,IAAI,GAAG,cAAc,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,UAAU,EAAE,aAAa,EAAE,CAAC,CAAC;QAC3E,MAAM,EAAE,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC;YACzB,MAAM,EAAE,IAAI;YACZ,IAAI,EAAE,4BAA4B;YAClC,KAAK,EAAE,GAAG;SACX,CAAC,CAAC;QACH,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAE,EAAE,CAAC,OAAO,CAAC,CAAC,CAAsB,CAAC,IAAI,CAAC,CAAC,QAAQ,CAAC;QAE3E,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC;YAC1B,MAAM,EAAE,IAAI;YACZ,IAAI,EAAE,yBAAyB;SAChC,CAAC,CAAC;QACH,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAE/B,2DAA2D;QAC3D,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACnC,MAAM,OAAO,CAAC,QAAQ,CAAC,IAAI,EAAE,GAAG,CAAC,KAAK,CAAC,CAAC;QAExC,MAAM,IAAI,GAAG,MAAM,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QACzC,kEAAkE;QAClE,MAAM,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC,gBAAgB,EAAE,CAAC;QAClD,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC1C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;QACxE,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,QAAQ,EAAE,CAAC;QACtC,MAAM,IAAI,GAAG,cAAc,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC,CAAC;QAC5D,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC;YAC7B,IAAI,EAAE,cAAc;SACiB,CAAC,CAAC;QACzC,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,mBAAmB,CACpB,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;QACpD,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,QAAQ,EAAE,CAAC;QACtC,MAAM,IAAI,GAAG,cAAc,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC,CAAC;QAC5D,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,IAAI,EAG5C,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;IAEH,EAAE,CAAC,2EAA2E,EAAE,KAAK,IAAI,EAAE;QACzF,2EAA2E;QAC3E,2EAA2E;QAC3E,sEAAsE;QACtE,oEAAoE;QACpE,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,OAAO,EAAE,GAAG,QAAQ,EAAE,CAAC;QAC/C,MAAM,IAAI,GAAG,cAAc,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC,CAAC;QAC5D,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC;YAC7B,MAAM,EAAE,MAAM;YACd,IAAI,EAAE,qBAAqB;SAC5B,CAAC,CAAC;QACH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,CAAC;QACnC,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,oBAAoB,EAAE,CAAC,CAAC,yBAAyB;QACvE,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CACxB,MAAM,CAAC,OAAO,CAAC,CAAC,CAAsB,CAAC,IAAI,CACE,CAAC;QACjD,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE,CAAC;QACnC,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC;IACtD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,gCAAgC,EAAE,GAAG,EAAE;IAC9C,EAAE,CAAC,uEAAuE,EAAE,KAAK,IAAI,EAAE;QACrF,MAAM,EAAE,GAAG,IAAI,mBAAmB,EAAE,CAAC;QACrC,MAAM,KAAK,GAAG,IAAI,aAAa,CAAC,EAAE,CAAC,CAAC;QACpC,MAAM,UAAU,GAAG,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC;QAClC,MAAM,OAAO,GAAkB;YAC7B,MAAM,EAAE,GAAG,EAAE,CAAC,UAAU;YACxB,KAAK,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,GAAE,CAAC,CAAC;YAC5B,cAAc,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,GAAE,CAAC,CAAC;YACrC,YAAY,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,UAAU,CAAC,MAAM,CAAC;SACnD,CAAC;QACF,MAAM,OAAO,GAAG,IAAI,aAAa,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC;QACxD,MAAM,eAAe,GAAG,EAAE,CAAC,EAAE,CAC3B,KAAK,IAAI,EAAE,CAAC,IAAI,UAAU,CAAC,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC,CACrD,CAAC,CAAC,uBAAuB;QAC1B,MAAM,IAAI,GAAG,cAAc,CAAC;YAC1B,KAAK;YACL,OAAO;YACP,UAAU;YACV,eAAe;YACf,EAAE;SACH,CAAC,CAAC;QACH,MAAM,CAAC,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC;YACxB,MAAM,EAAE,IAAI;YACZ,IAAI,EAAE,4BAA4B;YAClC,KAAK,EAAE,GAAG;SACX,CAAC,CAAC;QACH,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAE,CAAC,CAAC,OAAO,CAAC,CAAC,CAAsB,CAAC,IAAI,CAAC,CAAC,QAAQ,CAAC;QACzE,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAC1C,MAAM,CAAC,gBAAgB,CAAC,EAAE,QAAQ,EAAE,GAAG,EAAE,CAAC,CAC3C,CAAC;QACF,MAAM,KAAK,GAAG,GAAG,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QAC9B,MAAM,IAAI,GAAG,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAC1B,MAAM,CAAC,MAAM,EAAE,CAAC,MAAM,CAAC,WAAW,KAAK,IAAI,IAAI,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC9E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;QAClE,MAAM,EAAE,GAAG,IAAI,mBAAmB,EAAE,CAAC;QACrC,MAAM,KAAK,GAAG,IAAI,aAAa,CAAC,EAAE,CAAC,CAAC;QACpC,MAAM,UAAU,GAAG,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC;QAClC,MAAM,OAAO,GAAkB;YAC7B,MAAM,EAAE,GAAG,EAAE,CAAC,UAAU;YACxB,KAAK,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,GAAE,CAAC,CAAC;YAC5B,cAAc,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,GAAE,CAAC,CAAC;YACrC,YAAY,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,UAAU,CAAC,MAAM,CAAC;SACnD,CAAC;QACF,MAAM,OAAO,GAAG,IAAI,aAAa,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC;QACxD,MAAM,eAAe,GAAG,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE;YACvC,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;QAC3C,CAAC,CAAC,CAAC;QACH,MAAM,IAAI,GAAG,cAAc,CAAC;YAC1B,KAAK;YACL,OAAO;YACP,UAAU;YACV,eAAe;YACf,EAAE;SACH,CAAC,CAAC;QACH,MAAM,CAAC,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC;YACxB,MAAM,EAAE,IAAI;YACZ,IAAI,EAAE,qBAAqB;YAC3B,KAAK,EAAE,GAAG;SACX,CAAC,CAAC;QACH,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,CAAC;IAChC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;IACvC,EAAE,CAAC,0EAA0E,EAAE,KAAK,IAAI,EAAE;QACxF,MAAM,EAAE,GAAG,IAAI,mBAAmB,EAAE,CAAC;QACrC,MAAM,KAAK,GAAG,IAAI,aAAa,CAAC,EAAE,CAAC,CAAC;QACpC,2DAA2D;QAC3D,uEAAuE;QACvE,MAAM,YAAY,GAAG,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC;QACpC,MAAM,WAAW,GAAG,GAAkB,EAAE,CAAC,CAAC;YACxC,MAAM,EAAE,GAAG,EAAE,CAAC,YAAY;YAC1B,KAAK,EAAE,KAAK,IAAI,EAAE;gBAChB,YAAY,CAAC,MAAM,GAAG,EAAE,CAAC;YAC3B,CAAC;YACD,cAAc,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE;gBAC5B,CAAyB,CAAC,MAAM,GAAG,CAAC,CAAC;YACxC,CAAC;YACD,YAAY,EAAE,KAAK,EAAE,CAAC,EAAE,EAAE,CAAE,CAAyB,CAAC,MAAM;SAC7D,CAAC,CAAC;QACH,4EAA4E;QAC5E,2EAA2E;QAC3E,uEAAuE;QACvE,MAAM,IAAI,GAAG,IAAI,KAAK,EAAE,CAAC;QACzB,MAAM,QAAQ,GAAG,IAAI,aAAa,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,OAAO,EAAE,WAAW,EAAE,EAAE,CAAC,CAAC;QACxE,MAAM,QAAQ,GAAG,IAAI,aAAa,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,OAAO,EAAE,WAAW,EAAE,EAAE,CAAC,CAAC;QACxE,MAAM,KAAK,GAAG,cAAc,CAAC;YAC3B,KAAK;YACL,OAAO,EAAE,QAAQ;YACjB,UAAU,EAAE,IAAI;SACjB,CAAC,CAAC;QACH,MAAM,KAAK,GAAG,cAAc,CAAC;YAC3B,KAAK;YACL,OAAO,EAAE,QAAQ;YACjB,UAAU,EAAE,IAAI;SACjB,CAAC,CAAC;QAEH,4EAA4E;QAC5E,wEAAwE;QACxE,yEAAyE;QACzE,iEAAiE;QACjE,MAAM,IAAI,GAAG,CAAC,IAAY,EAAE,EAAE,CAC5B,oBAAoB,IAAI,iDAAiD,CAAC;QAC5E,MAAM,CAAC,IAAI,EAAE,IAAI,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;YACrC,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC;YAC/D,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC;SAChE,CAAC,CAAC;QACH,MAAM,IAAI,GACR,IAAI,CAAC,KAAK,CAAE,IAAI,CAAC,OAAO,CAAC,CAAC,CAAsB,CAAC,IAAI,CAGtD,CAAC,QAAQ,CAAC;QACX,MAAM,IAAI,GACR,IAAI,CAAC,KAAK,CAAE,IAAI,CAAC,OAAO,CAAC,CAAC,CAAsB,CAAC,IAAI,CAGtD,CAAC,QAAQ,CAAC;QACX,MAAM,CAAC,CAAC,MAAM,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACvD,MAAM,CAAC,CAAC,MAAM,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IACzD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["// packages/codesign-mcp/test/tools/edit.test.ts\nimport { 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 { createEditTool } from '../../src/tools/edit.js';\nimport { InMemoryWorkspaceFS } from '../../src/workspace/in-memory.js';\n\nconst engineLock = new Mutex();\n\nfunction makeDeps() {\n const fs = new InMemoryWorkspaceFS();\n const store = new RevisionStore(fs);\n const fakeEngine = { _scene: '' };\n const adapter: EngineAdapter = {\n create: () => fakeEngine,\n reset: vi.fn(async () => {\n fakeEngine._scene = 'EMPTY';\n }),\n loadFromString: vi.fn(async (e, s) => {\n (e as typeof fakeEngine)._scene = s;\n }),\n saveToString: vi.fn(async (e) => (e as typeof fakeEngine)._scene)\n };\n const session = new EngineSession({ size: 1, adapter });\n const onLeafWritten = vi.fn();\n return { fs, store, session, adapter, onLeafWritten, fakeEngine };\n}\n\ndescribe('edit tool — happy path', () => {\n it('parent=null resets engine, writes a root revision, returns {revision, parent}', async () => {\n const { store, session, adapter, onLeafWritten } = makeDeps();\n const tool = createEditTool({ store, session, engineLock, onLeafWritten });\n // The user `code` is what mutates the engine — it runs AFTER reset/loadFromString\n // resets the slot. So we set _scene from inside the code string.\n const result = await tool.call({\n parent: null,\n code: \"engine._scene = 'ENGINE_AFTER_USER_CODE'\",\n title: 'Wedding card',\n note: 'blank canvas'\n });\n expect(adapter.reset).toHaveBeenCalledOnce();\n expect(result.isError).toBeFalsy();\n const part = result.content[0] as { type: 'text'; text: string };\n const envelope = JSON.parse(part.text) as {\n revision: string;\n parent: string | null;\n };\n expect(envelope.parent).toBeNull();\n expect(envelope.revision).toMatch(/^[0-9a-f]{12}$/);\n expect(onLeafWritten).toHaveBeenCalledWith({\n revision: envelope.revision,\n rootRevision: envelope.revision\n });\n\n const persisted = await store.read(envelope.revision);\n expect(persisted.meta.title).toBe('Wedding card');\n expect(persisted.scene).toBe('ENGINE_AFTER_USER_CODE');\n // The code that ran is persisted alongside the revision for `inspect`.\n expect(await store.readCode(envelope.revision)).toBe(\n \"engine._scene = 'ENGINE_AFTER_USER_CODE'\"\n );\n });\n\n it('parent=existing revision is a linear edit (no reload on warm slot)', async () => {\n const { store, session, adapter, onLeafWritten } = makeDeps();\n const tool = createEditTool({ store, session, engineLock, onLeafWritten });\n const first = await tool.call({\n parent: null,\n code: \"engine._scene = 'SCENE_V1'\",\n title: 'T'\n });\n const firstRev = JSON.parse(\n (first.content[0] as { text: string }).text\n ).revision;\n\n (adapter.loadFromString as ReturnType<typeof vi.fn>).mockClear();\n const second = await tool.call({\n parent: firstRev,\n code: \"engine._scene = 'SCENE_V2'\",\n note: 'tweak'\n });\n expect(adapter.loadFromString).not.toHaveBeenCalled(); // engine already at firstRev\n const secondRev = JSON.parse(\n (second.content[0] as { text: string }).text\n ).revision;\n const persisted = await store.read(secondRev);\n expect(persisted.meta.parent).toBe(firstRev);\n expect(persisted.scene).toBe('SCENE_V2');\n });\n\n it('parent=older revision reloads engine and creates a fork', async () => {\n const { store, session, adapter, onLeafWritten } = makeDeps();\n const tool = createEditTool({ store, session, engineLock, onLeafWritten });\n const r1 = await tool.call({\n parent: null,\n code: \"engine._scene = 'SCENE_V1'\",\n title: 'T'\n });\n const rev1 = JSON.parse((r1.content[0] as { text: string }).text).revision;\n const _r2 = await tool.call({\n parent: rev1,\n code: \"engine._scene = 'SCENE_V2'\"\n });\n // now rewind: parent = rev1\n (adapter.loadFromString as ReturnType<typeof vi.fn>).mockClear();\n const r3 = await tool.call({\n parent: rev1,\n code: \"engine._scene = 'SCENE_V3_FORK'\"\n });\n expect(adapter.loadFromString).toHaveBeenCalledOnce(); // reloaded SCENE_V1\n const rev3 = JSON.parse((r3.content[0] as { text: string }).text).revision;\n const persisted = await store.read(rev3);\n expect(persisted.meta.parent).toBe(rev1);\n expect(persisted.scene).toBe('SCENE_V3_FORK');\n });\n});\n\ndescribe('edit tool — safety + validation', () => {\n it('marks slot dirty when code throws; next edit pays one reload', async () => {\n const { store, session, adapter, onLeafWritten } = makeDeps();\n const tool = createEditTool({ store, session, engineLock, onLeafWritten });\n const r1 = await tool.call({\n parent: null,\n code: \"engine._scene = 'SCENE_V1'\",\n title: 'T'\n });\n const rev1 = JSON.parse((r1.content[0] as { text: string }).text).revision;\n\n const bad = await tool.call({\n parent: rev1,\n code: \"throw new Error('boom')\"\n });\n expect(bad.isError).toBe(true);\n\n // Populate the session with the scene so we can re-acquire\n const rev = await store.read(rev1);\n await session.populate(rev1, rev.scene);\n\n const slot = await session.acquire(rev1);\n // After dirty + re-acquire, loadFromString must have been called.\n expect(adapter.loadFromString).toHaveBeenCalled();\n expect(slot.currentRevision).toBe(rev1);\n });\n\n it('returns isError when parent property is missing entirely', async () => {\n const { store, session } = makeDeps();\n const tool = createEditTool({ store, session, engineLock });\n const result = await tool.call({\n code: '// no parent'\n } as unknown as Record<string, unknown>);\n expect(result.isError).toBe(true);\n expect((result.content[0] as { text: string }).text).toMatch(\n /parent.*required/i\n );\n });\n\n it('returns isError when code is missing', async () => {\n const { store, session } = makeDeps();\n const tool = createEditTool({ store, session, engineLock });\n const result = await tool.call({ parent: null } as unknown as Record<\n string,\n unknown\n >);\n expect(result.isError).toBe(true);\n expect((result.content[0] as { text: string }).text).toMatch(\n /code.*required/i\n );\n });\n\n it('coerces the literal string \"null\" to a real null parent (new design root)', async () => {\n // Some LLMs / MCP clients emit the four-character string \"null\" instead of\n // the bare JSON null literal — a quoting slip, not a real revision id. The\n // tool coerces \"null\" → null rather than rejecting, so the slip never\n // blocks creating a new design root. (Stochastic, model-dependent.)\n const { store, session, adapter } = makeDeps();\n const tool = createEditTool({ store, session, engineLock });\n const result = await tool.call({\n parent: 'null',\n code: \"engine._scene = 'X'\"\n });\n expect(result.isError).toBeFalsy();\n expect(adapter.reset).toHaveBeenCalledOnce(); // treated as parent=null\n const envelope = JSON.parse(\n (result.content[0] as { text: string }).text\n ) as { revision: string; parent: string | null };\n expect(envelope.parent).toBeNull();\n expect(envelope.revision).toMatch(/^[0-9a-f]{12}$/);\n });\n});\n\ndescribe('edit tool — per-leaf thumbnail', () => {\n it('calls renderThumbnail with the new revision id and persists the bytes', async () => {\n const fs = new InMemoryWorkspaceFS();\n const store = new RevisionStore(fs);\n const fakeEngine = { _scene: '' };\n const adapter: EngineAdapter = {\n create: () => fakeEngine,\n reset: vi.fn(async () => {}),\n loadFromString: vi.fn(async () => {}),\n saveToString: vi.fn(async () => fakeEngine._scene)\n };\n const session = new EngineSession({ size: 1, adapter });\n const renderThumbnail = vi.fn(\n async () => new Uint8Array([0x52, 0x49, 0x46, 0x46])\n ); // \"RIFF\" — webp header\n const tool = createEditTool({\n store,\n session,\n engineLock,\n renderThumbnail,\n fs\n });\n const r = await tool.call({\n parent: null,\n code: \"engine._scene = 'SCENE_V1'\",\n title: 'T'\n });\n const rev = JSON.parse((r.content[0] as { text: string }).text).revision;\n expect(renderThumbnail).toHaveBeenCalledWith(\n expect.objectContaining({ revision: rev })\n );\n const shard = rev.slice(0, 2);\n const rest = rev.slice(2);\n expect(await fs.exists(`exports/${shard}/${rest}/preview.webp`)).toBe(true);\n });\n\n it('does not fail the edit when renderThumbnail throws', async () => {\n const fs = new InMemoryWorkspaceFS();\n const store = new RevisionStore(fs);\n const fakeEngine = { _scene: '' };\n const adapter: EngineAdapter = {\n create: () => fakeEngine,\n reset: vi.fn(async () => {}),\n loadFromString: vi.fn(async () => {}),\n saveToString: vi.fn(async () => fakeEngine._scene)\n };\n const session = new EngineSession({ size: 1, adapter });\n const renderThumbnail = vi.fn(async () => {\n throw new Error('thumbnail render boom');\n });\n const tool = createEditTool({\n store,\n session,\n engineLock,\n renderThumbnail,\n fs\n });\n const r = await tool.call({\n parent: null,\n code: \"engine._scene = 'S'\",\n title: 'T'\n });\n expect(r.isError).toBeFalsy();\n });\n});\n\ndescribe('edit tool — concurrency', () => {\n it('two sessions sharing one engine do not corrupt each other under the lock', async () => {\n const fs = new InMemoryWorkspaceFS();\n const store = new RevisionStore(fs);\n // ONE shared engine — exactly the production wiring, where\n // ensureHeadlessEngine is a process singleton handed to every session.\n const sharedEngine = { _scene: '' };\n const makeAdapter = (): EngineAdapter => ({\n create: () => sharedEngine,\n reset: async () => {\n sharedEngine._scene = '';\n },\n loadFromString: async (e, s) => {\n (e as typeof sharedEngine)._scene = s;\n },\n saveToString: async (e) => (e as typeof sharedEngine)._scene\n });\n // Each MCP session owns its EngineSession (per-session slot state) but they\n // share the engine. ONE lock is shared across sessions — the invariant the\n // bootstrap upholds by creating the Mutex once, outside createSession.\n const lock = new Mutex();\n const sessionA = new EngineSession({ size: 1, adapter: makeAdapter() });\n const sessionB = new EngineSession({ size: 1, adapter: makeAdapter() });\n const toolA = createEditTool({\n store,\n session: sessionA,\n engineLock: lock\n });\n const toolB = createEditTool({\n store,\n session: sessionB,\n engineLock: lock\n });\n\n // Each edit stamps the shared engine, then awaits — the exact window where,\n // WITHOUT the lock, the other session would overwrite the engine before\n // saveToString reads it back (so revA would persist SCENE_B). The shared\n // lock serializes the two, so each revision keeps its own scene.\n const code = (mark: string) =>\n `engine._scene = '${mark}'; await new Promise((r) => setTimeout(r, 10));`;\n const [resA, resB] = await Promise.all([\n toolA.call({ parent: null, code: code('SCENE_A'), title: 'A' }),\n toolB.call({ parent: null, code: code('SCENE_B'), title: 'B' })\n ]);\n const revA = (\n JSON.parse((resA.content[0] as { text: string }).text) as {\n revision: string;\n }\n ).revision;\n const revB = (\n JSON.parse((resB.content[0] as { text: string }).text) as {\n revision: string;\n }\n ).revision;\n expect((await store.read(revA)).scene).toBe('SCENE_A');\n expect((await store.read(revB)).scene).toBe('SCENE_B');\n });\n});\n"]}
|