@hibanacloud/core 0.3.15

package/dist/src/test-utils/config.js

@@ -0,0 +1,32 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { Config } from '../config/config.js';
+/**
+ * Default parameters used for {@link FAKE_CONFIG}
+ */
+export const DEFAULT_CONFIG_PARAMETERS = {
+    usageStatisticsEnabled: true,
+    debugMode: false,
+    sessionId: 'test-session-id',
+    proxy: undefined,
+    model: 'qwen-9001-super-duper',
+    targetDir: '/',
+    cwd: '/',
+};
+/**
+ * Produces a config.  Default paramters are set to
+ * {@link DEFAULT_CONFIG_PARAMETERS}, optionally, fields can be specified to
+ * override those defaults.
+ */
+export function makeFakeConfig(config = {
+    ...DEFAULT_CONFIG_PARAMETERS,
+}) {
+    return new Config({
+        ...DEFAULT_CONFIG_PARAMETERS,
+        ...config,
+    });
+}
+//# sourceMappingURL=config.js.map

package/dist/src/test-utils/config.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"config.js","sourceRoot":"","sources":["../../../src/test-utils/config.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAGH,OAAO,EAAE,MAAM,EAAE,MAAM,qBAAqB,CAAC;AAE7C;;GAEG;AACH,MAAM,CAAC,MAAM,yBAAyB,GAAqB;IACzD,sBAAsB,EAAE,IAAI;IAC5B,SAAS,EAAE,KAAK;IAChB,SAAS,EAAE,iBAAiB;IAC5B,KAAK,EAAE,SAAS;IAChB,KAAK,EAAE,uBAAuB;IAC9B,SAAS,EAAE,GAAG;IACd,GAAG,EAAE,GAAG;CACT,CAAC;AAEF;;;;GAIG;AACH,MAAM,UAAU,cAAc,CAC5B,SAAoC;IAClC,GAAG,yBAAyB;CAC7B;IAED,OAAO,IAAI,MAAM,CAAC;QAChB,GAAG,yBAAyB;QAC5B,GAAG,MAAM;KACV,CAAC,CAAC;AACL,CAAC"}

package/dist/src/test-utils/index.d.ts

@@ -0,0 +1,6 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export * from './mock-tool.js';

package/dist/src/test-utils/index.js

@@ -0,0 +1,7 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export * from './mock-tool.js';
+//# sourceMappingURL=index.js.map

package/dist/src/test-utils/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/test-utils/index.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,cAAc,gBAAgB,CAAC"}

package/dist/src/test-utils/mock-tool.d.ts

@@ -0,0 +1,66 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { ModifiableDeclarativeTool, ModifyContext } from '../tools/modifiable-tool.js';
+import type { ToolCallConfirmationDetails, ToolInvocation, ToolResult } from '../tools/tools.js';
+import { BaseDeclarativeTool, BaseToolInvocation } from '../tools/tools.js';
+interface MockToolOptions {
+    name: string;
+    displayName?: string;
+    description?: string;
+    canUpdateOutput?: boolean;
+    isOutputMarkdown?: boolean;
+    shouldConfirmExecute?: (params: {
+        [key: string]: unknown;
+    }, signal: AbortSignal) => Promise<ToolCallConfirmationDetails | false>;
+    execute?: (params: {
+        [key: string]: unknown;
+    }, signal?: AbortSignal, updateOutput?: (output: string) => void) => Promise<ToolResult>;
+    params?: object;
+}
+/**
+ * A highly configurable mock tool for testing purposes.
+ */
+export declare class MockTool extends BaseDeclarativeTool<{
+    [key: string]: unknown;
+}, ToolResult> {
+    shouldConfirmExecute: (params: {
+        [key: string]: unknown;
+    }, signal: AbortSignal) => Promise<ToolCallConfirmationDetails | false>;
+    execute: (params: {
+        [key: string]: unknown;
+    }, signal?: AbortSignal, updateOutput?: (output: string) => void) => Promise<ToolResult>;
+    constructor(options: MockToolOptions);
+    protected createInvocation(params: {
+        [key: string]: unknown;
+    }): ToolInvocation<{
+        [key: string]: unknown;
+    }, ToolResult>;
+}
+export declare const MOCK_TOOL_SHOULD_CONFIRM_EXECUTE: () => Promise<{
+    type: "exec";
+    title: string;
+    command: string;
+    rootCommand: string;
+    onConfirm: () => Promise<void>;
+}>;
+export declare class MockModifiableToolInvocation extends BaseToolInvocation<Record<string, unknown>, ToolResult> {
+    private readonly tool;
+    constructor(tool: MockModifiableTool, params: Record<string, unknown>);
+    execute(_abortSignal: AbortSignal): Promise<ToolResult>;
+    shouldConfirmExecute(_abortSignal: AbortSignal): Promise<ToolCallConfirmationDetails | false>;
+    getDescription(): string;
+}
+/**
+ * Configurable mock modifiable tool for testing.
+ */
+export declare class MockModifiableTool extends BaseDeclarativeTool<Record<string, unknown>, ToolResult> implements ModifiableDeclarativeTool<Record<string, unknown>> {
+    executeFn: (params: Record<string, unknown>) => ToolResult | undefined;
+    shouldConfirm: boolean;
+    constructor(name?: string);
+    getModifyContext(_abortSignal: AbortSignal): ModifyContext<Record<string, unknown>>;
+    protected createInvocation(params: Record<string, unknown>): ToolInvocation<Record<string, unknown>, ToolResult>;
+}
+export {};

