@machina.ai/cell-cli-core 1.0.13-rc9 → 1.0.17-rc1
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/package.json +4 -3
- package/dist/src/code_assist/converter.d.ts +2 -1
- package/dist/src/code_assist/converter.js +2 -1
- package/dist/src/code_assist/converter.js.map +1 -1
- package/dist/src/code_assist/converter.test.js +13 -10
- package/dist/src/code_assist/converter.test.js.map +1 -1
- package/dist/src/code_assist/oauth2.js +8 -0
- package/dist/src/code_assist/oauth2.js.map +1 -1
- package/dist/src/code_assist/oauth2.test.js +97 -7
- package/dist/src/code_assist/oauth2.test.js.map +1 -1
- package/dist/src/code_assist/server.d.ts +2 -2
- package/dist/src/code_assist/server.js +4 -4
- package/dist/src/code_assist/server.js.map +1 -1
- package/dist/src/code_assist/server.test.js +9 -9
- package/dist/src/code_assist/server.test.js.map +1 -1
- package/dist/src/code_assist/setup.js +1 -1
- package/dist/src/code_assist/setup.js.map +1 -1
- package/dist/src/code_assist/setup.test.js +2 -2
- package/dist/src/code_assist/setup.test.js.map +1 -1
- package/dist/src/config/config.d.ts +37 -9
- package/dist/src/config/config.js +84 -29
- package/dist/src/config/config.js.map +1 -1
- package/dist/src/config/config.test.js +95 -24
- package/dist/src/config/config.test.js.map +1 -1
- package/dist/src/config/flashFallback.test.js +23 -47
- package/dist/src/config/flashFallback.test.js.map +1 -1
- package/dist/src/config/models.d.ts +1 -0
- package/dist/src/config/models.js +1 -0
- package/dist/src/config/models.js.map +1 -1
- package/dist/src/core/client.d.ts +4 -1
- package/dist/src/core/client.js +82 -22
- package/dist/src/core/client.js.map +1 -1
- package/dist/src/core/client.test.js +188 -38
- package/dist/src/core/client.test.js.map +1 -1
- package/dist/src/core/contentGenerator.d.ts +2 -2
- package/dist/src/core/contentGenerator.js.map +1 -1
- package/dist/src/core/coreToolScheduler.d.ts +1 -3
- package/dist/src/core/coreToolScheduler.js +25 -16
- package/dist/src/core/coreToolScheduler.js.map +1 -1
- package/dist/src/core/coreToolScheduler.test.js +76 -1
- package/dist/src/core/coreToolScheduler.test.js.map +1 -1
- package/dist/src/core/geminiChat.d.ts +2 -1
- package/dist/src/core/geminiChat.js +9 -11
- package/dist/src/core/geminiChat.js.map +1 -1
- package/dist/src/core/geminiChat.test.js +2 -2
- package/dist/src/core/geminiChat.test.js.map +1 -1
- package/dist/src/core/geminiRequest.js +2 -37
- package/dist/src/core/geminiRequest.js.map +1 -1
- package/dist/src/core/logger.d.ts +1 -0
- package/dist/src/core/logger.js +24 -4
- package/dist/src/core/logger.js.map +1 -1
- package/dist/src/core/logger.test.js +60 -9
- package/dist/src/core/logger.test.js.map +1 -1
- package/dist/src/core/nonInteractiveToolExecutor.js +11 -3
- package/dist/src/core/nonInteractiveToolExecutor.js.map +1 -1
- package/dist/src/core/prompts.js +41 -17
- package/dist/src/core/prompts.js.map +1 -1
- package/dist/src/core/prompts.test.js +118 -1
- package/dist/src/core/prompts.test.js.map +1 -1
- package/dist/src/core/tokenLimits.js +1 -0
- package/dist/src/core/tokenLimits.js.map +1 -1
- package/dist/src/core/turn.d.ts +3 -0
- package/dist/src/core/turn.js +3 -0
- package/dist/src/core/turn.js.map +1 -1
- package/dist/src/ide/detect-ide.d.ts +10 -0
- package/dist/src/ide/detect-ide.js +27 -0
- package/dist/src/ide/detect-ide.js.map +1 -0
- package/dist/src/ide/ide-client.d.ts +40 -0
- package/dist/src/ide/ide-client.js +152 -0
- package/dist/src/ide/ide-client.js.map +1 -0
- package/dist/src/ide/ide-installer.d.ts +14 -0
- package/dist/src/ide/ide-installer.js +109 -0
- package/dist/src/ide/ide-installer.js.map +1 -0
- package/dist/src/ide/ide-installer.test.js +55 -0
- package/dist/src/ide/ide-installer.test.js.map +1 -0
- package/dist/src/ide/ideContext.d.ts +279 -0
- package/dist/src/ide/ideContext.js +102 -0
- package/dist/src/ide/ideContext.js.map +1 -0
- package/dist/src/ide/ideContext.test.js +265 -0
- package/dist/src/ide/ideContext.test.js.map +1 -0
- package/dist/src/index.d.ts +11 -1
- package/dist/src/index.js +14 -1
- package/dist/src/index.js.map +1 -1
- package/dist/src/mcp/google-auth-provider.d.ts +23 -0
- package/dist/src/mcp/google-auth-provider.js +63 -0
- package/dist/src/mcp/google-auth-provider.js.map +1 -0
- package/dist/src/mcp/google-auth-provider.test.d.ts +6 -0
- package/dist/src/mcp/google-auth-provider.test.js +54 -0
- package/dist/src/mcp/google-auth-provider.test.js.map +1 -0
- package/dist/src/mcp/oauth-provider.d.ts +5 -1
- package/dist/src/mcp/oauth-provider.js +44 -15
- package/dist/src/mcp/oauth-provider.js.map +1 -1
- package/dist/src/mcp/oauth-provider.test.js +94 -14
- package/dist/src/mcp/oauth-provider.test.js.map +1 -1
- package/dist/src/mcp/oauth-token-storage.d.ts +3 -1
- package/dist/src/mcp/oauth-token-storage.js +3 -1
- package/dist/src/mcp/oauth-token-storage.js.map +1 -1
- package/dist/src/mcp/oauth-utils.js +2 -2
- package/dist/src/mcp/oauth-utils.js.map +1 -1
- package/dist/src/mcp/oauth-utils.test.js +1 -1
- package/dist/src/mcp/oauth-utils.test.js.map +1 -1
- package/dist/src/prompts/mcp-prompts.d.ts +8 -0
- package/dist/src/prompts/mcp-prompts.js +13 -0
- package/dist/src/prompts/mcp-prompts.js.map +1 -0
- package/dist/src/prompts/prompt-registry.d.ts +26 -0
- package/dist/src/prompts/prompt-registry.js +47 -0
- package/dist/src/prompts/prompt-registry.js.map +1 -0
- package/dist/src/services/fileDiscoveryService.test.js +101 -60
- package/dist/src/services/fileDiscoveryService.test.js.map +1 -1
- package/dist/src/services/gitService.test.js +67 -86
- package/dist/src/services/gitService.test.js.map +1 -1
- package/dist/src/services/loopDetectionService.d.ts +51 -5
- package/dist/src/services/loopDetectionService.js +140 -36
- package/dist/src/services/loopDetectionService.js.map +1 -1
- package/dist/src/services/loopDetectionService.test.js +105 -99
- package/dist/src/services/loopDetectionService.test.js.map +1 -1
- package/dist/src/services/shellExecutionService.d.ts +70 -0
- package/dist/src/services/shellExecutionService.js +155 -0
- package/dist/src/services/shellExecutionService.js.map +1 -0
- package/dist/src/services/shellExecutionService.test.d.ts +6 -0
- package/dist/src/services/shellExecutionService.test.js +264 -0
- package/dist/src/services/shellExecutionService.test.js.map +1 -0
- package/dist/src/telemetry/clearcut-logger/clearcut-logger.d.ts +4 -1
- package/dist/src/telemetry/clearcut-logger/clearcut-logger.js +97 -34
- package/dist/src/telemetry/clearcut-logger/clearcut-logger.js.map +1 -1
- package/dist/src/telemetry/clearcut-logger/event-metadata-key.d.ts +6 -1
- package/dist/src/telemetry/clearcut-logger/event-metadata-key.js +19 -0
- package/dist/src/telemetry/clearcut-logger/event-metadata-key.js.map +1 -1
- package/dist/src/telemetry/constants.d.ts +2 -0
- package/dist/src/telemetry/constants.js +2 -0
- package/dist/src/telemetry/constants.js.map +1 -1
- package/dist/src/telemetry/file-exporters.d.ts +28 -0
- package/dist/src/telemetry/file-exporters.js +62 -0
- package/dist/src/telemetry/file-exporters.js.map +1 -0
- package/dist/src/telemetry/index.d.ts +2 -2
- package/dist/src/telemetry/index.js +2 -2
- package/dist/src/telemetry/index.js.map +1 -1
- package/dist/src/telemetry/loggers.d.ts +3 -1
- package/dist/src/telemetry/loggers.js +33 -1
- package/dist/src/telemetry/loggers.js.map +1 -1
- package/dist/src/telemetry/loggers.test.circular.js +2 -0
- package/dist/src/telemetry/loggers.test.circular.js.map +1 -1
- package/dist/src/telemetry/loggers.test.js +8 -3
- package/dist/src/telemetry/loggers.test.js.map +1 -1
- package/dist/src/telemetry/sdk.js +17 -6
- package/dist/src/telemetry/sdk.js.map +1 -1
- package/dist/src/telemetry/telemetry.test.js +2 -0
- package/dist/src/telemetry/telemetry.test.js.map +1 -1
- package/dist/src/telemetry/types.d.ts +24 -2
- package/dist/src/telemetry/types.js +40 -2
- package/dist/src/telemetry/types.js.map +1 -1
- package/dist/src/telemetry/uiTelemetry.test.js +3 -0
- package/dist/src/telemetry/uiTelemetry.test.js.map +1 -1
- package/dist/src/test-utils/mockWorkspaceContext.d.ts +13 -0
- package/dist/src/test-utils/mockWorkspaceContext.js +24 -0
- package/dist/src/test-utils/mockWorkspaceContext.js.map +1 -0
- package/dist/src/tools/edit.js +27 -3
- package/dist/src/tools/edit.js.map +1 -1
- package/dist/src/tools/edit.test.js +104 -1
- package/dist/src/tools/edit.test.js.map +1 -1
- package/dist/src/tools/glob.js +53 -17
- package/dist/src/tools/glob.js.map +1 -1
- package/dist/src/tools/glob.test.js +32 -6
- package/dist/src/tools/glob.test.js.map +1 -1
- package/dist/src/tools/grep.d.ts +1 -1
- package/dist/src/tools/grep.js +81 -29
- package/dist/src/tools/grep.js.map +1 -1
- package/dist/src/tools/grep.test.js +76 -9
- package/dist/src/tools/grep.test.js.map +1 -1
- package/dist/src/tools/ls.js +4 -3
- package/dist/src/tools/ls.js.map +1 -1
- package/dist/src/tools/ls.test.d.ts +6 -0
- package/dist/src/tools/ls.test.js +356 -0
- package/dist/src/tools/ls.test.js.map +1 -0
- package/dist/src/tools/mcp-client.d.ts +31 -3
- package/dist/src/tools/mcp-client.js +503 -57
- package/dist/src/tools/mcp-client.js.map +1 -1
- package/dist/src/tools/mcp-client.test.js +133 -7
- package/dist/src/tools/mcp-client.test.js.map +1 -1
- package/dist/src/tools/mcp-tool.js +104 -45
- package/dist/src/tools/mcp-tool.js.map +1 -1
- package/dist/src/tools/mcp-tool.test.js +314 -2
- package/dist/src/tools/mcp-tool.test.js.map +1 -1
- package/dist/src/tools/memoryTool.d.ts +17 -2
- package/dist/src/tools/memoryTool.js +130 -13
- package/dist/src/tools/memoryTool.js.map +1 -1
- package/dist/src/tools/memoryTool.test.js +88 -3
- package/dist/src/tools/memoryTool.test.js.map +1 -1
- package/dist/src/tools/modifiable-tool.test.js +51 -62
- package/dist/src/tools/modifiable-tool.test.js.map +1 -1
- package/dist/src/tools/read-file.js +8 -6
- package/dist/src/tools/read-file.js.map +1 -1
- package/dist/src/tools/read-file.test.js +125 -68
- package/dist/src/tools/read-file.test.js.map +1 -1
- package/dist/src/tools/read-many-files.d.ts +0 -1
- package/dist/src/tools/read-many-files.js +29 -20
- package/dist/src/tools/read-many-files.js.map +1 -1
- package/dist/src/tools/read-many-files.test.js +35 -2
- package/dist/src/tools/read-many-files.test.js.map +1 -1
- package/dist/src/tools/shell.d.ts +3 -23
- package/dist/src/tools/shell.js +173 -300
- package/dist/src/tools/shell.js.map +1 -1
- package/dist/src/tools/shell.test.js +278 -384
- package/dist/src/tools/shell.test.js.map +1 -1
- package/dist/src/tools/tool-error.d.ts +22 -0
- package/dist/src/tools/tool-error.js +27 -0
- package/dist/src/tools/tool-error.js.map +1 -0
- package/dist/src/tools/tool-registry.d.ts +13 -1
- package/dist/src/tools/tool-registry.js +46 -2
- package/dist/src/tools/tool-registry.js.map +1 -1
- package/dist/src/tools/tool-registry.test.js +13 -5
- package/dist/src/tools/tool-registry.test.js.map +1 -1
- package/dist/src/tools/tools.d.ts +8 -0
- package/dist/src/tools/tools.js.map +1 -1
- package/dist/src/tools/write-file.js +7 -5
- package/dist/src/tools/write-file.js.map +1 -1
- package/dist/src/tools/write-file.test.js +56 -2
- package/dist/src/tools/write-file.test.js.map +1 -1
- package/dist/src/utils/bfsFileSearch.js +51 -27
- package/dist/src/utils/bfsFileSearch.js.map +1 -1
- package/dist/src/utils/bfsFileSearch.test.js +163 -134
- package/dist/src/utils/bfsFileSearch.test.js.map +1 -1
- package/dist/src/utils/editCorrector.js +2 -2
- package/dist/src/utils/editCorrector.js.map +1 -1
- package/dist/src/utils/editor.d.ts +1 -1
- package/dist/src/utils/editor.js +9 -0
- package/dist/src/utils/editor.js.map +1 -1
- package/dist/src/utils/editor.test.js +21 -1
- package/dist/src/utils/editor.test.js.map +1 -1
- package/dist/src/utils/errorReporting.d.ts +1 -1
- package/dist/src/utils/errorReporting.js +2 -2
- package/dist/src/utils/errorReporting.js.map +1 -1
- package/dist/src/utils/errorReporting.test.js +44 -38
- package/dist/src/utils/errorReporting.test.js.map +1 -1
- package/dist/src/utils/fileUtils.js +13 -2
- package/dist/src/utils/fileUtils.js.map +1 -1
- package/dist/src/utils/fileUtils.test.js +42 -1
- package/dist/src/utils/fileUtils.test.js.map +1 -1
- package/dist/src/utils/flashFallback.integration.test.js +8 -0
- package/dist/src/utils/flashFallback.integration.test.js.map +1 -1
- package/dist/src/utils/formatters.d.ts +6 -0
- package/dist/src/utils/formatters.js +16 -0
- package/dist/src/utils/formatters.js.map +1 -0
- package/dist/src/utils/getFolderStructure.test.js +11 -13
- package/dist/src/utils/getFolderStructure.test.js.map +1 -1
- package/dist/src/utils/gitIgnoreParser.js +5 -11
- package/dist/src/utils/gitIgnoreParser.js.map +1 -1
- package/dist/src/utils/gitIgnoreParser.test.js +58 -61
- package/dist/src/utils/gitIgnoreParser.test.js.map +1 -1
- package/dist/src/utils/memoryDiscovery.d.ts +1 -1
- package/dist/src/utils/memoryDiscovery.js +69 -79
- package/dist/src/utils/memoryDiscovery.js.map +1 -1
- package/dist/src/utils/memoryDiscovery.test.js +119 -370
- package/dist/src/utils/memoryDiscovery.test.js.map +1 -1
- package/dist/src/utils/memoryImportProcessor.d.ts +19 -12
- package/dist/src/utils/memoryImportProcessor.js +241 -82
- package/dist/src/utils/memoryImportProcessor.js.map +1 -1
- package/dist/src/utils/memoryImportProcessor.test.js +595 -50
- package/dist/src/utils/memoryImportProcessor.test.js.map +1 -1
- package/dist/src/utils/nextSpeakerChecker.js +3 -23
- package/dist/src/utils/nextSpeakerChecker.js.map +1 -1
- package/dist/src/utils/nextSpeakerChecker.test.js +2 -2
- package/dist/src/utils/nextSpeakerChecker.test.js.map +1 -1
- package/dist/src/utils/partUtils.d.ts +14 -0
- package/dist/src/utils/partUtils.js +65 -0
- package/dist/src/utils/partUtils.js.map +1 -0
- package/dist/src/utils/partUtils.test.d.ts +6 -0
- package/dist/src/utils/partUtils.test.js +130 -0
- package/dist/src/utils/partUtils.test.js.map +1 -0
- package/dist/src/utils/paths.d.ts +11 -2
- package/dist/src/utils/paths.js +24 -7
- package/dist/src/utils/paths.js.map +1 -1
- package/dist/src/utils/paths.test.d.ts +6 -0
- package/dist/src/utils/paths.test.js +153 -0
- package/dist/src/utils/paths.test.js.map +1 -0
- package/dist/src/utils/retry.d.ts +3 -0
- package/dist/src/utils/retry.js.map +1 -1
- package/dist/src/utils/retry.test.js.map +1 -1
- package/dist/src/utils/secure-browser-launcher.d.ts +23 -0
- package/dist/src/utils/secure-browser-launcher.js +164 -0
- package/dist/src/utils/secure-browser-launcher.js.map +1 -0
- package/dist/src/utils/secure-browser-launcher.test.d.ts +6 -0
- package/dist/src/utils/secure-browser-launcher.test.js +149 -0
- package/dist/src/utils/secure-browser-launcher.test.js.map +1 -0
- package/dist/src/utils/shell-utils.d.ts +78 -0
- package/dist/src/utils/shell-utils.js +306 -0
- package/dist/src/utils/shell-utils.js.map +1 -0
- package/dist/src/utils/shell-utils.test.d.ts +6 -0
- package/dist/src/utils/shell-utils.test.js +200 -0
- package/dist/src/utils/shell-utils.test.js.map +1 -0
- package/dist/src/utils/summarizer.js +3 -32
- package/dist/src/utils/summarizer.js.map +1 -1
- package/dist/src/utils/textUtils.d.ts +13 -0
- package/dist/src/utils/textUtils.js +28 -0
- package/dist/src/utils/textUtils.js.map +1 -0
- package/dist/src/utils/workspaceContext.d.ts +51 -0
- package/dist/src/utils/workspaceContext.js +139 -0
- package/dist/src/utils/workspaceContext.js.map +1 -0
- package/dist/src/utils/workspaceContext.test.d.ts +6 -0
- package/dist/src/utils/workspaceContext.test.js +209 -0
- package/dist/src/utils/workspaceContext.test.js.map +1 -0
- package/dist/tsconfig.tsbuildinfo +1 -1
- package/package.json +4 -3
- package/dist/src/core/geminiRequest.test.js +0 -72
- package/dist/src/core/geminiRequest.test.js.map +0 -1
- package/dist/src/services/ideContext.d.ts +0 -178
- package/dist/src/services/ideContext.js +0 -105
- package/dist/src/services/ideContext.js.map +0 -1
- package/dist/src/services/ideContext.test.js +0 -111
- package/dist/src/services/ideContext.test.js.map +0 -1
- /package/dist/src/{core/geminiRequest.test.d.ts → ide/ide-installer.test.d.ts} +0 -0
- /package/dist/src/{services → ide}/ideContext.test.d.ts +0 -0
|
@@ -0,0 +1,13 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* @license
|
|
3
|
+
* Copyright 2025 Google LLC
|
|
4
|
+
* SPDX-License-Identifier: Apache-2.0
|
|
5
|
+
*/
|
|
6
|
+
export function getMCPServerPrompts(config, serverName) {
|
|
7
|
+
const promptRegistry = config.getPromptRegistry();
|
|
8
|
+
if (!promptRegistry) {
|
|
9
|
+
return [];
|
|
10
|
+
}
|
|
11
|
+
return promptRegistry.getPromptsByServer(serverName);
|
|
12
|
+
}
|
|
13
|
+
//# sourceMappingURL=mcp-prompts.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"mcp-prompts.js","sourceRoot":"","sources":["../../../src/prompts/mcp-prompts.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAKH,MAAM,UAAU,mBAAmB,CACjC,MAAc,EACd,UAAkB;IAElB,MAAM,cAAc,GAAG,MAAM,CAAC,iBAAiB,EAAE,CAAC;IAClD,IAAI,CAAC,cAAc,EAAE,CAAC;QACpB,OAAO,EAAE,CAAC;IACZ,CAAC;IACD,OAAO,cAAc,CAAC,kBAAkB,CAAC,UAAU,CAAC,CAAC;AACvD,CAAC"}
|
|
@@ -0,0 +1,26 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* @license
|
|
3
|
+
* Copyright 2025 Google LLC
|
|
4
|
+
* SPDX-License-Identifier: Apache-2.0
|
|
5
|
+
*/
|
|
6
|
+
import { DiscoveredMCPPrompt } from '../tools/mcp-client.js';
|
|
7
|
+
export declare class PromptRegistry {
|
|
8
|
+
private prompts;
|
|
9
|
+
/**
|
|
10
|
+
* Registers a prompt definition.
|
|
11
|
+
* @param prompt - The prompt object containing schema and execution logic.
|
|
12
|
+
*/
|
|
13
|
+
registerPrompt(prompt: DiscoveredMCPPrompt): void;
|
|
14
|
+
/**
|
|
15
|
+
* Returns an array of all registered and discovered prompt instances.
|
|
16
|
+
*/
|
|
17
|
+
getAllPrompts(): DiscoveredMCPPrompt[];
|
|
18
|
+
/**
|
|
19
|
+
* Get the definition of a specific prompt.
|
|
20
|
+
*/
|
|
21
|
+
getPrompt(name: string): DiscoveredMCPPrompt | undefined;
|
|
22
|
+
/**
|
|
23
|
+
* Returns an array of prompts registered from a specific MCP server.
|
|
24
|
+
*/
|
|
25
|
+
getPromptsByServer(serverName: string): DiscoveredMCPPrompt[];
|
|
26
|
+
}
|
|
@@ -0,0 +1,47 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* @license
|
|
3
|
+
* Copyright 2025 Google LLC
|
|
4
|
+
* SPDX-License-Identifier: Apache-2.0
|
|
5
|
+
*/
|
|
6
|
+
export class PromptRegistry {
|
|
7
|
+
prompts = new Map();
|
|
8
|
+
/**
|
|
9
|
+
* Registers a prompt definition.
|
|
10
|
+
* @param prompt - The prompt object containing schema and execution logic.
|
|
11
|
+
*/
|
|
12
|
+
registerPrompt(prompt) {
|
|
13
|
+
if (this.prompts.has(prompt.name)) {
|
|
14
|
+
const newName = `${prompt.serverName}_${prompt.name}`;
|
|
15
|
+
console.warn(`Prompt with name "${prompt.name}" is already registered. Renaming to "${newName}".`);
|
|
16
|
+
this.prompts.set(newName, { ...prompt, name: newName });
|
|
17
|
+
}
|
|
18
|
+
else {
|
|
19
|
+
this.prompts.set(prompt.name, prompt);
|
|
20
|
+
}
|
|
21
|
+
}
|
|
22
|
+
/**
|
|
23
|
+
* Returns an array of all registered and discovered prompt instances.
|
|
24
|
+
*/
|
|
25
|
+
getAllPrompts() {
|
|
26
|
+
return Array.from(this.prompts.values()).sort((a, b) => a.name.localeCompare(b.name));
|
|
27
|
+
}
|
|
28
|
+
/**
|
|
29
|
+
* Get the definition of a specific prompt.
|
|
30
|
+
*/
|
|
31
|
+
getPrompt(name) {
|
|
32
|
+
return this.prompts.get(name);
|
|
33
|
+
}
|
|
34
|
+
/**
|
|
35
|
+
* Returns an array of prompts registered from a specific MCP server.
|
|
36
|
+
*/
|
|
37
|
+
getPromptsByServer(serverName) {
|
|
38
|
+
const serverPrompts = [];
|
|
39
|
+
for (const prompt of this.prompts.values()) {
|
|
40
|
+
if (prompt.serverName === serverName) {
|
|
41
|
+
serverPrompts.push(prompt);
|
|
42
|
+
}
|
|
43
|
+
}
|
|
44
|
+
return serverPrompts.sort((a, b) => a.name.localeCompare(b.name));
|
|
45
|
+
}
|
|
46
|
+
}
|
|
47
|
+
//# sourceMappingURL=prompt-registry.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"prompt-registry.js","sourceRoot":"","sources":["../../../src/prompts/prompt-registry.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAIH,MAAM,OAAO,cAAc;IACjB,OAAO,GAAqC,IAAI,GAAG,EAAE,CAAC;IAE9D;;;OAGG;IACH,cAAc,CAAC,MAA2B;QACxC,IAAI,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC;YAClC,MAAM,OAAO,GAAG,GAAG,MAAM,CAAC,UAAU,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC;YACtD,OAAO,CAAC,IAAI,CACV,qBAAqB,MAAM,CAAC,IAAI,yCAAyC,OAAO,IAAI,CACrF,CAAC;YACF,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,EAAE,EAAE,GAAG,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC;QAC1D,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QACxC,CAAC;IACH,CAAC;IAED;;OAEG;IACH,aAAa;QACX,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CACrD,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,CAC7B,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,SAAS,CAAC,IAAY;QACpB,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IAChC,CAAC;IAED;;OAEG;IACH,kBAAkB,CAAC,UAAkB;QACnC,MAAM,aAAa,GAA0B,EAAE,CAAC;QAChD,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC;YAC3C,IAAI,MAAM,CAAC,UAAU,KAAK,UAAU,EAAE,CAAC;gBACrC,aAAa,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAC7B,CAAC;QACH,CAAC;QACD,OAAO,aAAa,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;IACpE,CAAC;CACF"}
|
|
@@ -3,99 +3,140 @@
|
|
|
3
3
|
* Copyright 2025 Google LLC
|
|
4
4
|
* SPDX-License-Identifier: Apache-2.0
|
|
5
5
|
*/
|
|
6
|
-
import { describe, it, expect, beforeEach,
|
|
6
|
+
import { describe, it, expect, beforeEach, afterEach } from 'vitest';
|
|
7
|
+
import * as fs from 'fs/promises';
|
|
8
|
+
import * as os from 'os';
|
|
9
|
+
import * as path from 'path';
|
|
7
10
|
import { FileDiscoveryService } from './fileDiscoveryService.js';
|
|
8
|
-
import { GitIgnoreParser } from '../utils/gitIgnoreParser.js';
|
|
9
|
-
import * as gitUtils from '../utils/gitUtils.js';
|
|
10
|
-
// Mock the GitIgnoreParser
|
|
11
|
-
vi.mock('../utils/gitIgnoreParser.js');
|
|
12
|
-
// Mock gitUtils module
|
|
13
|
-
vi.mock('../utils/gitUtils.js');
|
|
14
11
|
describe('FileDiscoveryService', () => {
|
|
15
|
-
let
|
|
16
|
-
let
|
|
17
|
-
|
|
18
|
-
|
|
19
|
-
|
|
20
|
-
|
|
21
|
-
|
|
22
|
-
|
|
23
|
-
|
|
24
|
-
|
|
25
|
-
|
|
26
|
-
|
|
27
|
-
vi.mocked(gitUtils.findGitRoot).mockReturnValue('/test/project');
|
|
28
|
-
vi.clearAllMocks();
|
|
12
|
+
let testRootDir;
|
|
13
|
+
let projectRoot;
|
|
14
|
+
async function createTestFile(filePath, content = '') {
|
|
15
|
+
const fullPath = path.join(projectRoot, filePath);
|
|
16
|
+
await fs.mkdir(path.dirname(fullPath), { recursive: true });
|
|
17
|
+
await fs.writeFile(fullPath, content);
|
|
18
|
+
return fullPath;
|
|
19
|
+
}
|
|
20
|
+
beforeEach(async () => {
|
|
21
|
+
testRootDir = await fs.mkdtemp(path.join(os.tmpdir(), 'file-discovery-test-'));
|
|
22
|
+
projectRoot = path.join(testRootDir, 'project');
|
|
23
|
+
await fs.mkdir(projectRoot, { recursive: true });
|
|
29
24
|
});
|
|
30
|
-
afterEach(() => {
|
|
31
|
-
|
|
25
|
+
afterEach(async () => {
|
|
26
|
+
await fs.rm(testRootDir, { recursive: true, force: true });
|
|
32
27
|
});
|
|
33
28
|
describe('initialization', () => {
|
|
34
|
-
it('should initialize git ignore parser by default', () => {
|
|
35
|
-
|
|
36
|
-
|
|
37
|
-
|
|
38
|
-
|
|
39
|
-
expect(
|
|
29
|
+
it('should initialize git ignore parser by default in a git repo', async () => {
|
|
30
|
+
await fs.mkdir(path.join(projectRoot, '.git'));
|
|
31
|
+
await createTestFile('.gitignore', 'node_modules/');
|
|
32
|
+
const service = new FileDiscoveryService(projectRoot);
|
|
33
|
+
// Let's check the effect of the parser instead of mocking it.
|
|
34
|
+
expect(service.shouldGitIgnoreFile('node_modules/foo.js')).toBe(true);
|
|
35
|
+
expect(service.shouldGitIgnoreFile('src/foo.js')).toBe(false);
|
|
40
36
|
});
|
|
41
|
-
it('should not
|
|
42
|
-
|
|
43
|
-
|
|
44
|
-
|
|
45
|
-
|
|
37
|
+
it('should not load git repo patterns when not in a git repo', async () => {
|
|
38
|
+
// No .git directory
|
|
39
|
+
await createTestFile('.gitignore', 'node_modules/');
|
|
40
|
+
const service = new FileDiscoveryService(projectRoot);
|
|
41
|
+
// .gitignore is not loaded in non-git repos
|
|
42
|
+
expect(service.shouldGitIgnoreFile('node_modules/foo.js')).toBe(false);
|
|
43
|
+
});
|
|
44
|
+
it('should load .geminiignore patterns even when not in a git repo', async () => {
|
|
45
|
+
await createTestFile('.geminiignore', 'secrets.txt');
|
|
46
|
+
const service = new FileDiscoveryService(projectRoot);
|
|
47
|
+
expect(service.shouldGeminiIgnoreFile('secrets.txt')).toBe(true);
|
|
48
|
+
expect(service.shouldGeminiIgnoreFile('src/index.js')).toBe(false);
|
|
46
49
|
});
|
|
47
50
|
});
|
|
48
51
|
describe('filterFiles', () => {
|
|
49
|
-
beforeEach(() => {
|
|
50
|
-
|
|
51
|
-
|
|
52
|
+
beforeEach(async () => {
|
|
53
|
+
await fs.mkdir(path.join(projectRoot, '.git'));
|
|
54
|
+
await createTestFile('.gitignore', 'node_modules/\n.git/\ndist');
|
|
55
|
+
await createTestFile('.geminiignore', 'logs/');
|
|
52
56
|
});
|
|
53
|
-
it('should filter out git-ignored files by default', () => {
|
|
57
|
+
it('should filter out git-ignored and gemini-ignored files by default', () => {
|
|
54
58
|
const files = [
|
|
55
59
|
'src/index.ts',
|
|
56
60
|
'node_modules/package/index.js',
|
|
57
61
|
'README.md',
|
|
58
62
|
'.git/config',
|
|
59
63
|
'dist/bundle.js',
|
|
60
|
-
|
|
61
|
-
|
|
62
|
-
|
|
64
|
+
'logs/latest.log',
|
|
65
|
+
].map((f) => path.join(projectRoot, f));
|
|
66
|
+
const service = new FileDiscoveryService(projectRoot);
|
|
67
|
+
expect(service.filterFiles(files)).toEqual(['src/index.ts', 'README.md'].map((f) => path.join(projectRoot, f)));
|
|
63
68
|
});
|
|
64
69
|
it('should not filter files when respectGitIgnore is false', () => {
|
|
65
70
|
const files = [
|
|
66
71
|
'src/index.ts',
|
|
67
72
|
'node_modules/package/index.js',
|
|
68
73
|
'.git/config',
|
|
69
|
-
|
|
70
|
-
|
|
71
|
-
|
|
74
|
+
'logs/latest.log',
|
|
75
|
+
].map((f) => path.join(projectRoot, f));
|
|
76
|
+
const service = new FileDiscoveryService(projectRoot);
|
|
77
|
+
const filtered = service.filterFiles(files, {
|
|
78
|
+
respectGitIgnore: false,
|
|
79
|
+
respectGeminiIgnore: true, // still respect this one
|
|
80
|
+
});
|
|
81
|
+
expect(filtered).toEqual(['src/index.ts', 'node_modules/package/index.js', '.git/config'].map((f) => path.join(projectRoot, f)));
|
|
82
|
+
});
|
|
83
|
+
it('should not filter files when respectGeminiIgnore is false', () => {
|
|
84
|
+
const files = [
|
|
85
|
+
'src/index.ts',
|
|
86
|
+
'node_modules/package/index.js',
|
|
87
|
+
'logs/latest.log',
|
|
88
|
+
].map((f) => path.join(projectRoot, f));
|
|
89
|
+
const service = new FileDiscoveryService(projectRoot);
|
|
90
|
+
const filtered = service.filterFiles(files, {
|
|
91
|
+
respectGitIgnore: true,
|
|
92
|
+
respectGeminiIgnore: false,
|
|
93
|
+
});
|
|
94
|
+
expect(filtered).toEqual(['src/index.ts', 'logs/latest.log'].map((f) => path.join(projectRoot, f)));
|
|
72
95
|
});
|
|
73
96
|
it('should handle empty file list', () => {
|
|
74
|
-
const
|
|
75
|
-
expect(
|
|
97
|
+
const service = new FileDiscoveryService(projectRoot);
|
|
98
|
+
expect(service.filterFiles([])).toEqual([]);
|
|
76
99
|
});
|
|
77
100
|
});
|
|
78
|
-
describe('shouldGitIgnoreFile', () => {
|
|
79
|
-
beforeEach(() => {
|
|
80
|
-
|
|
81
|
-
|
|
101
|
+
describe('shouldGitIgnoreFile & shouldGeminiIgnoreFile', () => {
|
|
102
|
+
beforeEach(async () => {
|
|
103
|
+
await fs.mkdir(path.join(projectRoot, '.git'));
|
|
104
|
+
await createTestFile('.gitignore', 'node_modules/');
|
|
105
|
+
await createTestFile('.geminiignore', '*.log');
|
|
82
106
|
});
|
|
83
107
|
it('should return true for git-ignored files', () => {
|
|
84
|
-
|
|
108
|
+
const service = new FileDiscoveryService(projectRoot);
|
|
109
|
+
expect(service.shouldGitIgnoreFile(path.join(projectRoot, 'node_modules/package/index.js'))).toBe(true);
|
|
110
|
+
});
|
|
111
|
+
it('should return false for non-git-ignored files', () => {
|
|
112
|
+
const service = new FileDiscoveryService(projectRoot);
|
|
113
|
+
expect(service.shouldGitIgnoreFile(path.join(projectRoot, 'src/index.ts'))).toBe(false);
|
|
114
|
+
});
|
|
115
|
+
it('should return true for gemini-ignored files', () => {
|
|
116
|
+
const service = new FileDiscoveryService(projectRoot);
|
|
117
|
+
expect(service.shouldGeminiIgnoreFile(path.join(projectRoot, 'debug.log'))).toBe(true);
|
|
85
118
|
});
|
|
86
|
-
it('should return false for non-ignored files', () => {
|
|
87
|
-
|
|
119
|
+
it('should return false for non-gemini-ignored files', () => {
|
|
120
|
+
const service = new FileDiscoveryService(projectRoot);
|
|
121
|
+
expect(service.shouldGeminiIgnoreFile(path.join(projectRoot, 'src/index.ts'))).toBe(false);
|
|
88
122
|
});
|
|
89
123
|
});
|
|
90
124
|
describe('edge cases', () => {
|
|
91
|
-
it('should handle relative project root paths', () => {
|
|
92
|
-
|
|
93
|
-
|
|
125
|
+
it('should handle relative project root paths', async () => {
|
|
126
|
+
await fs.mkdir(path.join(projectRoot, '.git'));
|
|
127
|
+
await createTestFile('.gitignore', 'ignored.txt');
|
|
128
|
+
const service = new FileDiscoveryService(path.relative(process.cwd(), projectRoot));
|
|
129
|
+
expect(service.shouldGitIgnoreFile(path.join(projectRoot, 'ignored.txt'))).toBe(true);
|
|
130
|
+
expect(service.shouldGitIgnoreFile(path.join(projectRoot, 'not-ignored.txt'))).toBe(false);
|
|
94
131
|
});
|
|
95
|
-
it('should handle filterFiles with undefined options', () => {
|
|
96
|
-
|
|
97
|
-
|
|
98
|
-
|
|
132
|
+
it('should handle filterFiles with undefined options', async () => {
|
|
133
|
+
await fs.mkdir(path.join(projectRoot, '.git'));
|
|
134
|
+
await createTestFile('.gitignore', 'ignored.txt');
|
|
135
|
+
const service = new FileDiscoveryService(projectRoot);
|
|
136
|
+
const files = ['src/index.ts', 'ignored.txt'].map((f) => path.join(projectRoot, f));
|
|
137
|
+
expect(service.filterFiles(files, undefined)).toEqual([
|
|
138
|
+
path.join(projectRoot, 'src/index.ts'),
|
|
139
|
+
]);
|
|
99
140
|
});
|
|
100
141
|
});
|
|
101
142
|
});
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"fileDiscoveryService.test.js","sourceRoot":"","sources":["../../../src/services/fileDiscoveryService.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,
|
|
1
|
+
{"version":3,"file":"fileDiscoveryService.test.js","sourceRoot":"","sources":["../../../src/services/fileDiscoveryService.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACrE,OAAO,KAAK,EAAE,MAAM,aAAa,CAAC;AAClC,OAAO,KAAK,EAAE,MAAM,IAAI,CAAC;AACzB,OAAO,KAAK,IAAI,MAAM,MAAM,CAAC;AAC7B,OAAO,EAAE,oBAAoB,EAAE,MAAM,2BAA2B,CAAC;AAEjE,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;IACpC,IAAI,WAAmB,CAAC;IACxB,IAAI,WAAmB,CAAC;IAExB,KAAK,UAAU,cAAc,CAAC,QAAgB,EAAE,OAAO,GAAG,EAAE;QAC1D,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,QAAQ,CAAC,CAAC;QAClD,MAAM,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAC5D,MAAM,EAAE,CAAC,SAAS,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QACtC,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,UAAU,CAAC,KAAK,IAAI,EAAE;QACpB,WAAW,GAAG,MAAM,EAAE,CAAC,OAAO,CAC5B,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,sBAAsB,CAAC,CAC/C,CAAC;QACF,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,SAAS,CAAC,CAAC;QAChD,MAAM,EAAE,CAAC,KAAK,CAAC,WAAW,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IACnD,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,KAAK,IAAI,EAAE;QACnB,MAAM,EAAE,CAAC,EAAE,CAAC,WAAW,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IAC7D,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,8DAA8D,EAAE,KAAK,IAAI,EAAE;YAC5E,MAAM,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,MAAM,CAAC,CAAC,CAAC;YAC/C,MAAM,cAAc,CAAC,YAAY,EAAE,eAAe,CAAC,CAAC;YAEpD,MAAM,OAAO,GAAG,IAAI,oBAAoB,CAAC,WAAW,CAAC,CAAC;YACtD,8DAA8D;YAC9D,MAAM,CAAC,OAAO,CAAC,mBAAmB,CAAC,qBAAqB,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACtE,MAAM,CAAC,OAAO,CAAC,mBAAmB,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;YACxE,oBAAoB;YACpB,MAAM,cAAc,CAAC,YAAY,EAAE,eAAe,CAAC,CAAC;YACpD,MAAM,OAAO,GAAG,IAAI,oBAAoB,CAAC,WAAW,CAAC,CAAC;YAEtD,4CAA4C;YAC5C,MAAM,CAAC,OAAO,CAAC,mBAAmB,CAAC,qBAAqB,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACzE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gEAAgE,EAAE,KAAK,IAAI,EAAE;YAC9E,MAAM,cAAc,CAAC,eAAe,EAAE,aAAa,CAAC,CAAC;YACrD,MAAM,OAAO,GAAG,IAAI,oBAAoB,CAAC,WAAW,CAAC,CAAC;YAEtD,MAAM,CAAC,OAAO,CAAC,sBAAsB,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACjE,MAAM,CAAC,OAAO,CAAC,sBAAsB,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACrE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;QAC3B,UAAU,CAAC,KAAK,IAAI,EAAE;YACpB,MAAM,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,MAAM,CAAC,CAAC,CAAC;YAC/C,MAAM,cAAc,CAAC,YAAY,EAAE,4BAA4B,CAAC,CAAC;YACjE,MAAM,cAAc,CAAC,eAAe,EAAE,OAAO,CAAC,CAAC;QACjD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mEAAmE,EAAE,GAAG,EAAE;YAC3E,MAAM,KAAK,GAAG;gBACZ,cAAc;gBACd,+BAA+B;gBAC/B,WAAW;gBACX,aAAa;gBACb,gBAAgB;gBAChB,iBAAiB;aAClB,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,CAAC;YAExC,MAAM,OAAO,GAAG,IAAI,oBAAoB,CAAC,WAAW,CAAC,CAAC;YAEtD,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CACxC,CAAC,cAAc,EAAE,WAAW,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,CACpE,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wDAAwD,EAAE,GAAG,EAAE;YAChE,MAAM,KAAK,GAAG;gBACZ,cAAc;gBACd,+BAA+B;gBAC/B,aAAa;gBACb,iBAAiB;aAClB,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,CAAC;YAExC,MAAM,OAAO,GAAG,IAAI,oBAAoB,CAAC,WAAW,CAAC,CAAC;YAEtD,MAAM,QAAQ,GAAG,OAAO,CAAC,WAAW,CAAC,KAAK,EAAE;gBAC1C,gBAAgB,EAAE,KAAK;gBACvB,mBAAmB,EAAE,IAAI,EAAE,yBAAyB;aACrD,CAAC,CAAC;YAEH,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CACtB,CAAC,cAAc,EAAE,+BAA+B,EAAE,aAAa,CAAC,CAAC,GAAG,CAClE,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC,CACjC,CACF,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2DAA2D,EAAE,GAAG,EAAE;YACnE,MAAM,KAAK,GAAG;gBACZ,cAAc;gBACd,+BAA+B;gBAC/B,iBAAiB;aAClB,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,CAAC;YAExC,MAAM,OAAO,GAAG,IAAI,oBAAoB,CAAC,WAAW,CAAC,CAAC;YAEtD,MAAM,QAAQ,GAAG,OAAO,CAAC,WAAW,CAAC,KAAK,EAAE;gBAC1C,gBAAgB,EAAE,IAAI;gBACtB,mBAAmB,EAAE,KAAK;aAC3B,CAAC,CAAC;YAEH,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CACtB,CAAC,cAAc,EAAE,iBAAiB,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAC5C,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC,CAC1B,CACF,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+BAA+B,EAAE,GAAG,EAAE;YACvC,MAAM,OAAO,GAAG,IAAI,oBAAoB,CAAC,WAAW,CAAC,CAAC;YAEtD,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,8CAA8C,EAAE,GAAG,EAAE;QAC5D,UAAU,CAAC,KAAK,IAAI,EAAE;YACpB,MAAM,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,MAAM,CAAC,CAAC,CAAC;YAC/C,MAAM,cAAc,CAAC,YAAY,EAAE,eAAe,CAAC,CAAC;YACpD,MAAM,cAAc,CAAC,eAAe,EAAE,OAAO,CAAC,CAAC;QACjD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;YAClD,MAAM,OAAO,GAAG,IAAI,oBAAoB,CAAC,WAAW,CAAC,CAAC;YAEtD,MAAM,CACJ,OAAO,CAAC,mBAAmB,CACzB,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,+BAA+B,CAAC,CACxD,CACF,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACf,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;YACvD,MAAM,OAAO,GAAG,IAAI,oBAAoB,CAAC,WAAW,CAAC,CAAC;YAEtD,MAAM,CACJ,OAAO,CAAC,mBAAmB,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,cAAc,CAAC,CAAC,CACpE,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAChB,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;YACrD,MAAM,OAAO,GAAG,IAAI,oBAAoB,CAAC,WAAW,CAAC,CAAC;YAEtD,MAAM,CACJ,OAAO,CAAC,sBAAsB,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,WAAW,CAAC,CAAC,CACpE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACf,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;YAC1D,MAAM,OAAO,GAAG,IAAI,oBAAoB,CAAC,WAAW,CAAC,CAAC;YAEtD,MAAM,CACJ,OAAO,CAAC,sBAAsB,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,cAAc,CAAC,CAAC,CACvE,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAChB,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;QAC1B,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;YACzD,MAAM,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,MAAM,CAAC,CAAC,CAAC;YAC/C,MAAM,cAAc,CAAC,YAAY,EAAE,aAAa,CAAC,CAAC;YAClD,MAAM,OAAO,GAAG,IAAI,oBAAoB,CACtC,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,WAAW,CAAC,CAC1C,CAAC;YAEF,MAAM,CACJ,OAAO,CAAC,mBAAmB,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,aAAa,CAAC,CAAC,CACnE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACb,MAAM,CACJ,OAAO,CAAC,mBAAmB,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,iBAAiB,CAAC,CAAC,CACvE,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAChB,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;YAChE,MAAM,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,MAAM,CAAC,CAAC,CAAC;YAC/C,MAAM,cAAc,CAAC,YAAY,EAAE,aAAa,CAAC,CAAC;YAClD,MAAM,OAAO,GAAG,IAAI,oBAAoB,CAAC,WAAW,CAAC,CAAC;YAEtD,MAAM,KAAK,GAAG,CAAC,cAAc,EAAE,aAAa,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CACtD,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC,CAC1B,CAAC;YAEF,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC;gBACpD,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,cAAc,CAAC;aACvC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -6,22 +6,13 @@
|
|
|
6
6
|
import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
|
|
7
7
|
import { GitService } from './gitService.js';
|
|
8
8
|
import * as path from 'path';
|
|
9
|
+
import * as fs from 'fs/promises';
|
|
10
|
+
import * as os from 'os';
|
|
11
|
+
import { getProjectHash, GEMINI_DIR } from '../utils/paths.js';
|
|
9
12
|
const hoistedMockExec = vi.hoisted(() => vi.fn());
|
|
10
13
|
vi.mock('node:child_process', () => ({
|
|
11
14
|
exec: hoistedMockExec,
|
|
12
15
|
}));
|
|
13
|
-
const hoistedMockMkdir = vi.hoisted(() => vi.fn());
|
|
14
|
-
const hoistedMockReadFile = vi.hoisted(() => vi.fn());
|
|
15
|
-
const hoistedMockWriteFile = vi.hoisted(() => vi.fn());
|
|
16
|
-
vi.mock('fs/promises', async (importOriginal) => {
|
|
17
|
-
const actual = (await importOriginal());
|
|
18
|
-
return {
|
|
19
|
-
...actual,
|
|
20
|
-
mkdir: hoistedMockMkdir,
|
|
21
|
-
readFile: hoistedMockReadFile,
|
|
22
|
-
writeFile: hoistedMockWriteFile,
|
|
23
|
-
};
|
|
24
|
-
});
|
|
25
16
|
const hoistedMockEnv = vi.hoisted(() => vi.fn());
|
|
26
17
|
const hoistedMockSimpleGit = vi.hoisted(() => vi.fn());
|
|
27
18
|
const hoistedMockCheckIsRepo = vi.hoisted(() => vi.fn());
|
|
@@ -44,34 +35,26 @@ const hoistedIsGitRepositoryMock = vi.hoisted(() => vi.fn());
|
|
|
44
35
|
vi.mock('../utils/gitUtils.js', () => ({
|
|
45
36
|
isGitRepository: hoistedIsGitRepositoryMock,
|
|
46
37
|
}));
|
|
47
|
-
const hoistedMockIsNodeError = vi.hoisted(() => vi.fn());
|
|
48
|
-
vi.mock('../utils/errors.js', () => ({
|
|
49
|
-
isNodeError: hoistedMockIsNodeError,
|
|
50
|
-
}));
|
|
51
38
|
const hoistedMockHomedir = vi.hoisted(() => vi.fn());
|
|
52
|
-
vi.mock('os', () =>
|
|
53
|
-
|
|
54
|
-
}));
|
|
55
|
-
const hoistedMockCreateHash = vi.hoisted(() => {
|
|
56
|
-
const mockUpdate = vi.fn().mockReturnThis();
|
|
57
|
-
const mockDigest = vi.fn();
|
|
39
|
+
vi.mock('os', async (importOriginal) => {
|
|
40
|
+
const actual = await importOriginal();
|
|
58
41
|
return {
|
|
59
|
-
|
|
60
|
-
|
|
61
|
-
digest: mockDigest,
|
|
62
|
-
})),
|
|
63
|
-
mockUpdate,
|
|
64
|
-
mockDigest,
|
|
42
|
+
...actual,
|
|
43
|
+
homedir: hoistedMockHomedir,
|
|
65
44
|
};
|
|
66
45
|
});
|
|
67
|
-
vi.mock('crypto', () => ({
|
|
68
|
-
createHash: hoistedMockCreateHash.createHash,
|
|
69
|
-
}));
|
|
70
46
|
describe('GitService', () => {
|
|
71
|
-
|
|
72
|
-
|
|
73
|
-
|
|
74
|
-
|
|
47
|
+
let testRootDir;
|
|
48
|
+
let projectRoot;
|
|
49
|
+
let homedir;
|
|
50
|
+
let hash;
|
|
51
|
+
beforeEach(async () => {
|
|
52
|
+
testRootDir = await fs.mkdtemp(path.join(os.tmpdir(), 'git-service-test-'));
|
|
53
|
+
projectRoot = path.join(testRootDir, 'project');
|
|
54
|
+
homedir = path.join(testRootDir, 'home');
|
|
55
|
+
await fs.mkdir(projectRoot, { recursive: true });
|
|
56
|
+
await fs.mkdir(homedir, { recursive: true });
|
|
57
|
+
hash = getProjectHash(projectRoot);
|
|
75
58
|
vi.clearAllMocks();
|
|
76
59
|
hoistedIsGitRepositoryMock.mockReturnValue(true);
|
|
77
60
|
hoistedMockExec.mockImplementation((command, callback) => {
|
|
@@ -83,13 +66,7 @@ describe('GitService', () => {
|
|
|
83
66
|
}
|
|
84
67
|
return {};
|
|
85
68
|
});
|
|
86
|
-
|
|
87
|
-
hoistedMockReadFile.mockResolvedValue('');
|
|
88
|
-
hoistedMockWriteFile.mockResolvedValue(undefined);
|
|
89
|
-
hoistedMockIsNodeError.mockImplementation((e) => e instanceof Error);
|
|
90
|
-
hoistedMockHomedir.mockReturnValue(mockHomedir);
|
|
91
|
-
hoistedMockCreateHash.mockUpdate.mockReturnThis();
|
|
92
|
-
hoistedMockCreateHash.mockDigest.mockReturnValue(mockHash);
|
|
69
|
+
hoistedMockHomedir.mockReturnValue(homedir);
|
|
93
70
|
hoistedMockEnv.mockImplementation(() => ({
|
|
94
71
|
checkIsRepo: hoistedMockCheckIsRepo,
|
|
95
72
|
init: hoistedMockInit,
|
|
@@ -113,17 +90,18 @@ describe('GitService', () => {
|
|
|
113
90
|
commit: 'initial',
|
|
114
91
|
});
|
|
115
92
|
});
|
|
116
|
-
afterEach(() => {
|
|
93
|
+
afterEach(async () => {
|
|
117
94
|
vi.restoreAllMocks();
|
|
95
|
+
await fs.rm(testRootDir, { recursive: true, force: true });
|
|
118
96
|
});
|
|
119
97
|
describe('constructor', () => {
|
|
120
|
-
it('should successfully create an instance
|
|
121
|
-
expect(() => new GitService(
|
|
98
|
+
it('should successfully create an instance', () => {
|
|
99
|
+
expect(() => new GitService(projectRoot)).not.toThrow();
|
|
122
100
|
});
|
|
123
101
|
});
|
|
124
102
|
describe('verifyGitAvailability', () => {
|
|
125
103
|
it('should resolve true if git --version command succeeds', async () => {
|
|
126
|
-
const service = new GitService(
|
|
104
|
+
const service = new GitService(projectRoot);
|
|
127
105
|
await expect(service.verifyGitAvailability()).resolves.toBe(true);
|
|
128
106
|
});
|
|
129
107
|
it('should resolve false if git --version command fails', async () => {
|
|
@@ -131,7 +109,7 @@ describe('GitService', () => {
|
|
|
131
109
|
callback(new Error('git not found'));
|
|
132
110
|
return {};
|
|
133
111
|
});
|
|
134
|
-
const service = new GitService(
|
|
112
|
+
const service = new GitService(projectRoot);
|
|
135
113
|
await expect(service.verifyGitAvailability()).resolves.toBe(false);
|
|
136
114
|
});
|
|
137
115
|
});
|
|
@@ -141,11 +119,11 @@ describe('GitService', () => {
|
|
|
141
119
|
callback(new Error('git not found'));
|
|
142
120
|
return {};
|
|
143
121
|
});
|
|
144
|
-
const service = new GitService(
|
|
122
|
+
const service = new GitService(projectRoot);
|
|
145
123
|
await expect(service.initialize()).rejects.toThrow('Checkpointing is enabled, but Git is not installed. Please install Git or disable checkpointing to continue.');
|
|
146
124
|
});
|
|
147
125
|
it('should call setupShadowGitRepository if Git is available', async () => {
|
|
148
|
-
const service = new GitService(
|
|
126
|
+
const service = new GitService(projectRoot);
|
|
149
127
|
const setupSpy = vi
|
|
150
128
|
.spyOn(service, 'setupShadowGitRepository')
|
|
151
129
|
.mockResolvedValue(undefined);
|
|
@@ -154,64 +132,67 @@ describe('GitService', () => {
|
|
|
154
132
|
});
|
|
155
133
|
});
|
|
156
134
|
describe('setupShadowGitRepository', () => {
|
|
157
|
-
|
|
158
|
-
|
|
159
|
-
|
|
160
|
-
|
|
161
|
-
|
|
162
|
-
const service = new GitService(mockProjectRoot);
|
|
163
|
-
await service.setupShadowGitRepository();
|
|
164
|
-
const expectedConfigContent = '[user]\n name = Gemini CLI\n email = gemini-cli@google.com\n[commit]\n gpgsign = false\n';
|
|
165
|
-
expect(hoistedMockWriteFile).toHaveBeenCalledWith(gitConfigPath, expectedConfigContent);
|
|
135
|
+
let repoDir;
|
|
136
|
+
let gitConfigPath;
|
|
137
|
+
beforeEach(() => {
|
|
138
|
+
repoDir = path.join(homedir, GEMINI_DIR, 'history', hash);
|
|
139
|
+
gitConfigPath = path.join(repoDir, '.gitconfig');
|
|
166
140
|
});
|
|
167
141
|
it('should create history and repository directories', async () => {
|
|
168
|
-
const service = new GitService(
|
|
142
|
+
const service = new GitService(projectRoot);
|
|
169
143
|
await service.setupShadowGitRepository();
|
|
170
|
-
|
|
171
|
-
|
|
172
|
-
|
|
144
|
+
const stats = await fs.stat(repoDir);
|
|
145
|
+
expect(stats.isDirectory()).toBe(true);
|
|
146
|
+
});
|
|
147
|
+
it('should create a .gitconfig file with the correct content', async () => {
|
|
148
|
+
const service = new GitService(projectRoot);
|
|
149
|
+
await service.setupShadowGitRepository();
|
|
150
|
+
const expectedConfigContent = '[user]\n name = Gemini CLI\n email = gemini-cli@google.com\n[commit]\n gpgsign = false\n';
|
|
151
|
+
const actualConfigContent = await fs.readFile(gitConfigPath, 'utf-8');
|
|
152
|
+
expect(actualConfigContent).toBe(expectedConfigContent);
|
|
173
153
|
});
|
|
174
154
|
it('should initialize git repo in historyDir if not already initialized', async () => {
|
|
175
155
|
hoistedMockCheckIsRepo.mockResolvedValue(false);
|
|
176
|
-
const service = new GitService(
|
|
156
|
+
const service = new GitService(projectRoot);
|
|
177
157
|
await service.setupShadowGitRepository();
|
|
178
158
|
expect(hoistedMockSimpleGit).toHaveBeenCalledWith(repoDir);
|
|
179
159
|
expect(hoistedMockInit).toHaveBeenCalled();
|
|
180
160
|
});
|
|
181
161
|
it('should not initialize git repo if already initialized', async () => {
|
|
182
162
|
hoistedMockCheckIsRepo.mockResolvedValue(true);
|
|
183
|
-
const service = new GitService(
|
|
163
|
+
const service = new GitService(projectRoot);
|
|
184
164
|
await service.setupShadowGitRepository();
|
|
185
165
|
expect(hoistedMockInit).not.toHaveBeenCalled();
|
|
186
166
|
});
|
|
187
167
|
it('should copy .gitignore from projectRoot if it exists', async () => {
|
|
188
|
-
const gitignoreContent =
|
|
189
|
-
|
|
190
|
-
|
|
191
|
-
|
|
192
|
-
}
|
|
193
|
-
return '';
|
|
194
|
-
});
|
|
195
|
-
const service = new GitService(mockProjectRoot);
|
|
168
|
+
const gitignoreContent = 'node_modules/\n.env';
|
|
169
|
+
const visibleGitIgnorePath = path.join(projectRoot, '.gitignore');
|
|
170
|
+
await fs.writeFile(visibleGitIgnorePath, gitignoreContent);
|
|
171
|
+
const service = new GitService(projectRoot);
|
|
196
172
|
await service.setupShadowGitRepository();
|
|
197
|
-
|
|
198
|
-
|
|
173
|
+
const hiddenGitIgnorePath = path.join(repoDir, '.gitignore');
|
|
174
|
+
const copiedContent = await fs.readFile(hiddenGitIgnorePath, 'utf-8');
|
|
175
|
+
expect(copiedContent).toBe(gitignoreContent);
|
|
176
|
+
});
|
|
177
|
+
it('should not create a .gitignore in shadow repo if project .gitignore does not exist', async () => {
|
|
178
|
+
const service = new GitService(projectRoot);
|
|
179
|
+
await service.setupShadowGitRepository();
|
|
180
|
+
const hiddenGitIgnorePath = path.join(repoDir, '.gitignore');
|
|
181
|
+
// An empty string is written if the file doesn't exist.
|
|
182
|
+
const content = await fs.readFile(hiddenGitIgnorePath, 'utf-8');
|
|
183
|
+
expect(content).toBe('');
|
|
199
184
|
});
|
|
200
185
|
it('should throw an error if reading projectRoot .gitignore fails with other errors', async () => {
|
|
201
|
-
const
|
|
202
|
-
|
|
203
|
-
|
|
204
|
-
|
|
205
|
-
|
|
206
|
-
|
|
207
|
-
});
|
|
208
|
-
hoistedMockIsNodeError.mockImplementation((e) => e instanceof Error);
|
|
209
|
-
const service = new GitService(mockProjectRoot);
|
|
210
|
-
await expect(service.setupShadowGitRepository()).rejects.toThrow('Read permission denied');
|
|
186
|
+
const visibleGitIgnorePath = path.join(projectRoot, '.gitignore');
|
|
187
|
+
// Create a directory instead of a file to cause a read error
|
|
188
|
+
await fs.mkdir(visibleGitIgnorePath);
|
|
189
|
+
const service = new GitService(projectRoot);
|
|
190
|
+
// EISDIR is the expected error code on Unix-like systems
|
|
191
|
+
await expect(service.setupShadowGitRepository()).rejects.toThrow(/EISDIR: illegal operation on a directory, read|EBUSY: resource busy or locked, read/);
|
|
211
192
|
});
|
|
212
193
|
it('should make an initial commit if no commits exist in history repo', async () => {
|
|
213
194
|
hoistedMockCheckIsRepo.mockResolvedValue(false);
|
|
214
|
-
const service = new GitService(
|
|
195
|
+
const service = new GitService(projectRoot);
|
|
215
196
|
await service.setupShadowGitRepository();
|
|
216
197
|
expect(hoistedMockCommit).toHaveBeenCalledWith('Initial commit', {
|
|
217
198
|
'--allow-empty': null,
|
|
@@ -219,7 +200,7 @@ describe('GitService', () => {
|
|
|
219
200
|
});
|
|
220
201
|
it('should not make an initial commit if commits already exist', async () => {
|
|
221
202
|
hoistedMockCheckIsRepo.mockResolvedValue(true);
|
|
222
|
-
const service = new GitService(
|
|
203
|
+
const service = new GitService(projectRoot);
|
|
223
204
|
await service.setupShadowGitRepository();
|
|
224
205
|
expect(hoistedMockCommit).not.toHaveBeenCalled();
|
|
225
206
|
});
|