Merge pull request #6012 from continuedev/feature/view-context/base-copy

[Updated PR] [Feature] Autocomplete context from other tabs open in the editor

Author: adarshramiyer

core/autocomplete/snippets/getAllSnippets.ts

@@ -3,6 +3,7 @@ import { findUriInDirs } from "../../util/uri";
 import { ContextRetrievalService } from "../context/ContextRetrievalService";
 import { GetLspDefinitionsFunction } from "../types";
 import { HelperVars } from "../util/HelperVars";
+import { openedFilesLruCache } from "../util/openedFilesLruCache";
 import { getDiffsFromCache } from "./gitDiffCache";
 
 import {
@@ -22,6 +23,7 @@ export interface SnippetPayload {
   recentlyVisitedRangesSnippets: AutocompleteCodeSnippet[];
   diffSnippets: AutocompleteDiffSnippet[];
   clipboardSnippets: AutocompleteClipboardSnippet[];
+  recentlyOpenedFileSnippets: AutocompleteCodeSnippet[];
 }
 
 function racePromise<T>(promise: Promise<T[]>, timeout = 100): Promise<T[]> {
@@ -103,6 +105,65 @@ const getDiffSnippets = async (
   });
 };
 
+const getSnippetsFromRecentlyOpenedFiles = async (
+  helper: HelperVars,
+  ide: IDE,
+): Promise<AutocompleteCodeSnippet[]> => {
+  if (helper.options.useRecentlyOpened === false) {
+    return [];
+  }
+
+  try {
+    const currentFileUri = `${helper.filepath}`;
+
+    // Get all file URIs excluding the current file
+    const fileUrisToRead = [...openedFilesLruCache.entriesDescending()]
+      .filter(([fileUri, _]) => fileUri !== currentFileUri)
+      .map(([fileUri, _]) => fileUri);
+
+    // Create an array of promises that each read a file with timeout
+    const fileReadPromises = fileUrisToRead.map((fileUri) => {
+      // Create a promise that resolves to a snippet or null
+      const readPromise = new Promise<AutocompleteCodeSnippet | null>(
+        (resolve) => {
+          ide
+            .readFile(fileUri)
+            .then((fileContent) => {
+              if (!fileContent || fileContent.trim() === "") {
+                resolve(null);
+                return;
+              }
+
+              resolve({
+                filepath: fileUri,
+                content: fileContent,
+                type: AutocompleteSnippetType.Code,
+              });
+            })
+            .catch((e) => {
+              console.error(`Failed to read file ${fileUri}:`, e);
+              resolve(null);
+            });
+        },
+      );
+      // Cut off at 80ms via racing promises
+      return Promise.race([
+        readPromise,
+        new Promise<null>((resolve) => setTimeout(() => resolve(null), 80)),
+      ]);
+    });
+
+    // Execute all file reads in parallel
+    const results = await Promise.all(fileReadPromises);
+
+    // Filter out null results
+    return results.filter(Boolean) as AutocompleteCodeSnippet[];
+  } catch (e) {
+    console.error("Error processing opened files cache:", e);
+    return [];
+  }
+};
+
 export const getAllSnippets = async ({
   helper,
   ide,
@@ -123,6 +184,7 @@ export const getAllSnippets = async ({
     ideSnippets,
     diffSnippets,
     clipboardSnippets,
+    recentlyOpenedFileSnippets,
   ] = await Promise.all([
     racePromise(contextRetrievalService.getRootPathSnippets(helper)),
     racePromise(
@@ -133,6 +195,7 @@ export const getAllSnippets = async ({
       : [],
     [], // racePromise(getDiffSnippets(ide)) // temporarily disabled, see https://github.com/continuedev/continue/pull/5882,
     racePromise(getClipboardSnippets(ide)),
+    racePromise(getSnippetsFromRecentlyOpenedFiles(helper, ide)), // giving this one a little more time to complete
   ]);
 
   return {
@@ -143,5 +206,6 @@ export const getAllSnippets = async ({
     diffSnippets,
     clipboardSnippets,
     recentlyVisitedRangesSnippets: helper.input.recentlyVisitedRanges,
+    recentlyOpenedFileSnippets,
   };
 };