Make server-side search respect Context Lines feature of Search Editor (#1290)

Author: gjsjohnmurray

src/providers/FileSystemProvider/TextSearchProvider.ts

@@ -323,12 +323,20 @@ export class TextSearchProvider implements vscode.TextSearchProvider {
 
       const uri = DocumentContentProvider.getUri(file.doc, "", "", true, options.folder);
       const content = decoder.decode(await vscode.workspace.fs.readFile(uri)).split("\n");
+      const contentLength = content.length;
       // Find all lines that we have matches on
       const lines = file.matches
-        .map((match: SearchMatch) => searchMatchToLine(content, match, file.doc, api.configName))
+        .map((match: SearchMatch) =>
+          token.isCancellationRequested ? null : searchMatchToLine(content, match, file.doc, api.configName)
+        )
         .filter(notNull);
-      // Filter out duplicates and compute all matches for each one
-      [...new Set(lines)].forEach((line) => {
+      // Remove duplicates and make them quickly searchable
+      const matchedLines = new Set(lines);
+      // Compute all matches for each one
+      matchedLines.forEach((line) => {
+        if (token.isCancellationRequested) {
+          return;
+        }
         const text = content[line];
         const regex = new RegExp(
           query.isRegExp ? query.pattern : query.pattern.replace(/[-[\]{}()*+?.,\\^$|#\s]/g, "\\$&"),
@@ -345,6 +353,19 @@ export class TextSearchProvider implements vscode.TextSearchProvider {
           counter++;
         }
         if (matchRanges.length && previewRanges.length) {
+          if (options.beforeContext) {
+            // Add preceding context lines that aren't themselves result lines
+            const previewFrom = Math.max(line - options.beforeContext, 0);
+            for (let i = previewFrom; i < line; i++) {
+              if (!matchedLines.has(i)) {
+                progress.report({
+                  uri,
+                  text: content[i],
+                  lineNumber: i + 1,
+                });
+              }
+            }
+          }
           progress.report({
             uri,
             ranges: matchRanges,
@@ -353,6 +374,19 @@ export class TextSearchProvider implements vscode.TextSearchProvider {
               matches: previewRanges,
             },
           });
+          if (options.afterContext) {
+            // Add following context lines that aren't themselves result lines
+            const previewTo = Math.min(line + options.afterContext, contentLength - 1);
+            for (let i = line + 1; i <= previewTo; i++) {
+              if (!matchedLines.has(i)) {
+                progress.report({
+                  uri,
+                  text: content[i],
+                  lineNumber: i + 1,
+                });
+              }
+            }
+          }
         }
       });
     };