package/dist/src/test-utils/mock-tool.js

@@ -0,0 +1,121 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { BaseDeclarativeTool, BaseToolInvocation, Kind, } from '../tools/tools.js';
+class MockToolInvocation extends BaseToolInvocation {
+    tool;
+    constructor(tool, params) {
+        super(params);
+        this.tool = tool;
+    }
+    execute(signal, updateOutput) {
+        if (updateOutput) {
+            return this.tool.execute(this.params, signal, updateOutput);
+        }
+        else {
+            return this.tool.execute(this.params);
+        }
+    }
+    shouldConfirmExecute(abortSignal) {
+        return this.tool.shouldConfirmExecute(this.params, abortSignal);
+    }
+    getDescription() {
+        return `A mock tool invocation for ${this.tool.name}`;
+    }
+}
+/**
+ * A highly configurable mock tool for testing purposes.
+ */
+export class MockTool extends BaseDeclarativeTool {
+    shouldConfirmExecute;
+    execute;
+    constructor(options) {
+        super(options.name, options.displayName ?? options.name, options.description ?? options.name, Kind.Other, options.params, options.isOutputMarkdown ?? false, options.canUpdateOutput ?? false);
+        if (options.shouldConfirmExecute) {
+            this.shouldConfirmExecute = options.shouldConfirmExecute;
+        }
+        else {
+            this.shouldConfirmExecute = () => Promise.resolve(false);
+        }
+        if (options.execute) {
+            this.execute = options.execute;
+        }
+        else {
+            this.execute = () => Promise.resolve({
+                llmContent: `Tool ${this.name} executed successfully.`,
+                returnDisplay: `Tool ${this.name} executed successfully.`,
+            });
+        }
+    }
+    createInvocation(params) {
+        return new MockToolInvocation(this, params);
+    }
+}
+export const MOCK_TOOL_SHOULD_CONFIRM_EXECUTE = () => Promise.resolve({
+    type: 'exec',
+    title: 'Confirm mockTool',
+    command: 'mockTool',
+    rootCommand: 'mockTool',
+    onConfirm: async () => { },
+});
+export class MockModifiableToolInvocation extends BaseToolInvocation {
+    tool;
+    constructor(tool, params) {
+        super(params);
+        this.tool = tool;
+    }
+    async execute(_abortSignal) {
+        const result = this.tool.executeFn(this.params);
+        return (result ?? {
+            llmContent: `Tool ${this.tool.name} executed successfully.`,
+            returnDisplay: `Tool ${this.tool.name} executed successfully.`,
+        });
+    }
+    async shouldConfirmExecute(_abortSignal) {
+        if (this.tool.shouldConfirm) {
+            return {
+                type: 'edit',
+                title: 'Confirm Mock Tool',
+                fileName: 'test.txt',
+                filePath: 'test.txt',
+                fileDiff: 'diff',
+                originalContent: 'originalContent',
+                newContent: 'newContent',
+                onConfirm: async () => { },
+            };
+        }
+        return false;
+    }
+    getDescription() {
+        return `A mock modifiable tool invocation for ${this.tool.name}`;
+    }
+}
+/**
+ * Configurable mock modifiable tool for testing.
+ */
+export class MockModifiableTool extends BaseDeclarativeTool {
+    // Should be overrided in test file. Functionality will be updated in follow
+    // up PR which has MockModifiableTool expect MockTool
+    executeFn = () => undefined;
+    shouldConfirm = true;
+    constructor(name = 'mockModifiableTool') {
+        super(name, name, 'A mock modifiable tool for testing.', Kind.Other, {
+            type: 'object',
+            properties: { param: { type: 'string' } },
+        });
+    }
+    getModifyContext(_abortSignal) {
+        return {
+            getFilePath: () => 'test.txt',
+            getCurrentContent: async () => 'old content',
+            getProposedContent: async () => 'new content',
+            createUpdatedParams: (_oldContent, modifiedProposedContent, _originalParams) => ({ newContent: modifiedProposedContent }),
+        };
+    }
+    createInvocation(params) {
+        return new MockModifiableToolInvocation(this, params);
+    }
+}
+//# sourceMappingURL=mock-tool.js.map

