feat: add ability to go to a file in a specific folder

Author: olafurpg

package.json

@@ -84,6 +84,11 @@
         "category": "Sourcegraph",
         "title": "Go to File"
       },
+      {
+        "command": "extension.goToFileInFolder",
+        "category": "Sourcegraph",
+        "title": "Go to File in Folder"
+      },
       {
         "command": "extension.goToRepository",
         "category": "Sourcegraph",
@@ -151,6 +156,9 @@
           "group": "navigation",
           "when": "view == sourcegraph.files"
         }
+      ],
+      "view/item/context": [
+        {"command": "extension.goToFileInFolder", "when": "view == sourcegraph.files && viewItem == directory"}
       ]
     },
     "keybindings": [

src/commands/goToFileCommand.ts

@@ -3,21 +3,27 @@ import { SourcegraphFileSystemProvider } from '../file-system/SourcegraphFileSys
 import { SourcegraphQuickPick } from './SourcegraphQuickPick'
 import { recentlyOpenFilesSetting } from '../settings/recentlyOpenFilesSetting'
 import { log } from '../log'
+import { SourcegraphUri } from '../file-system/SourcegraphUri'
 
-export async function goToFileCommand(fs: SourcegraphFileSystemProvider): Promise<void> {
+export async function goToFileCommand(fs: SourcegraphFileSystemProvider, folderUri?: string): Promise<void> {
     const quick = new SourcegraphQuickPick(fs)
     quick.pick.title = 'Go to a file from the open Sourcegraph repositories'
-    const recentlyOpenFiles = recentlyOpenFilesSetting.load()
+    const recentlyOpenFiles = folderUri ? [] : recentlyOpenFilesSetting.load()
     const fileItems = [...recentlyOpenFiles]
     quick.pick.items = fileItems
     quick.pick.busy = true
-    fs.allFileFromOpenRepositories().then(
+    const folder = folderUri ? SourcegraphUri.parse(folderUri) : undefined
+    const folderPath = folder?.path ? folder.path + '/' : undefined
+    fs.allFileFromOpenRepositories(folder).then(
         allFiles => {
             for (const repo of allFiles) {
                 for (const file of repo.fileNames) {
                     if (file === '') {
                         continue
                     }
+                    if (folderPath && !file.startsWith(folderPath)) {
+                        continue
+                    }
                     // Intentionally avoid using `SourcegraphUri.parse()` for
                     // performance reasons.  This loop is a hot path for large
                     // repositories like chromium/chromium with ~400k files.

src/extension.ts

@@ -80,6 +80,15 @@ export function activate(context: vscode.ExtensionContext): void {
     treeDataProvider.setTreeView(treeView)
     const semanticTokens = new SourcegraphSemanticTokenProvider()
     context.subscriptions.push(treeView)
+    context.subscriptions.push(
+        vscode.commands.registerCommand('extension.goToFileInFolder', async (uri: string | undefined) => {
+            if (typeof uri === 'string') {
+                await goToFileCommand(fs, uri)
+            } else {
+                log.error(`extension.goToFileInFolder - invalid argument ${uri || 'undefined'}`)
+            }
+        })
+    )
     context.subscriptions.push(
         vscode.commands.registerCommand(
             'extension.goToFile',

src/file-system/SourcegraphFileSystemProvider.ts

@@ -109,9 +109,13 @@ export class SourcegraphFileSystemProvider implements vscode.FileSystemProvider
         return [...this.fileNamesByRepository.keys()]
     }
 
-    public async allFileFromOpenRepositories(): Promise<RepositoryFileNames[]> {
+    public async allFileFromOpenRepositories(folder?: SourcegraphUri): Promise<RepositoryFileNames[]> {
         const promises: RepositoryFileNames[] = []
+        const folderRepositoryUri = folder?.repositoryUri()
         for (const [repositoryUri, downloadingFileNames] of this.fileNamesByRepository.entries()) {
+            if (folderRepositoryUri && repositoryUri !== folderRepositoryUri) {
+                continue
+            }
             try {
                 const fileNames = await downloadingFileNames
                 const uri = SourcegraphUri.parse(repositoryUri)

src/file-system/SourcegraphTreeDataProvider.ts

@@ -186,6 +186,7 @@ export class SourcegraphTreeDataProvider implements vscode.TreeDataProvider<stri
                 : vscode.TreeItemCollapsibleState.Collapsed,
             command,
             resourceUri: vscode.Uri.parse(uri.uri),
+            contextValue: uri.isFile() ? 'file' : 'directory',
         }
     }
 }