genaicode 0.0.37 → 0.2.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/README.md +3 -1
- package/bin/genaicode-dev.js +11 -0
- package/bin/genaicode.cjs +15 -1
- package/dist/ai-service/ai-studio.d.ts +6 -0
- package/dist/ai-service/ai-studio.js +135 -0
- package/dist/ai-service/ai-studio.js.map +1 -0
- package/dist/ai-service/anthropic.d.ts +6 -0
- package/dist/ai-service/anthropic.js +135 -0
- package/dist/ai-service/anthropic.js.map +1 -0
- package/dist/ai-service/chat-gpt.d.ts +5 -0
- package/dist/ai-service/chat-gpt.js +132 -0
- package/dist/ai-service/chat-gpt.js.map +1 -0
- package/dist/ai-service/common.d.ts +58 -0
- package/dist/ai-service/common.js +38 -0
- package/dist/ai-service/common.js.map +1 -0
- package/dist/ai-service/common.test.d.ts +1 -0
- package/dist/ai-service/common.test.js +85 -0
- package/dist/ai-service/common.test.js.map +1 -0
- package/dist/ai-service/dall-e.d.ts +14 -0
- package/dist/ai-service/dall-e.js +48 -0
- package/dist/ai-service/dall-e.js.map +1 -0
- package/dist/ai-service/vertex-ai-claude.d.ts +5 -0
- package/dist/ai-service/vertex-ai-claude.js +94 -0
- package/dist/ai-service/vertex-ai-claude.js.map +1 -0
- package/dist/ai-service/vertex-ai-imagen.d.ts +14 -0
- package/dist/ai-service/vertex-ai-imagen.js +77 -0
- package/dist/ai-service/vertex-ai-imagen.js.map +1 -0
- package/dist/ai-service/vertex-ai.d.ts +7 -0
- package/dist/ai-service/vertex-ai.js +142 -0
- package/dist/ai-service/vertex-ai.js.map +1 -0
- package/dist/cli/cli-options.d.ts +10 -0
- package/dist/cli/cli-options.js +124 -0
- package/dist/cli/cli-options.js.map +1 -0
- package/dist/cli/cli-options.test.d.ts +1 -0
- package/dist/cli/cli-options.test.js +34 -0
- package/dist/cli/cli-options.test.js.map +1 -0
- package/dist/cli/cli-params.d.ts +30 -0
- package/dist/cli/cli-params.js +109 -0
- package/dist/cli/cli-params.js.map +1 -0
- package/dist/cli/cli-params.test.d.ts +1 -0
- package/dist/cli/cli-params.test.js +38 -0
- package/dist/cli/cli-params.test.js.map +1 -0
- package/dist/cli/service-autodetect.d.ts +2 -0
- package/dist/cli/service-autodetect.js +17 -0
- package/dist/cli/service-autodetect.js.map +1 -0
- package/dist/cli/service-autodetect.test.d.ts +1 -0
- package/dist/cli/service-autodetect.test.js +38 -0
- package/dist/cli/service-autodetect.test.js.map +1 -0
- package/dist/cli/validate-cli-params.d.ts +18 -0
- package/dist/cli/validate-cli-params.js +129 -0
- package/dist/cli/validate-cli-params.js.map +1 -0
- package/dist/cli/validate-cli-params.test.d.ts +1 -0
- package/dist/cli/validate-cli-params.test.js +106 -0
- package/dist/cli/validate-cli-params.test.js.map +1 -0
- package/dist/files/file-utils.d.ts +2 -0
- package/dist/files/file-utils.js +7 -0
- package/dist/files/file-utils.js.map +1 -0
- package/dist/files/file-utils.test.d.ts +1 -0
- package/dist/files/file-utils.test.js +18 -0
- package/dist/files/file-utils.test.js.map +1 -0
- package/dist/files/find-files.d.ts +8 -0
- package/dist/files/find-files.js +76 -0
- package/dist/files/find-files.js.map +1 -0
- package/dist/files/read-files.d.ts +18 -0
- package/dist/files/read-files.js +66 -0
- package/dist/files/read-files.js.map +1 -0
- package/dist/files/read-files.test.d.ts +1 -0
- package/dist/files/read-files.test.js +88 -0
- package/dist/files/read-files.test.js.map +1 -0
- package/dist/files/temp-buffer.d.ts +2 -0
- package/dist/files/temp-buffer.js +12 -0
- package/dist/files/temp-buffer.js.map +1 -0
- package/dist/files/update-files.d.ts +6 -0
- package/dist/files/update-files.js +197 -0
- package/dist/files/update-files.js.map +1 -0
- package/dist/images/ensure-alpha.d.ts +1 -0
- package/dist/images/ensure-alpha.js +5 -0
- package/dist/images/ensure-alpha.js.map +1 -0
- package/dist/images/imgly-remove-background.d.ts +2 -0
- package/dist/images/imgly-remove-background.js +21 -0
- package/dist/images/imgly-remove-background.js.map +1 -0
- package/dist/images/resize-image.d.ts +19 -0
- package/dist/images/resize-image.js +22 -0
- package/dist/images/resize-image.js.map +1 -0
- package/dist/images/split-image.d.ts +18 -0
- package/dist/images/split-image.js +25 -0
- package/dist/images/split-image.js.map +1 -0
- package/dist/main/codegen-interactive-utils.d.ts +8 -0
- package/dist/main/codegen-interactive-utils.js +148 -0
- package/dist/main/codegen-interactive-utils.js.map +1 -0
- package/dist/main/codegen-interactive.d.ts +2 -0
- package/dist/main/codegen-interactive.js +68 -0
- package/dist/main/codegen-interactive.js.map +1 -0
- package/dist/main/codegen-types.d.ts +28 -0
- package/dist/main/codegen-types.js +2 -0
- package/dist/main/codegen-types.js.map +1 -0
- package/dist/main/codegen.d.ts +4 -0
- package/dist/main/codegen.js +164 -0
- package/dist/main/codegen.js.map +1 -0
- package/dist/main/codegen.test.d.ts +2 -0
- package/dist/main/codegen.test.js +273 -0
- package/dist/main/codegen.test.js.map +1 -0
- package/dist/main/config-lib.d.ts +16 -0
- package/dist/main/config-lib.js +28 -0
- package/dist/main/config-lib.js.map +1 -0
- package/dist/main/config-lib.test.d.ts +1 -0
- package/dist/main/config-lib.test.js +64 -0
- package/dist/main/config-lib.test.js.map +1 -0
- package/dist/main/config.d.ts +6 -0
- package/{src → dist}/main/config.js +28 -22
- package/dist/main/config.js.map +1 -0
- package/dist/main/interactive/codegen-interactive.d.ts +2 -0
- package/dist/main/interactive/codegen-interactive.js +55 -0
- package/dist/main/interactive/codegen-interactive.js.map +1 -0
- package/dist/main/interactive/codegen-interactive.test.d.ts +4 -0
- package/dist/main/interactive/codegen-interactive.test.js +77 -0
- package/dist/main/interactive/codegen-interactive.test.js.map +1 -0
- package/dist/main/interactive/codegen-worker.d.ts +3 -0
- package/dist/main/interactive/codegen-worker.js +47 -0
- package/dist/main/interactive/codegen-worker.js.map +1 -0
- package/dist/main/interactive/common.d.ts +5 -0
- package/dist/main/interactive/common.js +51 -0
- package/dist/main/interactive/common.js.map +1 -0
- package/dist/main/interactive/configure.d.ts +2 -0
- package/dist/main/interactive/configure.js +99 -0
- package/dist/main/interactive/configure.js.map +1 -0
- package/dist/main/interactive/error-handling.d.ts +4 -0
- package/dist/main/interactive/error-handling.js +49 -0
- package/dist/main/interactive/error-handling.js.map +1 -0
- package/dist/main/interactive/help.d.ts +1 -0
- package/dist/main/interactive/help.js +22 -0
- package/dist/main/interactive/help.js.map +1 -0
- package/dist/main/interactive/process-comments.d.ts +2 -0
- package/dist/main/interactive/process-comments.js +39 -0
- package/dist/main/interactive/process-comments.js.map +1 -0
- package/dist/main/interactive/select-ai-service.d.ts +2 -0
- package/dist/main/interactive/select-ai-service.js +27 -0
- package/dist/main/interactive/select-ai-service.js.map +1 -0
- package/dist/main/interactive/task-file.d.ts +2 -0
- package/dist/main/interactive/task-file.js +28 -0
- package/dist/main/interactive/task-file.js.map +1 -0
- package/dist/main/interactive/text-prompt.d.ts +2 -0
- package/dist/main/interactive/text-prompt.js +18 -0
- package/dist/main/interactive/text-prompt.js.map +1 -0
- package/dist/main/interactive/user-interrupt.d.ts +1 -0
- package/dist/main/interactive/user-interrupt.js +36 -0
- package/dist/main/interactive/user-interrupt.js.map +1 -0
- package/dist/main/interactive/worker-script.d.ts +1 -0
- package/dist/main/interactive/worker-script.js +24 -0
- package/dist/main/interactive/worker-script.js.map +1 -0
- package/dist/main/interactive/worker.d.ts +2 -0
- package/dist/main/interactive/worker.js +55 -0
- package/dist/main/interactive/worker.js.map +1 -0
- package/dist/prompt/function-calling-validate.d.ts +3 -0
- package/dist/prompt/function-calling-validate.js +33 -0
- package/dist/prompt/function-calling-validate.js.map +1 -0
- package/dist/prompt/function-calling.d.ts +5 -0
- package/{src → dist}/prompt/function-calling.js +25 -24
- package/dist/prompt/function-calling.js.map +1 -0
- package/dist/prompt/function-defs/ask-question.d.ts +65 -0
- package/dist/prompt/function-defs/ask-question.js +67 -0
- package/dist/prompt/function-defs/ask-question.js.map +1 -0
- package/dist/prompt/function-defs/codegen-summary.d.ts +63 -0
- package/dist/prompt/function-defs/codegen-summary.js +80 -0
- package/dist/prompt/function-defs/codegen-summary.js.map +1 -0
- package/dist/prompt/function-defs/create-directory.d.ts +21 -0
- package/dist/prompt/function-defs/create-directory.js +22 -0
- package/dist/prompt/function-defs/create-directory.js.map +1 -0
- package/dist/prompt/function-defs/create-file.d.ts +25 -0
- package/dist/prompt/function-defs/create-file.js +26 -0
- package/dist/prompt/function-defs/create-file.js.map +1 -0
- package/dist/prompt/function-defs/delete-file.d.ts +21 -0
- package/dist/prompt/function-defs/delete-file.js +22 -0
- package/dist/prompt/function-defs/delete-file.js.map +1 -0
- package/dist/prompt/function-defs/download-file.d.ts +25 -0
- package/dist/prompt/function-defs/download-file.js +26 -0
- package/dist/prompt/function-defs/download-file.js.map +1 -0
- package/dist/prompt/function-defs/explanation.d.ts +17 -0
- package/dist/prompt/function-defs/explanation.js +18 -0
- package/dist/prompt/function-defs/explanation.js.map +1 -0
- package/dist/prompt/function-defs/generate-image.d.ts +41 -0
- package/dist/prompt/function-defs/generate-image.js +42 -0
- package/dist/prompt/function-defs/generate-image.js.map +1 -0
- package/dist/prompt/function-defs/get-image-assets.d.ts +20 -0
- package/dist/prompt/function-defs/get-image-assets.js +21 -0
- package/dist/prompt/function-defs/get-image-assets.js.map +1 -0
- package/dist/prompt/function-defs/get-source-code.d.ts +20 -0
- package/dist/prompt/function-defs/get-source-code.js +21 -0
- package/dist/prompt/function-defs/get-source-code.js.map +1 -0
- package/dist/prompt/function-defs/imgly-remove-background.d.ts +25 -0
- package/dist/prompt/function-defs/imgly-remove-background.js +26 -0
- package/dist/prompt/function-defs/imgly-remove-background.js.map +1 -0
- package/dist/prompt/function-defs/move-file.d.ts +25 -0
- package/dist/prompt/function-defs/move-file.js +26 -0
- package/dist/prompt/function-defs/move-file.js.map +1 -0
- package/dist/prompt/function-defs/patch-file.d.ts +25 -0
- package/dist/prompt/function-defs/patch-file.js +39 -0
- package/dist/prompt/function-defs/patch-file.js.map +1 -0
- package/dist/prompt/function-defs/resize-image.d.ts +36 -0
- package/dist/prompt/function-defs/resize-image.js +37 -0
- package/dist/prompt/function-defs/resize-image.js.map +1 -0
- package/dist/prompt/function-defs/split-image.d.ts +57 -0
- package/dist/prompt/function-defs/split-image.js +46 -0
- package/dist/prompt/function-defs/split-image.js.map +1 -0
- package/dist/prompt/function-defs/update-file.d.ts +25 -0
- package/dist/prompt/function-defs/update-file.js +26 -0
- package/dist/prompt/function-defs/update-file.js.map +1 -0
- package/dist/prompt/limits.d.ts +3 -0
- package/dist/prompt/limits.js +19 -0
- package/dist/prompt/limits.js.map +1 -0
- package/dist/prompt/limits.test.d.ts +1 -0
- package/dist/prompt/limits.test.js +35 -0
- package/dist/prompt/limits.test.js.map +1 -0
- package/dist/prompt/prompt-codegen.d.ts +9 -0
- package/dist/prompt/prompt-codegen.js +90 -0
- package/dist/prompt/prompt-codegen.js.map +1 -0
- package/dist/prompt/prompt-codegen.test.d.ts +1 -0
- package/dist/prompt/prompt-codegen.test.js +98 -0
- package/dist/prompt/prompt-codegen.test.js.map +1 -0
- package/dist/prompt/prompt-consts.d.ts +1 -0
- package/{src → dist}/prompt/prompt-consts.js +1 -0
- package/dist/prompt/prompt-consts.js.map +1 -0
- package/dist/prompt/prompt-service-ask-question.test.d.ts +3 -0
- package/dist/prompt/prompt-service-ask-question.test.js +142 -0
- package/dist/prompt/prompt-service-ask-question.test.js.map +1 -0
- package/dist/prompt/prompt-service.d.ts +28 -0
- package/dist/prompt/prompt-service.js +163 -0
- package/dist/prompt/prompt-service.js.map +1 -0
- package/dist/prompt/prompt-service.test.d.ts +1 -0
- package/dist/prompt/prompt-service.test.js +531 -0
- package/dist/prompt/prompt-service.test.js.map +1 -0
- package/dist/prompt/steps/step-ask-question.d.ts +6 -0
- package/dist/prompt/steps/step-ask-question.js +82 -0
- package/dist/prompt/steps/step-ask-question.js.map +1 -0
- package/dist/prompt/steps/step-generate-image.d.ts +2 -0
- package/dist/prompt/steps/step-generate-image.js +29 -0
- package/dist/prompt/steps/step-generate-image.js.map +1 -0
- package/dist/prompt/steps/step-validate-recover.d.ts +13 -0
- package/dist/prompt/steps/step-validate-recover.js +51 -0
- package/dist/prompt/steps/step-validate-recover.js.map +1 -0
- package/dist/prompt/steps/step-verify-patch.d.ts +7 -0
- package/dist/prompt/steps/step-verify-patch.js +41 -0
- package/dist/prompt/steps/step-verify-patch.js.map +1 -0
- package/dist/prompt/steps/steps-types.d.ts +4 -0
- package/dist/prompt/steps/steps-types.js +6 -0
- package/dist/prompt/steps/steps-types.js.map +1 -0
- package/dist/prompt/systemprompt.d.ts +3 -0
- package/dist/prompt/systemprompt.js +47 -0
- package/dist/prompt/systemprompt.js.map +1 -0
- package/dist/prompt/systemprompt.test.d.ts +3 -0
- package/dist/prompt/systemprompt.test.js +46 -0
- package/dist/prompt/systemprompt.test.js.map +1 -0
- package/package.json +15 -5
- package/bin/genaicode.js +0 -5
- package/bin/vertex-monkey-patch.cjs +0 -33
- package/src/ai-service/anthropic.js +0 -126
- package/src/ai-service/chat-gpt.js +0 -126
- package/src/ai-service/common.js +0 -49
- package/src/ai-service/common.test.js +0 -107
- package/src/ai-service/dall-e.js +0 -50
- package/src/ai-service/vertex-ai-claude.js +0 -99
- package/src/ai-service/vertex-ai-imagen.js +0 -78
- package/src/ai-service/vertex-ai.js +0 -156
- package/src/cli/cli-options.js +0 -132
- package/src/cli/cli-options.test.js +0 -50
- package/src/cli/cli-params.js +0 -114
- package/src/cli/cli-params.test.js +0 -43
- package/src/cli/service-autodetect.js +0 -11
- package/src/cli/service-autodetect.test.js +0 -43
- package/src/cli/validate-cli-params.js +0 -137
- package/src/cli/validate-cli-params.test.js +0 -134
- package/src/files/file-utils.js +0 -7
- package/src/files/file-utils.test.js +0 -21
- package/src/files/find-files.js +0 -86
- package/src/files/read-files.js +0 -65
- package/src/files/read-files.test.js +0 -103
- package/src/files/temp-buffer.js +0 -13
- package/src/files/update-files.js +0 -205
- package/src/images/ensure-alpha.js +0 -5
- package/src/images/imgly-remove-background.js +0 -23
- package/src/images/resize-image.js +0 -23
- package/src/images/split-image.js +0 -26
- package/src/main/codegen.js +0 -120
- package/src/main/codegen.test.js +0 -335
- package/src/main/config-lib.js +0 -33
- package/src/main/config-lib.test.js +0 -80
- package/src/prompt/function-calling-validate.js +0 -25
- package/src/prompt/function-defs/ask-question.js +0 -33
- package/src/prompt/function-defs/codegen-summary.js +0 -87
- package/src/prompt/function-defs/create-directory.js +0 -21
- package/src/prompt/function-defs/create-file.js +0 -26
- package/src/prompt/function-defs/delete-file.js +0 -21
- package/src/prompt/function-defs/download-file.js +0 -25
- package/src/prompt/function-defs/explanation.js +0 -17
- package/src/prompt/function-defs/generate-image.js +0 -51
- package/src/prompt/function-defs/get-image-assets.js +0 -21
- package/src/prompt/function-defs/get-source-code.js +0 -21
- package/src/prompt/function-defs/imgly-remove-background.js +0 -25
- package/src/prompt/function-defs/move-file.js +0 -25
- package/src/prompt/function-defs/patch-file.js +0 -39
- package/src/prompt/function-defs/resize-image.js +0 -36
- package/src/prompt/function-defs/split-image.js +0 -45
- package/src/prompt/function-defs/update-file.js +0 -26
- package/src/prompt/limits.js +0 -23
- package/src/prompt/limits.test.js +0 -40
- package/src/prompt/prompt-codegen.js +0 -106
- package/src/prompt/prompt-codegen.test.js +0 -116
- package/src/prompt/prompt-service-ask-question.test.js +0 -163
- package/src/prompt/prompt-service.js +0 -335
- package/src/prompt/prompt-service.test.js +0 -602
- package/src/prompt/systemprompt.js +0 -39
- package/src/prompt/systemprompt.test.js +0 -64
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"codegen-types.js","sourceRoot":"","sources":["../../src/main/codegen-types.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,164 @@
|
|
|
1
|
+
import { exec } from 'child_process';
|
|
2
|
+
import util from 'util';
|
|
3
|
+
import assert from 'node:assert';
|
|
4
|
+
import * as cliParams from '../cli/cli-params.js';
|
|
5
|
+
import { validateCliParams } from '../cli/validate-cli-params.js';
|
|
6
|
+
import { generateContent as generateContentVertexAi } from '../ai-service/vertex-ai.js';
|
|
7
|
+
import { generateContent as generateContentGPT } from '../ai-service/chat-gpt.js';
|
|
8
|
+
import { generateContent as generateContentAnthropic } from '../ai-service/anthropic.js';
|
|
9
|
+
import { generateContent as generateContentVertexAiClaude } from '../ai-service/vertex-ai-claude.js';
|
|
10
|
+
import { generateContent as generateContentAiStudio } from '../ai-service/ai-studio.js';
|
|
11
|
+
import { generateImage as generateImageDallE } from '../ai-service/dall-e.js';
|
|
12
|
+
import { generateImage as generateImageVertexAi } from '../ai-service/vertex-ai-imagen.js';
|
|
13
|
+
import { promptService } from '../prompt/prompt-service.js';
|
|
14
|
+
import { updateFiles } from '../files/update-files.js';
|
|
15
|
+
import { rcConfig } from '../main/config.js';
|
|
16
|
+
import { getLintFixPrompt } from '../prompt/prompt-codegen.js';
|
|
17
|
+
import { printHelpMessage } from '../cli/cli-options.js';
|
|
18
|
+
import { getCodeGenPrompt } from '../prompt/prompt-codegen.js';
|
|
19
|
+
import { runInteractiveMode } from './interactive/codegen-interactive.js';
|
|
20
|
+
/** Executes codegen */
|
|
21
|
+
export async function runCodegen() {
|
|
22
|
+
// Print to console the received parameters
|
|
23
|
+
console.log(`Received parameters: ${process.argv.slice(2).join(' ')}`);
|
|
24
|
+
validateCliParams();
|
|
25
|
+
// Handle --help option
|
|
26
|
+
if (cliParams.helpRequested) {
|
|
27
|
+
printHelpMessage();
|
|
28
|
+
return;
|
|
29
|
+
}
|
|
30
|
+
const options = {
|
|
31
|
+
explicitPrompt: cliParams.explicitPrompt,
|
|
32
|
+
taskFile: cliParams.taskFile,
|
|
33
|
+
considerAllFiles: cliParams.considerAllFiles,
|
|
34
|
+
allowFileCreate: cliParams.allowFileCreate,
|
|
35
|
+
allowFileDelete: cliParams.allowFileDelete,
|
|
36
|
+
allowDirectoryCreate: cliParams.allowDirectoryCreate,
|
|
37
|
+
allowFileMove: cliParams.allowFileMove,
|
|
38
|
+
aiService: cliParamToAiService(),
|
|
39
|
+
vision: cliParams.vision,
|
|
40
|
+
imagen: cliParams.imagen,
|
|
41
|
+
disableContextOptimization: cliParams.disableContextOptimization,
|
|
42
|
+
temperature: cliParams.temperature,
|
|
43
|
+
cheap: cliParams.cheap,
|
|
44
|
+
dryRun: cliParams.dryRun,
|
|
45
|
+
verbose: cliParams.verbosePrompt,
|
|
46
|
+
requireExplanations: cliParams.requireExplanations,
|
|
47
|
+
geminiBlockNone: cliParams.geminiBlockNone,
|
|
48
|
+
disableInitialLint: cliParams.disableInitialLint,
|
|
49
|
+
contentMask: cliParams.contentMask,
|
|
50
|
+
ignorePatterns: cliParams.ignorePatterns,
|
|
51
|
+
askQuestion: cliParams.askQuestion,
|
|
52
|
+
disableCache: cliParams.disableCache,
|
|
53
|
+
interactive: cliParams.interactive,
|
|
54
|
+
};
|
|
55
|
+
// Handle interactive mode
|
|
56
|
+
if (cliParams.interactive) {
|
|
57
|
+
await runInteractiveMode(options);
|
|
58
|
+
}
|
|
59
|
+
else {
|
|
60
|
+
console.log('Executing codegen in non-interactive mode');
|
|
61
|
+
await runCodegenIteration(options);
|
|
62
|
+
}
|
|
63
|
+
}
|
|
64
|
+
export async function runCodegenIteration(options, abortSignal) {
|
|
65
|
+
if (rcConfig.lintCommand && !options.disableInitialLint) {
|
|
66
|
+
try {
|
|
67
|
+
console.log(`Executing lint command: ${rcConfig.lintCommand}`);
|
|
68
|
+
await execPromise(rcConfig.lintCommand);
|
|
69
|
+
console.log('Lint command executed successfully');
|
|
70
|
+
}
|
|
71
|
+
catch (error) {
|
|
72
|
+
const { stderr, stdout } = error;
|
|
73
|
+
console.log('Lint command failed. Aborting codegen, please fix lint issues before running codegen, or use --disable-initial-lint');
|
|
74
|
+
console.log('Lint errors:', stdout, stderr);
|
|
75
|
+
process.exit(1);
|
|
76
|
+
}
|
|
77
|
+
}
|
|
78
|
+
else if (rcConfig.lintCommand && options.disableInitialLint) {
|
|
79
|
+
console.log('Initial lint was skipped.');
|
|
80
|
+
}
|
|
81
|
+
if (abortSignal?.aborted) {
|
|
82
|
+
throw new Error('Codegen iteration aborted');
|
|
83
|
+
}
|
|
84
|
+
console.log('Generating response');
|
|
85
|
+
const functionCalls = await promptService(GENERATE_CONTENT_FNS, GENERATE_IMAGE_FNS, getCodeGenPrompt(options));
|
|
86
|
+
console.log('Received function calls:', functionCalls);
|
|
87
|
+
if (options.dryRun) {
|
|
88
|
+
console.log('Dry run mode, not updating files');
|
|
89
|
+
}
|
|
90
|
+
else {
|
|
91
|
+
console.log('Update files');
|
|
92
|
+
await updateFiles(functionCalls.filter((call) => call.name !== 'explanation' && call.name !== 'getSourceCode'), options);
|
|
93
|
+
console.log('Initial updates applied');
|
|
94
|
+
if (abortSignal?.aborted) {
|
|
95
|
+
throw new Error('Codegen iteration aborted after initial updates');
|
|
96
|
+
}
|
|
97
|
+
// Check if lintCommand is specified in .genaicoderc
|
|
98
|
+
if (rcConfig.lintCommand) {
|
|
99
|
+
try {
|
|
100
|
+
console.log(`Executing lint command: ${rcConfig.lintCommand}`);
|
|
101
|
+
await execPromise(rcConfig.lintCommand);
|
|
102
|
+
console.log('Lint command executed successfully');
|
|
103
|
+
}
|
|
104
|
+
catch (error) {
|
|
105
|
+
console.log('Lint command failed. Attempting to fix issues...');
|
|
106
|
+
// Prepare the lint error output for the second pass
|
|
107
|
+
const firstLintError = error;
|
|
108
|
+
const lintErrorPrompt = getLintFixPrompt(rcConfig.lintCommand, options, firstLintError.stdout, firstLintError.stderr);
|
|
109
|
+
console.log('Generating response for lint fixes');
|
|
110
|
+
const lintFixFunctionCalls = (await promptService(GENERATE_CONTENT_FNS, GENERATE_IMAGE_FNS, {
|
|
111
|
+
prompt: lintErrorPrompt,
|
|
112
|
+
options: { ...options, considerAllFiles: true },
|
|
113
|
+
}));
|
|
114
|
+
console.log('Received function calls for lint fixes:', lintFixFunctionCalls);
|
|
115
|
+
console.log('Applying lint fixes');
|
|
116
|
+
updateFiles(lintFixFunctionCalls.filter((call) => call.name !== 'explanation' && call.name !== 'getSourceCode'), options);
|
|
117
|
+
if (abortSignal?.aborted) {
|
|
118
|
+
throw new Error('Codegen iteration aborted after lint fixes');
|
|
119
|
+
}
|
|
120
|
+
// Run lint command again to verify fixes
|
|
121
|
+
try {
|
|
122
|
+
console.log(`Re-running lint command: ${rcConfig.lintCommand}`);
|
|
123
|
+
await execPromise(rcConfig.lintCommand);
|
|
124
|
+
console.log('Lint command executed successfully after fixes');
|
|
125
|
+
}
|
|
126
|
+
catch (secondLintError) {
|
|
127
|
+
const error = secondLintError;
|
|
128
|
+
console.log('Lint command still failing after fixes. Manual intervention may be required.');
|
|
129
|
+
console.log('Lint errors:', error.stdout, error.stderr);
|
|
130
|
+
}
|
|
131
|
+
}
|
|
132
|
+
}
|
|
133
|
+
console.log('Done!');
|
|
134
|
+
}
|
|
135
|
+
}
|
|
136
|
+
// helper functions and consts
|
|
137
|
+
const execPromise = util.promisify(exec);
|
|
138
|
+
const GENERATE_CONTENT_FNS = {
|
|
139
|
+
'vertex-ai-claude': generateContentVertexAiClaude,
|
|
140
|
+
'vertex-ai': generateContentVertexAi,
|
|
141
|
+
'ai-studio': generateContentAiStudio,
|
|
142
|
+
anthropic: generateContentAnthropic,
|
|
143
|
+
'chat-gpt': generateContentGPT,
|
|
144
|
+
};
|
|
145
|
+
const GENERATE_IMAGE_FNS = {
|
|
146
|
+
'dall-e': generateImageDallE,
|
|
147
|
+
'vertex-ai': generateImageVertexAi,
|
|
148
|
+
};
|
|
149
|
+
function cliParamToAiService() {
|
|
150
|
+
const result = cliParams.vertexAi
|
|
151
|
+
? 'vertex-ai'
|
|
152
|
+
: cliParams.aiStudio
|
|
153
|
+
? 'ai-studio'
|
|
154
|
+
: cliParams.vertexAiClaude
|
|
155
|
+
? 'vertex-ai-claude'
|
|
156
|
+
: cliParams.chatGpt
|
|
157
|
+
? 'chat-gpt'
|
|
158
|
+
: cliParams.anthropic
|
|
159
|
+
? 'anthropic'
|
|
160
|
+
: undefined;
|
|
161
|
+
assert(result, 'Please specify which AI service should be used');
|
|
162
|
+
return result;
|
|
163
|
+
}
|
|
164
|
+
//# sourceMappingURL=codegen.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"codegen.js","sourceRoot":"","sources":["../../src/main/codegen.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,eAAe,CAAC;AACrC,OAAO,IAAI,MAAM,MAAM,CAAC;AACxB,OAAO,MAAM,MAAM,aAAa,CAAC;AAEjC,OAAO,KAAK,SAAS,MAAM,sBAAsB,CAAC;AAClD,OAAO,EAAE,iBAAiB,EAAE,MAAM,+BAA+B,CAAC;AAClE,OAAO,EAAE,eAAe,IAAI,uBAAuB,EAAE,MAAM,4BAA4B,CAAC;AACxF,OAAO,EAAE,eAAe,IAAI,kBAAkB,EAAE,MAAM,2BAA2B,CAAC;AAClF,OAAO,EAAE,eAAe,IAAI,wBAAwB,EAAE,MAAM,4BAA4B,CAAC;AACzF,OAAO,EAAE,eAAe,IAAI,6BAA6B,EAAE,MAAM,mCAAmC,CAAC;AACrG,OAAO,EAAE,eAAe,IAAI,uBAAuB,EAAE,MAAM,4BAA4B,CAAC;AACxF,OAAO,EAAE,aAAa,IAAI,kBAAkB,EAAE,MAAM,yBAAyB,CAAC;AAC9E,OAAO,EAAE,aAAa,IAAI,qBAAqB,EAAE,MAAM,mCAAmC,CAAC;AAE3F,OAAO,EAAE,aAAa,EAAE,MAAM,6BAA6B,CAAC;AAC5D,OAAO,EAAE,WAAW,EAAE,MAAM,0BAA0B,CAAC;AACvD,OAAO,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAE7C,OAAO,EAAE,gBAAgB,EAAE,MAAM,6BAA6B,CAAC;AAC/D,OAAO,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAC;AAEzD,OAAO,EAAE,gBAAgB,EAAE,MAAM,6BAA6B,CAAC;AAE/D,OAAO,EAAE,kBAAkB,EAAE,MAAM,sCAAsC,CAAC;AAE1E,uBAAuB;AACvB,MAAM,CAAC,KAAK,UAAU,UAAU;IAC9B,2CAA2C;IAC3C,OAAO,CAAC,GAAG,CAAC,wBAAwB,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IAEvE,iBAAiB,EAAE,CAAC;IAEpB,uBAAuB;IACvB,IAAI,SAAS,CAAC,aAAa,EAAE,CAAC;QAC5B,gBAAgB,EAAE,CAAC;QACnB,OAAO;IACT,CAAC;IAED,MAAM,OAAO,GAAmB;QAC9B,cAAc,EAAE,SAAS,CAAC,cAAc;QACxC,QAAQ,EAAE,SAAS,CAAC,QAAQ;QAC5B,gBAAgB,EAAE,SAAS,CAAC,gBAAgB;QAE5C,eAAe,EAAE,SAAS,CAAC,eAAe;QAC1C,eAAe,EAAE,SAAS,CAAC,eAAe;QAC1C,oBAAoB,EAAE,SAAS,CAAC,oBAAoB;QACpD,aAAa,EAAE,SAAS,CAAC,aAAa;QAEtC,SAAS,EAAE,mBAAmB,EAAE;QAChC,MAAM,EAAE,SAAS,CAAC,MAAM;QACxB,MAAM,EAAE,SAAS,CAAC,MAAM;QAExB,0BAA0B,EAAE,SAAS,CAAC,0BAA0B;QAChE,WAAW,EAAE,SAAS,CAAC,WAAW;QAClC,KAAK,EAAE,SAAS,CAAC,KAAK;QACtB,MAAM,EAAE,SAAS,CAAC,MAAM;QACxB,OAAO,EAAE,SAAS,CAAC,aAAa;QAChC,mBAAmB,EAAE,SAAS,CAAC,mBAAmB;QAClD,eAAe,EAAE,SAAS,CAAC,eAAe;QAC1C,kBAAkB,EAAE,SAAS,CAAC,kBAAkB;QAChD,WAAW,EAAE,SAAS,CAAC,WAAW;QAClC,cAAc,EAAE,SAAS,CAAC,cAAc;QACxC,WAAW,EAAE,SAAS,CAAC,WAAW;QAClC,YAAY,EAAE,SAAS,CAAC,YAAY;QACpC,WAAW,EAAE,SAAS,CAAC,WAAW;KACnC,CAAC;IAEF,0BAA0B;IAC1B,IAAI,SAAS,CAAC,WAAW,EAAE,CAAC;QAC1B,MAAM,kBAAkB,CAAC,OAAO,CAAC,CAAC;IACpC,CAAC;SAAM,CAAC;QACN,OAAO,CAAC,GAAG,CAAC,2CAA2C,CAAC,CAAC;QACzD,MAAM,mBAAmB,CAAC,OAAO,CAAC,CAAC;IACrC,CAAC;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,mBAAmB,CAAC,OAAuB,EAAE,WAAyB;IAC1F,IAAI,QAAQ,CAAC,WAAW,IAAI,CAAC,OAAO,CAAC,kBAAkB,EAAE,CAAC;QACxD,IAAI,CAAC;YACH,OAAO,CAAC,GAAG,CAAC,2BAA2B,QAAQ,CAAC,WAAW,EAAE,CAAC,CAAC;YAC/D,MAAM,WAAW,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC;YACxC,OAAO,CAAC,GAAG,CAAC,oCAAoC,CAAC,CAAC;QACpD,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,KAA2C,CAAC;YACvE,OAAO,CAAC,GAAG,CACT,qHAAqH,CACtH,CAAC;YACF,OAAO,CAAC,GAAG,CAAC,cAAc,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC;YAC5C,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;IACH,CAAC;SAAM,IAAI,QAAQ,CAAC,WAAW,IAAI,OAAO,CAAC,kBAAkB,EAAE,CAAC;QAC9D,OAAO,CAAC,GAAG,CAAC,2BAA2B,CAAC,CAAC;IAC3C,CAAC;IAED,IAAI,WAAW,EAAE,OAAO,EAAE,CAAC;QACzB,MAAM,IAAI,KAAK,CAAC,2BAA2B,CAAC,CAAC;IAC/C,CAAC;IAED,OAAO,CAAC,GAAG,CAAC,qBAAqB,CAAC,CAAC;IACnC,MAAM,aAAa,GAAG,MAAM,aAAa,CAAC,oBAAoB,EAAE,kBAAkB,EAAE,gBAAgB,CAAC,OAAO,CAAC,CAAC,CAAC;IAE/G,OAAO,CAAC,GAAG,CAAC,0BAA0B,EAAE,aAAa,CAAC,CAAC;IAEvD,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;QACnB,OAAO,CAAC,GAAG,CAAC,kCAAkC,CAAC,CAAC;IAClD,CAAC;SAAM,CAAC;QACN,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;QAC5B,MAAM,WAAW,CACf,aAAa,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,aAAa,IAAI,IAAI,CAAC,IAAI,KAAK,eAAe,CAAC,EAC5F,OAAO,CACR,CAAC;QACF,OAAO,CAAC,GAAG,CAAC,yBAAyB,CAAC,CAAC;QAEvC,IAAI,WAAW,EAAE,OAAO,EAAE,CAAC;YACzB,MAAM,IAAI,KAAK,CAAC,iDAAiD,CAAC,CAAC;QACrE,CAAC;QAED,oDAAoD;QACpD,IAAI,QAAQ,CAAC,WAAW,EAAE,CAAC;YACzB,IAAI,CAAC;gBACH,OAAO,CAAC,GAAG,CAAC,2BAA2B,QAAQ,CAAC,WAAW,EAAE,CAAC,CAAC;gBAC/D,MAAM,WAAW,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC;gBACxC,OAAO,CAAC,GAAG,CAAC,oCAAoC,CAAC,CAAC;YACpD,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO,CAAC,GAAG,CAAC,kDAAkD,CAAC,CAAC;gBAEhE,oDAAoD;gBACpD,MAAM,cAAc,GAAG,KAA2C,CAAC;gBACnE,MAAM,eAAe,GAAG,gBAAgB,CACtC,QAAQ,CAAC,WAAW,EACpB,OAAO,EACP,cAAc,CAAC,MAAM,EACrB,cAAc,CAAC,MAAM,CACtB,CAAC;gBAEF,OAAO,CAAC,GAAG,CAAC,oCAAoC,CAAC,CAAC;gBAClD,MAAM,oBAAoB,GAAG,CAAC,MAAM,aAAa,CAAC,oBAAoB,EAAE,kBAAkB,EAAE;oBAC1F,MAAM,EAAE,eAAe;oBACvB,OAAO,EAAE,EAAE,GAAG,OAAO,EAAE,gBAAgB,EAAE,IAAI,EAAE;iBAChD,CAAC,CAAmB,CAAC;gBAEtB,OAAO,CAAC,GAAG,CAAC,yCAAyC,EAAE,oBAAoB,CAAC,CAAC;gBAE7E,OAAO,CAAC,GAAG,CAAC,qBAAqB,CAAC,CAAC;gBACnC,WAAW,CACT,oBAAoB,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,aAAa,IAAI,IAAI,CAAC,IAAI,KAAK,eAAe,CAAC,EACnG,OAAO,CACR,CAAC;gBAEF,IAAI,WAAW,EAAE,OAAO,EAAE,CAAC;oBACzB,MAAM,IAAI,KAAK,CAAC,4CAA4C,CAAC,CAAC;gBAChE,CAAC;gBAED,yCAAyC;gBACzC,IAAI,CAAC;oBACH,OAAO,CAAC,GAAG,CAAC,4BAA4B,QAAQ,CAAC,WAAW,EAAE,CAAC,CAAC;oBAChE,MAAM,WAAW,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC;oBACxC,OAAO,CAAC,GAAG,CAAC,gDAAgD,CAAC,CAAC;gBAChE,CAAC;gBAAC,OAAO,eAAe,EAAE,CAAC;oBACzB,MAAM,KAAK,GAAG,eAAqD,CAAC;oBACpE,OAAO,CAAC,GAAG,CAAC,8EAA8E,CAAC,CAAC;oBAC5F,OAAO,CAAC,GAAG,CAAC,cAAc,EAAE,KAAK,CAAC,MAAM,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;gBAC1D,CAAC;YACH,CAAC;QACH,CAAC;QAED,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;IACvB,CAAC;AACH,CAAC;AAED,8BAA8B;AAE9B,MAAM,WAAW,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;AAEzC,MAAM,oBAAoB,GAAmD;IAC3E,kBAAkB,EAAE,6BAA6B;IACjD,WAAW,EAAE,uBAAuB;IACpC,WAAW,EAAE,uBAAuB;IACpC,SAAS,EAAE,wBAAwB;IACnC,UAAU,EAAE,kBAAkB;CACtB,CAAC;AAEX,MAAM,kBAAkB,GAA8C;IACpE,QAAQ,EAAE,kBAAkB;IAC5B,WAAW,EAAE,qBAAqB;CAC1B,CAAC;AAEX,SAAS,mBAAmB;IAC1B,MAAM,MAAM,GAAG,SAAS,CAAC,QAAQ;QAC/B,CAAC,CAAC,WAAW;QACb,CAAC,CAAC,SAAS,CAAC,QAAQ;YAClB,CAAC,CAAC,WAAW;YACb,CAAC,CAAC,SAAS,CAAC,cAAc;gBACxB,CAAC,CAAC,kBAAkB;gBACpB,CAAC,CAAC,SAAS,CAAC,OAAO;oBACjB,CAAC,CAAC,UAAU;oBACZ,CAAC,CAAC,SAAS,CAAC,SAAS;wBACnB,CAAC,CAAC,WAAW;wBACb,CAAC,CAAC,SAAS,CAAC;IACtB,MAAM,CAAC,MAAM,EAAE,gDAAgD,CAAC,CAAC;IACjE,OAAO,MAAM,CAAC;AAChB,CAAC"}
|
|
@@ -0,0 +1,273 @@
|
|
|
1
|
+
import { describe, it, expect, beforeEach, vi } from 'vitest';
|
|
2
|
+
import { runCodegen } from './codegen.js';
|
|
3
|
+
import * as vertexAi from '../ai-service/vertex-ai.js';
|
|
4
|
+
import * as chatGpt from '../ai-service/chat-gpt.js';
|
|
5
|
+
import * as anthropic from '../ai-service/anthropic.js';
|
|
6
|
+
import * as vertexAiClaude from '../ai-service/vertex-ai-claude.js';
|
|
7
|
+
import * as updateFiles from '../files/update-files.js';
|
|
8
|
+
import '../files/find-files.js';
|
|
9
|
+
import * as cliParams from '../cli/cli-params.js';
|
|
10
|
+
import * as cliOptions from '../cli/cli-options.js';
|
|
11
|
+
import * as vertexAiImagen from '../ai-service/vertex-ai-imagen.js';
|
|
12
|
+
import * as dallE from '../ai-service/dall-e.js';
|
|
13
|
+
import './config.js';
|
|
14
|
+
vi.mock('../ai-service/vertex-ai.js', () => ({ generateContent: vi.fn() }));
|
|
15
|
+
vi.mock('../ai-service/chat-gpt.js', () => ({ generateContent: vi.fn() }));
|
|
16
|
+
vi.mock('../ai-service/anthropic.js', () => ({ generateContent: vi.fn() }));
|
|
17
|
+
vi.mock('../ai-service/vertex-ai-claude.js', () => ({ generateContent: vi.fn() }));
|
|
18
|
+
vi.mock('../files/update-files.js');
|
|
19
|
+
vi.mock('../cli/cli-params.js', () => ({
|
|
20
|
+
interactive: false,
|
|
21
|
+
requireExplanations: false,
|
|
22
|
+
considerAllFiles: false,
|
|
23
|
+
dependencyTree: false,
|
|
24
|
+
explicitPrompt: false,
|
|
25
|
+
allowFileCreate: false,
|
|
26
|
+
allowFileDelete: false,
|
|
27
|
+
allowDirectoryCreate: false,
|
|
28
|
+
allowFileMove: false,
|
|
29
|
+
verbosePrompt: false,
|
|
30
|
+
vertexAiClaude: false,
|
|
31
|
+
helpRequested: false,
|
|
32
|
+
aiStudio: false,
|
|
33
|
+
vision: false,
|
|
34
|
+
imagen: false,
|
|
35
|
+
temperature: 0.7,
|
|
36
|
+
cheap: false,
|
|
37
|
+
taskFile: undefined,
|
|
38
|
+
disableInitialLint: undefined,
|
|
39
|
+
askQuestion: false,
|
|
40
|
+
disableContextOptimization: undefined,
|
|
41
|
+
geminiBlockNone: undefined,
|
|
42
|
+
contentMask: undefined,
|
|
43
|
+
ignorePatterns: [],
|
|
44
|
+
disableCache: undefined,
|
|
45
|
+
}));
|
|
46
|
+
vi.mock('../files/find-files.js', () => ({
|
|
47
|
+
getSourceFiles: () => [],
|
|
48
|
+
getImageAssetFiles: () => [],
|
|
49
|
+
}));
|
|
50
|
+
vi.mock('../cli/cli-options.js', () => ({
|
|
51
|
+
printHelpMessage: vi.fn(),
|
|
52
|
+
}));
|
|
53
|
+
vi.mock('../ai-service/vertex-ai-imagen.js', () => ({ generateImage: vi.fn() }));
|
|
54
|
+
vi.mock('../ai-service/dall-e.js', () => ({ generateImage: vi.fn() }));
|
|
55
|
+
vi.mock('./config.js', () => ({
|
|
56
|
+
rootDir: '/mocked/root/dir',
|
|
57
|
+
rcConfig: {
|
|
58
|
+
rootDir: '.',
|
|
59
|
+
extensions: ['.js', '.ts', '.tsx', '.jsx'],
|
|
60
|
+
},
|
|
61
|
+
importantContext: {},
|
|
62
|
+
}));
|
|
63
|
+
describe('runCodegen', () => {
|
|
64
|
+
beforeEach(() => {
|
|
65
|
+
vi.resetAllMocks();
|
|
66
|
+
vi.mocked(cliParams).anthropic = false;
|
|
67
|
+
vi.mocked(cliParams).chatGpt = false;
|
|
68
|
+
vi.mocked(cliParams).vertexAi = false;
|
|
69
|
+
vi.mocked(cliParams).vertexAiClaude = false;
|
|
70
|
+
vi.mocked(cliParams).dryRun = false;
|
|
71
|
+
vi.mocked(cliParams).helpRequested = false;
|
|
72
|
+
vi.mocked(cliParams).vision = false;
|
|
73
|
+
vi.mocked(cliParams).imagen = undefined;
|
|
74
|
+
});
|
|
75
|
+
it('should run codegen with Vertex AI by default', async () => {
|
|
76
|
+
vi.mocked(cliParams).vertexAi = true;
|
|
77
|
+
const mockFunctionCalls = [
|
|
78
|
+
{ name: 'updateFile', args: { filePath: 'test.js', newContent: 'console.log("Hello");' } },
|
|
79
|
+
];
|
|
80
|
+
vi.mocked(vertexAi.generateContent).mockResolvedValueOnce(mockFunctionCalls);
|
|
81
|
+
await runCodegen();
|
|
82
|
+
expect(vertexAi.generateContent).toHaveBeenCalled();
|
|
83
|
+
expect(updateFiles.updateFiles).toHaveBeenCalledWith(mockFunctionCalls, expect.anything());
|
|
84
|
+
});
|
|
85
|
+
it('should run codegen with ChatGPT when chatGpt flag is true', async () => {
|
|
86
|
+
vi.mocked(cliParams).chatGpt = true;
|
|
87
|
+
const mockFunctionCalls = [{ name: 'createFile', args: { filePath: 'new.js', newContent: 'const x = 5;' } }];
|
|
88
|
+
vi.mocked(chatGpt).generateContent.mockResolvedValueOnce(mockFunctionCalls);
|
|
89
|
+
await runCodegen();
|
|
90
|
+
expect(chatGpt.generateContent).toHaveBeenCalled();
|
|
91
|
+
expect(updateFiles.updateFiles).toHaveBeenCalledWith(mockFunctionCalls, expect.anything());
|
|
92
|
+
});
|
|
93
|
+
it('should run codegen with Anthropic when anthropic flag is true', async () => {
|
|
94
|
+
vi.mocked(cliParams).anthropic = true;
|
|
95
|
+
const mockFunctionCalls = [{ name: 'deleteFile', args: { filePath: 'obsolete.js' } }];
|
|
96
|
+
vi.mocked(anthropic).generateContent.mockResolvedValueOnce(mockFunctionCalls);
|
|
97
|
+
await runCodegen();
|
|
98
|
+
expect(anthropic.generateContent).toHaveBeenCalled();
|
|
99
|
+
expect(updateFiles.updateFiles).toHaveBeenCalledWith(mockFunctionCalls, expect.anything());
|
|
100
|
+
});
|
|
101
|
+
it('should not update files in dry run mode', async () => {
|
|
102
|
+
vi.mocked(cliParams).vertexAi = true;
|
|
103
|
+
vi.mocked(cliParams).dryRun = true;
|
|
104
|
+
const mockFunctionCalls = [
|
|
105
|
+
{ name: 'updateFile', args: { filePath: 'test.js', newContent: 'console.log("Dry run");' } },
|
|
106
|
+
];
|
|
107
|
+
vi.mocked(vertexAi.generateContent).mockResolvedValueOnce(mockFunctionCalls);
|
|
108
|
+
await runCodegen();
|
|
109
|
+
expect(vertexAi.generateContent).toHaveBeenCalled();
|
|
110
|
+
expect(updateFiles.updateFiles).not.toHaveBeenCalled();
|
|
111
|
+
});
|
|
112
|
+
it('should run codegen with Vertex AI Claude when vertexAiClaude flag is true', async () => {
|
|
113
|
+
vi.mocked(cliParams).vertexAiClaude = true;
|
|
114
|
+
const mockFunctionCalls = [
|
|
115
|
+
{ name: 'updateFile', args: { filePath: 'test.js', newContent: 'console.log("Hello from Claude");' } },
|
|
116
|
+
];
|
|
117
|
+
vi.mocked(vertexAiClaude).generateContent.mockResolvedValueOnce(mockFunctionCalls);
|
|
118
|
+
await runCodegen();
|
|
119
|
+
expect(vertexAiClaude.generateContent).toHaveBeenCalled();
|
|
120
|
+
expect(updateFiles.updateFiles).toHaveBeenCalledWith(mockFunctionCalls, expect.anything());
|
|
121
|
+
});
|
|
122
|
+
it('should pass the temperature parameter to the AI service', async () => {
|
|
123
|
+
vi.mocked(cliParams).vertexAi = true;
|
|
124
|
+
vi.mocked(cliParams).temperature = 0.5;
|
|
125
|
+
const mockFunctionCalls = [
|
|
126
|
+
{ name: 'updateFile', args: { filePath: 'test.js', newContent: 'console.log("Temperature test");' } },
|
|
127
|
+
];
|
|
128
|
+
vi.mocked(vertexAi.generateContent).mockResolvedValueOnce(mockFunctionCalls);
|
|
129
|
+
await runCodegen();
|
|
130
|
+
expect(vertexAi.generateContent).toHaveBeenCalledWith(expect.anything(), expect.anything(), expect.anything(), 0.5, false, expect.anything());
|
|
131
|
+
expect(updateFiles.updateFiles).toHaveBeenCalledWith(mockFunctionCalls, expect.anything());
|
|
132
|
+
});
|
|
133
|
+
it('should print help message and not run codegen when --help option is provided', async () => {
|
|
134
|
+
vi.mocked(cliParams).helpRequested = true;
|
|
135
|
+
await runCodegen();
|
|
136
|
+
expect(cliOptions.printHelpMessage).toHaveBeenCalled();
|
|
137
|
+
expect(vertexAi.generateContent).not.toHaveBeenCalled();
|
|
138
|
+
expect(chatGpt.generateContent).not.toHaveBeenCalled();
|
|
139
|
+
expect(anthropic.generateContent).not.toHaveBeenCalled();
|
|
140
|
+
expect(vertexAiClaude.generateContent).not.toHaveBeenCalled();
|
|
141
|
+
expect(updateFiles.updateFiles).not.toHaveBeenCalled();
|
|
142
|
+
});
|
|
143
|
+
it('should run codegen with vision when vision flag is true', async () => {
|
|
144
|
+
vi.mocked(cliParams).chatGpt = true;
|
|
145
|
+
vi.mocked(cliParams).vision = true;
|
|
146
|
+
const mockFunctionCalls = [
|
|
147
|
+
{ name: 'updateFile', args: { filePath: 'test.js', newContent: 'console.log("Vision test");' } },
|
|
148
|
+
];
|
|
149
|
+
vi.mocked(chatGpt).generateContent.mockResolvedValueOnce(mockFunctionCalls);
|
|
150
|
+
await runCodegen();
|
|
151
|
+
expect(chatGpt.generateContent).toHaveBeenCalled();
|
|
152
|
+
expect(updateFiles.updateFiles).toHaveBeenCalledWith(mockFunctionCalls, expect.anything());
|
|
153
|
+
// Check if the generateContent function was called with the correct parameters
|
|
154
|
+
expect(vi.mocked(chatGpt).generateContent.mock.calls[0][0]).toEqual(expect.arrayContaining([
|
|
155
|
+
expect.objectContaining({
|
|
156
|
+
type: 'user',
|
|
157
|
+
text: expect.stringContaining('I should also provide you with a summary of application image assets'),
|
|
158
|
+
}),
|
|
159
|
+
expect.objectContaining({
|
|
160
|
+
type: 'assistant',
|
|
161
|
+
text: expect.stringContaining('Please provide summary of application image assets.'),
|
|
162
|
+
}),
|
|
163
|
+
expect.objectContaining({
|
|
164
|
+
type: 'user',
|
|
165
|
+
functionResponses: [{ name: 'getImageAssets', content: expect.any(String) }],
|
|
166
|
+
}),
|
|
167
|
+
]));
|
|
168
|
+
});
|
|
169
|
+
it('should use Vertex AI Imagen when imagen flag is set to vertex-ai', async () => {
|
|
170
|
+
vi.mocked(cliParams).imagen = 'vertex-ai';
|
|
171
|
+
vi.mocked(cliParams).vertexAi = true;
|
|
172
|
+
const mockCodegenSummary = [
|
|
173
|
+
{
|
|
174
|
+
name: 'codegenSummary',
|
|
175
|
+
args: {
|
|
176
|
+
fileUpdates: [{ path: 'landscape.png', updateToolName: 'generateImage' }],
|
|
177
|
+
contextPaths: [],
|
|
178
|
+
explanation: 'Mock summary with image generation failure',
|
|
179
|
+
},
|
|
180
|
+
},
|
|
181
|
+
];
|
|
182
|
+
const mockFunctionCalls = [
|
|
183
|
+
{
|
|
184
|
+
name: 'generateImage',
|
|
185
|
+
args: { prompt: 'A beautiful landscape', filePath: 'landscape.png', width: 512, height: 512 },
|
|
186
|
+
},
|
|
187
|
+
];
|
|
188
|
+
vi.mocked(vertexAi.generateContent).mockResolvedValueOnce(mockCodegenSummary);
|
|
189
|
+
vi.mocked(vertexAi.generateContent).mockResolvedValueOnce(mockFunctionCalls);
|
|
190
|
+
vi.mocked(vertexAiImagen).generateImage.mockResolvedValueOnce('mocked-image-data');
|
|
191
|
+
await runCodegen();
|
|
192
|
+
expect(vertexAi.generateContent).toHaveBeenCalled();
|
|
193
|
+
expect(vertexAiImagen.generateImage).toHaveBeenCalledWith('A beautiful landscape', undefined, {
|
|
194
|
+
height: 512,
|
|
195
|
+
width: 512,
|
|
196
|
+
}, false);
|
|
197
|
+
expect(updateFiles.updateFiles).toHaveBeenCalledWith(mockFunctionCalls, expect.anything());
|
|
198
|
+
});
|
|
199
|
+
it('should use DALL-E when imagen flag is set to dall-e', async () => {
|
|
200
|
+
vi.mocked(cliParams).imagen = 'dall-e';
|
|
201
|
+
vi.mocked(cliParams).chatGpt = true;
|
|
202
|
+
const mockCodegenSummary = [
|
|
203
|
+
{
|
|
204
|
+
name: 'codegenSummary',
|
|
205
|
+
args: {
|
|
206
|
+
fileUpdates: [{ path: 'city.png', updateToolName: 'generateImage' }],
|
|
207
|
+
contextPaths: [],
|
|
208
|
+
explanation: 'Mock summary with image generation failure',
|
|
209
|
+
},
|
|
210
|
+
},
|
|
211
|
+
];
|
|
212
|
+
const mockFunctionCalls = [
|
|
213
|
+
{ name: 'generateImage', args: { prompt: 'A futuristic city', filePath: 'city.png', width: 1024, height: 1024 } },
|
|
214
|
+
];
|
|
215
|
+
vi.mocked(chatGpt).generateContent.mockResolvedValueOnce(mockCodegenSummary);
|
|
216
|
+
vi.mocked(chatGpt).generateContent.mockResolvedValueOnce(mockFunctionCalls);
|
|
217
|
+
vi.mocked(dallE).generateImage.mockResolvedValueOnce('mocked-image-data');
|
|
218
|
+
await runCodegen();
|
|
219
|
+
expect(chatGpt.generateContent).toHaveBeenCalled();
|
|
220
|
+
expect(dallE.generateImage).toHaveBeenCalledWith('A futuristic city', undefined, { width: 1024, height: 1024 }, false);
|
|
221
|
+
expect(updateFiles.updateFiles).toHaveBeenCalledWith(mockFunctionCalls, expect.anything());
|
|
222
|
+
});
|
|
223
|
+
it('should throw an error when imagen flag is set but no AI service is specified', async () => {
|
|
224
|
+
vi.mocked(cliParams).imagen = 'vertex-ai';
|
|
225
|
+
await expect(runCodegen()).rejects.toThrow('Please specify which AI service should be used');
|
|
226
|
+
});
|
|
227
|
+
it('should pass the cheap parameter to the AI service when --cheap flag is true', async () => {
|
|
228
|
+
vi.mocked(cliParams).vertexAi = true;
|
|
229
|
+
vi.mocked(cliParams).cheap = true;
|
|
230
|
+
const mockFunctionCalls = [
|
|
231
|
+
{ name: 'updateFile', args: { filePath: 'test.js', newContent: 'console.log("Cheap test");' } },
|
|
232
|
+
];
|
|
233
|
+
vi.mocked(vertexAi.generateContent).mockResolvedValueOnce(mockFunctionCalls);
|
|
234
|
+
await runCodegen();
|
|
235
|
+
expect(vertexAi.generateContent).toHaveBeenCalledWith(expect.anything(), expect.anything(), expect.anything(), expect.anything(), true, expect.anything());
|
|
236
|
+
expect(updateFiles.updateFiles).toHaveBeenCalledWith(mockFunctionCalls, expect.anything());
|
|
237
|
+
});
|
|
238
|
+
it('should pass the cheap parameter to the image generation service when --cheap flag is true', async () => {
|
|
239
|
+
vi.mocked(cliParams).imagen = 'vertex-ai';
|
|
240
|
+
vi.mocked(cliParams).vertexAi = true;
|
|
241
|
+
vi.mocked(cliParams).cheap = true;
|
|
242
|
+
const mockCodegenSummary = [
|
|
243
|
+
{
|
|
244
|
+
name: 'codegenSummary',
|
|
245
|
+
args: {
|
|
246
|
+
fileUpdates: [{ path: 'landscape.png', updateToolName: 'generateImage' }],
|
|
247
|
+
contextPaths: [],
|
|
248
|
+
explanation: 'Mock summary with cheap image generation',
|
|
249
|
+
},
|
|
250
|
+
},
|
|
251
|
+
];
|
|
252
|
+
const mockFunctionCalls = [
|
|
253
|
+
{
|
|
254
|
+
name: 'generateImage',
|
|
255
|
+
args: {
|
|
256
|
+
prompt: 'A simple landscape',
|
|
257
|
+
filePath: 'landscape.png',
|
|
258
|
+
width: 256,
|
|
259
|
+
height: 256,
|
|
260
|
+
cheap: true,
|
|
261
|
+
},
|
|
262
|
+
},
|
|
263
|
+
];
|
|
264
|
+
vi.mocked(vertexAi.generateContent).mockResolvedValueOnce(mockCodegenSummary);
|
|
265
|
+
vi.mocked(vertexAi.generateContent).mockResolvedValueOnce(mockFunctionCalls);
|
|
266
|
+
vi.mocked(vertexAiImagen).generateImage.mockResolvedValueOnce('mocked-cheap-image-data');
|
|
267
|
+
await runCodegen();
|
|
268
|
+
expect(vertexAi.generateContent).toHaveBeenCalledWith(expect.anything(), expect.anything(), expect.anything(), expect.anything(), true, expect.anything());
|
|
269
|
+
expect(vertexAiImagen.generateImage).toHaveBeenCalledWith('A simple landscape', undefined, { width: 256, height: 256 }, true);
|
|
270
|
+
expect(updateFiles.updateFiles).toHaveBeenCalledWith(mockFunctionCalls, expect.anything());
|
|
271
|
+
});
|
|
272
|
+
});
|
|
273
|
+
//# sourceMappingURL=codegen.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"codegen.test.js","sourceRoot":"","sources":["../../src/main/codegen.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EAAE,UAAU,EAAE,MAAM,cAAc,CAAC;AAC1C,OAAO,KAAK,QAAQ,MAAM,4BAA4B,CAAC;AACvD,OAAO,KAAK,OAAO,MAAM,2BAA2B,CAAC;AACrD,OAAO,KAAK,SAAS,MAAM,4BAA4B,CAAC;AACxD,OAAO,KAAK,cAAc,MAAM,mCAAmC,CAAC;AACpE,OAAO,KAAK,WAAW,MAAM,0BAA0B,CAAC;AACxD,OAAO,wBAAwB,CAAC;AAChC,OAAO,KAAK,SAAS,MAAM,sBAAsB,CAAC;AAClD,OAAO,KAAK,UAAU,MAAM,uBAAuB,CAAC;AACpD,OAAO,KAAK,cAAc,MAAM,mCAAmC,CAAC;AACpE,OAAO,KAAK,KAAK,MAAM,yBAAyB,CAAC;AACjD,OAAO,aAAa,CAAC;AAErB,EAAE,CAAC,IAAI,CAAC,4BAA4B,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;AAC5E,EAAE,CAAC,IAAI,CAAC,2BAA2B,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;AAC3E,EAAE,CAAC,IAAI,CAAC,4BAA4B,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;AAC5E,EAAE,CAAC,IAAI,CAAC,mCAAmC,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;AACnF,EAAE,CAAC,IAAI,CAAC,0BAA0B,CAAC,CAAC;AACpC,EAAE,CAAC,IAAI,CAAC,sBAAsB,EAAE,GAAG,EAAE,CAAC,CAAC;IACrC,WAAW,EAAE,KAAK;IAClB,mBAAmB,EAAE,KAAK;IAC1B,gBAAgB,EAAE,KAAK;IACvB,cAAc,EAAE,KAAK;IACrB,cAAc,EAAE,KAAK;IACrB,eAAe,EAAE,KAAK;IACtB,eAAe,EAAE,KAAK;IACtB,oBAAoB,EAAE,KAAK;IAC3B,aAAa,EAAE,KAAK;IACpB,aAAa,EAAE,KAAK;IACpB,cAAc,EAAE,KAAK;IACrB,aAAa,EAAE,KAAK;IACpB,QAAQ,EAAE,KAAK;IACf,MAAM,EAAE,KAAK;IACb,MAAM,EAAE,KAAK;IACb,WAAW,EAAE,GAAG;IAChB,KAAK,EAAE,KAAK;IACZ,QAAQ,EAAE,SAAS;IACnB,kBAAkB,EAAE,SAAS;IAC7B,WAAW,EAAE,KAAK;IAClB,0BAA0B,EAAE,SAAS;IACrC,eAAe,EAAE,SAAS;IAC1B,WAAW,EAAE,SAAS;IACtB,cAAc,EAAE,EAAE;IAClB,YAAY,EAAE,SAAS;CACxB,CAAC,CAAC,CAAC;AACJ,EAAE,CAAC,IAAI,CAAC,wBAAwB,EAAE,GAAG,EAAE,CAAC,CAAC;IACvC,cAAc,EAAE,GAAG,EAAE,CAAC,EAAE;IACxB,kBAAkB,EAAE,GAAG,EAAE,CAAC,EAAE;CAC7B,CAAC,CAAC,CAAC;AACJ,EAAE,CAAC,IAAI,CAAC,uBAAuB,EAAE,GAAG,EAAE,CAAC,CAAC;IACtC,gBAAgB,EAAE,EAAE,CAAC,EAAE,EAAE;CAC1B,CAAC,CAAC,CAAC;AACJ,EAAE,CAAC,IAAI,CAAC,mCAAmC,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,aAAa,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;AACjF,EAAE,CAAC,IAAI,CAAC,yBAAyB,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,aAAa,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;AACvE,EAAE,CAAC,IAAI,CAAC,aAAa,EAAE,GAAG,EAAE,CAAC,CAAC;IAC5B,OAAO,EAAE,kBAAkB;IAC3B,QAAQ,EAAE;QACR,OAAO,EAAE,GAAG;QACZ,UAAU,EAAE,CAAC,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC;KAC3C;IACD,gBAAgB,EAAE,EAAE;CACrB,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;IAC1B,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,GAAG,KAAK,CAAC;QACvC,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,OAAO,GAAG,KAAK,CAAC;QACrC,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,QAAQ,GAAG,KAAK,CAAC;QACtC,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,cAAc,GAAG,KAAK,CAAC;QAC5C,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,MAAM,GAAG,KAAK,CAAC;QACpC,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,aAAa,GAAG,KAAK,CAAC;QAC3C,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,MAAM,GAAG,KAAK,CAAC;QACpC,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,MAAM,GAAG,SAAS,CAAC;IAC1C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;QAC5D,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,QAAQ,GAAG,IAAI,CAAC;QAErC,MAAM,iBAAiB,GAAG;YACxB,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,EAAE,QAAQ,EAAE,SAAS,EAAE,UAAU,EAAE,uBAAuB,EAAE,EAAE;SAC3F,CAAC;QACF,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,iBAAiB,CAAC,CAAC;QAE7E,MAAM,UAAU,EAAE,CAAC;QAEnB,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACpD,MAAM,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAAC,iBAAiB,EAAE,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC;IAC7F,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;QACzE,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,OAAO,GAAG,IAAI,CAAC;QAEpC,MAAM,iBAAiB,GAAG,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,EAAE,QAAQ,EAAE,QAAQ,EAAE,UAAU,EAAE,cAAc,EAAE,EAAE,CAAC,CAAC;QAC7G,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,eAAe,CAAC,qBAAqB,CAAC,iBAAiB,CAAC,CAAC;QAE5E,MAAM,UAAU,EAAE,CAAC;QAEnB,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACnD,MAAM,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAAC,iBAAiB,EAAE,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC;IAC7F,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;QAC7E,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,GAAG,IAAI,CAAC;QAEtC,MAAM,iBAAiB,GAAG,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,EAAE,QAAQ,EAAE,aAAa,EAAE,EAAE,CAAC,CAAC;QACtF,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,eAAe,CAAC,qBAAqB,CAAC,iBAAiB,CAAC,CAAC;QAE9E,MAAM,UAAU,EAAE,CAAC;QAEnB,MAAM,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACrD,MAAM,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAAC,iBAAiB,EAAE,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC;IAC7F,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;QACvD,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,QAAQ,GAAG,IAAI,CAAC;QACrC,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,MAAM,GAAG,IAAI,CAAC;QAEnC,MAAM,iBAAiB,GAAG;YACxB,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,EAAE,QAAQ,EAAE,SAAS,EAAE,UAAU,EAAE,yBAAyB,EAAE,EAAE;SAC7F,CAAC;QACF,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,iBAAiB,CAAC,CAAC;QAE7E,MAAM,UAAU,EAAE,CAAC;QAEnB,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACpD,MAAM,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IACzD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2EAA2E,EAAE,KAAK,IAAI,EAAE;QACzF,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,cAAc,GAAG,IAAI,CAAC;QAE3C,MAAM,iBAAiB,GAAG;YACxB,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,EAAE,QAAQ,EAAE,SAAS,EAAE,UAAU,EAAE,mCAAmC,EAAE,EAAE;SACvG,CAAC;QACF,EAAE,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,eAAe,CAAC,qBAAqB,CAAC,iBAAiB,CAAC,CAAC;QAEnF,MAAM,UAAU,EAAE,CAAC;QAEnB,MAAM,CAAC,cAAc,CAAC,eAAe,CAAC,CAAC,gBAAgB,EAAE,CAAC;QAC1D,MAAM,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAAC,iBAAiB,EAAE,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC;IAC7F,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;QACvE,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,QAAQ,GAAG,IAAI,CAAC;QACrC,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,WAAW,GAAG,GAAG,CAAC;QAEvC,MAAM,iBAAiB,GAAG;YACxB,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,EAAE,QAAQ,EAAE,SAAS,EAAE,UAAU,EAAE,kCAAkC,EAAE,EAAE;SACtG,CAAC;QACF,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,iBAAiB,CAAC,CAAC;QAE7E,MAAM,UAAU,EAAE,CAAC;QAEnB,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,oBAAoB,CACnD,MAAM,CAAC,QAAQ,EAAE,EACjB,MAAM,CAAC,QAAQ,EAAE,EACjB,MAAM,CAAC,QAAQ,EAAE,EACjB,GAAG,EACH,KAAK,EACL,MAAM,CAAC,QAAQ,EAAE,CAClB,CAAC;QACF,MAAM,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAAC,iBAAiB,EAAE,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC;IAC7F,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8EAA8E,EAAE,KAAK,IAAI,EAAE;QAC5F,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,aAAa,GAAG,IAAI,CAAC;QAE1C,MAAM,UAAU,EAAE,CAAC;QAEnB,MAAM,CAAC,UAAU,CAAC,gBAAgB,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACvD,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACxD,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACvD,MAAM,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACzD,MAAM,CAAC,cAAc,CAAC,eAAe,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC9D,MAAM,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IACzD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;QACvE,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,OAAO,GAAG,IAAI,CAAC;QACpC,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,MAAM,GAAG,IAAI,CAAC;QAEnC,MAAM,iBAAiB,GAAG;YACxB,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,EAAE,QAAQ,EAAE,SAAS,EAAE,UAAU,EAAE,6BAA6B,EAAE,EAAE;SACjG,CAAC;QACF,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,eAAe,CAAC,qBAAqB,CAAC,iBAAiB,CAAC,CAAC;QAE5E,MAAM,UAAU,EAAE,CAAC;QAEnB,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACnD,MAAM,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAAC,iBAAiB,EAAE,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC;QAC3F,+EAA+E;QAC/E,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CACjE,MAAM,CAAC,eAAe,CAAC;YACrB,MAAM,CAAC,gBAAgB,CAAC;gBACtB,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,MAAM,CAAC,gBAAgB,CAAC,sEAAsE,CAAC;aACtG,CAAC;YACF,MAAM,CAAC,gBAAgB,CAAC;gBACtB,IAAI,EAAE,WAAW;gBACjB,IAAI,EAAE,MAAM,CAAC,gBAAgB,CAAC,qDAAqD,CAAC;aACrF,CAAC;YACF,MAAM,CAAC,gBAAgB,CAAC;gBACtB,IAAI,EAAE,MAAM;gBACZ,iBAAiB,EAAE,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,OAAO,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC;aAC7E,CAAC;SACH,CAAC,CACH,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kEAAkE,EAAE,KAAK,IAAI,EAAE;QAChF,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,MAAM,GAAG,WAAW,CAAC;QAC1C,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,QAAQ,GAAG,IAAI,CAAC;QAErC,MAAM,kBAAkB,GAAG;YACzB;gBACE,IAAI,EAAE,gBAAgB;gBACtB,IAAI,EAAE;oBACJ,WAAW,EAAE,CAAC,EAAE,IAAI,EAAE,eAAe,EAAE,cAAc,EAAE,eAAe,EAAE,CAAC;oBACzE,YAAY,EAAE,EAAE;oBAChB,WAAW,EAAE,4CAA4C;iBAC1D;aACF;SACF,CAAC;QACF,MAAM,iBAAiB,GAAG;YACxB;gBACE,IAAI,EAAE,eAAe;gBACrB,IAAI,EAAE,EAAE,MAAM,EAAE,uBAAuB,EAAE,QAAQ,EAAE,eAAe,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE;aAC9F;SACF,CAAC;QACF,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,kBAAkB,CAAC,CAAC;QAC9E,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,iBAAiB,CAAC,CAAC;QAC7E,EAAE,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,aAAa,CAAC,qBAAqB,CAAC,mBAAmB,CAAC,CAAC;QAEnF,MAAM,UAAU,EAAE,CAAC;QAEnB,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACpD,MAAM,CAAC,cAAc,CAAC,aAAa,CAAC,CAAC,oBAAoB,CACvD,uBAAuB,EACvB,SAAS,EACT;YACE,MAAM,EAAE,GAAG;YACX,KAAK,EAAE,GAAG;SACX,EACD,KAAK,CACN,CAAC;QACF,MAAM,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAAC,iBAAiB,EAAE,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC;IAC7F,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;QACnE,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,MAAM,GAAG,QAAQ,CAAC;QACvC,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,OAAO,GAAG,IAAI,CAAC;QAEpC,MAAM,kBAAkB,GAAG;YACzB;gBACE,IAAI,EAAE,gBAAgB;gBACtB,IAAI,EAAE;oBACJ,WAAW,EAAE,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,cAAc,EAAE,eAAe,EAAE,CAAC;oBACpE,YAAY,EAAE,EAAE;oBAChB,WAAW,EAAE,4CAA4C;iBAC1D;aACF;SACF,CAAC;QACF,MAAM,iBAAiB,GAAG;YACxB,EAAE,IAAI,EAAE,eAAe,EAAE,IAAI,EAAE,EAAE,MAAM,EAAE,mBAAmB,EAAE,QAAQ,EAAE,UAAU,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE;SAClH,CAAC;QACF,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,eAAe,CAAC,qBAAqB,CAAC,kBAAkB,CAAC,CAAC;QAC7E,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,eAAe,CAAC,qBAAqB,CAAC,iBAAiB,CAAC,CAAC;QAC5E,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,aAAa,CAAC,qBAAqB,CAAC,mBAAmB,CAAC,CAAC;QAE1E,MAAM,UAAU,EAAE,CAAC;QAEnB,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACnD,MAAM,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC,oBAAoB,CAC9C,mBAAmB,EACnB,SAAS,EACT,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,EAC7B,KAAK,CACN,CAAC;QACF,MAAM,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAAC,iBAAiB,EAAE,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC;IAC7F,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8EAA8E,EAAE,KAAK,IAAI,EAAE;QAC5F,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,MAAM,GAAG,WAAW,CAAC;QAE1C,MAAM,MAAM,CAAC,UAAU,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,gDAAgD,CAAC,CAAC;IAC/F,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6EAA6E,EAAE,KAAK,IAAI,EAAE;QAC3F,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,QAAQ,GAAG,IAAI,CAAC;QACrC,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,KAAK,GAAG,IAAI,CAAC;QAElC,MAAM,iBAAiB,GAAG;YACxB,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,EAAE,QAAQ,EAAE,SAAS,EAAE,UAAU,EAAE,4BAA4B,EAAE,EAAE;SAChG,CAAC;QACF,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,iBAAiB,CAAC,CAAC;QAE7E,MAAM,UAAU,EAAE,CAAC;QAEnB,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,oBAAoB,CACnD,MAAM,CAAC,QAAQ,EAAE,EACjB,MAAM,CAAC,QAAQ,EAAE,EACjB,MAAM,CAAC,QAAQ,EAAE,EACjB,MAAM,CAAC,QAAQ,EAAE,EACjB,IAAI,EACJ,MAAM,CAAC,QAAQ,EAAE,CAClB,CAAC;QACF,MAAM,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAAC,iBAAiB,EAAE,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC;IAC7F,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2FAA2F,EAAE,KAAK,IAAI,EAAE;QACzG,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,MAAM,GAAG,WAAW,CAAC;QAC1C,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,QAAQ,GAAG,IAAI,CAAC;QACrC,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,KAAK,GAAG,IAAI,CAAC;QAElC,MAAM,kBAAkB,GAAG;YACzB;gBACE,IAAI,EAAE,gBAAgB;gBACtB,IAAI,EAAE;oBACJ,WAAW,EAAE,CAAC,EAAE,IAAI,EAAE,eAAe,EAAE,cAAc,EAAE,eAAe,EAAE,CAAC;oBACzE,YAAY,EAAE,EAAE;oBAChB,WAAW,EAAE,0CAA0C;iBACxD;aACF;SACF,CAAC;QACF,MAAM,iBAAiB,GAAG;YACxB;gBACE,IAAI,EAAE,eAAe;gBACrB,IAAI,EAAE;oBACJ,MAAM,EAAE,oBAAoB;oBAC5B,QAAQ,EAAE,eAAe;oBACzB,KAAK,EAAE,GAAG;oBACV,MAAM,EAAE,GAAG;oBACX,KAAK,EAAE,IAAI;iBACZ;aACF;SACF,CAAC;QACF,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,kBAAkB,CAAC,CAAC;QAC9E,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,iBAAiB,CAAC,CAAC;QAC7E,EAAE,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,aAAa,CAAC,qBAAqB,CAAC,yBAAyB,CAAC,CAAC;QAEzF,MAAM,UAAU,EAAE,CAAC;QAEnB,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,oBAAoB,CACnD,MAAM,CAAC,QAAQ,EAAE,EACjB,MAAM,CAAC,QAAQ,EAAE,EACjB,MAAM,CAAC,QAAQ,EAAE,EACjB,MAAM,CAAC,QAAQ,EAAE,EACjB,IAAI,EACJ,MAAM,CAAC,QAAQ,EAAE,CAClB,CAAC;QACF,MAAM,CAAC,cAAc,CAAC,aAAa,CAAC,CAAC,oBAAoB,CACvD,oBAAoB,EACpB,SAAS,EACT,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE,EAC3B,IAAI,CACL,CAAC;QACF,MAAM,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAAC,iBAAiB,EAAE,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC;IAC7F,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1,16 @@
|
|
|
1
|
+
export interface TextPrompt {
|
|
2
|
+
content: string;
|
|
3
|
+
}
|
|
4
|
+
export interface ImportantContext {
|
|
5
|
+
textPrompts?: TextPrompt[];
|
|
6
|
+
files?: string[];
|
|
7
|
+
}
|
|
8
|
+
export interface RcConfig {
|
|
9
|
+
rootDir: string;
|
|
10
|
+
lintCommand?: string;
|
|
11
|
+
extensions?: string[];
|
|
12
|
+
ignorePaths?: string[];
|
|
13
|
+
importantContext?: ImportantContext;
|
|
14
|
+
}
|
|
15
|
+
export declare function findRcFile(): string;
|
|
16
|
+
export declare function parseRcFile(rcFilePath: string): RcConfig;
|
|
@@ -0,0 +1,28 @@
|
|
|
1
|
+
import fs from 'fs';
|
|
2
|
+
import path from 'path';
|
|
3
|
+
import assert from 'node:assert';
|
|
4
|
+
import { isAncestorDirectory } from '../files/file-utils.js';
|
|
5
|
+
// This file contains project codegen configuration
|
|
6
|
+
const CODEGENRC_FILENAME = '.genaicoderc';
|
|
7
|
+
// Find .genaicoderc file
|
|
8
|
+
export function findRcFile() {
|
|
9
|
+
let rcFilePath = process.cwd();
|
|
10
|
+
while (!fs.existsSync(path.join(rcFilePath, CODEGENRC_FILENAME))) {
|
|
11
|
+
const parentDir = path.dirname(rcFilePath);
|
|
12
|
+
if (parentDir === rcFilePath) {
|
|
13
|
+
throw new Error(`${CODEGENRC_FILENAME} not found in any parent directory`);
|
|
14
|
+
}
|
|
15
|
+
rcFilePath = parentDir;
|
|
16
|
+
}
|
|
17
|
+
return path.join(rcFilePath, CODEGENRC_FILENAME);
|
|
18
|
+
}
|
|
19
|
+
// Read and parse .genaicoderc file
|
|
20
|
+
export function parseRcFile(rcFilePath) {
|
|
21
|
+
assert(fs.existsSync(rcFilePath), `${CODEGENRC_FILENAME} not found`);
|
|
22
|
+
const rcConfig = JSON.parse(fs.readFileSync(rcFilePath, 'utf-8'));
|
|
23
|
+
assert(rcConfig.rootDir, 'Root dir not configured');
|
|
24
|
+
const rootDir = path.resolve(path.dirname(rcFilePath), rcConfig.rootDir);
|
|
25
|
+
assert(isAncestorDirectory(path.dirname(rcFilePath), rootDir), 'Root dir is not located inside project directory');
|
|
26
|
+
return { ...rcConfig, rootDir };
|
|
27
|
+
}
|
|
28
|
+
//# sourceMappingURL=config-lib.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"config-lib.js","sourceRoot":"","sources":["../../src/main/config-lib.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,IAAI,MAAM,MAAM,CAAC;AACxB,OAAO,MAAM,MAAM,aAAa,CAAC;AAEjC,OAAO,EAAE,mBAAmB,EAAE,MAAM,wBAAwB,CAAC;AAE7D,mDAAmD;AACnD,MAAM,kBAAkB,GAAG,cAAc,CAAC;AAmB1C,yBAAyB;AACzB,MAAM,UAAU,UAAU;IACxB,IAAI,UAAU,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC;IAC/B,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,kBAAkB,CAAC,CAAC,EAAE,CAAC;QACjE,MAAM,SAAS,GAAG,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;QAC3C,IAAI,SAAS,KAAK,UAAU,EAAE,CAAC;YAC7B,MAAM,IAAI,KAAK,CAAC,GAAG,kBAAkB,oCAAoC,CAAC,CAAC;QAC7E,CAAC;QACD,UAAU,GAAG,SAAS,CAAC;IACzB,CAAC;IACD,OAAO,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,kBAAkB,CAAC,CAAC;AACnD,CAAC;AAED,mCAAmC;AACnC,MAAM,UAAU,WAAW,CAAC,UAAkB;IAC5C,MAAM,CAAC,EAAE,CAAC,UAAU,CAAC,UAAU,CAAC,EAAE,GAAG,kBAAkB,YAAY,CAAC,CAAC;IACrE,MAAM,QAAQ,GAAa,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,YAAY,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC,CAAC;IAC5E,MAAM,CAAC,QAAQ,CAAC,OAAO,EAAE,yBAAyB,CAAC,CAAC;IAEpD,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,QAAQ,CAAC,OAAO,CAAC,CAAC;IACzE,MAAM,CAAC,mBAAmB,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,OAAO,CAAC,EAAE,kDAAkD,CAAC,CAAC;IAEnH,OAAO,EAAE,GAAG,QAAQ,EAAE,OAAO,EAAE,CAAC;AAClC,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
export {};
|
|
@@ -0,0 +1,64 @@
|
|
|
1
|
+
import { describe, it, expect, vi } from 'vitest';
|
|
2
|
+
import fs from 'fs';
|
|
3
|
+
import path from 'path';
|
|
4
|
+
import { isAncestorDirectory } from '../files/file-utils.js';
|
|
5
|
+
import { findRcFile, parseRcFile } from './config-lib.js';
|
|
6
|
+
vi.mock('fs');
|
|
7
|
+
vi.mock('path');
|
|
8
|
+
vi.mock('../files/file-utils.js');
|
|
9
|
+
const CODEGENRC_FILENAME = '.genaicoderc';
|
|
10
|
+
// Mock data
|
|
11
|
+
const mockRcContent = JSON.stringify({ rootDir: 'src' });
|
|
12
|
+
const mockRootDir = '/project/src';
|
|
13
|
+
const mockRcFilePath = `/project/${CODEGENRC_FILENAME}`;
|
|
14
|
+
// Helper function to mock fs.existsSync
|
|
15
|
+
function mockExistsSync(paths) {
|
|
16
|
+
return (p) => paths.includes(p);
|
|
17
|
+
}
|
|
18
|
+
// Tests
|
|
19
|
+
// Test for findRcFile
|
|
20
|
+
describe('findRcFile', () => {
|
|
21
|
+
it('should find .genaicoderc in the current or parent directories', () => {
|
|
22
|
+
vi.mocked(fs).existsSync.mockImplementation(mockExistsSync([mockRcFilePath]));
|
|
23
|
+
vi.mocked(path).dirname.mockImplementation((p) => (p === '/project' ? '/' : '/project'));
|
|
24
|
+
vi.mocked(path).join.mockImplementation((...args) => args.join('/'));
|
|
25
|
+
const result = findRcFile();
|
|
26
|
+
expect(result).toBe(mockRcFilePath);
|
|
27
|
+
});
|
|
28
|
+
it('should throw an error if .genaicoderc is not found', () => {
|
|
29
|
+
vi.mocked(fs).existsSync.mockReturnValue(false);
|
|
30
|
+
vi.mocked(path).dirname.mockImplementation((p) => (p === '/' ? '/' : '/project'));
|
|
31
|
+
expect(() => findRcFile()).toThrowError(`${CODEGENRC_FILENAME} not found in any parent directory`);
|
|
32
|
+
});
|
|
33
|
+
});
|
|
34
|
+
// Test for parseRcFile
|
|
35
|
+
describe('parseRcFile', () => {
|
|
36
|
+
it('should parse .genaicoderc and return config with rootDir', () => {
|
|
37
|
+
vi.mocked(fs).existsSync.mockReturnValue(true);
|
|
38
|
+
vi.mocked(fs).readFileSync.mockReturnValue(mockRcContent);
|
|
39
|
+
vi.mocked(path).resolve.mockImplementation((...args) => args.join('/'));
|
|
40
|
+
vi.mocked(path).dirname.mockReturnValue('/project');
|
|
41
|
+
vi.mocked(isAncestorDirectory).mockReturnValue(true);
|
|
42
|
+
const result = parseRcFile(mockRcFilePath);
|
|
43
|
+
expect(result).toEqual({ ...JSON.parse(mockRcContent), rootDir: mockRootDir });
|
|
44
|
+
});
|
|
45
|
+
it('should throw an error if .genaicoderc is missing', () => {
|
|
46
|
+
vi.mocked(fs).existsSync.mockReturnValue(false);
|
|
47
|
+
expect(() => parseRcFile(mockRcFilePath)).toThrowError(`${CODEGENRC_FILENAME} not found`);
|
|
48
|
+
});
|
|
49
|
+
it('should throw an error if rootDir is not configured', () => {
|
|
50
|
+
const invalidContent = JSON.stringify({});
|
|
51
|
+
vi.mocked(fs).existsSync.mockReturnValue(true);
|
|
52
|
+
vi.mocked(fs).readFileSync.mockReturnValue(invalidContent);
|
|
53
|
+
expect(() => parseRcFile(mockRcFilePath)).toThrowError('Root dir not configured');
|
|
54
|
+
});
|
|
55
|
+
it('should throw an error if rootDir is not located inside project directory', () => {
|
|
56
|
+
vi.mocked(fs).existsSync.mockReturnValue(true);
|
|
57
|
+
vi.mocked(fs).readFileSync.mockReturnValue(mockRcContent);
|
|
58
|
+
vi.mocked(path).resolve.mockImplementation((...args) => args.join('/'));
|
|
59
|
+
vi.mocked(path).dirname.mockReturnValue('/project');
|
|
60
|
+
vi.mocked(isAncestorDirectory).mockReturnValue(false);
|
|
61
|
+
expect(() => parseRcFile(mockRcFilePath)).toThrowError('Root dir is not located inside project directory');
|
|
62
|
+
});
|
|
63
|
+
});
|
|
64
|
+
//# sourceMappingURL=config-lib.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"config-lib.test.js","sourceRoot":"","sources":["../../src/main/config-lib.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAClD,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,IAAI,MAAM,MAAM,CAAC;AACxB,OAAO,EAAE,mBAAmB,EAAE,MAAM,wBAAwB,CAAC;AAC7D,OAAO,EAAE,UAAU,EAAE,WAAW,EAAE,MAAM,iBAAiB,CAAC;AAE1D,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AACd,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;AAChB,EAAE,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;AAElC,MAAM,kBAAkB,GAAG,cAAc,CAAC;AAE1C,YAAY;AACZ,MAAM,aAAa,GAAG,IAAI,CAAC,SAAS,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC;AACzD,MAAM,WAAW,GAAG,cAAc,CAAC;AACnC,MAAM,cAAc,GAAG,YAAY,kBAAkB,EAAE,CAAC;AAExD,wCAAwC;AACxC,SAAS,cAAc,CAAC,KAAe;IACrC,OAAO,CAAC,CAAwB,EAAE,EAAE,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAW,CAAC,CAAC;AACnE,CAAC;AAED,QAAQ;AAER,sBAAsB;AACtB,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;IAC1B,EAAE,CAAC,+DAA+D,EAAE,GAAG,EAAE;QACvE,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,UAAU,CAAC,kBAAkB,CAAC,cAAc,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC;QAC9E,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,kBAAkB,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,UAAU,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC;QACzF,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC,GAAG,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;QAErE,MAAM,MAAM,GAAG,UAAU,EAAE,CAAC;QAC5B,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;IACtC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;QAC5D,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,UAAU,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;QAChD,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,kBAAkB,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC;QAElF,MAAM,CAAC,GAAG,EAAE,CAAC,UAAU,EAAE,CAAC,CAAC,YAAY,CAAC,GAAG,kBAAkB,oCAAoC,CAAC,CAAC;IACrG,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,uBAAuB;AACvB,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;IAC3B,EAAE,CAAC,0DAA0D,EAAE,GAAG,EAAE;QAClE,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,UAAU,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;QAC/C,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,YAAY,CAAC,eAAe,CAAC,aAAa,CAAC,CAAC;QAC1D,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,kBAAkB,CAAC,CAAC,GAAG,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;QACxE,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,eAAe,CAAC,UAAU,CAAC,CAAC;QACpD,EAAE,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;QAErD,MAAM,MAAM,GAAG,WAAW,CAAC,cAAc,CAAC,CAAC;QAC3C,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,EAAE,OAAO,EAAE,WAAW,EAAE,CAAC,CAAC;IACjF,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;QAC1D,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,UAAU,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;QAEhD,MAAM,CAAC,GAAG,EAAE,CAAC,WAAW,CAAC,cAAc,CAAC,CAAC,CAAC,YAAY,CAAC,GAAG,kBAAkB,YAAY,CAAC,CAAC;IAC5F,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;QAC5D,MAAM,cAAc,GAAG,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;QAC1C,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,UAAU,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;QAC/C,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,YAAY,CAAC,eAAe,CAAC,cAAc,CAAC,CAAC;QAE3D,MAAM,CAAC,GAAG,EAAE,CAAC,WAAW,CAAC,cAAc,CAAC,CAAC,CAAC,YAAY,CAAC,yBAAyB,CAAC,CAAC;IACpF,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0EAA0E,EAAE,GAAG,EAAE;QAClF,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,UAAU,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;QAC/C,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,YAAY,CAAC,eAAe,CAAC,aAAa,CAAC,CAAC;QAC1D,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,kBAAkB,CAAC,CAAC,GAAG,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;QACxE,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,eAAe,CAAC,UAAU,CAAC,CAAC;QACpD,EAAE,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;QAEtD,MAAM,CAAC,GAAG,EAAE,CAAC,WAAW,CAAC,cAAc,CAAC,CAAC,CAAC,YAAY,CAAC,kDAAkD,CAAC,CAAC;IAC7G,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1,6 @@
|
|
|
1
|
+
import { RcConfig, ImportantContext } from './config-lib.js';
|
|
2
|
+
export declare const rcConfig: RcConfig;
|
|
3
|
+
export declare const sourceExtensions: string[];
|
|
4
|
+
export declare const IMAGE_ASSET_EXTENSIONS: string[];
|
|
5
|
+
export declare const ignorePaths: string[];
|
|
6
|
+
export declare const importantContext: ImportantContext;
|