package/dist/src/test-utils/mock-tool.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mock-tool.js","sourceRoot":"","sources":["../../../src/test-utils/mock-tool.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAWH,OAAO,EACL,mBAAmB,EACnB,kBAAkB,EAClB,IAAI,GACL,MAAM,mBAAmB,CAAC;AAoB3B,MAAM,kBAAmB,SAAQ,kBAGhC;IAEoB;IADnB,YACmB,IAAc,EAC/B,MAAkC;QAElC,KAAK,CAAC,MAAM,CAAC,CAAC;QAHG,SAAI,GAAJ,IAAI,CAAU;IAIjC,CAAC;IAED,OAAO,CACL,MAAmB,EACnB,YAAuC;QAEvC,IAAI,YAAY,EAAE,CAAC;YACjB,OAAO,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,EAAE,MAAM,EAAE,YAAY,CAAC,CAAC;QAC9D,CAAC;aAAM,CAAC;YACN,OAAO,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACxC,CAAC;IACH,CAAC;IAEQ,oBAAoB,CAC3B,WAAwB;QAExB,OAAO,IAAI,CAAC,IAAI,CAAC,oBAAoB,CAAC,IAAI,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;IAClE,CAAC;IAED,cAAc;QACZ,OAAO,8BAA8B,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;IACxD,CAAC;CACF;AAED;;GAEG;AACH,MAAM,OAAO,QAAS,SAAQ,mBAG7B;IACC,oBAAoB,CAG8B;IAClD,OAAO,CAIkB;IAEzB,YAAY,OAAwB;QAClC,KAAK,CACH,OAAO,CAAC,IAAI,EACZ,OAAO,CAAC,WAAW,IAAI,OAAO,CAAC,IAAI,EACnC,OAAO,CAAC,WAAW,IAAI,OAAO,CAAC,IAAI,EACnC,IAAI,CAAC,KAAK,EACV,OAAO,CAAC,MAAM,EACd,OAAO,CAAC,gBAAgB,IAAI,KAAK,EACjC,OAAO,CAAC,eAAe,IAAI,KAAK,CACjC,CAAC;QAEF,IAAI,OAAO,CAAC,oBAAoB,EAAE,CAAC;YACjC,IAAI,CAAC,oBAAoB,GAAG,OAAO,CAAC,oBAAoB,CAAC;QAC3D,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,oBAAoB,GAAG,GAAG,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QAC3D,CAAC;QAED,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;YACpB,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC;QACjC,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,OAAO,GAAG,GAAG,EAAE,CAClB,OAAO,CAAC,OAAO,CAAC;gBACd,UAAU,EAAE,QAAQ,IAAI,CAAC,IAAI,yBAAyB;gBACtD,aAAa,EAAE,QAAQ,IAAI,CAAC,IAAI,yBAAyB;aAC1D,CAAC,CAAC;QACP,CAAC;IACH,CAAC;IAES,gBAAgB,CAAC,MAE1B;QACC,OAAO,IAAI,kBAAkB,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;IAC9C,CAAC;CACF;AAED,MAAM,CAAC,MAAM,gCAAgC,GAAG,GAAG,EAAE,CACnD,OAAO,CAAC,OAAO,CAAC;IACd,IAAI,EAAE,MAAe;IACrB,KAAK,EAAE,kBAAkB;IACzB,OAAO,EAAE,UAAU;IACnB,WAAW,EAAE,UAAU;IACvB,SAAS,EAAE,KAAK,IAAI,EAAE,GAAE,CAAC;CAC1B,CAAC,CAAC;AAEL,MAAM,OAAO,4BAA6B,SAAQ,kBAGjD;IAEoB;IADnB,YACmB,IAAwB,EACzC,MAA+B;QAE/B,KAAK,CAAC,MAAM,CAAC,CAAC;QAHG,SAAI,GAAJ,IAAI,CAAoB;IAI3C,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,YAAyB;QACrC,MAAM,MAAM,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAChD,OAAO,CACL,MAAM,IAAI;YACR,UAAU,EAAE,QAAQ,IAAI,CAAC,IAAI,CAAC,IAAI,yBAAyB;YAC3D,aAAa,EAAE,QAAQ,IAAI,CAAC,IAAI,CAAC,IAAI,yBAAyB;SAC/D,CACF,CAAC;IACJ,CAAC;IAEQ,KAAK,CAAC,oBAAoB,CACjC,YAAyB;QAEzB,IAAI,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,CAAC;YAC5B,OAAO;gBACL,IAAI,EAAE,MAAM;gBACZ,KAAK,EAAE,mBAAmB;gBAC1B,QAAQ,EAAE,UAAU;gBACpB,QAAQ,EAAE,UAAU;gBACpB,QAAQ,EAAE,MAAM;gBAChB,eAAe,EAAE,iBAAiB;gBAClC,UAAU,EAAE,YAAY;gBACxB,SAAS,EAAE,KAAK,IAAI,EAAE,GAAE,CAAC;aAC1B,CAAC;QACJ,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,cAAc;QACZ,OAAO,yCAAyC,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;IACnE,CAAC;CACF;AAED;;GAEG;AACH,MAAM,OAAO,kBACX,SAAQ,mBAAwD;IAGhE,4EAA4E;IAC5E,qDAAqD;IACrD,SAAS,GAAgE,GAAG,EAAE,CAC5E,SAAS,CAAC;IACZ,aAAa,GAAG,IAAI,CAAC;IAErB,YAAY,IAAI,GAAG,oBAAoB;QACrC,KAAK,CAAC,IAAI,EAAE,IAAI,EAAE,qCAAqC,EAAE,IAAI,CAAC,KAAK,EAAE;YACnE,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE;SAC1C,CAAC,CAAC;IACL,CAAC;IAED,gBAAgB,CACd,YAAyB;QAEzB,OAAO;YACL,WAAW,EAAE,GAAG,EAAE,CAAC,UAAU;YAC7B,iBAAiB,EAAE,KAAK,IAAI,EAAE,CAAC,aAAa;YAC5C,kBAAkB,EAAE,KAAK,IAAI,EAAE,CAAC,aAAa;YAC7C,mBAAmB,EAAE,CACnB,WAAmB,EACnB,uBAA+B,EAC/B,eAAwC,EACxC,EAAE,CAAC,CAAC,EAAE,UAAU,EAAE,uBAAuB,EAAE,CAAC;SAC/C,CAAC;IACJ,CAAC;IAES,gBAAgB,CACxB,MAA+B;QAE/B,OAAO,IAAI,4BAA4B,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;IACxD,CAAC;CACF"}

