@office-ai/aioncli-core 0.26.0 → 0.30.0

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

@@ -4,7 +4,7 @@
  * SPDX-License-Identifier: Apache-2.0
  */
 import { describe, it, expect } from 'vitest';
-import { escapeRegex, buildArgsPatterns } from './utils.js';
+import { escapeRegex, buildArgsPatterns, isSafeRegExp } from './utils.js';
 describe('policy/utils', () => {
     describe('escapeRegex', () => {
         it('should escape special regex characters', () => {
@@ -17,6 +17,38 @@ describe('policy/utils', () => {
             expect(escapeRegex(input)).toBe(input);
         });
     });
+    describe('isSafeRegExp', () => {
+        it('should return true for simple regexes', () => {
+            expect(isSafeRegExp('abc')).toBe(true);
+            expect(isSafeRegExp('^abc$')).toBe(true);
+            expect(isSafeRegExp('a|b')).toBe(true);
+        });
+        it('should return true for safe quantifiers', () => {
+            expect(isSafeRegExp('a+')).toBe(true);
+            expect(isSafeRegExp('a*')).toBe(true);
+            expect(isSafeRegExp('a?')).toBe(true);
+            expect(isSafeRegExp('a{1,3}')).toBe(true);
+        });
+        it('should return true for safe groups', () => {
+            expect(isSafeRegExp('(abc)*')).toBe(true);
+            expect(isSafeRegExp('(a|b)+')).toBe(true);
+        });
+        it('should return false for invalid regexes', () => {
+            expect(isSafeRegExp('([a-z)')).toBe(false);
+            expect(isSafeRegExp('*')).toBe(false);
+        });
+        it('should return false for extremely long regexes', () => {
+            expect(isSafeRegExp('a'.repeat(2049))).toBe(false);
+        });
+        it('should return false for nested quantifiers (potential ReDoS)', () => {
+            expect(isSafeRegExp('(a+)+')).toBe(false);
+            expect(isSafeRegExp('(a+)*')).toBe(false);
+            expect(isSafeRegExp('(a*)+')).toBe(false);
+            expect(isSafeRegExp('(a*)*')).toBe(false);
+            expect(isSafeRegExp('(a|b+)+')).toBe(false);
+            expect(isSafeRegExp('(.*)+')).toBe(false);
+        });
+    });
     describe('buildArgsPatterns', () => {
         it('should return argsPattern if provided and no commandPrefix/regex', () => {
             const result = buildArgsPatterns('my-pattern', undefined, undefined);
@@ -24,13 +56,13 @@ describe('policy/utils', () => {
         });
         it('should build pattern from a single commandPrefix', () => {
             const result = buildArgsPatterns(undefined, 'ls', undefined);
-            expect(result).toEqual(['"command":"ls(?:[\\s"]|$)']);
+            expect(result).toEqual(['"command":"ls(?:[\\s"]|\\\\")']);
         });
         it('should build patterns from an array of commandPrefixes', () => {
             const result = buildArgsPatterns(undefined, ['ls', 'cd'], undefined);
             expect(result).toEqual([
-                '"command":"ls(?:[\\s"]|$)',
-                '"command":"cd(?:[\\s"]|$)',
+                '"command":"ls(?:[\\s"]|\\\\")',
+                '"command":"cd(?:[\\s"]|\\\\")',
             ]);
         });
         it('should build pattern from commandRegex', () => {
@@ -39,7 +71,7 @@ describe('policy/utils', () => {
         });
         it('should prioritize commandPrefix over commandRegex and argsPattern', () => {
             const result = buildArgsPatterns('raw', 'prefix', 'regex');
-            expect(result).toEqual(['"command":"prefix(?:[\\s"]|$)']);
+            expect(result).toEqual(['"command":"prefix(?:[\\s"]|\\\\")']);
         });
         it('should prioritize commandRegex over argsPattern if no commandPrefix', () => {
             const result = buildArgsPatterns('raw', undefined, 'regex');
@@ -47,18 +79,46 @@ describe('policy/utils', () => {
         });
         it('should escape characters in commandPrefix', () => {
             const result = buildArgsPatterns(undefined, 'git checkout -b', undefined);
-            expect(result).toEqual(['"command":"git\\ checkout\\ \\-b(?:[\\s"]|$)']);
+            expect(result).toEqual([
+                '"command":"git\\ checkout\\ \\-b(?:[\\s"]|\\\\")',
+            ]);
         });
         it('should correctly escape quotes in commandPrefix', () => {
             const result = buildArgsPatterns(undefined, 'git "fix"', undefined);
             expect(result).toEqual([
-                '"command":"git\\ \\\\\\"fix\\\\\\"(?:[\\s"]|$)',
+                '"command":"git\\ \\\\\\"fix\\\\\\"(?:[\\s"]|\\\\")',
             ]);
         });
         it('should handle undefined correctly when no inputs are provided', () => {
             const result = buildArgsPatterns(undefined, undefined, undefined);
             expect(result).toEqual([undefined]);
         });
+        it('should match prefixes followed by JSON escaped quotes', () => {
+            // Testing the security fix logic: allowing "echo \"foo\""
+            const prefix = 'echo ';
+            const patterns = buildArgsPatterns(undefined, prefix, undefined);
+            const regex = new RegExp(patterns[0]);
+            // Mimic JSON stringified args
+            // echo "foo" -> {"command":"echo \"foo\""}
+            const validJsonArgs = '{"command":"echo \\"foo\\""}';
+            expect(regex.test(validJsonArgs)).toBe(true);
+        });
+        it('should NOT match prefixes followed by raw backslashes (security check)', () => {
+            // Testing that we blocked the hole: "echo\foo"
+            const prefix = 'echo ';
+            const patterns = buildArgsPatterns(undefined, prefix, undefined);
+            const regex = new RegExp(patterns[0]);
+            // echo\foo -> {"command":"echo\\foo"}
+            // In regex matching: "echo " is followed by "\" which is NOT in [\s"] and is not \"
+            const attackJsonArgs = '{"command":"echo\\\\foo"}';
+            expect(regex.test(attackJsonArgs)).toBe(false);
+            // Also validation for "git " matching "git\status"
+            const gitPatterns = buildArgsPatterns(undefined, 'git ', undefined);
+            const gitRegex = new RegExp(gitPatterns[0]);
+            // git\status -> {"command":"git\\status"}
+            const gitAttack = '{"command":"git\\\\status"}';
+            expect(gitRegex.test(gitAttack)).toBe(false);
+        });
     });
 });
 //# sourceMappingURL=utils.test.js.map

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

@@ -1 +1 @@
-{"version":3,"file":"utils.test.js","sourceRoot":"","sources":["../../../src/policy/utils.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,WAAW,EAAE,iBAAiB,EAAE,MAAM,YAAY,CAAC;AAE5D,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;IAC5B,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;QAC3B,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;YAChD,MAAM,KAAK,GAAG,oBAAoB,CAAC;YACnC,MAAM,OAAO,GAAG,WAAW,CAAC,KAAK,CAAC,CAAC;YACnC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAClB,sDAAsD,CACvD,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oEAAoE,EAAE,GAAG,EAAE;YAC5E,MAAM,KAAK,GAAG,WAAW,CAAC;YAC1B,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;QACjC,EAAE,CAAC,kEAAkE,EAAE,GAAG,EAAE;YAC1E,MAAM,MAAM,GAAG,iBAAiB,CAAC,YAAY,EAAE,SAAS,EAAE,SAAS,CAAC,CAAC;YACrE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;YAC1D,MAAM,MAAM,GAAG,iBAAiB,CAAC,SAAS,EAAE,IAAI,EAAE,SAAS,CAAC,CAAC;YAC7D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,CAAC,2BAA2B,CAAC,CAAC,CAAC;QACxD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wDAAwD,EAAE,GAAG,EAAE;YAChE,MAAM,MAAM,GAAG,iBAAiB,CAAC,SAAS,EAAE,CAAC,IAAI,EAAE,IAAI,CAAC,EAAE,SAAS,CAAC,CAAC;YACrE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,2BAA2B;gBAC3B,2BAA2B;aAC5B,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;YAChD,MAAM,MAAM,GAAG,iBAAiB,CAAC,SAAS,EAAE,SAAS,EAAE,WAAW,CAAC,CAAC;YACpE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC;QACnD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mEAAmE,EAAE,GAAG,EAAE;YAC3E,MAAM,MAAM,GAAG,iBAAiB,CAAC,KAAK,EAAE,QAAQ,EAAE,OAAO,CAAC,CAAC;YAC3D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,CAAC,+BAA+B,CAAC,CAAC,CAAC;QAC5D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qEAAqE,EAAE,GAAG,EAAE;YAC7E,MAAM,MAAM,GAAG,iBAAiB,CAAC,KAAK,EAAE,SAAS,EAAE,OAAO,CAAC,CAAC;YAC5D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,CAAC,kBAAkB,CAAC,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;YACnD,MAAM,MAAM,GAAG,iBAAiB,CAAC,SAAS,EAAE,iBAAiB,EAAE,SAAS,CAAC,CAAC;YAC1E,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,CAAC,8CAA8C,CAAC,CAAC,CAAC;QAC3E,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;YACzD,MAAM,MAAM,GAAG,iBAAiB,CAAC,SAAS,EAAE,WAAW,EAAE,SAAS,CAAC,CAAC;YACpE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,gDAAgD;aACjD,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+DAA+D,EAAE,GAAG,EAAE;YACvE,MAAM,MAAM,GAAG,iBAAiB,CAAC,SAAS,EAAE,SAAS,EAAE,SAAS,CAAC,CAAC;YAClE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC;QACtC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
+{"version":3,"file":"utils.test.js","sourceRoot":"","sources":["../../../src/policy/utils.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,WAAW,EAAE,iBAAiB,EAAE,YAAY,EAAE,MAAM,YAAY,CAAC;AAE1E,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;IAC5B,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;QAC3B,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;YAChD,MAAM,KAAK,GAAG,oBAAoB,CAAC;YACnC,MAAM,OAAO,GAAG,WAAW,CAAC,KAAK,CAAC,CAAC;YACnC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAClB,sDAAsD,CACvD,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oEAAoE,EAAE,GAAG,EAAE;YAC5E,MAAM,KAAK,GAAG,WAAW,CAAC;YAC1B,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;QAC5B,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;YAC/C,MAAM,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACvC,MAAM,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACzC,MAAM,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;YACjD,MAAM,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACtC,MAAM,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACtC,MAAM,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACtC,MAAM,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oCAAoC,EAAE,GAAG,EAAE;YAC5C,MAAM,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC1C,MAAM,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;YACjD,MAAM,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC3C,MAAM,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACxC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;YACxD,MAAM,CAAC,YAAY,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACrD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8DAA8D,EAAE,GAAG,EAAE;YACtE,MAAM,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC1C,MAAM,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC1C,MAAM,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC1C,MAAM,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC1C,MAAM,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC5C,MAAM,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;QACjC,EAAE,CAAC,kEAAkE,EAAE,GAAG,EAAE;YAC1E,MAAM,MAAM,GAAG,iBAAiB,CAAC,YAAY,EAAE,SAAS,EAAE,SAAS,CAAC,CAAC;YACrE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;YAC1D,MAAM,MAAM,GAAG,iBAAiB,CAAC,SAAS,EAAE,IAAI,EAAE,SAAS,CAAC,CAAC;YAC7D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,CAAC,+BAA+B,CAAC,CAAC,CAAC;QAC5D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wDAAwD,EAAE,GAAG,EAAE;YAChE,MAAM,MAAM,GAAG,iBAAiB,CAAC,SAAS,EAAE,CAAC,IAAI,EAAE,IAAI,CAAC,EAAE,SAAS,CAAC,CAAC;YACrE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,+BAA+B;gBAC/B,+BAA+B;aAChC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;YAChD,MAAM,MAAM,GAAG,iBAAiB,CAAC,SAAS,EAAE,SAAS,EAAE,WAAW,CAAC,CAAC;YACpE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC;QACnD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mEAAmE,EAAE,GAAG,EAAE;YAC3E,MAAM,MAAM,GAAG,iBAAiB,CAAC,KAAK,EAAE,QAAQ,EAAE,OAAO,CAAC,CAAC;YAC3D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,CAAC,mCAAmC,CAAC,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qEAAqE,EAAE,GAAG,EAAE;YAC7E,MAAM,MAAM,GAAG,iBAAiB,CAAC,KAAK,EAAE,SAAS,EAAE,OAAO,CAAC,CAAC;YAC5D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,CAAC,kBAAkB,CAAC,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;YACnD,MAAM,MAAM,GAAG,iBAAiB,CAAC,SAAS,EAAE,iBAAiB,EAAE,SAAS,CAAC,CAAC;YAC1E,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,kDAAkD;aACnD,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;YACzD,MAAM,MAAM,GAAG,iBAAiB,CAAC,SAAS,EAAE,WAAW,EAAE,SAAS,CAAC,CAAC;YACpE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,oDAAoD;aACrD,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+DAA+D,EAAE,GAAG,EAAE;YACvE,MAAM,MAAM,GAAG,iBAAiB,CAAC,SAAS,EAAE,SAAS,EAAE,SAAS,CAAC,CAAC;YAClE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC;QACtC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uDAAuD,EAAE,GAAG,EAAE;YAC/D,0DAA0D;YAC1D,MAAM,MAAM,GAAG,OAAO,CAAC;YACvB,MAAM,QAAQ,GAAG,iBAAiB,CAAC,SAAS,EAAE,MAAM,EAAE,SAAS,CAAC,CAAC;YACjE,MAAM,KAAK,GAAG,IAAI,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAE,CAAC,CAAC;YAEvC,8BAA8B;YAC9B,2CAA2C;YAC3C,MAAM,aAAa,GAAG,8BAA8B,CAAC;YACrD,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wEAAwE,EAAE,GAAG,EAAE;YAChF,+CAA+C;YAC/C,MAAM,MAAM,GAAG,OAAO,CAAC;YACvB,MAAM,QAAQ,GAAG,iBAAiB,CAAC,SAAS,EAAE,MAAM,EAAE,SAAS,CAAC,CAAC;YACjE,MAAM,KAAK,GAAG,IAAI,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAE,CAAC,CAAC;YAEvC,sCAAsC;YACtC,oFAAoF;YACpF,MAAM,cAAc,GAAG,2BAA2B,CAAC;YACnD,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAE/C,mDAAmD;YACnD,MAAM,WAAW,GAAG,iBAAiB,CAAC,SAAS,EAAE,MAAM,EAAE,SAAS,CAAC,CAAC;YACpE,MAAM,QAAQ,GAAG,IAAI,MAAM,CAAC,WAAW,CAAC,CAAC,CAAE,CAAC,CAAC;YAC7C,0CAA0C;YAC1C,MAAM,SAAS,GAAG,6BAA6B,CAAC;YAChD,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/prompts/promptProvider.d.ts

@@ -0,0 +1,19 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { Config } from '../config/config.js';
+import type { HierarchicalMemory } from '../config/memory.js';
+/**
+ * Orchestrates prompt generation by gathering context and building options.
+ */
+export declare class PromptProvider {
+    /**
+     * Generates the core system prompt.
+     */
+    getCoreSystemPrompt(config: Config, userMemory?: string | HierarchicalMemory, interactiveOverride?: boolean): string;
+    getCompressionPrompt(config: Config): string;
+    private withSection;
+    private maybeWriteSystemMd;
+}

package/dist/src/prompts/promptProvider.js

@@ -0,0 +1,174 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import fs from 'node:fs';
+import path from 'node:path';
+import process from 'node:process';
+import { GEMINI_DIR } from '../utils/paths.js';
+import { ApprovalMode } from '../policy/types.js';
+import * as snippets from './snippets.js';
+import * as legacySnippets from './snippets.legacy.js';
+import { resolvePathFromEnv, applySubstitutions, isSectionEnabled, } from './utils.js';
+import { CodebaseInvestigatorAgent } from '../agents/codebase-investigator.js';
+import { isGitRepository } from '../utils/gitUtils.js';
+import { PLAN_MODE_TOOLS, WRITE_TODOS_TOOL_NAME, READ_FILE_TOOL_NAME, ENTER_PLAN_MODE_TOOL_NAME, GLOB_TOOL_NAME, GREP_TOOL_NAME, } from '../tools/tool-names.js';
+import { resolveModel, supportsModernFeatures } from '../config/models.js';
+import { DiscoveredMCPTool } from '../tools/mcp-tool.js';
+import { getAllGeminiMdFilenames } from '../tools/memoryTool.js';
+/**
+ * Orchestrates prompt generation by gathering context and building options.
+ */
+export class PromptProvider {
+    /**
+     * Generates the core system prompt.
+     */
+    getCoreSystemPrompt(config, userMemory, interactiveOverride) {
+        const systemMdResolution = resolvePathFromEnv(process.env['GEMINI_SYSTEM_MD']);
+        const interactiveMode = interactiveOverride ?? config.isInteractive();
+        const approvalMode = config.getApprovalMode?.() ?? ApprovalMode.DEFAULT;
+        const isPlanMode = approvalMode === ApprovalMode.PLAN;
+        const isYoloMode = approvalMode === ApprovalMode.YOLO;
+        const skills = config.getSkillManager().getSkills();
+        const toolNames = config.getToolRegistry().getAllToolNames();
+        const enabledToolNames = new Set(toolNames);
+        const approvedPlanPath = config.getApprovedPlanPath();
+        const desiredModel = resolveModel(config.getActiveModel(), config.getGemini31LaunchedSync?.() ?? false);
+        const isModernModel = supportsModernFeatures(desiredModel);
+        const activeSnippets = isModernModel ? snippets : legacySnippets;
+        const contextFilenames = getAllGeminiMdFilenames();
+        // --- Context Gathering ---
+        let planModeToolsList = PLAN_MODE_TOOLS.filter((t) => enabledToolNames.has(t))
+            .map((t) => `  <tool>\`${t}\`</tool>`)
+            .join('\n');
+        // Add read-only MCP tools to the list
+        if (isPlanMode) {
+            const allTools = config.getToolRegistry().getAllTools();
+            const readOnlyMcpTools = allTools.filter((t) => t instanceof DiscoveredMCPTool && !!t.isReadOnly);
+            if (readOnlyMcpTools.length > 0) {
+                const mcpToolsList = readOnlyMcpTools
+                    .map((t) => `  <tool>\`${t.name}\` (${t.serverName})</tool>`)
+                    .join('\n');
+                planModeToolsList += `\n${mcpToolsList}`;
+            }
+        }
+        let basePrompt;
+        // --- Template File Override ---
+        if (systemMdResolution.value && !systemMdResolution.isDisabled) {
+            let systemMdPath = path.resolve(path.join(GEMINI_DIR, 'system.md'));
+            if (!systemMdResolution.isSwitch) {
+                systemMdPath = systemMdResolution.value;
+            }
+            if (!fs.existsSync(systemMdPath)) {
+                throw new Error(`missing system prompt file '${systemMdPath}'`);
+            }
+            basePrompt = fs.readFileSync(systemMdPath, 'utf8');
+            const skillsPrompt = activeSnippets.renderAgentSkills(skills.map((s) => ({
+                name: s.name,
+                description: s.description,
+                location: s.location,
+            })));
+            basePrompt = applySubstitutions(basePrompt, config, skillsPrompt, isModernModel);
+        }
+        else {
+            // --- Standard Composition ---
+            const hasHierarchicalMemory = typeof userMemory === 'object' &&
+                userMemory !== null &&
+                (!!userMemory.global?.trim() ||
+                    !!userMemory.extension?.trim() ||
+                    !!userMemory.project?.trim());
+            const options = {
+                preamble: this.withSection('preamble', () => ({
+                    interactive: interactiveMode,
+                })),
+                coreMandates: this.withSection('coreMandates', () => ({
+                    interactive: interactiveMode,
+                    hasSkills: skills.length > 0,
+                    hasHierarchicalMemory,
+                    contextFilenames,
+                })),
+                subAgents: this.withSection('agentContexts', () => config
+                    .getAgentRegistry()
+                    .getAllDefinitions()
+                    .map((d) => ({
+                    name: d.name,
+                    description: d.description,
+                }))),
+                agentSkills: this.withSection('agentSkills', () => skills.map((s) => ({
+                    name: s.name,
+                    description: s.description,
+                    location: s.location,
+                })), skills.length > 0),
+                hookContext: isSectionEnabled('hookContext') || undefined,
+                primaryWorkflows: this.withSection('primaryWorkflows', () => ({
+                    interactive: interactiveMode,
+                    enableCodebaseInvestigator: enabledToolNames.has(CodebaseInvestigatorAgent.name),
+                    enableWriteTodosTool: enabledToolNames.has(WRITE_TODOS_TOOL_NAME),
+                    enableEnterPlanModeTool: enabledToolNames.has(ENTER_PLAN_MODE_TOOL_NAME),
+                    enableGrep: enabledToolNames.has(GREP_TOOL_NAME),
+                    enableGlob: enabledToolNames.has(GLOB_TOOL_NAME),
+                    approvedPlan: approvedPlanPath
+                        ? { path: approvedPlanPath }
+                        : undefined,
+                }), !isPlanMode),
+                planningWorkflow: this.withSection('planningWorkflow', () => ({
+                    planModeToolsList,
+                    plansDir: config.storage.getProjectTempPlansDir(),
+                    approvedPlanPath: config.getApprovedPlanPath(),
+                }), isPlanMode),
+                operationalGuidelines: this.withSection('operationalGuidelines', () => ({
+                    interactive: interactiveMode,
+                    enableShellEfficiency: config.getEnableShellOutputEfficiency(),
+                    interactiveShellEnabled: config.isInteractiveShellEnabled(),
+                })),
+                sandbox: this.withSection('sandbox', () => getSandboxMode()),
+                interactiveYoloMode: this.withSection('interactiveYoloMode', () => true, isYoloMode && interactiveMode),
+                gitRepo: this.withSection('git', () => ({ interactive: interactiveMode }), isGitRepository(process.cwd()) ? true : false),
+                finalReminder: isModernModel
+                    ? undefined
+                    : this.withSection('finalReminder', () => ({
+                        readFileToolName: READ_FILE_TOOL_NAME,
+                    })),
+            };
+            // eslint-disable-next-line @typescript-eslint/no-unsafe-type-assertion
+            const getCoreSystemPrompt = activeSnippets.getCoreSystemPrompt;
+            basePrompt = getCoreSystemPrompt(options);
+        }
+        // --- Finalization (Shell) ---
+        const finalPrompt = activeSnippets.renderFinalShell(basePrompt, userMemory, contextFilenames);
+        // Sanitize erratic newlines from composition
+        const sanitizedPrompt = finalPrompt.replace(/\n{3,}/g, '\n\n');
+        // Write back to file if requested
+        this.maybeWriteSystemMd(sanitizedPrompt, systemMdResolution, path.resolve(path.join(GEMINI_DIR, 'system.md')));
+        return sanitizedPrompt;
+    }
+    getCompressionPrompt(config) {
+        const desiredModel = resolveModel(config.getActiveModel(), config.getGemini31LaunchedSync?.() ?? false);
+        const isModernModel = supportsModernFeatures(desiredModel);
+        const activeSnippets = isModernModel ? snippets : legacySnippets;
+        return activeSnippets.getCompressionPrompt();
+    }
+    withSection(key, factory, guard = true) {
+        return guard && isSectionEnabled(key) ? factory() : undefined;
+    }
+    maybeWriteSystemMd(basePrompt, resolution, defaultPath) {
+        const writeSystemMdResolution = resolvePathFromEnv(process.env['GEMINI_WRITE_SYSTEM_MD']);
+        if (writeSystemMdResolution.value && !writeSystemMdResolution.isDisabled) {
+            const writePath = writeSystemMdResolution.isSwitch
+                ? defaultPath
+                : writeSystemMdResolution.value;
+            fs.mkdirSync(path.dirname(writePath), { recursive: true });
+            fs.writeFileSync(writePath, basePrompt);
+        }
+    }
+}
+// --- Internal Context Helpers ---
+function getSandboxMode() {
+    if (process.env['SANDBOX'] === 'sandbox-exec')
+        return 'macos-seatbelt';
+    if (process.env['SANDBOX'])
+        return 'generic';
+    return 'outside';
+}
+//# sourceMappingURL=promptProvider.js.map

package/dist/src/prompts/promptProvider.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"promptProvider.js","sourceRoot":"","sources":["../../../src/prompts/promptProvider.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,OAAO,MAAM,cAAc,CAAC;AAGnC,OAAO,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAC/C,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAClD,OAAO,KAAK,QAAQ,MAAM,eAAe,CAAC;AAC1C,OAAO,KAAK,cAAc,MAAM,sBAAsB,CAAC;AACvD,OAAO,EACL,kBAAkB,EAClB,kBAAkB,EAClB,gBAAgB,GAEjB,MAAM,YAAY,CAAC;AACpB,OAAO,EAAE,yBAAyB,EAAE,MAAM,oCAAoC,CAAC;AAC/E,OAAO,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAC;AACvD,OAAO,EACL,eAAe,EACf,qBAAqB,EACrB,mBAAmB,EACnB,yBAAyB,EACzB,cAAc,EACd,cAAc,GACf,MAAM,wBAAwB,CAAC;AAChC,OAAO,EAAE,YAAY,EAAE,sBAAsB,EAAE,MAAM,qBAAqB,CAAC;AAC3E,OAAO,EAAE,iBAAiB,EAAE,MAAM,sBAAsB,CAAC;AACzD,OAAO,EAAE,uBAAuB,EAAE,MAAM,wBAAwB,CAAC;AAEjE;;GAEG;AACH,MAAM,OAAO,cAAc;IACzB;;OAEG;IACH,mBAAmB,CACjB,MAAc,EACd,UAAwC,EACxC,mBAA6B;QAE7B,MAAM,kBAAkB,GAAG,kBAAkB,CAC3C,OAAO,CAAC,GAAG,CAAC,kBAAkB,CAAC,CAChC,CAAC;QAEF,MAAM,eAAe,GAAG,mBAAmB,IAAI,MAAM,CAAC,aAAa,EAAE,CAAC;QACtE,MAAM,YAAY,GAAG,MAAM,CAAC,eAAe,EAAE,EAAE,IAAI,YAAY,CAAC,OAAO,CAAC;QACxE,MAAM,UAAU,GAAG,YAAY,KAAK,YAAY,CAAC,IAAI,CAAC;QACtD,MAAM,UAAU,GAAG,YAAY,KAAK,YAAY,CAAC,IAAI,CAAC;QACtD,MAAM,MAAM,GAAG,MAAM,CAAC,eAAe,EAAE,CAAC,SAAS,EAAE,CAAC;QACpD,MAAM,SAAS,GAAG,MAAM,CAAC,eAAe,EAAE,CAAC,eAAe,EAAE,CAAC;QAC7D,MAAM,gBAAgB,GAAG,IAAI,GAAG,CAAC,SAAS,CAAC,CAAC;QAC5C,MAAM,gBAAgB,GAAG,MAAM,CAAC,mBAAmB,EAAE,CAAC;QAEtD,MAAM,YAAY,GAAG,YAAY,CAC/B,MAAM,CAAC,cAAc,EAAE,EACvB,MAAM,CAAC,uBAAuB,EAAE,EAAE,IAAI,KAAK,CAC5C,CAAC;QACF,MAAM,aAAa,GAAG,sBAAsB,CAAC,YAAY,CAAC,CAAC;QAC3D,MAAM,cAAc,GAAG,aAAa,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,cAAc,CAAC;QACjE,MAAM,gBAAgB,GAAG,uBAAuB,EAAE,CAAC;QAEnD,4BAA4B;QAC5B,IAAI,iBAAiB,GAAG,eAAe,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CACnD,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC,CACxB;aACE,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,aAAa,CAAC,WAAW,CAAC;aACrC,IAAI,CAAC,IAAI,CAAC,CAAC;QAEd,sCAAsC;QACtC,IAAI,UAAU,EAAE,CAAC;YACf,MAAM,QAAQ,GAAG,MAAM,CAAC,eAAe,EAAE,CAAC,WAAW,EAAE,CAAC;YACxD,MAAM,gBAAgB,GAAG,QAAQ,CAAC,MAAM,CACtC,CAAC,CAAC,EAA0B,EAAE,CAC5B,CAAC,YAAY,iBAAiB,IAAI,CAAC,CAAC,CAAC,CAAC,UAAU,CACnD,CAAC;YACF,IAAI,gBAAgB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAChC,MAAM,YAAY,GAAG,gBAAgB;qBAClC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,aAAa,CAAC,CAAC,IAAI,OAAO,CAAC,CAAC,UAAU,UAAU,CAAC;qBAC5D,IAAI,CAAC,IAAI,CAAC,CAAC;gBACd,iBAAiB,IAAI,KAAK,YAAY,EAAE,CAAC;YAC3C,CAAC;QACH,CAAC;QAED,IAAI,UAAkB,CAAC;QAEvB,iCAAiC;QACjC,IAAI,kBAAkB,CAAC,KAAK,IAAI,CAAC,kBAAkB,CAAC,UAAU,EAAE,CAAC;YAC/D,IAAI,YAAY,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,WAAW,CAAC,CAAC,CAAC;YACpE,IAAI,CAAC,kBAAkB,CAAC,QAAQ,EAAE,CAAC;gBACjC,YAAY,GAAG,kBAAkB,CAAC,KAAK,CAAC;YAC1C,CAAC;YACD,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,YAAY,CAAC,EAAE,CAAC;gBACjC,MAAM,IAAI,KAAK,CAAC,+BAA+B,YAAY,GAAG,CAAC,CAAC;YAClE,CAAC;YACD,UAAU,GAAG,EAAE,CAAC,YAAY,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC;YACnD,MAAM,YAAY,GAAG,cAAc,CAAC,iBAAiB,CACnD,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;gBACjB,IAAI,EAAE,CAAC,CAAC,IAAI;gBACZ,WAAW,EAAE,CAAC,CAAC,WAAW;gBAC1B,QAAQ,EAAE,CAAC,CAAC,QAAQ;aACrB,CAAC,CAAC,CACJ,CAAC;YACF,UAAU,GAAG,kBAAkB,CAC7B,UAAU,EACV,MAAM,EACN,YAAY,EACZ,aAAa,CACd,CAAC;QACJ,CAAC;aAAM,CAAC;YACN,+BAA+B;YAC/B,MAAM,qBAAqB,GACzB,OAAO,UAAU,KAAK,QAAQ;gBAC9B,UAAU,KAAK,IAAI;gBACnB,CAAC,CAAC,CAAC,UAAU,CAAC,MAAM,EAAE,IAAI,EAAE;oBAC1B,CAAC,CAAC,UAAU,CAAC,SAAS,EAAE,IAAI,EAAE;oBAC9B,CAAC,CAAC,UAAU,CAAC,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;YAElC,MAAM,OAAO,GAAiC;gBAC5C,QAAQ,EAAE,IAAI,CAAC,WAAW,CAAC,UAAU,EAAE,GAAG,EAAE,CAAC,CAAC;oBAC5C,WAAW,EAAE,eAAe;iBAC7B,CAAC,CAAC;gBACH,YAAY,EAAE,IAAI,CAAC,WAAW,CAAC,cAAc,EAAE,GAAG,EAAE,CAAC,CAAC;oBACpD,WAAW,EAAE,eAAe;oBAC5B,SAAS,EAAE,MAAM,CAAC,MAAM,GAAG,CAAC;oBAC5B,qBAAqB;oBACrB,gBAAgB;iBACjB,CAAC,CAAC;gBACH,SAAS,EAAE,IAAI,CAAC,WAAW,CAAC,eAAe,EAAE,GAAG,EAAE,CAChD,MAAM;qBACH,gBAAgB,EAAE;qBAClB,iBAAiB,EAAE;qBACnB,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;oBACX,IAAI,EAAE,CAAC,CAAC,IAAI;oBACZ,WAAW,EAAE,CAAC,CAAC,WAAW;iBAC3B,CAAC,CAAC,CACN;gBACD,WAAW,EAAE,IAAI,CAAC,WAAW,CAC3B,aAAa,EACb,GAAG,EAAE,CACH,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;oBACjB,IAAI,EAAE,CAAC,CAAC,IAAI;oBACZ,WAAW,EAAE,CAAC,CAAC,WAAW;oBAC1B,QAAQ,EAAE,CAAC,CAAC,QAAQ;iBACrB,CAAC,CAAC,EACL,MAAM,CAAC,MAAM,GAAG,CAAC,CAClB;gBACD,WAAW,EAAE,gBAAgB,CAAC,aAAa,CAAC,IAAI,SAAS;gBACzD,gBAAgB,EAAE,IAAI,CAAC,WAAW,CAChC,kBAAkB,EAClB,GAAG,EAAE,CAAC,CAAC;oBACL,WAAW,EAAE,eAAe;oBAC5B,0BAA0B,EAAE,gBAAgB,CAAC,GAAG,CAC9C,yBAAyB,CAAC,IAAI,CAC/B;oBACD,oBAAoB,EAAE,gBAAgB,CAAC,GAAG,CAAC,qBAAqB,CAAC;oBACjE,uBAAuB,EAAE,gBAAgB,CAAC,GAAG,CAC3C,yBAAyB,CAC1B;oBACD,UAAU,EAAE,gBAAgB,CAAC,GAAG,CAAC,cAAc,CAAC;oBAChD,UAAU,EAAE,gBAAgB,CAAC,GAAG,CAAC,cAAc,CAAC;oBAChD,YAAY,EAAE,gBAAgB;wBAC5B,CAAC,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE;wBAC5B,CAAC,CAAC,SAAS;iBACd,CAAC,EACF,CAAC,UAAU,CACZ;gBACD,gBAAgB,EAAE,IAAI,CAAC,WAAW,CAChC,kBAAkB,EAClB,GAAG,EAAE,CAAC,CAAC;oBACL,iBAAiB;oBACjB,QAAQ,EAAE,MAAM,CAAC,OAAO,CAAC,sBAAsB,EAAE;oBACjD,gBAAgB,EAAE,MAAM,CAAC,mBAAmB,EAAE;iBAC/C,CAAC,EACF,UAAU,CACX;gBACD,qBAAqB,EAAE,IAAI,CAAC,WAAW,CACrC,uBAAuB,EACvB,GAAG,EAAE,CAAC,CAAC;oBACL,WAAW,EAAE,eAAe;oBAC5B,qBAAqB,EAAE,MAAM,CAAC,8BAA8B,EAAE;oBAC9D,uBAAuB,EAAE,MAAM,CAAC,yBAAyB,EAAE;iBAC5D,CAAC,CACH;gBACD,OAAO,EAAE,IAAI,CAAC,WAAW,CAAC,SAAS,EAAE,GAAG,EAAE,CAAC,cAAc,EAAE,CAAC;gBAC5D,mBAAmB,EAAE,IAAI,CAAC,WAAW,CACnC,qBAAqB,EACrB,GAAG,EAAE,CAAC,IAAI,EACV,UAAU,IAAI,eAAe,CAC9B;gBACD,OAAO,EAAE,IAAI,CAAC,WAAW,CACvB,KAAK,EACL,GAAG,EAAE,CAAC,CAAC,EAAE,WAAW,EAAE,eAAe,EAAE,CAAC,EACxC,eAAe,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,KAAK,CAC9C;gBACD,aAAa,EAAE,aAAa;oBAC1B,CAAC,CAAC,SAAS;oBACX,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,eAAe,EAAE,GAAG,EAAE,CAAC,CAAC;wBACvC,gBAAgB,EAAE,mBAAmB;qBACtC,CAAC,CAAC;aACwB,CAAC;YAElC,uEAAuE;YACvE,MAAM,mBAAmB,GAAG,cAAc,CAAC,mBAEhC,CAAC;YACZ,UAAU,GAAG,mBAAmB,CAAC,OAAO,CAAC,CAAC;QAC5C,CAAC;QAED,+BAA+B;QAC/B,MAAM,WAAW,GAAG,cAAc,CAAC,gBAAgB,CACjD,UAAU,EACV,UAAU,EACV,gBAAgB,CACjB,CAAC;QAEF,6CAA6C;QAC7C,MAAM,eAAe,GAAG,WAAW,CAAC,OAAO,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;QAE/D,kCAAkC;QAClC,IAAI,CAAC,kBAAkB,CACrB,eAAe,EACf,kBAAkB,EAClB,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,WAAW,CAAC,CAAC,CACjD,CAAC;QAEF,OAAO,eAAe,CAAC;IACzB,CAAC;IAED,oBAAoB,CAAC,MAAc;QACjC,MAAM,YAAY,GAAG,YAAY,CAC/B,MAAM,CAAC,cAAc,EAAE,EACvB,MAAM,CAAC,uBAAuB,EAAE,EAAE,IAAI,KAAK,CAC5C,CAAC;QACF,MAAM,aAAa,GAAG,sBAAsB,CAAC,YAAY,CAAC,CAAC;QAC3D,MAAM,cAAc,GAAG,aAAa,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,cAAc,CAAC;QACjE,OAAO,cAAc,CAAC,oBAAoB,EAAE,CAAC;IAC/C,CAAC;IAEO,WAAW,CACjB,GAAW,EACX,OAAgB,EAChB,QAAiB,IAAI;QAErB,OAAO,KAAK,IAAI,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;IAChE,CAAC;IAEO,kBAAkB,CACxB,UAAkB,EAClB,UAAwB,EACxB,WAAmB;QAEnB,MAAM,uBAAuB,GAAG,kBAAkB,CAChD,OAAO,CAAC,GAAG,CAAC,wBAAwB,CAAC,CACtC,CAAC;QACF,IAAI,uBAAuB,CAAC,KAAK,IAAI,CAAC,uBAAuB,CAAC,UAAU,EAAE,CAAC;YACzE,MAAM,SAAS,GAAG,uBAAuB,CAAC,QAAQ;gBAChD,CAAC,CAAC,WAAW;gBACb,CAAC,CAAC,uBAAuB,CAAC,KAAK,CAAC;YAClC,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YAC3D,EAAE,CAAC,aAAa,CAAC,SAAS,EAAE,UAAU,CAAC,CAAC;QAC1C,CAAC;IACH,CAAC;CACF;AAED,mCAAmC;AAEnC,SAAS,cAAc;IACrB,IAAI,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,KAAK,cAAc;QAAE,OAAO,gBAAgB,CAAC;IACvE,IAAI,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC;QAAE,OAAO,SAAS,CAAC;IAC7C,OAAO,SAAS,CAAC;AACnB,CAAC"}

package/dist/src/prompts/promptProvider.test.d.ts

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

package/dist/src/prompts/promptProvider.test.js

@@ -0,0 +1,71 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, vi, beforeEach } from 'vitest';
+import { PromptProvider } from './promptProvider.js';
+import { getAllGeminiMdFilenames, DEFAULT_CONTEXT_FILENAME, } from '../tools/memoryTool.js';
+import { PREVIEW_GEMINI_MODEL } from '../config/models.js';
+vi.mock('../tools/memoryTool.js', async (importOriginal) => {
+    const actual = await importOriginal();
+    return {
+        ...actual,
+        getAllGeminiMdFilenames: vi.fn(),
+    };
+});
+vi.mock('../utils/gitUtils', () => ({
+    isGitRepository: vi.fn().mockReturnValue(false),
+}));
+describe('PromptProvider', () => {
+    let mockConfig;
+    beforeEach(() => {
+        vi.resetAllMocks();
+        mockConfig = {
+            getToolRegistry: vi.fn().mockReturnValue({
+                getAllToolNames: vi.fn().mockReturnValue([]),
+                getAllTools: vi.fn().mockReturnValue([]),
+            }),
+            getEnableShellOutputEfficiency: vi.fn().mockReturnValue(true),
+            storage: {
+                getProjectTempDir: vi.fn().mockReturnValue('/tmp/project-temp'),
+                getProjectTempPlansDir: vi
+                    .fn()
+                    .mockReturnValue('/tmp/project-temp/plans'),
+            },
+            isInteractive: vi.fn().mockReturnValue(true),
+            isInteractiveShellEnabled: vi.fn().mockReturnValue(true),
+            getSkillManager: vi.fn().mockReturnValue({
+                getSkills: vi.fn().mockReturnValue([]),
+            }),
+            getActiveModel: vi.fn().mockReturnValue(PREVIEW_GEMINI_MODEL),
+            getAgentRegistry: vi.fn().mockReturnValue({
+                getAllDefinitions: vi.fn().mockReturnValue([]),
+            }),
+            getApprovedPlanPath: vi.fn().mockReturnValue(undefined),
+            getApprovalMode: vi.fn(),
+        };
+    });
+    it('should handle multiple context filenames in the system prompt', () => {
+        vi.mocked(getAllGeminiMdFilenames).mockReturnValue([
+            DEFAULT_CONTEXT_FILENAME,
+            'CUSTOM.md',
+            'ANOTHER.md',
+        ]);
+        const provider = new PromptProvider();
+        const prompt = provider.getCoreSystemPrompt(mockConfig);
+        // Verify renderCoreMandates usage
+        expect(prompt).toContain(`Instructions found in \`${DEFAULT_CONTEXT_FILENAME}\`, \`CUSTOM.md\` or \`ANOTHER.md\` files are foundational mandates.`);
+    });
+    it('should handle multiple context filenames in user memory section', () => {
+        vi.mocked(getAllGeminiMdFilenames).mockReturnValue([
+            DEFAULT_CONTEXT_FILENAME,
+            'CUSTOM.md',
+        ]);
+        const provider = new PromptProvider();
+        const prompt = provider.getCoreSystemPrompt(mockConfig, 'Some memory content');
+        // Verify renderUserMemory usage
+        expect(prompt).toContain(`# Contextual Instructions (${DEFAULT_CONTEXT_FILENAME}, CUSTOM.md)`);
+    });
+});
+//# sourceMappingURL=promptProvider.test.js.map

package/dist/src/prompts/promptProvider.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"promptProvider.test.js","sourceRoot":"","sources":["../../../src/prompts/promptProvider.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AAErD,OAAO,EACL,uBAAuB,EACvB,wBAAwB,GACzB,MAAM,wBAAwB,CAAC;AAChC,OAAO,EAAE,oBAAoB,EAAE,MAAM,qBAAqB,CAAC;AAE3D,EAAE,CAAC,IAAI,CAAC,wBAAwB,EAAE,KAAK,EAAE,cAAc,EAAE,EAAE;IACzD,MAAM,MAAM,GAAG,MAAM,cAAc,EAAE,CAAC;IACtC,OAAO;QACL,GAAI,MAAiB;QACrB,uBAAuB,EAAE,EAAE,CAAC,EAAE,EAAE;KACjC,CAAC;AACJ,CAAC,CAAC,CAAC;AAEH,EAAE,CAAC,IAAI,CAAC,mBAAmB,EAAE,GAAG,EAAE,CAAC,CAAC;IAClC,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,KAAK,CAAC;CAChD,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;IAC9B,IAAI,UAAkB,CAAC;IAEvB,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,UAAU,GAAG;YACX,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC;gBACvC,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,EAAE,CAAC;gBAC5C,WAAW,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,EAAE,CAAC;aACzC,CAAC;YACF,8BAA8B,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,IAAI,CAAC;YAC7D,OAAO,EAAE;gBACP,iBAAiB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,mBAAmB,CAAC;gBAC/D,sBAAsB,EAAE,EAAE;qBACvB,EAAE,EAAE;qBACJ,eAAe,CAAC,yBAAyB,CAAC;aAC9C;YACD,aAAa,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,IAAI,CAAC;YAC5C,yBAAyB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,IAAI,CAAC;YACxD,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC;gBACvC,SAAS,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,EAAE,CAAC;aACvC,CAAC;YACF,cAAc,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,oBAAoB,CAAC;YAC7D,gBAAgB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC;gBACxC,iBAAiB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,EAAE,CAAC;aAC/C,CAAC;YACF,mBAAmB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,SAAS,CAAC;YACvD,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE;SACJ,CAAC;IACzB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+DAA+D,EAAE,GAAG,EAAE;QACvE,EAAE,CAAC,MAAM,CAAC,uBAAuB,CAAC,CAAC,eAAe,CAAC;YACjD,wBAAwB;YACxB,WAAW;YACX,YAAY;SACb,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,IAAI,cAAc,EAAE,CAAC;QACtC,MAAM,MAAM,GAAG,QAAQ,CAAC,mBAAmB,CAAC,UAAU,CAAC,CAAC;QAExD,kCAAkC;QAClC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CACtB,2BAA2B,wBAAwB,sEAAsE,CAC1H,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iEAAiE,EAAE,GAAG,EAAE;QACzE,EAAE,CAAC,MAAM,CAAC,uBAAuB,CAAC,CAAC,eAAe,CAAC;YACjD,wBAAwB;YACxB,WAAW;SACZ,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,IAAI,cAAc,EAAE,CAAC;QACtC,MAAM,MAAM,GAAG,QAAQ,CAAC,mBAAmB,CACzC,UAAU,EACV,qBAAqB,CACtB,CAAC;QAEF,gCAAgC;QAChC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CACtB,8BAA8B,wBAAwB,cAAc,CACrE,CAAC;IACJ,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/prompts/snippets.d.ts

@@ -0,0 +1,86 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { HierarchicalMemory } from '../config/memory.js';
+export interface SystemPromptOptions {
+    preamble?: PreambleOptions;
+    coreMandates?: CoreMandatesOptions;
+    subAgents?: SubAgentOptions[];
+    agentSkills?: AgentSkillOptions[];
+    hookContext?: boolean;
+    primaryWorkflows?: PrimaryWorkflowsOptions;
+    planningWorkflow?: PlanningWorkflowOptions;
+    operationalGuidelines?: OperationalGuidelinesOptions;
+    sandbox?: SandboxMode;
+    interactiveYoloMode?: boolean;
+    gitRepo?: GitRepoOptions;
+}
+export interface PreambleOptions {
+    interactive: boolean;
+}
+export interface CoreMandatesOptions {
+    interactive: boolean;
+    hasSkills: boolean;
+    hasHierarchicalMemory: boolean;
+    contextFilenames?: string[];
+}
+export interface PrimaryWorkflowsOptions {
+    interactive: boolean;
+    enableCodebaseInvestigator: boolean;
+    enableWriteTodosTool: boolean;
+    enableEnterPlanModeTool: boolean;
+    enableGrep: boolean;
+    enableGlob: boolean;
+    approvedPlan?: {
+        path: string;
+    };
+}
+export interface OperationalGuidelinesOptions {
+    interactive: boolean;
+    interactiveShellEnabled: boolean;
+}
+export type SandboxMode = 'macos-seatbelt' | 'generic' | 'outside';
+export interface GitRepoOptions {
+    interactive: boolean;
+}
+export interface PlanningWorkflowOptions {
+    planModeToolsList: string;
+    plansDir: string;
+    approvedPlanPath?: string;
+}
+export interface AgentSkillOptions {
+    name: string;
+    description: string;
+    location: string;
+}
+export interface SubAgentOptions {
+    name: string;
+    description: string;
+}
+/**
+ * Composes the core system prompt from its constituent subsections.
+ * Adheres to the minimal complexity principle by using simple interpolation of function calls.
+ */
+export declare function getCoreSystemPrompt(options: SystemPromptOptions): string;
+/**
+ * Wraps the base prompt with user memory and approval mode plans.
+ */
+export declare function renderFinalShell(basePrompt: string, userMemory?: string | HierarchicalMemory, contextFilenames?: string[]): string;
+export declare function renderPreamble(options?: PreambleOptions): string;
+export declare function renderCoreMandates(options?: CoreMandatesOptions): string;
+export declare function renderSubAgents(subAgents?: SubAgentOptions[]): string;
+export declare function renderAgentSkills(skills?: AgentSkillOptions[]): string;
+export declare function renderHookContext(enabled?: boolean): string;
+export declare function renderPrimaryWorkflows(options?: PrimaryWorkflowsOptions): string;
+export declare function renderOperationalGuidelines(options?: OperationalGuidelinesOptions): string;
+export declare function renderSandbox(mode?: SandboxMode): string;
+export declare function renderInteractiveYoloMode(enabled?: boolean): string;
+export declare function renderGitRepo(options?: GitRepoOptions): string;
+export declare function renderUserMemory(memory?: string | HierarchicalMemory, contextFilenames?: string[]): string;
+export declare function renderPlanningWorkflow(options?: PlanningWorkflowOptions): string;
+/**
+ * Provides the system prompt for history compression.
+ */
+export declare function getCompressionPrompt(): string;