@hibanacloud/core 0.3.15

package/dist/src/utils/testUtils.js

@@ -0,0 +1,70 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+/**
+ * Testing utilities for simulating 429 errors in unit tests
+ */
+let requestCounter = 0;
+let simulate429Enabled = false;
+let simulate429AfterRequests = 0;
+let simulate429ForAuthType;
+let fallbackOccurred = false;
+/**
+ * Check if we should simulate a 429 error for the current request
+ */
+export function shouldSimulate429(authType) {
+    if (!simulate429Enabled || fallbackOccurred) {
+        return false;
+    }
+    // If auth type filter is set, only simulate for that auth type
+    if (simulate429ForAuthType && authType !== simulate429ForAuthType) {
+        return false;
+    }
+    requestCounter++;
+    // If afterRequests is set, only simulate after that many requests
+    if (simulate429AfterRequests > 0) {
+        return requestCounter > simulate429AfterRequests;
+    }
+    // Otherwise, simulate for every request
+    return true;
+}
+/**
+ * Reset the request counter (useful for tests)
+ */
+export function resetRequestCounter() {
+    requestCounter = 0;
+}
+/**
+ * Disable 429 simulation after successful fallback
+ */
+export function disableSimulationAfterFallback() {
+    fallbackOccurred = true;
+}
+/**
+ * Create a simulated 429 error response
+ */
+export function createSimulated429Error() {
+    const error = new Error('Rate limit exceeded (simulated)');
+    error.status = 429;
+    return error;
+}
+/**
+ * Reset simulation state when switching auth methods
+ */
+export function resetSimulationState() {
+    fallbackOccurred = false;
+    resetRequestCounter();
+}
+/**
+ * Enable/disable 429 simulation programmatically (for tests)
+ */
+export function setSimulate429(enabled, afterRequests = 0, forAuthType) {
+    simulate429Enabled = enabled;
+    simulate429AfterRequests = afterRequests;
+    simulate429ForAuthType = forAuthType;
+    fallbackOccurred = false; // Reset fallback state when simulation is re-enabled
+    resetRequestCounter();
+}
+//# sourceMappingURL=testUtils.js.map

package/dist/src/utils/testUtils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"testUtils.js","sourceRoot":"","sources":["../../../src/utils/testUtils.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH;;GAEG;AAEH,IAAI,cAAc,GAAG,CAAC,CAAC;AACvB,IAAI,kBAAkB,GAAG,KAAK,CAAC;AAC/B,IAAI,wBAAwB,GAAG,CAAC,CAAC;AACjC,IAAI,sBAA0C,CAAC;AAC/C,IAAI,gBAAgB,GAAG,KAAK,CAAC;AAE7B;;GAEG;AACH,MAAM,UAAU,iBAAiB,CAAC,QAAiB;IACjD,IAAI,CAAC,kBAAkB,IAAI,gBAAgB,EAAE,CAAC;QAC5C,OAAO,KAAK,CAAC;IACf,CAAC;IAED,+DAA+D;IAC/D,IAAI,sBAAsB,IAAI,QAAQ,KAAK,sBAAsB,EAAE,CAAC;QAClE,OAAO,KAAK,CAAC;IACf,CAAC;IAED,cAAc,EAAE,CAAC;IAEjB,kEAAkE;IAClE,IAAI,wBAAwB,GAAG,CAAC,EAAE,CAAC;QACjC,OAAO,cAAc,GAAG,wBAAwB,CAAC;IACnD,CAAC;IAED,wCAAwC;IACxC,OAAO,IAAI,CAAC;AACd,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,mBAAmB;IACjC,cAAc,GAAG,CAAC,CAAC;AACrB,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,8BAA8B;IAC5C,gBAAgB,GAAG,IAAI,CAAC;AAC1B,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,uBAAuB;IACrC,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,iCAAiC,CAExD,CAAC;IACF,KAAK,CAAC,MAAM,GAAG,GAAG,CAAC;IACnB,OAAO,KAAK,CAAC;AACf,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,oBAAoB;IAClC,gBAAgB,GAAG,KAAK,CAAC;IACzB,mBAAmB,EAAE,CAAC;AACxB,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,cAAc,CAC5B,OAAgB,EAChB,aAAa,GAAG,CAAC,EACjB,WAAoB;IAEpB,kBAAkB,GAAG,OAAO,CAAC;IAC7B,wBAAwB,GAAG,aAAa,CAAC;IACzC,sBAAsB,GAAG,WAAW,CAAC;IACrC,gBAAgB,GAAG,KAAK,CAAC,CAAC,qDAAqD;IAC/E,mBAAmB,EAAE,CAAC;AACxB,CAAC"}

package/dist/src/utils/textUtils.d.ts

@@ -0,0 +1,18 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+/**
+ * Safely replaces text with literal strings, avoiding ECMAScript GetSubstitution issues.
+ * Escapes $ characters to prevent template interpretation.
+ */
+export declare function safeLiteralReplace(str: string, oldString: string, newString: string): string;
+/**
+ * Checks if a Buffer is likely binary by testing for the presence of a NULL byte.
+ * The presence of a NULL byte is a strong indicator that the data is not plain text.
+ * @param data The Buffer to check.
+ * @param sampleSize The number of bytes from the start of the buffer to test.
+ * @returns True if a NULL byte is found, false otherwise.
+ */
+export declare function isBinary(data: Buffer | null | undefined, sampleSize?: number): boolean;

package/dist/src/utils/textUtils.js

@@ -0,0 +1,42 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+/**
+ * Safely replaces text with literal strings, avoiding ECMAScript GetSubstitution issues.
+ * Escapes $ characters to prevent template interpretation.
+ */
+export function safeLiteralReplace(str, oldString, newString) {
+    if (oldString === '' || !str.includes(oldString)) {
+        return str;
+    }
+    if (!newString.includes('$')) {
+        return str.replaceAll(oldString, newString);
+    }
+    const escapedNewString = newString.replaceAll('$', '$$$$');
+    return str.replaceAll(oldString, escapedNewString);
+}
+/**
+ * Checks if a Buffer is likely binary by testing for the presence of a NULL byte.
+ * The presence of a NULL byte is a strong indicator that the data is not plain text.
+ * @param data The Buffer to check.
+ * @param sampleSize The number of bytes from the start of the buffer to test.
+ * @returns True if a NULL byte is found, false otherwise.
+ */
+export function isBinary(data, sampleSize = 512) {
+    if (!data) {
+        return false;
+    }
+    const sample = data.length > sampleSize ? data.subarray(0, sampleSize) : data;
+    for (const byte of sample) {
+        // The presence of a NULL byte (0x00) is one of the most reliable
+        // indicators of a binary file. Text files should not contain them.
+        if (byte === 0) {
+            return true;
+        }
+    }
+    // If no NULL bytes were found in the sample, we assume it's text.
+    return false;
+}
+//# sourceMappingURL=textUtils.js.map