package/dist/src/test-utils/mockWorkspaceContext.d.ts

@@ -0,0 +1,13 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { WorkspaceContext } from '../utils/workspaceContext.js';
+/**
+ * Creates a mock WorkspaceContext for testing
+ * @param rootDir The root directory to use for the mock
+ * @param additionalDirs Optional additional directories to include in the workspace
+ * @returns A mock WorkspaceContext instance
+ */
+export declare function createMockWorkspaceContext(rootDir: string, additionalDirs?: string[]): WorkspaceContext;

package/dist/src/test-utils/mockWorkspaceContext.js

@@ -0,0 +1,24 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { vi } from 'vitest';
+/**
+ * Creates a mock WorkspaceContext for testing
+ * @param rootDir The root directory to use for the mock
+ * @param additionalDirs Optional additional directories to include in the workspace
+ * @returns A mock WorkspaceContext instance
+ */
+export function createMockWorkspaceContext(rootDir, additionalDirs = []) {
+    const allDirs = [rootDir, ...additionalDirs];
+    const mockWorkspaceContext = {
+        addDirectory: vi.fn(),
+        getDirectories: vi.fn().mockReturnValue(allDirs),
+        isPathWithinWorkspace: vi
+            .fn()
+            .mockImplementation((path) => allDirs.some((dir) => path.startsWith(dir))),
+    };
+    return mockWorkspaceContext;
+}
+//# sourceMappingURL=mockWorkspaceContext.js.map

package/dist/src/test-utils/mockWorkspaceContext.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mockWorkspaceContext.js","sourceRoot":"","sources":["../../../src/test-utils/mockWorkspaceContext.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAG5B;;;;;GAKG;AACH,MAAM,UAAU,0BAA0B,CACxC,OAAe,EACf,iBAA2B,EAAE;IAE7B,MAAM,OAAO,GAAG,CAAC,OAAO,EAAE,GAAG,cAAc,CAAC,CAAC;IAE7C,MAAM,oBAAoB,GAAG;QAC3B,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE;QACrB,cAAc,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,OAAO,CAAC;QAChD,qBAAqB,EAAE,EAAE;aACtB,EAAE,EAAE;aACJ,kBAAkB,CAAC,CAAC,IAAY,EAAE,EAAE,CACnC,OAAO,CAAC,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAC5C;KAC2B,CAAC;IAEjC,OAAO,oBAAoB,CAAC;AAC9B,CAAC"}

package/dist/src/tools/diffOptions.d.ts

@@ -0,0 +1,9 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import * as Diff from 'diff';
+import type { DiffStat } from './tools.js';
+export declare const DEFAULT_DIFF_OPTIONS: Diff.PatchOptions;
+export declare function getDiffStat(fileName: string, oldStr: string, aiStr: string, userStr: string): DiffStat;

package/dist/src/tools/diffOptions.js

