Fix auto context to exclude file paths inside liquid comment tags

src/services/AutoContext.js

@@ -11,11 +11,26 @@ export default class AutoContext {
     let match;
 
     while ((match = regex.exec(prompt)) !== null) {
-      if (!filePaths.includes(match[1])) {
+      if (!filePaths.includes(match[1]) && !this.isInsideLiquidComment(prompt, match.index)) {
         filePaths.push(match[1]);
       }
     }
 
     return filePaths;
   }
-}
+
+  /**
+   * Checks if a file path is inside a liquid comment tag.
+   *
+   * @param {string} prompt - The input prompt containing file paths.
+   * @param {number} index - The index of the file path in the prompt.
+   * @returns {boolean} True if the file path is inside a liquid comment tag, false otherwise.
+   */
+  static isInsideLiquidComment(prompt, index) {
+    const before = prompt.slice(0, index);
+    const after = prompt.slice(index);
+    const openTag = before.lastIndexOf('{% comment %}');
+    const closeTag = before.lastIndexOf('{% endcomment %}');
+    return openTag > closeTag && after.includes('{% endcomment %}');
+  }
+}

test/AutoContext.test.js

@@ -1,3 +1,4 @@
+import { describe, it } from 'mocha'
 import assert from 'assert';
 import AutoContext from '../src/services/AutoContext.js';
 
@@ -51,6 +52,21 @@ describe('AutoContext.call', () => {
       assert.deepStrictEqual([
         '/src/services/AutoContext.js'
     ], result)
+  })
+
+  describe('when the prompt mentions paths inside liquid comment tags', () => {
+    let prompt = `
+      Instructions here
+
+      {% comment %}
+      // this file shouldn't be included in the prompt sent to the LLM
+      See file /abc/test.txt
+      {% endcomment %}
+    `
+
+    it('excludes paths inside liquid comment tags', () => {
+      const result = AutoContext.call(prompt)
+      assert.deepStrictEqual([], result)
     })
   })
 })

test/ExtractOperationsService.test.js

@@ -1,3 +1,4 @@
+import { describe, it } from 'mocha'
 import assert from 'assert'
 import { ExtractOperationsService } from '../src/services/ExtractOperationsService.js'
 

test/FileService.test.js

@@ -1,3 +1,4 @@
+import { describe, it } from 'mocha'
 import { FileService } from '../src/services/FileService.js';
 import fs from 'fs';
 import assert from 'assert';
@@ -57,4 +58,4 @@ describe('FileService', () => {
       assert.strictEqual(exists, false);
     });
   });
-});
+});

test/Main.test.js

@@ -1,3 +1,4 @@
+import { describe, it } from 'mocha'
 import assert from 'assert'
 import Main from '../src/Main.js'
 import PromptrService from '../src/services/PromptrService.js'

test/OpenAiGptService.test.js

@@ -1,3 +1,4 @@
+import { describe, it } from 'mocha'
 import assert from 'assert';
 import sinon from 'sinon';
 import OpenAiGptService from '../src/services/OpenAiGptService.js';
@@ -121,4 +122,4 @@ describe('OpenAiGptService', () => {
 
     sinon.assert.callCount(openaiStub, models.length);
   });
-});
+});

test/PromptContext.test.js

@@ -1,3 +1,4 @@
+import { describe, it } from 'mocha'
 import assert from 'assert';
 import sinon from 'sinon';
 import PromptContext from '../src/services/PromptContext.js';

test/PromptrService_call.test.js

@@ -1,3 +1,4 @@
+import { describe, it } from 'mocha'
 import assert from 'assert';
 import sinon from 'sinon';
 import PromptrService from '../src/services/PromptrService.js';
@@ -103,4 +104,4 @@ describe('PromptrService', () => {
     });
   });
 
-});
+});

test/PromptrService_executeMode.test.js

@@ -1,3 +1,4 @@
+import { describe, it } from 'mocha'
 import assert from 'assert';
 import sinon from 'sinon';
 import PromptrService from '../src/services/PromptrService.js';
@@ -88,4 +89,4 @@ describe('PromptrService', () => {
     });
   });
 
-});
+});

test/PromptrService_shouldRefactor.test.js

@@ -1,3 +1,4 @@
+import { describe, it } from 'mocha'
 import assert from 'assert';
 import sinon from 'sinon';
 import PromptrService from '../src/services/PromptrService.js';
@@ -32,4 +33,4 @@ describe('PromptrService', () => {
     })
   })
 
-});
+});

test/cliState.test.js

@@ -1,3 +1,4 @@
+import { describe, it } from 'mocha'
 import assert from 'assert';
 import CliState from '../src/CliState.js';
 
@@ -10,4 +11,4 @@ describe('CliState', () => {
     assert.strictEqual(opts.prompt, 'Test prompt');
     assert.strictEqual(opts.verbose, true);
   });
-});
+});

test/refactorResultProcessor.test.js

@@ -1,3 +1,4 @@
+import { describe, it } from 'mocha'
 import fs from 'fs';
 import path from 'path';
 import assert from 'assert';
@@ -146,4 +147,4 @@ describe('RefactorResultProcessor', () => {
       if (fs.existsSync(path.resolve('test/testDirNew/nested'))) fs.rmdirSync(path.resolve('test/testDirNew'), { recursive: true })
     });
   });
-});
+});