package/dist/src/utils/textUtils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"textUtils.js","sourceRoot":"","sources":["../../../src/utils/textUtils.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH;;;GAGG;AACH,MAAM,UAAU,kBAAkB,CAChC,GAAW,EACX,SAAiB,EACjB,SAAiB;IAEjB,IAAI,SAAS,KAAK,EAAE,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;QACjD,OAAO,GAAG,CAAC;IACb,CAAC;IAED,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;QAC7B,OAAO,GAAG,CAAC,UAAU,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC;IAC9C,CAAC;IAED,MAAM,gBAAgB,GAAG,SAAS,CAAC,UAAU,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;IAC3D,OAAO,GAAG,CAAC,UAAU,CAAC,SAAS,EAAE,gBAAgB,CAAC,CAAC;AACrD,CAAC;AAED;;;;;;GAMG;AACH,MAAM,UAAU,QAAQ,CACtB,IAA+B,EAC/B,UAAU,GAAG,GAAG;IAEhB,IAAI,CAAC,IAAI,EAAE,CAAC;QACV,OAAO,KAAK,CAAC;IACf,CAAC;IAED,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,GAAG,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,EAAE,UAAU,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;IAE9E,KAAK,MAAM,IAAI,IAAI,MAAM,EAAE,CAAC;QAC1B,iEAAiE;QACjE,mEAAmE;QACnE,IAAI,IAAI,KAAK,CAAC,EAAE,CAAC;YACf,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;IAED,kEAAkE;IAClE,OAAO,KAAK,CAAC;AACf,CAAC"}

package/dist/src/utils/textUtils.test.d.ts

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

package/dist/src/utils/textUtils.test.js

@@ -0,0 +1,59 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect } from 'vitest';
+import { safeLiteralReplace } from './textUtils.js';
+describe('safeLiteralReplace', () => {
+    it('returns original string when oldString empty or not found', () => {
+        expect(safeLiteralReplace('abc', '', 'X')).toBe('abc');
+        expect(safeLiteralReplace('abc', 'z', 'X')).toBe('abc');
+    });
+    it('fast path when newString has no $', () => {
+        expect(safeLiteralReplace('abc', 'b', 'X')).toBe('aXc');
+    });
+    it('treats $ literally', () => {
+        expect(safeLiteralReplace('foo', 'foo', "bar$'baz")).toBe("bar$'baz");
+    });
+    it("does not interpret replacement patterns like $&, $', $` and $1", () => {
+        expect(safeLiteralReplace('hello', 'hello', '$&-replacement')).toBe('$&-replacement');
+        expect(safeLiteralReplace('mid', 'mid', 'new$`content')).toBe('new$`content');
+        expect(safeLiteralReplace('test', 'test', '$1$2value')).toBe('$1$2value');
+    });
+    it('preserves end-of-line $ in regex-like text', () => {
+        const current = "| select('match', '^[sv]d[a-z]$')";
+        const oldStr = "'^[sv]d[a-z]$'";
+        const newStr = "'^[sv]d[a-z]$' # updated";
+        const expected = "| select('match', '^[sv]d[a-z]$' # updated)";
+        expect(safeLiteralReplace(current, oldStr, newStr)).toBe(expected);
+    });
+    it('handles multiple $ characters', () => {
+        expect(safeLiteralReplace('x', 'x', '$$$')).toBe('$$$');
+    });
+    it('preserves pre-escaped $$ literally', () => {
+        expect(safeLiteralReplace('x', 'x', '$$value')).toBe('$$value');
+    });
+    it('handles complex malicious patterns from PR #7871', () => {
+        const original = 'The price is PRICE.';
+        const result = safeLiteralReplace(original, 'PRICE', "$& Wow, that's a lot! $'");
+        expect(result).toBe("The price is $& Wow, that's a lot! $'.");
+    });
+    it('handles multiple replacements correctly', () => {
+        const text = 'Replace FOO and FOO again';
+        const result = safeLiteralReplace(text, 'FOO', '$100');
+        expect(result).toBe('Replace $100 and $100 again');
+    });
+    it('preserves $ at different positions', () => {
+        expect(safeLiteralReplace('test', 'test', '$')).toBe('$');
+        expect(safeLiteralReplace('test', 'test', 'prefix$')).toBe('prefix$');
+        expect(safeLiteralReplace('test', 'test', '$suffix')).toBe('$suffix');
+    });
+    it('handles edge case with $$$$', () => {
+        expect(safeLiteralReplace('x', 'x', '$$$$')).toBe('$$$$');
+    });
+    it('handles newString with only dollar signs', () => {
+        expect(safeLiteralReplace('abc', 'b', '$$')).toBe('a$$c');
+    });
+});
+//# sourceMappingURL=textUtils.test.js.map