@@ -0,0 +1,46 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import * as Diff from 'diff';
+export const DEFAULT_DIFF_OPTIONS = {
+    context: 3,
+    ignoreWhitespace: true,
+};
+export function getDiffStat(fileName, oldStr, aiStr, userStr) {
+    const getStats = (patch) => {
+        let addedLines = 0;
+        let removedLines = 0;
+        let addedChars = 0;
+        let removedChars = 0;
+        patch.hunks.forEach((hunk) => {
+            hunk.lines.forEach((line) => {
+                if (line.startsWith('+')) {
+                    addedLines++;
+                    addedChars += line.length - 1;
+                }
+                else if (line.startsWith('-')) {
+                    removedLines++;
+                    removedChars += line.length - 1;
+                }
+            });
+        });
+        return { addedLines, removedLines, addedChars, removedChars };
+    };
+    const modelPatch = Diff.structuredPatch(fileName, fileName, oldStr, aiStr, 'Current', 'Proposed', DEFAULT_DIFF_OPTIONS);
+    const modelStats = getStats(modelPatch);
+    const userPatch = Diff.structuredPatch(fileName, fileName, aiStr, userStr, 'Proposed', 'User', DEFAULT_DIFF_OPTIONS);
+    const userStats = getStats(userPatch);
+    return {
+        model_added_lines: modelStats.addedLines,
+        model_removed_lines: modelStats.removedLines,
+        model_added_chars: modelStats.addedChars,
+        model_removed_chars: modelStats.removedChars,
+        user_added_lines: userStats.addedLines,
+        user_removed_lines: userStats.removedLines,
+        user_added_chars: userStats.addedChars,
+        user_removed_chars: userStats.removedChars,
+    };
+}
+//# sourceMappingURL=diffOptions.js.map

package/dist/src/tools/diffOptions.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"diffOptions.js","sourceRoot":"","sources":["../../../src/tools/diffOptions.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,KAAK,IAAI,MAAM,MAAM,CAAC;AAG7B,MAAM,CAAC,MAAM,oBAAoB,GAAsB;IACrD,OAAO,EAAE,CAAC;IACV,gBAAgB,EAAE,IAAI;CACvB,CAAC;AAEF,MAAM,UAAU,WAAW,CACzB,QAAgB,EAChB,MAAc,EACd,KAAa,EACb,OAAe;IAEf,MAAM,QAAQ,GAAG,CAAC,KAAsB,EAAE,EAAE;QAC1C,IAAI,UAAU,GAAG,CAAC,CAAC;QACnB,IAAI,YAAY,GAAG,CAAC,CAAC;QACrB,IAAI,UAAU,GAAG,CAAC,CAAC;QACnB,IAAI,YAAY,GAAG,CAAC,CAAC;QAErB,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,IAAe,EAAE,EAAE;YACtC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,IAAY,EAAE,EAAE;gBAClC,IAAI,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;oBACzB,UAAU,EAAE,CAAC;oBACb,UAAU,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC;gBAChC,CAAC;qBAAM,IAAI,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;oBAChC,YAAY,EAAE,CAAC;oBACf,YAAY,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC;gBAClC,CAAC;YACH,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QACH,OAAO,EAAE,UAAU,EAAE,YAAY,EAAE,UAAU,EAAE,YAAY,EAAE,CAAC;IAChE,CAAC,CAAC;IAEF,MAAM,UAAU,GAAG,IAAI,CAAC,eAAe,CACrC,QAAQ,EACR,QAAQ,EACR,MAAM,EACN,KAAK,EACL,SAAS,EACT,UAAU,EACV,oBAAoB,CACrB,CAAC;IACF,MAAM,UAAU,GAAG,QAAQ,CAAC,UAAU,CAAC,CAAC;IAExC,MAAM,SAAS,GAAG,IAAI,CAAC,eAAe,CACpC,QAAQ,EACR,QAAQ,EACR,KAAK,EACL,OAAO,EACP,UAAU,EACV,MAAM,EACN,oBAAoB,CACrB,CAAC;IACF,MAAM,SAAS,GAAG,QAAQ,CAAC,SAAS,CAAC,CAAC;IAEtC,OAAO;QACL,iBAAiB,EAAE,UAAU,CAAC,UAAU;QACxC,mBAAmB,EAAE,UAAU,CAAC,YAAY;QAC5C,iBAAiB,EAAE,UAAU,CAAC,UAAU;QACxC,mBAAmB,EAAE,UAAU,CAAC,YAAY;QAC5C,gBAAgB,EAAE,SAAS,CAAC,UAAU;QACtC,kBAAkB,EAAE,SAAS,CAAC,YAAY;QAC1C,gBAAgB,EAAE,SAAS,CAAC,UAAU;QACtC,kBAAkB,EAAE,SAAS,CAAC,YAAY;KAC3C,CAAC;AACJ,CAAC"}

package/dist/src/tools/diffOptions.test.d.ts

@@ -0,0 +1,6 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export {};

package/dist/src/tools/diffOptions.test.js

