@auto-engineer/component-implementor-react 1.98.0 → 1.99.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/.turbo/turbo-build.log +1 -1
- package/.turbo/turbo-test.log +6 -6
- package/.turbo/turbo-type-check.log +1 -1
- package/CHANGELOG.md +74 -0
- package/dist/src/commands/implement-component.d.ts +19 -0
- package/dist/src/commands/implement-component.d.ts.map +1 -1
- package/dist/src/commands/implement-component.js +109 -30
- package/dist/src/commands/implement-component.js.map +1 -1
- package/dist/src/commands/implement-component.test.js +259 -69
- package/dist/src/commands/implement-component.test.js.map +1 -1
- package/dist/src/extract-exports.d.ts +6 -0
- package/dist/src/extract-exports.d.ts.map +1 -0
- package/dist/src/extract-exports.js +46 -0
- package/dist/src/extract-exports.js.map +1 -0
- package/dist/src/generate-story-deterministic.d.ts +30 -0
- package/dist/src/generate-story-deterministic.d.ts.map +1 -0
- package/dist/src/generate-story-deterministic.js +229 -0
- package/dist/src/generate-story-deterministic.js.map +1 -0
- package/dist/src/index.d.ts +4 -0
- package/dist/src/index.d.ts.map +1 -1
- package/dist/src/index.js +3 -0
- package/dist/src/index.js.map +1 -1
- package/dist/src/pipeline/run-pipeline.d.ts +69 -0
- package/dist/src/pipeline/run-pipeline.d.ts.map +1 -0
- package/dist/src/pipeline/run-pipeline.js +78 -0
- package/dist/src/pipeline/run-pipeline.js.map +1 -0
- package/dist/src/pipeline/run-pipeline.test.d.ts +2 -0
- package/dist/src/pipeline/run-pipeline.test.d.ts.map +1 -0
- package/dist/src/pipeline/run-pipeline.test.js +247 -0
- package/dist/src/pipeline/run-pipeline.test.js.map +1 -0
- package/dist/src/pipeline/steps/generate-component.d.ts +4 -0
- package/dist/src/pipeline/steps/generate-component.d.ts.map +1 -0
- package/dist/src/pipeline/steps/generate-component.js +50 -0
- package/dist/src/pipeline/steps/generate-component.js.map +1 -0
- package/dist/src/pipeline/steps/generate-component.test.d.ts.map +1 -0
- package/dist/src/pipeline/steps/generate-component.test.js +106 -0
- package/dist/src/pipeline/steps/generate-component.test.js.map +1 -0
- package/dist/src/pipeline/steps/generate-story.d.ts +3 -0
- package/dist/src/pipeline/steps/generate-story.d.ts.map +1 -0
- package/dist/src/pipeline/steps/generate-story.js +14 -0
- package/dist/src/pipeline/steps/generate-story.js.map +1 -0
- package/dist/src/pipeline/steps/generate-story.test.d.ts.map +1 -0
- package/dist/src/pipeline/steps/generate-story.test.js +41 -0
- package/dist/src/pipeline/steps/generate-story.test.js.map +1 -0
- package/dist/src/pipeline/steps/generate-test.d.ts +4 -0
- package/dist/src/pipeline/steps/generate-test.d.ts.map +1 -0
- package/dist/src/pipeline/steps/generate-test.js +19 -0
- package/dist/src/pipeline/steps/generate-test.js.map +1 -0
- package/dist/src/pipeline/steps/generate-test.test.d.ts.map +1 -0
- package/dist/src/pipeline/steps/generate-test.test.js +60 -0
- package/dist/src/pipeline/steps/generate-test.test.js.map +1 -0
- package/dist/src/pipeline/steps/lint-fix-loop.d.ts +4 -0
- package/dist/src/pipeline/steps/lint-fix-loop.d.ts.map +1 -0
- package/dist/src/pipeline/steps/lint-fix-loop.js +45 -0
- package/dist/src/pipeline/steps/lint-fix-loop.js.map +1 -0
- package/dist/src/pipeline/steps/lint-fix-loop.test.d.ts +2 -0
- package/dist/src/pipeline/steps/lint-fix-loop.test.d.ts.map +1 -0
- package/dist/src/pipeline/steps/lint-fix-loop.test.js +119 -0
- package/dist/src/pipeline/steps/lint-fix-loop.test.js.map +1 -0
- package/dist/src/pipeline/steps/story-fix-loop.d.ts +4 -0
- package/dist/src/pipeline/steps/story-fix-loop.d.ts.map +1 -0
- package/dist/src/pipeline/steps/story-fix-loop.js +34 -0
- package/dist/src/pipeline/steps/story-fix-loop.js.map +1 -0
- package/dist/src/pipeline/steps/story-fix-loop.test.d.ts +2 -0
- package/dist/src/pipeline/steps/story-fix-loop.test.d.ts.map +1 -0
- package/dist/src/pipeline/steps/story-fix-loop.test.js +94 -0
- package/dist/src/pipeline/steps/story-fix-loop.test.js.map +1 -0
- package/dist/src/pipeline/steps/storybook-test.d.ts +3 -0
- package/dist/src/pipeline/steps/storybook-test.d.ts.map +1 -0
- package/dist/src/pipeline/steps/storybook-test.js +22 -0
- package/dist/src/pipeline/steps/storybook-test.js.map +1 -0
- package/dist/src/pipeline/steps/storybook-test.test.d.ts +2 -0
- package/dist/src/pipeline/steps/storybook-test.test.d.ts.map +1 -0
- package/dist/src/pipeline/steps/storybook-test.test.js +66 -0
- package/dist/src/pipeline/steps/storybook-test.test.js.map +1 -0
- package/dist/src/pipeline/steps/test-fix-loop.d.ts +4 -0
- package/dist/src/pipeline/steps/test-fix-loop.d.ts.map +1 -0
- package/dist/src/pipeline/steps/test-fix-loop.js +44 -0
- package/dist/src/pipeline/steps/test-fix-loop.js.map +1 -0
- package/dist/src/pipeline/steps/test-fix-loop.test.d.ts +2 -0
- package/dist/src/pipeline/steps/test-fix-loop.test.d.ts.map +1 -0
- package/dist/src/pipeline/steps/test-fix-loop.test.js +168 -0
- package/dist/src/pipeline/steps/test-fix-loop.test.js.map +1 -0
- package/dist/src/pipeline/steps/type-fix-loop.d.ts +4 -0
- package/dist/src/pipeline/steps/type-fix-loop.d.ts.map +1 -0
- package/dist/src/pipeline/steps/type-fix-loop.js +43 -0
- package/dist/src/pipeline/steps/type-fix-loop.js.map +1 -0
- package/dist/src/pipeline/steps/type-fix-loop.test.d.ts +2 -0
- package/dist/src/pipeline/steps/type-fix-loop.test.d.ts.map +1 -0
- package/dist/src/pipeline/steps/type-fix-loop.test.js +112 -0
- package/dist/src/pipeline/steps/type-fix-loop.test.js.map +1 -0
- package/dist/src/pipeline/steps/visual-test.d.ts +3 -0
- package/dist/src/pipeline/steps/visual-test.d.ts.map +1 -0
- package/dist/src/pipeline/steps/visual-test.js +4 -0
- package/dist/src/pipeline/steps/visual-test.js.map +1 -0
- package/dist/src/pipeline/steps/visual-test.test.d.ts +2 -0
- package/dist/src/pipeline/steps/visual-test.test.d.ts.map +1 -0
- package/dist/src/pipeline/steps/visual-test.test.js +9 -0
- package/dist/src/pipeline/steps/visual-test.test.js.map +1 -0
- package/dist/src/project-context.d.ts +10 -0
- package/dist/src/project-context.d.ts.map +1 -0
- package/dist/src/project-context.js +178 -0
- package/dist/src/project-context.js.map +1 -0
- package/dist/src/prompt.d.ts +39 -7
- package/dist/src/prompt.d.ts.map +1 -1
- package/dist/src/prompt.js +233 -23
- package/dist/src/prompt.js.map +1 -1
- package/dist/src/prompt.test.js +154 -9
- package/dist/src/prompt.test.js.map +1 -1
- package/dist/src/scaffold.d.ts +49 -0
- package/dist/src/scaffold.d.ts.map +1 -0
- package/dist/src/scaffold.js +208 -0
- package/dist/src/scaffold.js.map +1 -0
- package/dist/src/tools/lint-runner.d.ts +7 -0
- package/dist/src/tools/lint-runner.d.ts.map +1 -0
- package/dist/src/tools/lint-runner.js +48 -0
- package/dist/src/tools/lint-runner.js.map +1 -0
- package/dist/src/tools/lint-runner.test.d.ts +2 -0
- package/dist/src/tools/lint-runner.test.d.ts.map +1 -0
- package/dist/src/tools/lint-runner.test.js +90 -0
- package/dist/src/tools/lint-runner.test.js.map +1 -0
- package/dist/src/tools/storybook-runner.d.ts +6 -0
- package/dist/src/tools/storybook-runner.d.ts.map +1 -0
- package/dist/src/tools/storybook-runner.js +25 -0
- package/dist/src/tools/storybook-runner.js.map +1 -0
- package/dist/src/tools/storybook-runner.test.d.ts +2 -0
- package/dist/src/tools/storybook-runner.test.d.ts.map +1 -0
- package/dist/src/tools/storybook-runner.test.js +43 -0
- package/dist/src/tools/storybook-runner.test.js.map +1 -0
- package/dist/src/tools/test-runner.d.ts +9 -0
- package/dist/src/tools/test-runner.d.ts.map +1 -0
- package/dist/src/tools/test-runner.js +74 -0
- package/dist/src/tools/test-runner.js.map +1 -0
- package/dist/src/tools/test-runner.test.d.ts +2 -0
- package/dist/src/tools/test-runner.test.d.ts.map +1 -0
- package/dist/src/tools/test-runner.test.js +177 -0
- package/dist/src/tools/test-runner.test.js.map +1 -0
- package/dist/src/tools/type-checker.d.ts +6 -0
- package/dist/src/tools/type-checker.d.ts.map +1 -0
- package/dist/src/tools/type-checker.js +36 -0
- package/dist/src/tools/type-checker.js.map +1 -0
- package/dist/src/tools/type-checker.test.d.ts +2 -0
- package/dist/src/tools/type-checker.test.d.ts.map +1 -0
- package/dist/src/tools/type-checker.test.js +96 -0
- package/dist/src/tools/type-checker.test.js.map +1 -0
- package/dist/tsconfig.tsbuildinfo +1 -1
- package/inputs/model-a/spec-deltas.json +1460 -0
- package/inputs/model-b/spec-deltas.json +1424 -0
- package/inputs/model-c/spec-deltas.json +1432 -0
- package/inputs/model-d/spec-deltas.json +967 -0
- package/inputs/model-e/spec-deltas.json +2292 -0
- package/ketchup-plan.md +43 -8
- package/package.json +3 -3
- package/scoring-heuristic.md +138 -0
- package/scripts/improve.ts +23 -18
- package/src/commands/implement-component.test.ts +309 -76
- package/src/commands/implement-component.ts +155 -31
- package/src/extract-exports.ts +53 -0
- package/src/generate-story-deterministic.ts +267 -0
- package/src/index.ts +12 -0
- package/src/pipeline/run-pipeline.test.ts +292 -0
- package/src/pipeline/run-pipeline.ts +160 -0
- package/src/pipeline/steps/generate-component.test.ts +130 -0
- package/src/pipeline/steps/generate-component.ts +60 -0
- package/src/pipeline/steps/generate-story.test.ts +54 -0
- package/src/pipeline/steps/generate-story.ts +17 -0
- package/src/pipeline/steps/generate-test.test.ts +75 -0
- package/src/pipeline/steps/generate-test.ts +25 -0
- package/src/pipeline/steps/lint-fix-loop.test.ts +155 -0
- package/src/pipeline/steps/lint-fix-loop.ts +59 -0
- package/src/pipeline/steps/story-fix-loop.test.ts +123 -0
- package/src/pipeline/steps/story-fix-loop.ts +47 -0
- package/src/pipeline/steps/storybook-test.test.ts +82 -0
- package/src/pipeline/steps/storybook-test.ts +27 -0
- package/src/pipeline/steps/test-fix-loop.test.ts +201 -0
- package/src/pipeline/steps/test-fix-loop.ts +56 -0
- package/src/pipeline/steps/type-fix-loop.test.ts +145 -0
- package/src/pipeline/steps/type-fix-loop.ts +55 -0
- package/src/pipeline/steps/visual-test.test.ts +10 -0
- package/src/pipeline/steps/visual-test.ts +5 -0
- package/src/project-context.ts +205 -0
- package/src/prompt.test.ts +174 -8
- package/src/prompt.ts +301 -23
- package/src/scaffold.ts +281 -0
- package/src/tools/lint-runner.test.ts +112 -0
- package/src/tools/lint-runner.ts +52 -0
- package/src/tools/storybook-runner.test.ts +53 -0
- package/src/tools/storybook-runner.ts +29 -0
- package/src/tools/test-runner.test.ts +213 -0
- package/src/tools/test-runner.ts +84 -0
- package/src/tools/type-checker.test.ts +120 -0
- package/src/tools/type-checker.ts +42 -0
- package/vitest.config.ts +9 -1
- package/dist/src/generate-component.d.ts +0 -4
- package/dist/src/generate-component.d.ts.map +0 -1
- package/dist/src/generate-component.js +0 -14
- package/dist/src/generate-component.js.map +0 -1
- package/dist/src/generate-component.test.d.ts.map +0 -1
- package/dist/src/generate-component.test.js +0 -73
- package/dist/src/generate-component.test.js.map +0 -1
- package/dist/src/generate-story.d.ts +0 -4
- package/dist/src/generate-story.d.ts.map +0 -1
- package/dist/src/generate-story.js +0 -14
- package/dist/src/generate-story.js.map +0 -1
- package/dist/src/generate-story.test.d.ts.map +0 -1
- package/dist/src/generate-story.test.js +0 -58
- package/dist/src/generate-story.test.js.map +0 -1
- package/dist/src/generate-test.d.ts +0 -4
- package/dist/src/generate-test.d.ts.map +0 -1
- package/dist/src/generate-test.js +0 -14
- package/dist/src/generate-test.js.map +0 -1
- package/dist/src/generate-test.test.d.ts.map +0 -1
- package/dist/src/generate-test.test.js +0 -77
- package/dist/src/generate-test.test.js.map +0 -1
- package/dist/src/reconcile.d.ts +0 -8
- package/dist/src/reconcile.d.ts.map +0 -1
- package/dist/src/reconcile.js +0 -18
- package/dist/src/reconcile.js.map +0 -1
- package/dist/src/reconcile.test.d.ts +0 -2
- package/dist/src/reconcile.test.d.ts.map +0 -1
- package/dist/src/reconcile.test.js +0 -108
- package/dist/src/reconcile.test.js.map +0 -1
- package/src/generate-component.test.ts +0 -89
- package/src/generate-component.ts +0 -16
- package/src/generate-story.test.ts +0 -71
- package/src/generate-story.ts +0 -16
- package/src/generate-test.test.ts +0 -93
- package/src/generate-test.ts +0 -16
- package/src/reconcile.test.ts +0 -127
- package/src/reconcile.ts +0 -27
- /package/dist/src/{generate-component.test.d.ts → pipeline/steps/generate-component.test.d.ts} +0 -0
- /package/dist/src/{generate-story.test.d.ts → pipeline/steps/generate-story.test.d.ts} +0 -0
- /package/dist/src/{generate-test.test.d.ts → pipeline/steps/generate-test.test.d.ts} +0 -0
|
@@ -0,0 +1,168 @@
|
|
|
1
|
+
import { afterEach, describe, expect, it, vi } from 'vitest';
|
|
2
|
+
vi.mock('ai', () => ({
|
|
3
|
+
generateText: vi.fn(),
|
|
4
|
+
}));
|
|
5
|
+
vi.mock('node:fs/promises', () => ({
|
|
6
|
+
readFile: vi.fn(),
|
|
7
|
+
writeFile: vi.fn(),
|
|
8
|
+
}));
|
|
9
|
+
vi.mock('../../tools/test-runner', () => ({
|
|
10
|
+
runTests: vi.fn(),
|
|
11
|
+
}));
|
|
12
|
+
import { readFile, writeFile } from 'node:fs/promises';
|
|
13
|
+
import { generateText } from 'ai';
|
|
14
|
+
import { runTests } from '../../tools/test-runner.js';
|
|
15
|
+
import { testFixLoop } from './test-fix-loop.js';
|
|
16
|
+
afterEach(() => {
|
|
17
|
+
vi.clearAllMocks();
|
|
18
|
+
});
|
|
19
|
+
function makeCtx(overrides = {}) {
|
|
20
|
+
return {
|
|
21
|
+
componentName: 'MyButton',
|
|
22
|
+
componentPath: '/project/src/MyButton.tsx',
|
|
23
|
+
testPath: '/project/src/MyButton.test.tsx',
|
|
24
|
+
storyPath: '/project/src/MyButton.stories.tsx',
|
|
25
|
+
componentImportPath: '@/components/ui/MyButton',
|
|
26
|
+
targetDir: '/project',
|
|
27
|
+
specDeltas: { structure: [], rendering: [], interaction: [], styling: [] },
|
|
28
|
+
projectSection: '',
|
|
29
|
+
composes: [],
|
|
30
|
+
isModify: false,
|
|
31
|
+
llmCalls: 0,
|
|
32
|
+
fixIterations: 0,
|
|
33
|
+
componentCode: 'original component',
|
|
34
|
+
testCode: 'original test',
|
|
35
|
+
...overrides,
|
|
36
|
+
};
|
|
37
|
+
}
|
|
38
|
+
describe('testFixLoop', () => {
|
|
39
|
+
it('returns success immediately when all tests pass', async () => {
|
|
40
|
+
vi.mocked(runTests).mockReturnValue({
|
|
41
|
+
passed: true,
|
|
42
|
+
numPassed: 3,
|
|
43
|
+
numFailed: 0,
|
|
44
|
+
failures: [],
|
|
45
|
+
output: '',
|
|
46
|
+
});
|
|
47
|
+
const result = await testFixLoop('mock-model', makeCtx(), 3);
|
|
48
|
+
expect(result).toEqual({ success: true });
|
|
49
|
+
expect(generateText).not.toHaveBeenCalled();
|
|
50
|
+
});
|
|
51
|
+
it('calls LLM to fix failures and writes fixed files', async () => {
|
|
52
|
+
vi.mocked(runTests)
|
|
53
|
+
.mockReturnValueOnce({
|
|
54
|
+
passed: false,
|
|
55
|
+
numPassed: 1,
|
|
56
|
+
numFailed: 1,
|
|
57
|
+
failures: ['renders button: element not found'],
|
|
58
|
+
output: 'FAIL',
|
|
59
|
+
})
|
|
60
|
+
.mockReturnValueOnce({
|
|
61
|
+
passed: true,
|
|
62
|
+
numPassed: 2,
|
|
63
|
+
numFailed: 0,
|
|
64
|
+
failures: [],
|
|
65
|
+
output: 'PASS',
|
|
66
|
+
});
|
|
67
|
+
vi.mocked(generateText).mockResolvedValue({
|
|
68
|
+
text: '```tsx\nfixed component\n```\n```tsx\nfixed test\n```',
|
|
69
|
+
});
|
|
70
|
+
vi.mocked(writeFile).mockResolvedValue(undefined);
|
|
71
|
+
const result = await testFixLoop('mock-model', makeCtx(), 3);
|
|
72
|
+
expect(result).toEqual({ success: true });
|
|
73
|
+
expect(generateText).toHaveBeenCalledTimes(1);
|
|
74
|
+
expect(writeFile).toHaveBeenCalledWith('/project/src/MyButton.tsx', 'fixed component', 'utf-8');
|
|
75
|
+
expect(writeFile).toHaveBeenCalledWith('/project/src/MyButton.test.tsx', 'fixed test', 'utf-8');
|
|
76
|
+
});
|
|
77
|
+
it('returns failure after exhausting max iterations', async () => {
|
|
78
|
+
vi.mocked(runTests).mockReturnValue({
|
|
79
|
+
passed: false,
|
|
80
|
+
numPassed: 0,
|
|
81
|
+
numFailed: 1,
|
|
82
|
+
failures: ['persistent failure'],
|
|
83
|
+
output: 'FAIL',
|
|
84
|
+
});
|
|
85
|
+
vi.mocked(generateText).mockResolvedValue({
|
|
86
|
+
text: '```tsx\nstill broken\n```\n```tsx\nstill broken test\n```',
|
|
87
|
+
});
|
|
88
|
+
vi.mocked(writeFile).mockResolvedValue(undefined);
|
|
89
|
+
vi.mocked(readFile).mockResolvedValue('still broken');
|
|
90
|
+
const result = await testFixLoop('mock-model', makeCtx(), 1);
|
|
91
|
+
expect(result).toEqual({
|
|
92
|
+
success: false,
|
|
93
|
+
error: expect.stringContaining('Test failures remain after 1 iterations'),
|
|
94
|
+
});
|
|
95
|
+
});
|
|
96
|
+
it('handles single code block response by updating only component', async () => {
|
|
97
|
+
vi.mocked(runTests)
|
|
98
|
+
.mockReturnValueOnce({
|
|
99
|
+
passed: false,
|
|
100
|
+
numPassed: 0,
|
|
101
|
+
numFailed: 1,
|
|
102
|
+
failures: ['failure'],
|
|
103
|
+
output: 'FAIL',
|
|
104
|
+
})
|
|
105
|
+
.mockReturnValueOnce({
|
|
106
|
+
passed: true,
|
|
107
|
+
numPassed: 1,
|
|
108
|
+
numFailed: 0,
|
|
109
|
+
failures: [],
|
|
110
|
+
output: 'PASS',
|
|
111
|
+
});
|
|
112
|
+
vi.mocked(generateText).mockResolvedValue({
|
|
113
|
+
text: '```tsx\nfixed component only\n```',
|
|
114
|
+
});
|
|
115
|
+
vi.mocked(writeFile).mockResolvedValue(undefined);
|
|
116
|
+
const result = await testFixLoop('mock-model', makeCtx(), 3);
|
|
117
|
+
expect(result).toEqual({ success: true });
|
|
118
|
+
expect(writeFile).toHaveBeenCalledWith('/project/src/MyButton.tsx', 'fixed component only', 'utf-8');
|
|
119
|
+
expect(writeFile).toHaveBeenCalledTimes(1);
|
|
120
|
+
});
|
|
121
|
+
it('returns success when final check passes after last iteration fix', async () => {
|
|
122
|
+
vi.mocked(runTests)
|
|
123
|
+
.mockReturnValueOnce({
|
|
124
|
+
passed: false,
|
|
125
|
+
numPassed: 0,
|
|
126
|
+
numFailed: 1,
|
|
127
|
+
failures: ['failure'],
|
|
128
|
+
output: 'FAIL',
|
|
129
|
+
})
|
|
130
|
+
.mockReturnValueOnce({
|
|
131
|
+
passed: true,
|
|
132
|
+
numPassed: 1,
|
|
133
|
+
numFailed: 0,
|
|
134
|
+
failures: [],
|
|
135
|
+
output: 'PASS',
|
|
136
|
+
});
|
|
137
|
+
vi.mocked(generateText).mockResolvedValue({
|
|
138
|
+
text: '```tsx\nfixed\n```\n```tsx\nfixed test\n```',
|
|
139
|
+
});
|
|
140
|
+
vi.mocked(writeFile).mockResolvedValue(undefined);
|
|
141
|
+
const result = await testFixLoop('mock-model', makeCtx(), 1);
|
|
142
|
+
expect(result).toEqual({ success: true });
|
|
143
|
+
});
|
|
144
|
+
it('uses empty string fallback when componentCode and testCode are undefined', async () => {
|
|
145
|
+
vi.mocked(runTests)
|
|
146
|
+
.mockReturnValueOnce({
|
|
147
|
+
passed: false,
|
|
148
|
+
numPassed: 0,
|
|
149
|
+
numFailed: 1,
|
|
150
|
+
failures: ['failure'],
|
|
151
|
+
output: 'FAIL',
|
|
152
|
+
})
|
|
153
|
+
.mockReturnValueOnce({
|
|
154
|
+
passed: true,
|
|
155
|
+
numPassed: 1,
|
|
156
|
+
numFailed: 0,
|
|
157
|
+
failures: [],
|
|
158
|
+
output: 'PASS',
|
|
159
|
+
});
|
|
160
|
+
vi.mocked(generateText).mockResolvedValue({
|
|
161
|
+
text: '```tsx\nfixed\n```\n```tsx\nfixed test\n```',
|
|
162
|
+
});
|
|
163
|
+
vi.mocked(writeFile).mockResolvedValue(undefined);
|
|
164
|
+
const result = await testFixLoop('mock-model', makeCtx({ componentCode: undefined, testCode: undefined }), 1);
|
|
165
|
+
expect(result).toEqual({ success: true });
|
|
166
|
+
});
|
|
167
|
+
});
|
|
168
|
+
//# sourceMappingURL=test-fix-loop.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"test-fix-loop.test.js","sourceRoot":"","sources":["../../../../src/pipeline/steps/test-fix-loop.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAE7D,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,GAAG,EAAE,CAAC,CAAC;IACnB,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE;CACtB,CAAC,CAAC,CAAC;AAEJ,EAAE,CAAC,IAAI,CAAC,kBAAkB,EAAE,GAAG,EAAE,CAAC,CAAC;IACjC,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE;IACjB,SAAS,EAAE,EAAE,CAAC,EAAE,EAAE;CACnB,CAAC,CAAC,CAAC;AAEJ,EAAE,CAAC,IAAI,CAAC,yBAAyB,EAAE,GAAG,EAAE,CAAC,CAAC;IACxC,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE;CAClB,CAAC,CAAC,CAAC;AAEJ,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AACvD,OAAO,EAAE,YAAY,EAAE,MAAM,IAAI,CAAC;AAClC,OAAO,EAAE,QAAQ,EAAE,MAAM,yBAAyB,CAAC;AAEnD,OAAO,EAAE,WAAW,EAAE,MAAM,iBAAiB,CAAC;AAE9C,SAAS,CAAC,GAAG,EAAE;IACb,EAAE,CAAC,aAAa,EAAE,CAAC;AACrB,CAAC,CAAC,CAAC;AAEH,SAAS,OAAO,CAAC,YAAsC,EAAE;IACvD,OAAO;QACL,aAAa,EAAE,UAAU;QACzB,aAAa,EAAE,2BAA2B;QAC1C,QAAQ,EAAE,gCAAgC;QAC1C,SAAS,EAAE,mCAAmC;QAC9C,mBAAmB,EAAE,0BAA0B;QAC/C,SAAS,EAAE,UAAU;QACrB,UAAU,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,WAAW,EAAE,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE;QAC1E,cAAc,EAAE,EAAE;QAClB,QAAQ,EAAE,EAAE;QACZ,QAAQ,EAAE,KAAK;QACf,QAAQ,EAAE,CAAC;QACX,aAAa,EAAE,CAAC;QAChB,aAAa,EAAE,oBAAoB;QACnC,QAAQ,EAAE,eAAe;QACzB,GAAG,SAAS;KACb,CAAC;AACJ,CAAC;AAED,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;IAC3B,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;QAC/D,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,eAAe,CAAC;YAClC,MAAM,EAAE,IAAI;YACZ,SAAS,EAAE,CAAC;YACZ,SAAS,EAAE,CAAC;YACZ,QAAQ,EAAE,EAAE;YACZ,MAAM,EAAE,EAAE;SACX,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,YAAqB,EAAE,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;QAEtE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;QAC1C,MAAM,CAAC,YAAY,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;QAChE,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC;aAChB,mBAAmB,CAAC;YACnB,MAAM,EAAE,KAAK;YACb,SAAS,EAAE,CAAC;YACZ,SAAS,EAAE,CAAC;YACZ,QAAQ,EAAE,CAAC,mCAAmC,CAAC;YAC/C,MAAM,EAAE,MAAM;SACf,CAAC;aACD,mBAAmB,CAAC;YACnB,MAAM,EAAE,IAAI;YACZ,SAAS,EAAE,CAAC;YACZ,SAAS,EAAE,CAAC;YACZ,QAAQ,EAAE,EAAE;YACZ,MAAM,EAAE,MAAM;SACf,CAAC,CAAC;QAEL,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,iBAAiB,CAAC;YACxC,IAAI,EAAE,uDAAuD;SAClB,CAAC,CAAC;QAC/C,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;QAElD,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,YAAqB,EAAE,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;QAEtE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;QAC1C,MAAM,CAAC,YAAY,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC9C,MAAM,CAAC,SAAS,CAAC,CAAC,oBAAoB,CAAC,2BAA2B,EAAE,iBAAiB,EAAE,OAAO,CAAC,CAAC;QAChG,MAAM,CAAC,SAAS,CAAC,CAAC,oBAAoB,CAAC,gCAAgC,EAAE,YAAY,EAAE,OAAO,CAAC,CAAC;IAClG,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;QAC/D,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,eAAe,CAAC;YAClC,MAAM,EAAE,KAAK;YACb,SAAS,EAAE,CAAC;YACZ,SAAS,EAAE,CAAC;YACZ,QAAQ,EAAE,CAAC,oBAAoB,CAAC;YAChC,MAAM,EAAE,MAAM;SACf,CAAC,CAAC;QAEH,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,iBAAiB,CAAC;YACxC,IAAI,EAAE,2DAA2D;SACtB,CAAC,CAAC;QAC/C,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;QAClD,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,iBAAiB,CAAC,cAAuB,CAAC,CAAC;QAE/D,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,YAAqB,EAAE,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;QAEtE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;YACrB,OAAO,EAAE,KAAK;YACd,KAAK,EAAE,MAAM,CAAC,gBAAgB,CAAC,yCAAyC,CAAC;SAC1E,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;QAC7E,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC;aAChB,mBAAmB,CAAC;YACnB,MAAM,EAAE,KAAK;YACb,SAAS,EAAE,CAAC;YACZ,SAAS,EAAE,CAAC;YACZ,QAAQ,EAAE,CAAC,SAAS,CAAC;YACrB,MAAM,EAAE,MAAM;SACf,CAAC;aACD,mBAAmB,CAAC;YACnB,MAAM,EAAE,IAAI;YACZ,SAAS,EAAE,CAAC;YACZ,SAAS,EAAE,CAAC;YACZ,QAAQ,EAAE,EAAE;YACZ,MAAM,EAAE,MAAM;SACf,CAAC,CAAC;QAEL,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,iBAAiB,CAAC;YACxC,IAAI,EAAE,mCAAmC;SACE,CAAC,CAAC;QAC/C,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;QAElD,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,YAAqB,EAAE,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;QAEtE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;QAC1C,MAAM,CAAC,SAAS,CAAC,CAAC,oBAAoB,CAAC,2BAA2B,EAAE,sBAAsB,EAAE,OAAO,CAAC,CAAC;QACrG,MAAM,CAAC,SAAS,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IAC7C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kEAAkE,EAAE,KAAK,IAAI,EAAE;QAChF,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC;aAChB,mBAAmB,CAAC;YACnB,MAAM,EAAE,KAAK;YACb,SAAS,EAAE,CAAC;YACZ,SAAS,EAAE,CAAC;YACZ,QAAQ,EAAE,CAAC,SAAS,CAAC;YACrB,MAAM,EAAE,MAAM;SACf,CAAC;aACD,mBAAmB,CAAC;YACnB,MAAM,EAAE,IAAI;YACZ,SAAS,EAAE,CAAC;YACZ,SAAS,EAAE,CAAC;YACZ,QAAQ,EAAE,EAAE;YACZ,MAAM,EAAE,MAAM;SACf,CAAC,CAAC;QAEL,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,iBAAiB,CAAC;YACxC,IAAI,EAAE,6CAA6C;SACR,CAAC,CAAC;QAC/C,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;QAElD,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,YAAqB,EAAE,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;QAEtE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;IAC5C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0EAA0E,EAAE,KAAK,IAAI,EAAE;QACxF,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC;aAChB,mBAAmB,CAAC;YACnB,MAAM,EAAE,KAAK;YACb,SAAS,EAAE,CAAC;YACZ,SAAS,EAAE,CAAC;YACZ,QAAQ,EAAE,CAAC,SAAS,CAAC;YACrB,MAAM,EAAE,MAAM;SACf,CAAC;aACD,mBAAmB,CAAC;YACnB,MAAM,EAAE,IAAI;YACZ,SAAS,EAAE,CAAC;YACZ,SAAS,EAAE,CAAC;YACZ,QAAQ,EAAE,EAAE;YACZ,MAAM,EAAE,MAAM;SACf,CAAC,CAAC;QAEL,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,iBAAiB,CAAC;YACxC,IAAI,EAAE,6CAA6C;SACR,CAAC,CAAC;QAC/C,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;QAElD,MAAM,MAAM,GAAG,MAAM,WAAW,CAC9B,YAAqB,EACrB,OAAO,CAAC,EAAE,aAAa,EAAE,SAAS,EAAE,QAAQ,EAAE,SAAS,EAAE,CAAC,EAC1D,CAAC,CACF,CAAC;QAEF,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;IAC5C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1,4 @@
|
|
|
1
|
+
import type { LanguageModel } from 'ai';
|
|
2
|
+
import type { PipelineContext, StepResult } from '../run-pipeline';
|
|
3
|
+
export declare function typeFixLoop(model: LanguageModel, ctx: PipelineContext, maxIterations: number): Promise<StepResult>;
|
|
4
|
+
//# sourceMappingURL=type-fix-loop.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"type-fix-loop.d.ts","sourceRoot":"","sources":["../../../../src/pipeline/steps/type-fix-loop.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,IAAI,CAAC;AAKxC,OAAO,KAAK,EAAE,eAAe,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAEnE,wBAAsB,WAAW,CAC/B,KAAK,EAAE,aAAa,EACpB,GAAG,EAAE,eAAe,EACpB,aAAa,EAAE,MAAM,GACpB,OAAO,CAAC,UAAU,CAAC,CA0CrB"}
|
|
@@ -0,0 +1,43 @@
|
|
|
1
|
+
import { readFile, writeFile } from 'node:fs/promises';
|
|
2
|
+
import { generateText } from 'ai';
|
|
3
|
+
import { extractCodeBlocks } from '../../extract-code-block.js';
|
|
4
|
+
import { buildTypeFixPrompt } from '../../prompt.js';
|
|
5
|
+
import { runTypeCheck } from '../../tools/type-checker.js';
|
|
6
|
+
export async function typeFixLoop(model, ctx, maxIterations) {
|
|
7
|
+
for (let i = 0; i < maxIterations; i++) {
|
|
8
|
+
const result = runTypeCheck(ctx.targetDir, [ctx.componentPath, ctx.testPath]);
|
|
9
|
+
if (result.passed) {
|
|
10
|
+
return { success: true };
|
|
11
|
+
}
|
|
12
|
+
const { system, prompt } = buildTypeFixPrompt({
|
|
13
|
+
componentCode: ctx.componentCode ?? '',
|
|
14
|
+
testCode: ctx.testCode ?? '',
|
|
15
|
+
errors: result.errors,
|
|
16
|
+
});
|
|
17
|
+
const { text } = await generateText({ model, system, prompt });
|
|
18
|
+
ctx.llmCalls++;
|
|
19
|
+
ctx.fixIterations++;
|
|
20
|
+
const blocks = extractCodeBlocks(text);
|
|
21
|
+
if (blocks.length >= 2) {
|
|
22
|
+
ctx.componentCode = blocks[0];
|
|
23
|
+
ctx.testCode = blocks[1];
|
|
24
|
+
await writeFile(ctx.componentPath, blocks[0], 'utf-8');
|
|
25
|
+
await writeFile(ctx.testPath, blocks[1], 'utf-8');
|
|
26
|
+
}
|
|
27
|
+
else if (blocks.length === 1) {
|
|
28
|
+
ctx.componentCode = blocks[0];
|
|
29
|
+
await writeFile(ctx.componentPath, blocks[0], 'utf-8');
|
|
30
|
+
}
|
|
31
|
+
}
|
|
32
|
+
const finalResult = runTypeCheck(ctx.targetDir, [ctx.componentPath, ctx.testPath]);
|
|
33
|
+
if (finalResult.passed) {
|
|
34
|
+
return { success: true };
|
|
35
|
+
}
|
|
36
|
+
ctx.componentCode = await readFile(ctx.componentPath, 'utf-8');
|
|
37
|
+
ctx.testCode = await readFile(ctx.testPath, 'utf-8');
|
|
38
|
+
return {
|
|
39
|
+
success: false,
|
|
40
|
+
error: `Type errors remain after ${maxIterations} iterations: ${finalResult.errors.join('; ')}`,
|
|
41
|
+
};
|
|
42
|
+
}
|
|
43
|
+
//# sourceMappingURL=type-fix-loop.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"type-fix-loop.js","sourceRoot":"","sources":["../../../../src/pipeline/steps/type-fix-loop.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AAEvD,OAAO,EAAE,YAAY,EAAE,MAAM,IAAI,CAAC;AAClC,OAAO,EAAE,iBAAiB,EAAE,MAAM,0BAA0B,CAAC;AAC7D,OAAO,EAAE,kBAAkB,EAAE,MAAM,cAAc,CAAC;AAClD,OAAO,EAAE,YAAY,EAAE,MAAM,0BAA0B,CAAC;AAGxD,MAAM,CAAC,KAAK,UAAU,WAAW,CAC/B,KAAoB,EACpB,GAAoB,EACpB,aAAqB;IAErB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,aAAa,EAAE,CAAC,EAAE,EAAE,CAAC;QACvC,MAAM,MAAM,GAAG,YAAY,CAAC,GAAG,CAAC,SAAS,EAAE,CAAC,GAAG,CAAC,aAAa,EAAE,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC;QAE9E,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;YAClB,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;QAC3B,CAAC;QAED,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,kBAAkB,CAAC;YAC5C,aAAa,EAAE,GAAG,CAAC,aAAa,IAAI,EAAE;YACtC,QAAQ,EAAE,GAAG,CAAC,QAAQ,IAAI,EAAE;YAC5B,MAAM,EAAE,MAAM,CAAC,MAAM;SACtB,CAAC,CAAC;QAEH,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,YAAY,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;QAC/D,GAAG,CAAC,QAAQ,EAAE,CAAC;QACf,GAAG,CAAC,aAAa,EAAE,CAAC;QAEpB,MAAM,MAAM,GAAG,iBAAiB,CAAC,IAAI,CAAC,CAAC;QACvC,IAAI,MAAM,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC;YACvB,GAAG,CAAC,aAAa,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC;YAC9B,GAAG,CAAC,QAAQ,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC;YACzB,MAAM,SAAS,CAAC,GAAG,CAAC,aAAa,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC;YACvD,MAAM,SAAS,CAAC,GAAG,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC;QACpD,CAAC;aAAM,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC/B,GAAG,CAAC,aAAa,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC;YAC9B,MAAM,SAAS,CAAC,GAAG,CAAC,aAAa,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC;QACzD,CAAC;IACH,CAAC;IAED,MAAM,WAAW,GAAG,YAAY,CAAC,GAAG,CAAC,SAAS,EAAE,CAAC,GAAG,CAAC,aAAa,EAAE,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC;IACnF,IAAI,WAAW,CAAC,MAAM,EAAE,CAAC;QACvB,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;IAC3B,CAAC;IAED,GAAG,CAAC,aAAa,GAAG,MAAM,QAAQ,CAAC,GAAG,CAAC,aAAa,EAAE,OAAO,CAAC,CAAC;IAC/D,GAAG,CAAC,QAAQ,GAAG,MAAM,QAAQ,CAAC,GAAG,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;IAErD,OAAO;QACL,OAAO,EAAE,KAAK;QACd,KAAK,EAAE,4BAA4B,aAAa,gBAAgB,WAAW,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE;KAChG,CAAC;AACJ,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"type-fix-loop.test.d.ts","sourceRoot":"","sources":["../../../../src/pipeline/steps/type-fix-loop.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,112 @@
|
|
|
1
|
+
import { afterEach, describe, expect, it, vi } from 'vitest';
|
|
2
|
+
vi.mock('ai', () => ({
|
|
3
|
+
generateText: vi.fn(),
|
|
4
|
+
}));
|
|
5
|
+
vi.mock('node:fs/promises', () => ({
|
|
6
|
+
readFile: vi.fn(),
|
|
7
|
+
writeFile: vi.fn(),
|
|
8
|
+
}));
|
|
9
|
+
vi.mock('../../tools/type-checker', () => ({
|
|
10
|
+
runTypeCheck: vi.fn(),
|
|
11
|
+
}));
|
|
12
|
+
import { readFile, writeFile } from 'node:fs/promises';
|
|
13
|
+
import { generateText } from 'ai';
|
|
14
|
+
import { runTypeCheck } from '../../tools/type-checker.js';
|
|
15
|
+
import { typeFixLoop } from './type-fix-loop.js';
|
|
16
|
+
afterEach(() => {
|
|
17
|
+
vi.clearAllMocks();
|
|
18
|
+
});
|
|
19
|
+
function makeCtx(overrides = {}) {
|
|
20
|
+
return {
|
|
21
|
+
componentName: 'MyButton',
|
|
22
|
+
componentPath: '/project/src/MyButton.tsx',
|
|
23
|
+
testPath: '/project/src/MyButton.test.tsx',
|
|
24
|
+
storyPath: '/project/src/MyButton.stories.tsx',
|
|
25
|
+
componentImportPath: '@/components/ui/MyButton',
|
|
26
|
+
targetDir: '/project',
|
|
27
|
+
specDeltas: { structure: [], rendering: [], interaction: [], styling: [] },
|
|
28
|
+
projectSection: '',
|
|
29
|
+
composes: [],
|
|
30
|
+
isModify: false,
|
|
31
|
+
llmCalls: 0,
|
|
32
|
+
fixIterations: 0,
|
|
33
|
+
componentCode: 'original component',
|
|
34
|
+
testCode: 'original test',
|
|
35
|
+
...overrides,
|
|
36
|
+
};
|
|
37
|
+
}
|
|
38
|
+
describe('typeFixLoop', () => {
|
|
39
|
+
it('returns success immediately when type check passes', async () => {
|
|
40
|
+
vi.mocked(runTypeCheck).mockReturnValue({ passed: true, errors: [] });
|
|
41
|
+
const result = await typeFixLoop('mock-model', makeCtx(), 3);
|
|
42
|
+
expect(result).toEqual({ success: true });
|
|
43
|
+
expect(generateText).not.toHaveBeenCalled();
|
|
44
|
+
});
|
|
45
|
+
it('calls LLM to fix type errors and writes fixed files', async () => {
|
|
46
|
+
vi.mocked(runTypeCheck)
|
|
47
|
+
.mockReturnValueOnce({ passed: false, errors: ['error TS2307: Module not found'] })
|
|
48
|
+
.mockReturnValueOnce({ passed: true, errors: [] });
|
|
49
|
+
vi.mocked(generateText).mockResolvedValue({
|
|
50
|
+
text: '```tsx\nfixed component\n```\n```tsx\nfixed test\n```',
|
|
51
|
+
});
|
|
52
|
+
vi.mocked(writeFile).mockResolvedValue(undefined);
|
|
53
|
+
const result = await typeFixLoop('mock-model', makeCtx(), 3);
|
|
54
|
+
expect(result).toEqual({ success: true });
|
|
55
|
+
expect(generateText).toHaveBeenCalledTimes(1);
|
|
56
|
+
expect(writeFile).toHaveBeenCalledWith('/project/src/MyButton.tsx', 'fixed component', 'utf-8');
|
|
57
|
+
expect(writeFile).toHaveBeenCalledWith('/project/src/MyButton.test.tsx', 'fixed test', 'utf-8');
|
|
58
|
+
});
|
|
59
|
+
it('returns failure after exhausting max iterations', async () => {
|
|
60
|
+
vi.mocked(runTypeCheck).mockReturnValue({
|
|
61
|
+
passed: false,
|
|
62
|
+
errors: ['persistent error'],
|
|
63
|
+
});
|
|
64
|
+
vi.mocked(generateText).mockResolvedValue({
|
|
65
|
+
text: '```tsx\nstill broken\n```\n```tsx\nstill broken test\n```',
|
|
66
|
+
});
|
|
67
|
+
vi.mocked(writeFile).mockResolvedValue(undefined);
|
|
68
|
+
vi.mocked(readFile).mockResolvedValue('still broken');
|
|
69
|
+
const result = await typeFixLoop('mock-model', makeCtx(), 2);
|
|
70
|
+
expect(result).toEqual({
|
|
71
|
+
success: false,
|
|
72
|
+
error: expect.stringContaining('Type errors remain after 2 iterations'),
|
|
73
|
+
});
|
|
74
|
+
expect(generateText).toHaveBeenCalledTimes(2);
|
|
75
|
+
});
|
|
76
|
+
it('handles single code block response by updating only component', async () => {
|
|
77
|
+
vi.mocked(runTypeCheck)
|
|
78
|
+
.mockReturnValueOnce({ passed: false, errors: ['error in component'] })
|
|
79
|
+
.mockReturnValueOnce({ passed: true, errors: [] });
|
|
80
|
+
vi.mocked(generateText).mockResolvedValue({
|
|
81
|
+
text: '```tsx\nfixed component only\n```',
|
|
82
|
+
});
|
|
83
|
+
vi.mocked(writeFile).mockResolvedValue(undefined);
|
|
84
|
+
const result = await typeFixLoop('mock-model', makeCtx(), 3);
|
|
85
|
+
expect(result).toEqual({ success: true });
|
|
86
|
+
expect(writeFile).toHaveBeenCalledWith('/project/src/MyButton.tsx', 'fixed component only', 'utf-8');
|
|
87
|
+
expect(writeFile).toHaveBeenCalledTimes(1);
|
|
88
|
+
});
|
|
89
|
+
it('returns success when final check passes after last iteration fix', async () => {
|
|
90
|
+
vi.mocked(runTypeCheck)
|
|
91
|
+
.mockReturnValueOnce({ passed: false, errors: ['error'] })
|
|
92
|
+
.mockReturnValueOnce({ passed: true, errors: [] });
|
|
93
|
+
vi.mocked(generateText).mockResolvedValue({
|
|
94
|
+
text: '```tsx\nfixed\n```\n```tsx\nfixed test\n```',
|
|
95
|
+
});
|
|
96
|
+
vi.mocked(writeFile).mockResolvedValue(undefined);
|
|
97
|
+
const result = await typeFixLoop('mock-model', makeCtx(), 1);
|
|
98
|
+
expect(result).toEqual({ success: true });
|
|
99
|
+
});
|
|
100
|
+
it('uses empty string fallback when componentCode and testCode are undefined', async () => {
|
|
101
|
+
vi.mocked(runTypeCheck)
|
|
102
|
+
.mockReturnValueOnce({ passed: false, errors: ['error'] })
|
|
103
|
+
.mockReturnValueOnce({ passed: true, errors: [] });
|
|
104
|
+
vi.mocked(generateText).mockResolvedValue({
|
|
105
|
+
text: '```tsx\nfixed\n```\n```tsx\nfixed test\n```',
|
|
106
|
+
});
|
|
107
|
+
vi.mocked(writeFile).mockResolvedValue(undefined);
|
|
108
|
+
const result = await typeFixLoop('mock-model', makeCtx({ componentCode: undefined, testCode: undefined }), 1);
|
|
109
|
+
expect(result).toEqual({ success: true });
|
|
110
|
+
});
|
|
111
|
+
});
|
|
112
|
+
//# sourceMappingURL=type-fix-loop.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"type-fix-loop.test.js","sourceRoot":"","sources":["../../../../src/pipeline/steps/type-fix-loop.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAE7D,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,GAAG,EAAE,CAAC,CAAC;IACnB,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE;CACtB,CAAC,CAAC,CAAC;AAEJ,EAAE,CAAC,IAAI,CAAC,kBAAkB,EAAE,GAAG,EAAE,CAAC,CAAC;IACjC,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE;IACjB,SAAS,EAAE,EAAE,CAAC,EAAE,EAAE;CACnB,CAAC,CAAC,CAAC;AAEJ,EAAE,CAAC,IAAI,CAAC,0BAA0B,EAAE,GAAG,EAAE,CAAC,CAAC;IACzC,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE;CACtB,CAAC,CAAC,CAAC;AAEJ,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AACvD,OAAO,EAAE,YAAY,EAAE,MAAM,IAAI,CAAC;AAClC,OAAO,EAAE,YAAY,EAAE,MAAM,0BAA0B,CAAC;AAExD,OAAO,EAAE,WAAW,EAAE,MAAM,iBAAiB,CAAC;AAE9C,SAAS,CAAC,GAAG,EAAE;IACb,EAAE,CAAC,aAAa,EAAE,CAAC;AACrB,CAAC,CAAC,CAAC;AAEH,SAAS,OAAO,CAAC,YAAsC,EAAE;IACvD,OAAO;QACL,aAAa,EAAE,UAAU;QACzB,aAAa,EAAE,2BAA2B;QAC1C,QAAQ,EAAE,gCAAgC;QAC1C,SAAS,EAAE,mCAAmC;QAC9C,mBAAmB,EAAE,0BAA0B;QAC/C,SAAS,EAAE,UAAU;QACrB,UAAU,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,WAAW,EAAE,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE;QAC1E,cAAc,EAAE,EAAE;QAClB,QAAQ,EAAE,EAAE;QACZ,QAAQ,EAAE,KAAK;QACf,QAAQ,EAAE,CAAC;QACX,aAAa,EAAE,CAAC;QAChB,aAAa,EAAE,oBAAoB;QACnC,QAAQ,EAAE,eAAe;QACzB,GAAG,SAAS;KACb,CAAC;AACJ,CAAC;AAED,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;IAC3B,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;QAClE,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,eAAe,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CAAC;QAEtE,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,YAAqB,EAAE,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;QAEtE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;QAC1C,MAAM,CAAC,YAAY,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;QACnE,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC;aACpB,mBAAmB,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,gCAAgC,CAAC,EAAE,CAAC;aAClF,mBAAmB,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CAAC;QAErD,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,iBAAiB,CAAC;YACxC,IAAI,EAAE,uDAAuD;SAClB,CAAC,CAAC;QAC/C,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;QAElD,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,YAAqB,EAAE,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;QAEtE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;QAC1C,MAAM,CAAC,YAAY,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC9C,MAAM,CAAC,SAAS,CAAC,CAAC,oBAAoB,CAAC,2BAA2B,EAAE,iBAAiB,EAAE,OAAO,CAAC,CAAC;QAChG,MAAM,CAAC,SAAS,CAAC,CAAC,oBAAoB,CAAC,gCAAgC,EAAE,YAAY,EAAE,OAAO,CAAC,CAAC;IAClG,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;QAC/D,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,eAAe,CAAC;YACtC,MAAM,EAAE,KAAK;YACb,MAAM,EAAE,CAAC,kBAAkB,CAAC;SAC7B,CAAC,CAAC;QAEH,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,iBAAiB,CAAC;YACxC,IAAI,EAAE,2DAA2D;SACtB,CAAC,CAAC;QAC/C,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;QAClD,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,iBAAiB,CAAC,cAAuB,CAAC,CAAC;QAE/D,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,YAAqB,EAAE,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;QAEtE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;YACrB,OAAO,EAAE,KAAK;YACd,KAAK,EAAE,MAAM,CAAC,gBAAgB,CAAC,uCAAuC,CAAC;SACxE,CAAC,CAAC;QACH,MAAM,CAAC,YAAY,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;QAC7E,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC;aACpB,mBAAmB,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,oBAAoB,CAAC,EAAE,CAAC;aACtE,mBAAmB,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CAAC;QAErD,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,iBAAiB,CAAC;YACxC,IAAI,EAAE,mCAAmC;SACE,CAAC,CAAC;QAC/C,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;QAElD,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,YAAqB,EAAE,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;QAEtE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;QAC1C,MAAM,CAAC,SAAS,CAAC,CAAC,oBAAoB,CAAC,2BAA2B,EAAE,sBAAsB,EAAE,OAAO,CAAC,CAAC;QACrG,MAAM,CAAC,SAAS,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IAC7C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kEAAkE,EAAE,KAAK,IAAI,EAAE;QAChF,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC;aACpB,mBAAmB,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC;aACzD,mBAAmB,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CAAC;QAErD,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,iBAAiB,CAAC;YACxC,IAAI,EAAE,6CAA6C;SACR,CAAC,CAAC;QAC/C,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;QAElD,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,YAAqB,EAAE,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;QAEtE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;IAC5C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0EAA0E,EAAE,KAAK,IAAI,EAAE;QACxF,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC;aACpB,mBAAmB,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC;aACzD,mBAAmB,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CAAC;QAErD,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,iBAAiB,CAAC;YACxC,IAAI,EAAE,6CAA6C;SACR,CAAC,CAAC;QAC/C,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;QAElD,MAAM,MAAM,GAAG,MAAM,WAAW,CAC9B,YAAqB,EACrB,OAAO,CAAC,EAAE,aAAa,EAAE,SAAS,EAAE,QAAQ,EAAE,SAAS,EAAE,CAAC,EAC1D,CAAC,CACF,CAAC;QAEF,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;IAC5C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"visual-test.d.ts","sourceRoot":"","sources":["../../../../src/pipeline/steps/visual-test.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAElD,wBAAsB,cAAc,IAAI,OAAO,CAAC,UAAU,CAAC,CAE1D"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"visual-test.js","sourceRoot":"","sources":["../../../../src/pipeline/steps/visual-test.ts"],"names":[],"mappings":"AAEA,MAAM,CAAC,KAAK,UAAU,cAAc;IAClC,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;AAC3B,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"visual-test.test.d.ts","sourceRoot":"","sources":["../../../../src/pipeline/steps/visual-test.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,9 @@
|
|
|
1
|
+
import { describe, expect, it } from 'vitest';
|
|
2
|
+
import { visualTestStep } from './visual-test.js';
|
|
3
|
+
describe('visualTestStep', () => {
|
|
4
|
+
it('returns success (placeholder)', async () => {
|
|
5
|
+
const result = await visualTestStep();
|
|
6
|
+
expect(result).toEqual({ success: true });
|
|
7
|
+
});
|
|
8
|
+
});
|
|
9
|
+
//# sourceMappingURL=visual-test.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"visual-test.test.js","sourceRoot":"","sources":["../../../../src/pipeline/steps/visual-test.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AAE/C,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;IAC9B,EAAE,CAAC,+BAA+B,EAAE,KAAK,IAAI,EAAE;QAC7C,MAAM,MAAM,GAAG,MAAM,cAAc,EAAE,CAAC;QAEtC,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;IAC5C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1,10 @@
|
|
|
1
|
+
export type ComposedComponent = {
|
|
2
|
+
id: string;
|
|
3
|
+
path: string;
|
|
4
|
+
};
|
|
5
|
+
export declare function buildPackageJsonSection(targetDir: string): Promise<string>;
|
|
6
|
+
export declare function buildBarrelFileSection(targetDir: string): Promise<string>;
|
|
7
|
+
export declare function buildStylesSection(targetDir: string): Promise<string>;
|
|
8
|
+
export declare function buildComposedComponentsSection(targetDir: string, composes: ComposedComponent[]): Promise<string>;
|
|
9
|
+
export declare function buildFullProjectSection(targetDir: string, composes: ComposedComponent[]): Promise<string>;
|
|
10
|
+
//# sourceMappingURL=project-context.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"project-context.d.ts","sourceRoot":"","sources":["../../src/project-context.ts"],"names":[],"mappings":"AAGA,MAAM,MAAM,iBAAiB,GAAG;IAAE,EAAE,EAAE,MAAM,CAAC;IAAC,IAAI,EAAE,MAAM,CAAA;CAAE,CAAC;AA8C7D,wBAAsB,uBAAuB,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CAUhF;AAED,wBAAsB,sBAAsB,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CAU/E;AAED,wBAAsB,kBAAkB,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CAU3E;AAiED,wBAAsB,8BAA8B,CAClD,SAAS,EAAE,MAAM,EACjB,QAAQ,EAAE,iBAAiB,EAAE,GAC5B,OAAO,CAAC,MAAM,CAAC,CA0CjB;AAED,wBAAsB,uBAAuB,CAAC,SAAS,EAAE,MAAM,EAAE,QAAQ,EAAE,iBAAiB,EAAE,GAAG,OAAO,CAAC,MAAM,CAAC,CAS/G"}
|
|
@@ -0,0 +1,178 @@
|
|
|
1
|
+
import { readFile } from 'node:fs/promises';
|
|
2
|
+
import path from 'node:path';
|
|
3
|
+
function kebabToPascal(name) {
|
|
4
|
+
return name
|
|
5
|
+
.split('-')
|
|
6
|
+
.map((s) => s.charAt(0).toUpperCase() + s.slice(1))
|
|
7
|
+
.join('');
|
|
8
|
+
}
|
|
9
|
+
/**
|
|
10
|
+
* Extract the component name from a file path.
|
|
11
|
+
* e.g. "./src/components/ui/Button.tsx" → "Button"
|
|
12
|
+
* "src/components/MyWidget.tsx" → "MyWidget"
|
|
13
|
+
*/
|
|
14
|
+
function componentNameFromPath(filePath) {
|
|
15
|
+
const lastSlash = filePath.lastIndexOf('/');
|
|
16
|
+
const filename = lastSlash === -1 ? filePath : filePath.slice(lastSlash + 1);
|
|
17
|
+
const dotIndex = filename.lastIndexOf('.');
|
|
18
|
+
return dotIndex === -1 ? filename : filename.slice(0, dotIndex);
|
|
19
|
+
}
|
|
20
|
+
/**
|
|
21
|
+
* Converts a file path like `src/components/ui/Button.tsx` or `src/components/MyFeature`
|
|
22
|
+
* to a `@/` import alias like `@/components/ui/Button` or `@/components/MyFeature`.
|
|
23
|
+
*/
|
|
24
|
+
function filePathToImportAlias(filePath) {
|
|
25
|
+
let rel = filePath;
|
|
26
|
+
if (rel.startsWith('./')) {
|
|
27
|
+
rel = rel.slice(2);
|
|
28
|
+
}
|
|
29
|
+
if (rel.startsWith('src/')) {
|
|
30
|
+
rel = rel.slice(4);
|
|
31
|
+
}
|
|
32
|
+
// Strip .tsx/.ts extension
|
|
33
|
+
rel = rel.replace(/\.tsx?$/, '');
|
|
34
|
+
return `@/${rel}`;
|
|
35
|
+
}
|
|
36
|
+
async function readFileSafe(filePath) {
|
|
37
|
+
try {
|
|
38
|
+
return await readFile(filePath, 'utf-8');
|
|
39
|
+
}
|
|
40
|
+
catch {
|
|
41
|
+
return null;
|
|
42
|
+
}
|
|
43
|
+
}
|
|
44
|
+
export async function buildPackageJsonSection(targetDir) {
|
|
45
|
+
const content = await readFileSafe(path.join(targetDir, 'package.json'));
|
|
46
|
+
if (!content)
|
|
47
|
+
return '';
|
|
48
|
+
return [
|
|
49
|
+
'## Project Dependencies (package.json)',
|
|
50
|
+
'Only use packages listed here. Do not import packages that are not installed.',
|
|
51
|
+
'',
|
|
52
|
+
content,
|
|
53
|
+
].join('\n');
|
|
54
|
+
}
|
|
55
|
+
export async function buildBarrelFileSection(targetDir) {
|
|
56
|
+
const content = await readFileSafe(path.join(targetDir, 'src', 'components', 'ui', 'index.ts'));
|
|
57
|
+
if (!content)
|
|
58
|
+
return '';
|
|
59
|
+
return [
|
|
60
|
+
'## Component Library (src/components/ui/index.ts)',
|
|
61
|
+
"These are all available UI components. Import from '@/components/ui'.",
|
|
62
|
+
'',
|
|
63
|
+
content,
|
|
64
|
+
].join('\n');
|
|
65
|
+
}
|
|
66
|
+
export async function buildStylesSection(targetDir) {
|
|
67
|
+
const content = await readFileSafe(path.join(targetDir, 'src', 'index.css'));
|
|
68
|
+
if (!content)
|
|
69
|
+
return '';
|
|
70
|
+
return [
|
|
71
|
+
'## Project Styles (Tailwind CSS v4)',
|
|
72
|
+
"This project uses Tailwind CSS v4. Here is the project's stylesheet with design tokens and base utilities:",
|
|
73
|
+
'',
|
|
74
|
+
content,
|
|
75
|
+
].join('\n');
|
|
76
|
+
}
|
|
77
|
+
/**
|
|
78
|
+
* Extract only the exported interface from a component file:
|
|
79
|
+
* - export type/interface declarations (props types)
|
|
80
|
+
* - export function/const signatures (without function body)
|
|
81
|
+
* - import statements (to understand dependencies)
|
|
82
|
+
*
|
|
83
|
+
* This reduces token usage by ~70% compared to sending full source.
|
|
84
|
+
*/
|
|
85
|
+
function extractSignature(content) {
|
|
86
|
+
const lines = content.split('\n');
|
|
87
|
+
const signatureLines = [];
|
|
88
|
+
let inExportBlock = false;
|
|
89
|
+
let braceDepth = 0;
|
|
90
|
+
for (const line of lines) {
|
|
91
|
+
const trimmed = line.trim();
|
|
92
|
+
// Always include import statements
|
|
93
|
+
if (trimmed.startsWith('import ')) {
|
|
94
|
+
signatureLines.push(line);
|
|
95
|
+
continue;
|
|
96
|
+
}
|
|
97
|
+
// Capture export type / export interface blocks
|
|
98
|
+
if (trimmed.startsWith('export type ') || trimmed.startsWith('export interface ')) {
|
|
99
|
+
inExportBlock = true;
|
|
100
|
+
braceDepth = 0;
|
|
101
|
+
}
|
|
102
|
+
// Capture export { ... } re-export blocks (e.g. `export { Select, SelectTrigger, ... }`)
|
|
103
|
+
if (!inExportBlock && trimmed.startsWith('export {')) {
|
|
104
|
+
inExportBlock = true;
|
|
105
|
+
braceDepth = 0;
|
|
106
|
+
}
|
|
107
|
+
// Track export function/const signatures (just the signature line)
|
|
108
|
+
if (!inExportBlock &&
|
|
109
|
+
(trimmed.startsWith('export function ') ||
|
|
110
|
+
trimmed.startsWith('export const ') ||
|
|
111
|
+
trimmed.startsWith('export default '))) {
|
|
112
|
+
// Grab just the signature — stop at opening brace
|
|
113
|
+
const sigLine = trimmed.replace(/\{[\s\S]*$/, '{ ... }');
|
|
114
|
+
signatureLines.push(sigLine);
|
|
115
|
+
continue;
|
|
116
|
+
}
|
|
117
|
+
if (inExportBlock) {
|
|
118
|
+
signatureLines.push(line);
|
|
119
|
+
for (const ch of trimmed) {
|
|
120
|
+
if (ch === '{')
|
|
121
|
+
braceDepth++;
|
|
122
|
+
if (ch === '}')
|
|
123
|
+
braceDepth--;
|
|
124
|
+
}
|
|
125
|
+
if (braceDepth <= 0 && trimmed.includes('}')) {
|
|
126
|
+
inExportBlock = false;
|
|
127
|
+
}
|
|
128
|
+
}
|
|
129
|
+
}
|
|
130
|
+
return signatureLines.join('\n');
|
|
131
|
+
}
|
|
132
|
+
export async function buildComposedComponentsSection(targetDir, composes) {
|
|
133
|
+
if (composes.length === 0)
|
|
134
|
+
return '';
|
|
135
|
+
const sections = [
|
|
136
|
+
'## Composed Components',
|
|
137
|
+
'This component uses the following components. Import each using the exact path shown.',
|
|
138
|
+
];
|
|
139
|
+
for (const { id, path: componentPath } of composes) {
|
|
140
|
+
const pascalName = componentNameFromPath(componentPath);
|
|
141
|
+
const importAlias = filePathToImportAlias(componentPath);
|
|
142
|
+
// Try .tsx then without extension
|
|
143
|
+
const candidates = [
|
|
144
|
+
path.join(targetDir, componentPath + '.tsx'),
|
|
145
|
+
path.join(targetDir, componentPath + '.ts'),
|
|
146
|
+
path.join(targetDir, componentPath),
|
|
147
|
+
];
|
|
148
|
+
let content = null;
|
|
149
|
+
let resolvedRelative = componentPath;
|
|
150
|
+
for (const candidate of candidates) {
|
|
151
|
+
content = await readFileSafe(candidate);
|
|
152
|
+
if (content) {
|
|
153
|
+
resolvedRelative = path.relative(targetDir, candidate);
|
|
154
|
+
break;
|
|
155
|
+
}
|
|
156
|
+
}
|
|
157
|
+
if (!content)
|
|
158
|
+
continue;
|
|
159
|
+
// Extract only the signature (types + exports), not full implementation
|
|
160
|
+
const signature = extractSignature(content);
|
|
161
|
+
sections.push('');
|
|
162
|
+
sections.push(`### ${pascalName} — import from '${importAlias}' (${resolvedRelative})`);
|
|
163
|
+
sections.push('```tsx');
|
|
164
|
+
sections.push(signature);
|
|
165
|
+
sections.push('```');
|
|
166
|
+
}
|
|
167
|
+
return sections.join('\n');
|
|
168
|
+
}
|
|
169
|
+
export async function buildFullProjectSection(targetDir, composes) {
|
|
170
|
+
const parts = await Promise.all([
|
|
171
|
+
buildPackageJsonSection(targetDir),
|
|
172
|
+
buildBarrelFileSection(targetDir),
|
|
173
|
+
buildStylesSection(targetDir),
|
|
174
|
+
buildComposedComponentsSection(targetDir, composes),
|
|
175
|
+
]);
|
|
176
|
+
return parts.filter(Boolean).join('\n\n');
|
|
177
|
+
}
|
|
178
|
+
//# sourceMappingURL=project-context.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"project-context.js","sourceRoot":"","sources":["../../src/project-context.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,kBAAkB,CAAC;AAC5C,OAAO,IAAI,MAAM,WAAW,CAAC;AAI7B,SAAS,aAAa,CAAC,IAAY;IACjC,OAAO,IAAI;SACR,KAAK,CAAC,GAAG,CAAC;SACV,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;SAClD,IAAI,CAAC,EAAE,CAAC,CAAC;AACd,CAAC;AAED;;;;GAIG;AACH,SAAS,qBAAqB,CAAC,QAAgB;IAC7C,MAAM,SAAS,GAAG,QAAQ,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;IAC5C,MAAM,QAAQ,GAAG,SAAS,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,SAAS,GAAG,CAAC,CAAC,CAAC;IAC7E,MAAM,QAAQ,GAAG,QAAQ,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;IAC3C,OAAO,QAAQ,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,QAAQ,CAAC,CAAC;AAClE,CAAC;AAED;;;GAGG;AACH,SAAS,qBAAqB,CAAC,QAAgB;IAC7C,IAAI,GAAG,GAAG,QAAQ,CAAC;IACnB,IAAI,GAAG,CAAC,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC;QACzB,GAAG,GAAG,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;IACrB,CAAC;IACD,IAAI,GAAG,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC;QAC3B,GAAG,GAAG,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;IACrB,CAAC;IACD,2BAA2B;IAC3B,GAAG,GAAG,GAAG,CAAC,OAAO,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;IACjC,OAAO,KAAK,GAAG,EAAE,CAAC;AACpB,CAAC;AAED,KAAK,UAAU,YAAY,CAAC,QAAgB;IAC1C,IAAI,CAAC;QACH,OAAO,MAAM,QAAQ,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;IAC3C,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,IAAI,CAAC;IACd,CAAC;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,uBAAuB,CAAC,SAAiB;IAC7D,MAAM,OAAO,GAAG,MAAM,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,cAAc,CAAC,CAAC,CAAC;IACzE,IAAI,CAAC,OAAO;QAAE,OAAO,EAAE,CAAC;IAExB,OAAO;QACL,wCAAwC;QACxC,+EAA+E;QAC/E,EAAE;QACF,OAAO;KACR,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AACf,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,sBAAsB,CAAC,SAAiB;IAC5D,MAAM,OAAO,GAAG,MAAM,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,KAAK,EAAE,YAAY,EAAE,IAAI,EAAE,UAAU,CAAC,CAAC,CAAC;IAChG,IAAI,CAAC,OAAO;QAAE,OAAO,EAAE,CAAC;IAExB,OAAO;QACL,mDAAmD;QACnD,uEAAuE;QACvE,EAAE;QACF,OAAO;KACR,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AACf,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,kBAAkB,CAAC,SAAiB;IACxD,MAAM,OAAO,GAAG,MAAM,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,KAAK,EAAE,WAAW,CAAC,CAAC,CAAC;IAC7E,IAAI,CAAC,OAAO;QAAE,OAAO,EAAE,CAAC;IAExB,OAAO;QACL,qCAAqC;QACrC,4GAA4G;QAC5G,EAAE;QACF,OAAO;KACR,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AACf,CAAC;AAED;;;;;;;GAOG;AACH,SAAS,gBAAgB,CAAC,OAAe;IACvC,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IAClC,MAAM,cAAc,GAAa,EAAE,CAAC;IACpC,IAAI,aAAa,GAAG,KAAK,CAAC;IAC1B,IAAI,UAAU,GAAG,CAAC,CAAC;IAEnB,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;QAE5B,mCAAmC;QACnC,IAAI,OAAO,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;YAClC,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC1B,SAAS;QACX,CAAC;QAED,gDAAgD;QAChD,IAAI,OAAO,CAAC,UAAU,CAAC,cAAc,CAAC,IAAI,OAAO,CAAC,UAAU,CAAC,mBAAmB,CAAC,EAAE,CAAC;YAClF,aAAa,GAAG,IAAI,CAAC;YACrB,UAAU,GAAG,CAAC,CAAC;QACjB,CAAC;QAED,yFAAyF;QACzF,IAAI,CAAC,aAAa,IAAI,OAAO,CAAC,UAAU,CAAC,UAAU,CAAC,EAAE,CAAC;YACrD,aAAa,GAAG,IAAI,CAAC;YACrB,UAAU,GAAG,CAAC,CAAC;QACjB,CAAC;QAED,mEAAmE;QACnE,IACE,CAAC,aAAa;YACd,CAAC,OAAO,CAAC,UAAU,CAAC,kBAAkB,CAAC;gBACrC,OAAO,CAAC,UAAU,CAAC,eAAe,CAAC;gBACnC,OAAO,CAAC,UAAU,CAAC,iBAAiB,CAAC,CAAC,EACxC,CAAC;YACD,kDAAkD;YAClD,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,YAAY,EAAE,SAAS,CAAC,CAAC;YACzD,cAAc,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAC7B,SAAS;QACX,CAAC;QAED,IAAI,aAAa,EAAE,CAAC;YAClB,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC1B,KAAK,MAAM,EAAE,IAAI,OAAO,EAAE,CAAC;gBACzB,IAAI,EAAE,KAAK,GAAG;oBAAE,UAAU,EAAE,CAAC;gBAC7B,IAAI,EAAE,KAAK,GAAG;oBAAE,UAAU,EAAE,CAAC;YAC/B,CAAC;YACD,IAAI,UAAU,IAAI,CAAC,IAAI,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;gBAC7C,aAAa,GAAG,KAAK,CAAC;YACxB,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AACnC,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,8BAA8B,CAClD,SAAiB,EACjB,QAA6B;IAE7B,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,EAAE,CAAC;IAErC,MAAM,QAAQ,GAAa;QACzB,wBAAwB;QACxB,uFAAuF;KACxF,CAAC;IAEF,KAAK,MAAM,EAAE,EAAE,EAAE,IAAI,EAAE,aAAa,EAAE,IAAI,QAAQ,EAAE,CAAC;QACnD,MAAM,UAAU,GAAG,qBAAqB,CAAC,aAAa,CAAC,CAAC;QACxD,MAAM,WAAW,GAAG,qBAAqB,CAAC,aAAa,CAAC,CAAC;QAEzD,kCAAkC;QAClC,MAAM,UAAU,GAAG;YACjB,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,aAAa,GAAG,MAAM,CAAC;YAC5C,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,aAAa,GAAG,KAAK,CAAC;YAC3C,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,aAAa,CAAC;SACpC,CAAC;QAEF,IAAI,OAAO,GAAkB,IAAI,CAAC;QAClC,IAAI,gBAAgB,GAAG,aAAa,CAAC;QACrC,KAAK,MAAM,SAAS,IAAI,UAAU,EAAE,CAAC;YACnC,OAAO,GAAG,MAAM,YAAY,CAAC,SAAS,CAAC,CAAC;YACxC,IAAI,OAAO,EAAE,CAAC;gBACZ,gBAAgB,GAAG,IAAI,CAAC,QAAQ,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC;gBACvD,MAAM;YACR,CAAC;QACH,CAAC;QAED,IAAI,CAAC,OAAO;YAAE,SAAS;QAEvB,wEAAwE;QACxE,MAAM,SAAS,GAAG,gBAAgB,CAAC,OAAO,CAAC,CAAC;QAE5C,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAClB,QAAQ,CAAC,IAAI,CAAC,OAAO,UAAU,mBAAmB,WAAW,MAAM,gBAAgB,GAAG,CAAC,CAAC;QACxF,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACxB,QAAQ,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACzB,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACvB,CAAC;IAED,OAAO,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC7B,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,uBAAuB,CAAC,SAAiB,EAAE,QAA6B;IAC5F,MAAM,KAAK,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;QAC9B,uBAAuB,CAAC,SAAS,CAAC;QAClC,sBAAsB,CAAC,SAAS,CAAC;QACjC,kBAAkB,CAAC,SAAS,CAAC;QAC7B,8BAA8B,CAAC,SAAS,EAAE,QAAQ,CAAC;KACpD,CAAC,CAAC;IAEH,OAAO,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;AAC5C,CAAC"}
|