package/dist/src/utils/textUtils.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"textUtils.test.js","sourceRoot":"","sources":["../../../src/utils/textUtils.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,kBAAkB,EAAE,MAAM,gBAAgB,CAAC;AAEpD,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;IAClC,EAAE,CAAC,2DAA2D,EAAE,GAAG,EAAE;QACnE,MAAM,CAAC,kBAAkB,CAAC,KAAK,EAAE,EAAE,EAAE,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACvD,MAAM,CAAC,kBAAkB,CAAC,KAAK,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC1D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;QAC3C,MAAM,CAAC,kBAAkB,CAAC,KAAK,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC1D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAC5B,MAAM,CAAC,kBAAkB,CAAC,KAAK,EAAE,KAAK,EAAE,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IACxE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gEAAgE,EAAE,GAAG,EAAE;QACxE,MAAM,CAAC,kBAAkB,CAAC,OAAO,EAAE,OAAO,EAAE,gBAAgB,CAAC,CAAC,CAAC,IAAI,CACjE,gBAAgB,CACjB,CAAC;QACF,MAAM,CAAC,kBAAkB,CAAC,KAAK,EAAE,KAAK,EAAE,cAAc,CAAC,CAAC,CAAC,IAAI,CAC3D,cAAc,CACf,CAAC;QACF,MAAM,CAAC,kBAAkB,CAAC,MAAM,EAAE,MAAM,EAAE,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IAC5E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;QACpD,MAAM,OAAO,GAAG,mCAAmC,CAAC;QACpD,MAAM,MAAM,GAAG,gBAAgB,CAAC;QAChC,MAAM,MAAM,GAAG,0BAA0B,CAAC;QAC1C,MAAM,QAAQ,GAAG,6CAA6C,CAAC;QAC/D,MAAM,CAAC,kBAAkB,CAAC,OAAO,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IACrE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+BAA+B,EAAE,GAAG,EAAE;QACvC,MAAM,CAAC,kBAAkB,CAAC,GAAG,EAAE,GAAG,EAAE,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC1D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oCAAoC,EAAE,GAAG,EAAE;QAC5C,MAAM,CAAC,kBAAkB,CAAC,GAAG,EAAE,GAAG,EAAE,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IAClE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;QAC1D,MAAM,QAAQ,GAAG,qBAAqB,CAAC;QACvC,MAAM,MAAM,GAAG,kBAAkB,CAC/B,QAAQ,EACR,OAAO,EACP,0BAA0B,CAC3B,CAAC;QACF,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,wCAAwC,CAAC,CAAC;IAChE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;QACjD,MAAM,IAAI,GAAG,2BAA2B,CAAC;QACzC,MAAM,MAAM,GAAG,kBAAkB,CAAC,IAAI,EAAE,KAAK,EAAE,MAAM,CAAC,CAAC;QACvD,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,6BAA6B,CAAC,CAAC;IACrD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oCAAoC,EAAE,GAAG,EAAE;QAC5C,MAAM,CAAC,kBAAkB,CAAC,MAAM,EAAE,MAAM,EAAE,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC1D,MAAM,CAAC,kBAAkB,CAAC,MAAM,EAAE,MAAM,EAAE,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACtE,MAAM,CAAC,kBAAkB,CAAC,MAAM,EAAE,MAAM,EAAE,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IACxE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6BAA6B,EAAE,GAAG,EAAE;QACrC,MAAM,CAAC,kBAAkB,CAAC,GAAG,EAAE,GAAG,EAAE,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IAC5D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;QAClD,MAAM,CAAC,kBAAkB,CAAC,KAAK,EAAE,GAAG,EAAE,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IAC5D,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/utils/thoughtUtils.d.ts

@@ -0,0 +1,21 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export type ThoughtSummary = {
+    subject: string;
+    description: string;
+};
+/**
+ * Parses a raw thought string into a structured ThoughtSummary object.
+ *
+ * Thoughts are expected to have a bold "subject" part enclosed in double
+ * asterisks (e.g., **Subject**). The rest of the string is considered
+ * the description. This function only parses the first valid subject found.
+ *
+ * @param rawText The raw text of the thought.
+ * @returns A ThoughtSummary object. If no valid subject is found, the entire
+ * string is treated as the description.
+ */
+export declare function parseThought(rawText: string): ThoughtSummary;

package/dist/src/utils/thoughtUtils.js

@@ -0,0 +1,39 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+const START_DELIMITER = '**';
+const END_DELIMITER = '**';
+/**
+ * Parses a raw thought string into a structured ThoughtSummary object.
+ *
+ * Thoughts are expected to have a bold "subject" part enclosed in double
+ * asterisks (e.g., **Subject**). The rest of the string is considered
+ * the description. This function only parses the first valid subject found.
+ *
+ * @param rawText The raw text of the thought.
+ * @returns A ThoughtSummary object. If no valid subject is found, the entire
+ * string is treated as the description.
+ */
+export function parseThought(rawText) {
+    const startIndex = rawText.indexOf(START_DELIMITER);
+    if (startIndex === -1) {
+        // No start delimiter found, the whole text is the description.
+        return { subject: '', description: rawText.trim() };
+    }
+    const endIndex = rawText.indexOf(END_DELIMITER, startIndex + START_DELIMITER.length);
+    if (endIndex === -1) {
+        // Start delimiter found but no end delimiter, so it's not a valid subject.
+        // Treat the entire string as the description.
+        return { subject: '', description: rawText.trim() };
+    }
+    const subject = rawText
+        .substring(startIndex + START_DELIMITER.length, endIndex)
+        .trim();
+    // The description is everything before the start delimiter and after the end delimiter.
+    const description = (rawText.substring(0, startIndex) +
+        rawText.substring(endIndex + END_DELIMITER.length)).trim();
+    return { subject, description };
+}
+//# sourceMappingURL=thoughtUtils.js.map

package/dist/src/utils/thoughtUtils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"thoughtUtils.js","sourceRoot":"","sources":["../../../src/utils/thoughtUtils.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAOH,MAAM,eAAe,GAAG,IAAI,CAAC;AAC7B,MAAM,aAAa,GAAG,IAAI,CAAC;AAE3B;;;;;;;;;;GAUG;AACH,MAAM,UAAU,YAAY,CAAC,OAAe;IAC1C,MAAM,UAAU,GAAG,OAAO,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;IACpD,IAAI,UAAU,KAAK,CAAC,CAAC,EAAE,CAAC;QACtB,+DAA+D;QAC/D,OAAO,EAAE,OAAO,EAAE,EAAE,EAAE,WAAW,EAAE,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC;IACtD,CAAC;IAED,MAAM,QAAQ,GAAG,OAAO,CAAC,OAAO,CAC9B,aAAa,EACb,UAAU,GAAG,eAAe,CAAC,MAAM,CACpC,CAAC;IACF,IAAI,QAAQ,KAAK,CAAC,CAAC,EAAE,CAAC;QACpB,2EAA2E;QAC3E,8CAA8C;QAC9C,OAAO,EAAE,OAAO,EAAE,EAAE,EAAE,WAAW,EAAE,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC;IACtD,CAAC;IAED,MAAM,OAAO,GAAG,OAAO;SACpB,SAAS,CAAC,UAAU,GAAG,eAAe,CAAC,MAAM,EAAE,QAAQ,CAAC;SACxD,IAAI,EAAE,CAAC;IAEV,wFAAwF;IACxF,MAAM,WAAW,GAAG,CAClB,OAAO,CAAC,SAAS,CAAC,CAAC,EAAE,UAAU,CAAC;QAChC,OAAO,CAAC,SAAS,CAAC,QAAQ,GAAG,aAAa,CAAC,MAAM,CAAC,CACnD,CAAC,IAAI,EAAE,CAAC;IAET,OAAO,EAAE,OAAO,EAAE,WAAW,EAAE,CAAC;AAClC,CAAC"}

package/dist/src/utils/thoughtUtils.test.d.ts

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

package/dist/src/utils/thoughtUtils.test.js

@@ -0,0 +1,78 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect } from 'vitest';
+import { parseThought } from './thoughtUtils.js';
+describe('parseThought', () => {
+    it.each([
+        {
+            name: 'a standard thought with subject and description',
+            rawText: '**Subject:** This is the description.',
+            expected: {
+                subject: 'Subject:',
+                description: 'This is the description.',
+            },
+        },
+        {
+            name: 'leading and trailing whitespace in the raw string',
+            rawText: '  **Subject** description with spaces   ',
+            expected: { subject: 'Subject', description: 'description with spaces' },
+        },
+        {
+            name: 'whitespace surrounding the subject content',
+            rawText: '** Subject  **',
+            expected: { subject: 'Subject', description: '' },
+        },
+        {
+            name: 'a thought with only a subject',
+            rawText: '**Only Subject**',
+            expected: { subject: 'Only Subject', description: '' },
+        },
+        {
+            name: 'a thought with only a description (no subject)',
+            rawText: 'This is just a description.',
+            expected: { subject: '', description: 'This is just a description.' },
+        },
+        {
+            name: 'an empty string input',
+            rawText: '',
+            expected: { subject: '', description: '' },
+        },
+        {
+            name: 'newlines within the subject and description',
+            rawText: '**Multi-line\nSubject**\nHere is a description\nspread across lines.',
+            expected: {
+                subject: 'Multi-line\nSubject',
+                description: 'Here is a description\nspread across lines.',
+            },
+        },
+        {
+            name: 'only the first subject if multiple are present',
+            rawText: '**First** some text **Second**',
+            expected: { subject: 'First', description: 'some text **Second**' },
+        },
+        {
+            name: 'text before and after the subject',
+            rawText: 'Prefix text **Subject** Suffix text.',
+            expected: {
+                subject: 'Subject',
+                description: 'Prefix text  Suffix text.',
+            },
+        },
+        {
+            name: 'an unclosed subject tag',
+            rawText: 'Text with **an unclosed subject',
+            expected: { subject: '', description: 'Text with **an unclosed subject' },
+        },
+        {
+            name: 'an empty subject tag',
+            rawText: 'A thought with **** in the middle.',
+            expected: { subject: '', description: 'A thought with  in the middle.' },
+        },
+    ])('should correctly parse $name', ({ rawText, expected }) => {
+        expect(parseThought(rawText)).toEqual(expected);
+    });
+});
+//# sourceMappingURL=thoughtUtils.test.js.map