@@ -0,0 +1,155 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, expect, it } from 'vitest';
+import { getDiffStat } from './diffOptions.js';
+describe('getDiffStat', () => {
+    const fileName = 'test.txt';
+    it('should return 0 for all stats when there are no changes', () => {
+        const oldStr = 'line1\nline2\n';
+        const aiStr = 'line1\nline2\n';
+        const userStr = 'line1\nline2\n';
+        const diffStat = getDiffStat(fileName, oldStr, aiStr, userStr);
+        expect(diffStat).toEqual({
+            model_added_lines: 0,
+            model_removed_lines: 0,
+            model_added_chars: 0,
+            model_removed_chars: 0,
+            user_added_lines: 0,
+            user_removed_lines: 0,
+            user_added_chars: 0,
+            user_removed_chars: 0,
+        });
+    });
+    it('should correctly report model additions', () => {
+        const oldStr = 'line1\nline2\n';
+        const aiStr = 'line1\nline2\nline3\n';
+        const userStr = 'line1\nline2\nline3\n';
+        const diffStat = getDiffStat(fileName, oldStr, aiStr, userStr);
+        expect(diffStat).toEqual({
+            model_added_lines: 1,
+            model_removed_lines: 0,
+            model_added_chars: 5,
+            model_removed_chars: 0,
+            user_added_lines: 0,
+            user_removed_lines: 0,
+            user_added_chars: 0,
+            user_removed_chars: 0,
+        });
+    });
+    it('should correctly report model removals', () => {
+        const oldStr = 'line1\nline2\nline3\n';
+        const aiStr = 'line1\nline3\n';
+        const userStr = 'line1\nline3\n';
+        const diffStat = getDiffStat(fileName, oldStr, aiStr, userStr);
+        expect(diffStat).toEqual({
+            model_added_lines: 0,
+            model_removed_lines: 1,
+            model_added_chars: 0,
+            model_removed_chars: 5,
+            user_added_lines: 0,
+            user_removed_lines: 0,
+            user_added_chars: 0,
+            user_removed_chars: 0,
+        });
+    });
+    it('should correctly report model modifications', () => {
+        const oldStr = 'line1\nline2\nline3\n';
+        const aiStr = 'line1\nline_two\nline3\n';
+        const userStr = 'line1\nline_two\nline3\n';
+        const diffStat = getDiffStat(fileName, oldStr, aiStr, userStr);
+        expect(diffStat).toEqual({
+            model_added_lines: 1,
+            model_removed_lines: 1,
+            model_added_chars: 8,
+            model_removed_chars: 5,
+            user_added_lines: 0,
+            user_removed_lines: 0,
+            user_added_chars: 0,
+            user_removed_chars: 0,
+        });
+    });
+    it('should correctly report user additions', () => {
+        const oldStr = 'line1\nline2\n';
+        const aiStr = 'line1\nline2\nline3\n';
+        const userStr = 'line1\nline2\nline3\nline4\n';
+        const diffStat = getDiffStat(fileName, oldStr, aiStr, userStr);
+        expect(diffStat).toEqual({
+            model_added_lines: 1,
+            model_removed_lines: 0,
+            model_added_chars: 5,
+            model_removed_chars: 0,
+            user_added_lines: 1,
+            user_removed_lines: 0,
+            user_added_chars: 5,
+            user_removed_chars: 0,
+        });
+    });
+    it('should correctly report user removals', () => {
+        const oldStr = 'line1\nline2\n';
+        const aiStr = 'line1\nline2\nline3\n';
+        const userStr = 'line1\nline2\n';
+        const diffStat = getDiffStat(fileName, oldStr, aiStr, userStr);
+        expect(diffStat).toEqual({
+            model_added_lines: 1,
+            model_removed_lines: 0,
+            model_added_chars: 5,
+            model_removed_chars: 0,
+            user_added_lines: 0,
+            user_removed_lines: 1,
+            user_added_chars: 0,
+            user_removed_chars: 5,
+        });
+    });
+    it('should correctly report user modifications', () => {
+        const oldStr = 'line1\nline2\n';
+        const aiStr = 'line1\nline2\nline3\n';
+        const userStr = 'line1\nline2\nline_three\n';
+        const diffStat = getDiffStat(fileName, oldStr, aiStr, userStr);
+        expect(diffStat).toEqual({
+            model_added_lines: 1,
+            model_removed_lines: 0,
+            model_added_chars: 5,
+            model_removed_chars: 0,
+            user_added_lines: 1,
+            user_removed_lines: 1,
+            user_added_chars: 10,
+            user_removed_chars: 5,
+        });
+    });
+    it('should handle complex changes from both model and user', () => {
+        const oldStr = 'line1\nline2\nline3\nline4\n';
+        const aiStr = 'line_one\nline2\nline_three\nline4\n';
+        const userStr = 'line_one\nline_two\nline_three\nline4\nline5\n';
+        const diffStat = getDiffStat(fileName, oldStr, aiStr, userStr);
+        expect(diffStat).toEqual({
+            model_added_lines: 2,
+            model_removed_lines: 2,
+            model_added_chars: 18,
+            model_removed_chars: 10,
+            user_added_lines: 2,
+            user_removed_lines: 1,
+            user_added_chars: 13,
+            user_removed_chars: 5,
+        });
+    });
+    it('should report a single line modification as one addition and one removal', () => {
+        const oldStr = 'hello world';
+        const aiStr = 'hello universe';
+        const userStr = 'hello universe';
+        const diffStat = getDiffStat(fileName, oldStr, aiStr, userStr);
+        expect(diffStat).toEqual({
+            model_added_lines: 1,
+            model_removed_lines: 1,
+            model_added_chars: 14,
+            model_removed_chars: 11,
+            user_added_lines: 0,
+            user_removed_lines: 0,
+            user_added_chars: 0,
+            user_removed_chars: 0,
+        });
+    });
+});
+//# sourceMappingURL=diffOptions.test.js.map

package/dist/src/tools/diffOptions.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"diffOptions.test.js","sourceRoot":"","sources":["../../../src/tools/diffOptions.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAE/C,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;IAC3B,MAAM,QAAQ,GAAG,UAAU,CAAC;IAE5B,EAAE,CAAC,yDAAyD,EAAE,GAAG,EAAE;QACjE,MAAM,MAAM,GAAG,gBAAgB,CAAC;QAChC,MAAM,KAAK,GAAG,gBAAgB,CAAC;QAC/B,MAAM,OAAO,GAAG,gBAAgB,CAAC;QACjC,MAAM,QAAQ,GAAG,WAAW,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,OAAO,CAAC,CAAC;QAC/D,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC;YACvB,iBAAiB,EAAE,CAAC;YACpB,mBAAmB,EAAE,CAAC;YACtB,iBAAiB,EAAE,CAAC;YACpB,mBAAmB,EAAE,CAAC;YACtB,gBAAgB,EAAE,CAAC;YACnB,kBAAkB,EAAE,CAAC;YACrB,gBAAgB,EAAE,CAAC;YACnB,kBAAkB,EAAE,CAAC;SACtB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;QACjD,MAAM,MAAM,GAAG,gBAAgB,CAAC;QAChC,MAAM,KAAK,GAAG,uBAAuB,CAAC;QACtC,MAAM,OAAO,GAAG,uBAAuB,CAAC;QACxC,MAAM,QAAQ,GAAG,WAAW,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,OAAO,CAAC,CAAC;QAC/D,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC;YACvB,iBAAiB,EAAE,CAAC;YACpB,mBAAmB,EAAE,CAAC;YACtB,iBAAiB,EAAE,CAAC;YACpB,mBAAmB,EAAE,CAAC;YACtB,gBAAgB,EAAE,CAAC;YACnB,kBAAkB,EAAE,CAAC;YACrB,gBAAgB,EAAE,CAAC;YACnB,kBAAkB,EAAE,CAAC;SACtB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;QAChD,MAAM,MAAM,GAAG,uBAAuB,CAAC;QACvC,MAAM,KAAK,GAAG,gBAAgB,CAAC;QAC/B,MAAM,OAAO,GAAG,gBAAgB,CAAC;QACjC,MAAM,QAAQ,GAAG,WAAW,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,OAAO,CAAC,CAAC;QAC/D,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC;YACvB,iBAAiB,EAAE,CAAC;YACpB,mBAAmB,EAAE,CAAC;YACtB,iBAAiB,EAAE,CAAC;YACpB,mBAAmB,EAAE,CAAC;YACtB,gBAAgB,EAAE,CAAC;YACnB,kBAAkB,EAAE,CAAC;YACrB,gBAAgB,EAAE,CAAC;YACnB,kBAAkB,EAAE,CAAC;SACtB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;QACrD,MAAM,MAAM,GAAG,uBAAuB,CAAC;QACvC,MAAM,KAAK,GAAG,0BAA0B,CAAC;QACzC,MAAM,OAAO,GAAG,0BAA0B,CAAC;QAC3C,MAAM,QAAQ,GAAG,WAAW,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,OAAO,CAAC,CAAC;QAC/D,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC;YACvB,iBAAiB,EAAE,CAAC;YACpB,mBAAmB,EAAE,CAAC;YACtB,iBAAiB,EAAE,CAAC;YACpB,mBAAmB,EAAE,CAAC;YACtB,gBAAgB,EAAE,CAAC;YACnB,kBAAkB,EAAE,CAAC;YACrB,gBAAgB,EAAE,CAAC;YACnB,kBAAkB,EAAE,CAAC;SACtB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;QAChD,MAAM,MAAM,GAAG,gBAAgB,CAAC;QAChC,MAAM,KAAK,GAAG,uBAAuB,CAAC;QACtC,MAAM,OAAO,GAAG,8BAA8B,CAAC;QAC/C,MAAM,QAAQ,GAAG,WAAW,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,OAAO,CAAC,CAAC;QAC/D,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC;YACvB,iBAAiB,EAAE,CAAC;YACpB,mBAAmB,EAAE,CAAC;YACtB,iBAAiB,EAAE,CAAC;YACpB,mBAAmB,EAAE,CAAC;YACtB,gBAAgB,EAAE,CAAC;YACnB,kBAAkB,EAAE,CAAC;YACrB,gBAAgB,EAAE,CAAC;YACnB,kBAAkB,EAAE,CAAC;SACtB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;QAC/C,MAAM,MAAM,GAAG,gBAAgB,CAAC;QAChC,MAAM,KAAK,GAAG,uBAAuB,CAAC;QACtC,MAAM,OAAO,GAAG,gBAAgB,CAAC;QACjC,MAAM,QAAQ,GAAG,WAAW,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,OAAO,CAAC,CAAC;QAC/D,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC;YACvB,iBAAiB,EAAE,CAAC;YACpB,mBAAmB,EAAE,CAAC;YACtB,iBAAiB,EAAE,CAAC;YACpB,mBAAmB,EAAE,CAAC;YACtB,gBAAgB,EAAE,CAAC;YACnB,kBAAkB,EAAE,CAAC;YACrB,gBAAgB,EAAE,CAAC;YACnB,kBAAkB,EAAE,CAAC;SACtB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;QACpD,MAAM,MAAM,GAAG,gBAAgB,CAAC;QAChC,MAAM,KAAK,GAAG,uBAAuB,CAAC;QACtC,MAAM,OAAO,GAAG,4BAA4B,CAAC;QAC7C,MAAM,QAAQ,GAAG,WAAW,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,OAAO,CAAC,CAAC;QAC/D,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC;YACvB,iBAAiB,EAAE,CAAC;YACpB,mBAAmB,EAAE,CAAC;YACtB,iBAAiB,EAAE,CAAC;YACpB,mBAAmB,EAAE,CAAC;YACtB,gBAAgB,EAAE,CAAC;YACnB,kBAAkB,EAAE,CAAC;YACrB,gBAAgB,EAAE,EAAE;YACpB,kBAAkB,EAAE,CAAC;SACtB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wDAAwD,EAAE,GAAG,EAAE;QAChE,MAAM,MAAM,GAAG,8BAA8B,CAAC;QAC9C,MAAM,KAAK,GAAG,sCAAsC,CAAC;QACrD,MAAM,OAAO,GAAG,gDAAgD,CAAC;QACjE,MAAM,QAAQ,GAAG,WAAW,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,OAAO,CAAC,CAAC;QAC/D,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC;YACvB,iBAAiB,EAAE,CAAC;YACpB,mBAAmB,EAAE,CAAC;YACtB,iBAAiB,EAAE,EAAE;YACrB,mBAAmB,EAAE,EAAE;YACvB,gBAAgB,EAAE,CAAC;YACnB,kBAAkB,EAAE,CAAC;YACrB,gBAAgB,EAAE,EAAE;YACpB,kBAAkB,EAAE,CAAC;SACtB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0EAA0E,EAAE,GAAG,EAAE;QAClF,MAAM,MAAM,GAAG,aAAa,CAAC;QAC7B,MAAM,KAAK,GAAG,gBAAgB,CAAC;QAC/B,MAAM,OAAO,GAAG,gBAAgB,CAAC;QACjC,MAAM,QAAQ,GAAG,WAAW,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,OAAO,CAAC,CAAC;QAC/D,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC;YACvB,iBAAiB,EAAE,CAAC;YACpB,mBAAmB,EAAE,CAAC;YACtB,iBAAiB,EAAE,EAAE;YACrB,mBAAmB,EAAE,EAAE;YACvB,gBAAgB,EAAE,CAAC;YACnB,kBAAkB,EAAE,CAAC;YACrB,gBAAgB,EAAE,CAAC;YACnB,kBAAkB,EAAE,CAAC;SACtB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/tools/edit.d.ts

@@ -0,0 +1,55 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { ToolInvocation, ToolResult } from './tools.js';
+import { BaseDeclarativeTool } from './tools.js';
+import type { Config } from '../config/config.js';
+import type { ModifiableDeclarativeTool, ModifyContext } from './modifiable-tool.js';
+export declare function applyReplacement(currentContent: string | null, oldString: string, newString: string, isNewFile: boolean): string;
+/**
+ * Parameters for the Edit tool
+ */
+export interface EditToolParams {
+    /**
+     * The absolute path to the file to modify
+     */
+    file_path: string;
+    /**
+     * The text to replace
+     */
+    old_string: string;
+    /**
+     * The text to replace it with
+     */
+    new_string: string;
+    /**
+     * Replace every occurrence of old_string instead of requiring a unique match.
+     */
+    replace_all?: boolean;
+    /**
+     * Whether the edit was modified manually by the user.
+     */
+    modified_by_user?: boolean;
+    /**
+     * Initially proposed content.
+     */
+    ai_proposed_content?: string;
+}
+/**
+ * Implementation of the Edit tool logic
+ */
+export declare class EditTool extends BaseDeclarativeTool<EditToolParams, ToolResult> implements ModifiableDeclarativeTool<EditToolParams> {
+    private readonly config;
+    static readonly Name: "edit";
+    constructor(config: Config);
+    /**
+     * Validates the parameters for the Edit tool
+     * @param params Parameters to validate
+     * @returns Error message string or null if valid
+     */
+    protected validateToolParamValues(params: EditToolParams): string | null;
+    protected createInvocation(params: EditToolParams): ToolInvocation<EditToolParams, ToolResult>;
+    getModifyContext(_: AbortSignal): ModifyContext<EditToolParams>;
+}