package/dist/src/utils/thoughtUtils.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"thoughtUtils.test.js","sourceRoot":"","sources":["../../../src/utils/thoughtUtils.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AAEjD,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;IAC5B,EAAE,CAAC,IAAI,CAAC;QACN;YACE,IAAI,EAAE,iDAAiD;YACvD,OAAO,EAAE,uCAAuC;YAChD,QAAQ,EAAE;gBACR,OAAO,EAAE,UAAU;gBACnB,WAAW,EAAE,0BAA0B;aACxC;SACF;QACD;YACE,IAAI,EAAE,mDAAmD;YACzD,OAAO,EAAE,0CAA0C;YACnD,QAAQ,EAAE,EAAE,OAAO,EAAE,SAAS,EAAE,WAAW,EAAE,yBAAyB,EAAE;SACzE;QACD;YACE,IAAI,EAAE,4CAA4C;YAClD,OAAO,EAAE,gBAAgB;YACzB,QAAQ,EAAE,EAAE,OAAO,EAAE,SAAS,EAAE,WAAW,EAAE,EAAE,EAAE;SAClD;QACD;YACE,IAAI,EAAE,+BAA+B;YACrC,OAAO,EAAE,kBAAkB;YAC3B,QAAQ,EAAE,EAAE,OAAO,EAAE,cAAc,EAAE,WAAW,EAAE,EAAE,EAAE;SACvD;QACD;YACE,IAAI,EAAE,gDAAgD;YACtD,OAAO,EAAE,6BAA6B;YACtC,QAAQ,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE,WAAW,EAAE,6BAA6B,EAAE;SACtE;QACD;YACE,IAAI,EAAE,uBAAuB;YAC7B,OAAO,EAAE,EAAE;YACX,QAAQ,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE,WAAW,EAAE,EAAE,EAAE;SAC3C;QACD;YACE,IAAI,EAAE,6CAA6C;YACnD,OAAO,EACL,sEAAsE;YACxE,QAAQ,EAAE;gBACR,OAAO,EAAE,qBAAqB;gBAC9B,WAAW,EAAE,6CAA6C;aAC3D;SACF;QACD;YACE,IAAI,EAAE,gDAAgD;YACtD,OAAO,EAAE,gCAAgC;YACzC,QAAQ,EAAE,EAAE,OAAO,EAAE,OAAO,EAAE,WAAW,EAAE,sBAAsB,EAAE;SACpE;QACD;YACE,IAAI,EAAE,mCAAmC;YACzC,OAAO,EAAE,sCAAsC;YAC/C,QAAQ,EAAE;gBACR,OAAO,EAAE,SAAS;gBAClB,WAAW,EAAE,2BAA2B;aACzC;SACF;QACD;YACE,IAAI,EAAE,yBAAyB;YAC/B,OAAO,EAAE,iCAAiC;YAC1C,QAAQ,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE,WAAW,EAAE,iCAAiC,EAAE;SAC1E;QACD;YACE,IAAI,EAAE,sBAAsB;YAC5B,OAAO,EAAE,oCAAoC;YAC7C,QAAQ,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE,WAAW,EAAE,gCAAgC,EAAE;SACzE;KACF,CAAC,CAAC,8BAA8B,EAAE,CAAC,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE;QAC3D,MAAM,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IAClD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

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

@@ -0,0 +1,22 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { AnyDeclarativeTool, AnyToolInvocation } from '../index.js';
+import { ToolNames } from '../tools/tool-names.js';
+export type ToolName = (typeof ToolNames)[keyof typeof ToolNames];
+export declare function isToolEnabled(toolName: ToolName, coreTools?: string[], excludeTools?: string[]): boolean;
+/**
+ * Checks if a tool invocation matches any of a list of patterns.
+ *
+ * @param toolOrToolName The tool object or the name of the tool being invoked.
+ * @param invocation The invocation object for the tool.
+ * @param patterns A list of patterns to match against.
+ *   Patterns can be:
+ *   - A tool name (e.g., "ReadFileTool") to match any invocation of that tool.
+ *   - A tool name with a prefix (e.g., "ShellTool(git status)") to match
+ *     invocations where the arguments start with that prefix.
+ * @returns True if the invocation matches any pattern, false otherwise.
+ */
+export declare function doesToolInvocationMatch(toolOrToolName: AnyDeclarativeTool | string, invocation: AnyToolInvocation, patterns: string[]): boolean;

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

@@ -0,0 +1,121 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { isTool } from '../index.js';
+import { ToolNames, ToolDisplayNames, ToolNamesMigration, ToolDisplayNamesMigration, } from '../tools/tool-names.js';
+const normalizeIdentifier = (identifier) => identifier.trim().replace(/^_+/, '');
+const toolNameKeys = Object.keys(ToolNames);
+const TOOL_ALIAS_MAP = (() => {
+    const map = new Map();
+    const addAlias = (set, alias) => {
+        if (!alias) {
+            return;
+        }
+        set.add(normalizeIdentifier(alias));
+    };
+    for (const key of toolNameKeys) {
+        const canonicalName = ToolNames[key];
+        const displayName = ToolDisplayNames[key];
+        const aliases = new Set();
+        addAlias(aliases, canonicalName);
+        addAlias(aliases, displayName);
+        addAlias(aliases, `${displayName}Tool`);
+        for (const [legacyName, mappedName] of Object.entries(ToolNamesMigration)) {
+            if (mappedName === canonicalName) {
+                addAlias(aliases, legacyName);
+            }
+        }
+        for (const [legacyDisplay, mappedDisplay] of Object.entries(ToolDisplayNamesMigration)) {
+            if (mappedDisplay === displayName) {
+                addAlias(aliases, legacyDisplay);
+            }
+        }
+        map.set(canonicalName, aliases);
+    }
+    return map;
+})();
+const getAliasSetForTool = (toolName) => {
+    const aliases = TOOL_ALIAS_MAP.get(toolName);
+    if (!aliases) {
+        return new Set([normalizeIdentifier(toolName)]);
+    }
+    return aliases;
+};
+const sanitizeExactIdentifier = (value) => normalizeIdentifier(value);
+const sanitizePatternIdentifier = (value) => {
+    const openParenIndex = value.indexOf('(');
+    if (openParenIndex === -1) {
+        return normalizeIdentifier(value);
+    }
+    return normalizeIdentifier(value.slice(0, openParenIndex));
+};
+const filterList = (list) => (list ?? []).filter((entry) => Boolean(entry && entry.trim()));
+export function isToolEnabled(toolName, coreTools, excludeTools) {
+    const aliasSet = getAliasSetForTool(toolName);
+    const matchesIdentifier = (value) => aliasSet.has(sanitizeExactIdentifier(value));
+    const matchesIdentifierWithArgs = (value) => aliasSet.has(sanitizePatternIdentifier(value));
+    const filteredCore = filterList(coreTools);
+    const filteredExclude = filterList(excludeTools);
+    if (filteredCore.length === 0) {
+        return !filteredExclude.some((entry) => matchesIdentifier(entry));
+    }
+    const isExplicitlyEnabled = filteredCore.some((entry) => matchesIdentifier(entry) || matchesIdentifierWithArgs(entry));
+    if (!isExplicitlyEnabled) {
+        return false;
+    }
+    return !filteredExclude.some((entry) => matchesIdentifier(entry));
+}
+const SHELL_TOOL_NAMES = ['run_shell_command', 'ShellTool'];
+/**
+ * Checks if a tool invocation matches any of a list of patterns.
+ *
+ * @param toolOrToolName The tool object or the name of the tool being invoked.
+ * @param invocation The invocation object for the tool.
+ * @param patterns A list of patterns to match against.
+ *   Patterns can be:
+ *   - A tool name (e.g., "ReadFileTool") to match any invocation of that tool.
+ *   - A tool name with a prefix (e.g., "ShellTool(git status)") to match
+ *     invocations where the arguments start with that prefix.
+ * @returns True if the invocation matches any pattern, false otherwise.
+ */
+export function doesToolInvocationMatch(toolOrToolName, invocation, patterns) {
+    let toolNames;
+    if (isTool(toolOrToolName)) {
+        toolNames = [toolOrToolName.name, toolOrToolName.constructor.name];
+    }
+    else {
+        toolNames = [toolOrToolName];
+    }
+    if (toolNames.some((name) => SHELL_TOOL_NAMES.includes(name))) {
+        toolNames = [...new Set([...toolNames, ...SHELL_TOOL_NAMES])];
+    }
+    for (const pattern of patterns) {
+        const openParen = pattern.indexOf('(');
+        if (openParen === -1) {
+            // No arguments, just a tool name
+            if (toolNames.includes(pattern)) {
+                return true;
+            }
+            continue;
+        }
+        const patternToolName = pattern.substring(0, openParen);
+        if (!toolNames.includes(patternToolName)) {
+            continue;
+        }
+        if (!pattern.endsWith(')')) {
+            continue;
+        }
+        const argPattern = pattern.substring(openParen + 1, pattern.length - 1);
+        if ('command' in invocation.params &&
+            toolNames.includes('run_shell_command')) {
+            const argValue = String(invocation.params.command);
+            if (argValue === argPattern || argValue.startsWith(argPattern + ' ')) {
+                return true;
+            }
+        }
+    }
+    return false;
+}
+//# sourceMappingURL=tool-utils.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"tool-utils.js","sourceRoot":"","sources":["../../../src/utils/tool-utils.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAGH,OAAO,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AACrC,OAAO,EACL,SAAS,EACT,gBAAgB,EAChB,kBAAkB,EAClB,yBAAyB,GAC1B,MAAM,wBAAwB,CAAC;AAIhC,MAAM,mBAAmB,GAAG,CAAC,UAAkB,EAAU,EAAE,CACzD,UAAU,CAAC,IAAI,EAAE,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;AAEvC,MAAM,YAAY,GAAG,MAAM,CAAC,IAAI,CAAC,SAAS,CAAkC,CAAC;AAE7E,MAAM,cAAc,GAA+B,CAAC,GAAG,EAAE;IACvD,MAAM,GAAG,GAAG,IAAI,GAAG,EAAyB,CAAC;IAE7C,MAAM,QAAQ,GAAG,CAAC,GAAgB,EAAE,KAAc,EAAE,EAAE;QACpD,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,OAAO;QACT,CAAC;QACD,GAAG,CAAC,GAAG,CAAC,mBAAmB,CAAC,KAAK,CAAC,CAAC,CAAC;IACtC,CAAC,CAAC;IAEF,KAAK,MAAM,GAAG,IAAI,YAAY,EAAE,CAAC;QAC/B,MAAM,aAAa,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC;QACrC,MAAM,WAAW,GAAG,gBAAgB,CAAC,GAAG,CAAC,CAAC;QAC1C,MAAM,OAAO,GAAG,IAAI,GAAG,EAAU,CAAC;QAElC,QAAQ,CAAC,OAAO,EAAE,aAAa,CAAC,CAAC;QACjC,QAAQ,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC;QAC/B,QAAQ,CAAC,OAAO,EAAE,GAAG,WAAW,MAAM,CAAC,CAAC;QAExC,KAAK,MAAM,CAAC,UAAU,EAAE,UAAU,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,kBAAkB,CAAC,EAAE,CAAC;YAC1E,IAAI,UAAU,KAAK,aAAa,EAAE,CAAC;gBACjC,QAAQ,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;YAChC,CAAC;QACH,CAAC;QAED,KAAK,MAAM,CAAC,aAAa,EAAE,aAAa,CAAC,IAAI,MAAM,CAAC,OAAO,CACzD,yBAAyB,CAC1B,EAAE,CAAC;YACF,IAAI,aAAa,KAAK,WAAW,EAAE,CAAC;gBAClC,QAAQ,CAAC,OAAO,EAAE,aAAa,CAAC,CAAC;YACnC,CAAC;QACH,CAAC;QAED,GAAG,CAAC,GAAG,CAAC,aAAa,EAAE,OAAO,CAAC,CAAC;IAClC,CAAC;IAED,OAAO,GAAG,CAAC;AACb,CAAC,CAAC,EAAE,CAAC;AAEL,MAAM,kBAAkB,GAAG,CAAC,QAAkB,EAAe,EAAE;IAC7D,MAAM,OAAO,GAAG,cAAc,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;IAC7C,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,OAAO,IAAI,GAAG,CAAC,CAAC,mBAAmB,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC;IAClD,CAAC;IACD,OAAO,OAAO,CAAC;AACjB,CAAC,CAAC;AAEF,MAAM,uBAAuB,GAAG,CAAC,KAAa,EAAU,EAAE,CACxD,mBAAmB,CAAC,KAAK,CAAC,CAAC;AAE7B,MAAM,yBAAyB,GAAG,CAAC,KAAa,EAAU,EAAE;IAC1D,MAAM,cAAc,GAAG,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;IAC1C,IAAI,cAAc,KAAK,CAAC,CAAC,EAAE,CAAC;QAC1B,OAAO,mBAAmB,CAAC,KAAK,CAAC,CAAC;IACpC,CAAC;IACD,OAAO,mBAAmB,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,cAAc,CAAC,CAAC,CAAC;AAC7D,CAAC,CAAC;AAEF,MAAM,UAAU,GAAG,CAAC,IAAe,EAAY,EAAE,CAC/C,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,KAAK,EAAmB,EAAE,CAC7C,OAAO,CAAC,KAAK,IAAI,KAAK,CAAC,IAAI,EAAE,CAAC,CAC/B,CAAC;AAEJ,MAAM,UAAU,aAAa,CAC3B,QAAkB,EAClB,SAAoB,EACpB,YAAuB;IAEvB,MAAM,QAAQ,GAAG,kBAAkB,CAAC,QAAQ,CAAC,CAAC;IAC9C,MAAM,iBAAiB,GAAG,CAAC,KAAa,EAAW,EAAE,CACnD,QAAQ,CAAC,GAAG,CAAC,uBAAuB,CAAC,KAAK,CAAC,CAAC,CAAC;IAC/C,MAAM,yBAAyB,GAAG,CAAC,KAAa,EAAW,EAAE,CAC3D,QAAQ,CAAC,GAAG,CAAC,yBAAyB,CAAC,KAAK,CAAC,CAAC,CAAC;IAEjD,MAAM,YAAY,GAAG,UAAU,CAAC,SAAS,CAAC,CAAC;IAC3C,MAAM,eAAe,GAAG,UAAU,CAAC,YAAY,CAAC,CAAC;IAEjD,IAAI,YAAY,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC9B,OAAO,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC,CAAC;IACpE,CAAC;IAED,MAAM,mBAAmB,GAAG,YAAY,CAAC,IAAI,CAC3C,CAAC,KAAK,EAAE,EAAE,CAAC,iBAAiB,CAAC,KAAK,CAAC,IAAI,yBAAyB,CAAC,KAAK,CAAC,CACxE,CAAC;IAEF,IAAI,CAAC,mBAAmB,EAAE,CAAC;QACzB,OAAO,KAAK,CAAC;IACf,CAAC;IAED,OAAO,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC,CAAC;AACpE,CAAC;AAED,MAAM,gBAAgB,GAAG,CAAC,mBAAmB,EAAE,WAAW,CAAC,CAAC;AAE5D;;;;;;;;;;;GAWG;AACH,MAAM,UAAU,uBAAuB,CACrC,cAA2C,EAC3C,UAA6B,EAC7B,QAAkB;IAElB,IAAI,SAAmB,CAAC;IACxB,IAAI,MAAM,CAAC,cAAc,CAAC,EAAE,CAAC;QAC3B,SAAS,GAAG,CAAC,cAAc,CAAC,IAAI,EAAE,cAAc,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;IACrE,CAAC;SAAM,CAAC;QACN,SAAS,GAAG,CAAC,cAAwB,CAAC,CAAC;IACzC,CAAC;IAED,IAAI,SAAS,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,gBAAgB,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC;QAC9D,SAAS,GAAG,CAAC,GAAG,IAAI,GAAG,CAAC,CAAC,GAAG,SAAS,EAAE,GAAG,gBAAgB,CAAC,CAAC,CAAC,CAAC;IAChE,CAAC;IAED,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;QAC/B,MAAM,SAAS,GAAG,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QAEvC,IAAI,SAAS,KAAK,CAAC,CAAC,EAAE,CAAC;YACrB,iCAAiC;YACjC,IAAI,SAAS,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;gBAChC,OAAO,IAAI,CAAC;YACd,CAAC;YACD,SAAS;QACX,CAAC;QAED,MAAM,eAAe,GAAG,OAAO,CAAC,SAAS,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC;QACxD,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,eAAe,CAAC,EAAE,CAAC;YACzC,SAAS;QACX,CAAC;QAED,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;YAC3B,SAAS;QACX,CAAC;QAED,MAAM,UAAU,GAAG,OAAO,CAAC,SAAS,CAAC,SAAS,GAAG,CAAC,EAAE,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QAExE,IACE,SAAS,IAAI,UAAU,CAAC,MAAM;YAC9B,SAAS,CAAC,QAAQ,CAAC,mBAAmB,CAAC,EACvC,CAAC;YACD,MAAM,QAAQ,GAAG,MAAM,CACpB,UAAU,CAAC,MAA8B,CAAC,OAAO,CACnD,CAAC;YACF,IAAI,QAAQ,KAAK,UAAU,IAAI,QAAQ,CAAC,UAAU,CAAC,UAAU,GAAG,GAAG,CAAC,EAAE,CAAC;gBACrE,OAAO,IAAI,CAAC;YACd,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,KAAK,CAAC;AACf,CAAC"}

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

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

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

@@ -0,0 +1,100 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { expect, describe, it } from 'vitest';
+import { doesToolInvocationMatch, isToolEnabled } from './tool-utils.js';
+import { ReadFileTool } from '../tools/read-file.js';
+import { ToolNames } from '../tools/tool-names.js';
+describe('doesToolInvocationMatch', () => {
+    it('should not match a partial command prefix', () => {
+        const invocation = {
+            params: { command: 'git commitsomething' },
+        };
+        const patterns = ['ShellTool(git commit)'];
+        const result = doesToolInvocationMatch('run_shell_command', invocation, patterns);
+        expect(result).toBe(false);
+    });
+    it('should match an exact command', () => {
+        const invocation = {
+            params: { command: 'git status' },
+        };
+        const patterns = ['ShellTool(git status)'];
+        const result = doesToolInvocationMatch('run_shell_command', invocation, patterns);
+        expect(result).toBe(true);
+    });
+    it('should match a command that is a prefix', () => {
+        const invocation = {
+            params: { command: 'git status -v' },
+        };
+        const patterns = ['ShellTool(git status)'];
+        const result = doesToolInvocationMatch('run_shell_command', invocation, patterns);
+        expect(result).toBe(true);
+    });
+    describe('for non-shell tools', () => {
+        const readFileTool = new ReadFileTool({});
+        const invocation = {
+            params: { file: 'test.txt' },
+        };
+        it('should match by tool name', () => {
+            const patterns = ['read_file'];
+            const result = doesToolInvocationMatch(readFileTool, invocation, patterns);
+            expect(result).toBe(true);
+        });
+        it('should match by tool class name', () => {
+            const patterns = ['ReadFileTool'];
+            const result = doesToolInvocationMatch(readFileTool, invocation, patterns);
+            expect(result).toBe(true);
+        });
+        it('should not match if neither name is in the patterns', () => {
+            const patterns = ['some_other_tool', 'AnotherToolClass'];
+            const result = doesToolInvocationMatch(readFileTool, invocation, patterns);
+            expect(result).toBe(false);
+        });
+        it('should match by tool name when passed as a string', () => {
+            const patterns = ['read_file'];
+            const result = doesToolInvocationMatch('read_file', invocation, patterns);
+            expect(result).toBe(true);
+        });
+    });
+});
+describe('isToolEnabled', () => {
+    it('enables tool when coreTools is undefined and tool is not excluded', () => {
+        expect(isToolEnabled(ToolNames.SHELL, undefined, undefined)).toBe(true);
+    });
+    it('disables tool when excluded by canonical tool name', () => {
+        expect(isToolEnabled(ToolNames.SHELL, undefined, ['run_shell_command'])).toBe(false);
+    });
+    it('enables tool when explicitly listed by display name', () => {
+        expect(isToolEnabled(ToolNames.SHELL, ['Shell'], undefined)).toBe(true);
+    });
+    it('enables tool when explicitly listed by class name', () => {
+        expect(isToolEnabled(ToolNames.SHELL, ['ShellTool'], undefined)).toBe(true);
+    });
+    it('supports class names with leading underscores', () => {
+        expect(isToolEnabled(ToolNames.SHELL, ['__ShellTool'], undefined)).toBe(true);
+    });
+    it('enables tool when coreTools contains a legacy tool name alias', () => {
+        expect(isToolEnabled(ToolNames.GREP, ['search_file_content'], undefined)).toBe(true);
+    });
+    it('enables tool when coreTools contains a legacy display name alias', () => {
+        expect(isToolEnabled(ToolNames.GLOB, ['FindFiles'], undefined)).toBe(true);
+    });
+    it('enables tool when coreTools contains an argument-specific pattern', () => {
+        expect(isToolEnabled(ToolNames.SHELL, ['Shell(git status)'], undefined)).toBe(true);
+    });
+    it('disables tool when not present in coreTools', () => {
+        expect(isToolEnabled(ToolNames.SHELL, ['Edit'], undefined)).toBe(false);
+    });
+    it('uses legacy display name aliases when excluding tools', () => {
+        expect(isToolEnabled(ToolNames.GREP, undefined, ['SearchFiles'])).toBe(false);
+    });
+    it('does not treat argument-specific exclusions as matches', () => {
+        expect(isToolEnabled(ToolNames.SHELL, undefined, ['Shell(git status)'])).toBe(true);
+    });
+    it('considers excludeTools even when tool is explicitly enabled', () => {
+        expect(isToolEnabled(ToolNames.SHELL, ['Shell'], ['ShellTool'])).toBe(false);
+    });
+});
+//# sourceMappingURL=tool-utils.test.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"tool-utils.test.js","sourceRoot":"","sources":["../../../src/utils/tool-utils.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,uBAAuB,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC;AAEzE,OAAO,EAAE,YAAY,EAAE,MAAM,uBAAuB,CAAC;AACrD,OAAO,EAAE,SAAS,EAAE,MAAM,wBAAwB,CAAC;AAEnD,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;IACvC,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;QACnD,MAAM,UAAU,GAAG;YACjB,MAAM,EAAE,EAAE,OAAO,EAAE,qBAAqB,EAAE;SACtB,CAAC;QACvB,MAAM,QAAQ,GAAG,CAAC,uBAAuB,CAAC,CAAC;QAC3C,MAAM,MAAM,GAAG,uBAAuB,CACpC,mBAAmB,EACnB,UAAU,EACV,QAAQ,CACT,CAAC;QACF,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC7B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+BAA+B,EAAE,GAAG,EAAE;QACvC,MAAM,UAAU,GAAG;YACjB,MAAM,EAAE,EAAE,OAAO,EAAE,YAAY,EAAE;SACb,CAAC;QACvB,MAAM,QAAQ,GAAG,CAAC,uBAAuB,CAAC,CAAC;QAC3C,MAAM,MAAM,GAAG,uBAAuB,CACpC,mBAAmB,EACnB,UAAU,EACV,QAAQ,CACT,CAAC;QACF,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC5B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;QACjD,MAAM,UAAU,GAAG;YACjB,MAAM,EAAE,EAAE,OAAO,EAAE,eAAe,EAAE;SAChB,CAAC;QACvB,MAAM,QAAQ,GAAG,CAAC,uBAAuB,CAAC,CAAC;QAC3C,MAAM,MAAM,GAAG,uBAAuB,CACpC,mBAAmB,EACnB,UAAU,EACV,QAAQ,CACT,CAAC;QACF,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC5B,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;QACnC,MAAM,YAAY,GAAG,IAAI,YAAY,CAAC,EAAY,CAAC,CAAC;QACpD,MAAM,UAAU,GAAG;YACjB,MAAM,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE;SACR,CAAC;QAEvB,EAAE,CAAC,2BAA2B,EAAE,GAAG,EAAE;YACnC,MAAM,QAAQ,GAAG,CAAC,WAAW,CAAC,CAAC;YAC/B,MAAM,MAAM,GAAG,uBAAuB,CACpC,YAAY,EACZ,UAAU,EACV,QAAQ,CACT,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC5B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;YACzC,MAAM,QAAQ,GAAG,CAAC,cAAc,CAAC,CAAC;YAClC,MAAM,MAAM,GAAG,uBAAuB,CACpC,YAAY,EACZ,UAAU,EACV,QAAQ,CACT,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC5B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;YAC7D,MAAM,QAAQ,GAAG,CAAC,iBAAiB,EAAE,kBAAkB,CAAC,CAAC;YACzD,MAAM,MAAM,GAAG,uBAAuB,CACpC,YAAY,EACZ,UAAU,EACV,QAAQ,CACT,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC7B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mDAAmD,EAAE,GAAG,EAAE;YAC3D,MAAM,QAAQ,GAAG,CAAC,WAAW,CAAC,CAAC;YAC/B,MAAM,MAAM,GAAG,uBAAuB,CAAC,WAAW,EAAE,UAAU,EAAE,QAAQ,CAAC,CAAC;YAC1E,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC5B,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;IAC7B,EAAE,CAAC,mEAAmE,EAAE,GAAG,EAAE;QAC3E,MAAM,CAAC,aAAa,CAAC,SAAS,CAAC,KAAK,EAAE,SAAS,EAAE,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC1E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;QAC5D,MAAM,CACJ,aAAa,CAAC,SAAS,CAAC,KAAK,EAAE,SAAS,EAAE,CAAC,mBAAmB,CAAC,CAAC,CACjE,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAChB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;QAC7D,MAAM,CAAC,aAAa,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC,OAAO,CAAC,EAAE,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC1E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mDAAmD,EAAE,GAAG,EAAE;QAC3D,MAAM,CAAC,aAAa,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC,WAAW,CAAC,EAAE,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC9E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;QACvD,MAAM,CAAC,aAAa,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC,aAAa,CAAC,EAAE,SAAS,CAAC,CAAC,CAAC,IAAI,CACrE,IAAI,CACL,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+DAA+D,EAAE,GAAG,EAAE;QACvE,MAAM,CACJ,aAAa,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC,qBAAqB,CAAC,EAAE,SAAS,CAAC,CAClE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACf,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kEAAkE,EAAE,GAAG,EAAE;QAC1E,MAAM,CAAC,aAAa,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC,WAAW,CAAC,EAAE,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC7E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mEAAmE,EAAE,GAAG,EAAE;QAC3E,MAAM,CACJ,aAAa,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC,mBAAmB,CAAC,EAAE,SAAS,CAAC,CACjE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACf,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;QACrD,MAAM,CAAC,aAAa,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC,MAAM,CAAC,EAAE,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC1E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uDAAuD,EAAE,GAAG,EAAE;QAC/D,MAAM,CAAC,aAAa,CAAC,SAAS,CAAC,IAAI,EAAE,SAAS,EAAE,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,IAAI,CACpE,KAAK,CACN,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wDAAwD,EAAE,GAAG,EAAE;QAChE,MAAM,CACJ,aAAa,CAAC,SAAS,CAAC,KAAK,EAAE,SAAS,EAAE,CAAC,mBAAmB,CAAC,CAAC,CACjE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACf,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6DAA6D,EAAE,GAAG,EAAE;QACrE,MAAM,CAAC,aAAa,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,IAAI,CACnE,KAAK,CACN,CAAC;IACJ,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}