feat: add support to change git revision

Author: olafurpg

package.json

@@ -94,6 +94,11 @@
         "category": "Sourcegraph",
         "title": "Go to Repository"
       },
+      {
+        "command": "extension.switchGitRevision",
+        "category": "Sourcegraph",
+        "title": "Switch git revision"
+      },
       {
         "command": "extension.newNotebook",
         "category": "Sourcegraph",

src/commands/switchGitRevisionCommand.ts

@@ -0,0 +1,45 @@
+import { SourcegraphTreeDataProvider } from '../file-system/SourcegraphTreeDataProvider'
+import { log } from '../log'
+import { GitReference, gitReferencesQuery } from '../queries/gitReferencesQuery'
+import { openSourcegraphUriCommand } from './openSourcegraphUriCommand'
+import { SourcegraphQuickPick } from './SourcegraphQuickPick'
+
+export async function switchGitRevisionCommand(tree: SourcegraphTreeDataProvider): Promise<void> {
+    const quick = new SourcegraphQuickPick(tree.fs)
+    quick.pick.title = 'Search for a git branch, git tag or a git commit'
+    const activeTextDocument = tree.activeTextDocument()
+    if (!activeTextDocument || !activeTextDocument.path) {
+        return
+    }
+    const activeTextDocumentPath = activeTextDocument.path
+    const metadata = await tree.fs.repositoryMetadata(activeTextDocument.repositoryName)
+    quick.onDidChangeValue(async query => {
+        quick.pick.busy = true
+        log.appendLine(`gitReferences: ${query.text}`)
+        const references = await gitReferencesQuery(
+            { query: query.text, repositoryId: metadata?.id || '' },
+            query.token
+        )
+        log.appendLine(`gitReferences: ${query.text} ${JSON.stringify(references)}`)
+        quick.pick.busy = false
+        quick.pick.items = references.map(reference => ({
+            label: gitReferenceTag(reference) + reference.displayName,
+            uri: `sourcegraph://${activeTextDocument.host}${reference.url}/-/blob/${activeTextDocumentPath}`,
+        }))
+    })
+    const uri = await quick.showQuickPickAndGetUserInput()
+    await openSourcegraphUriCommand(tree.fs, uri)
+}
+
+function gitReferenceTag(reference: GitReference): string {
+    switch (reference.type) {
+        case 'GIT_TAG':
+            return ' $(tag)'
+        case 'GIT_BRANCH':
+            return ' $(git-branch)'
+        case 'GIT_COMMIT':
+            return ' $(git-commit)'
+        default:
+            return ''
+    }
+}

src/extension.ts

@@ -16,6 +16,7 @@ import { SourcegraphHoverProvider } from './code-intel/SourcegraphHoverProvider'
 import { SourcegraphDefinitionProvider } from './code-intel/SourcegraphDefinitionProvider'
 import { SourcegraphReferenceProvider } from './code-intel/SourcegraphReferenceProvider'
 import { SourcegraphTreeDataProvider } from './file-system/SourcegraphTreeDataProvider'
+import { switchGitRevisionCommand } from './commands/switchGitRevisionCommand'
 
 // eslint-disable-next-line @typescript-eslint/no-require-imports, @typescript-eslint/no-var-requires
 const { version } = require('../package.json')
@@ -101,6 +102,12 @@ export function activate(context: vscode.ExtensionContext): void {
             handleCommandErrors('extension.goToRepository', () => goToRepositoryCommand(fs))
         )
     )
+    context.subscriptions.push(
+        vscode.commands.registerCommand(
+            'extension.switchGitRevision',
+            handleCommandErrors('extension.switchGitRevision', () => switchGitRevisionCommand(treeDataProvider))
+        )
+    )
     context.subscriptions.push(
         vscode.commands.registerCommand(
             'extension.newNotebook',

src/file-system/SourcegraphTreeDataProvider.ts

@@ -4,7 +4,7 @@ import { SourcegraphFileSystemProvider } from './SourcegraphFileSystemProvider'
 import { SourcegraphUri } from './SourcegraphUri'
 
 export class SourcegraphTreeDataProvider implements vscode.TreeDataProvider<string> {
-    constructor(private readonly fs: SourcegraphFileSystemProvider) {
+    constructor(public readonly fs: SourcegraphFileSystemProvider) {
         fs.onDidDownloadRepositoryFilenames(() => this.didChangeTreeData.fire(undefined))
     }
 
@@ -17,6 +17,11 @@ export class SourcegraphTreeDataProvider implements vscode.TreeDataProvider<stri
     private readonly didChangeTreeData = new vscode.EventEmitter<string | undefined>()
     public readonly onDidChangeTreeData: vscode.Event<string | undefined> = this.didChangeTreeData.event
 
+    public activeTextDocument(): SourcegraphUri | undefined {
+        return this.activeUri && this.activeUri.scheme === 'sourcegraph'
+            ? this.fs.sourcegraphUri(this.activeUri)
+            : undefined
+    }
     public setTreeView(treeView: vscode.TreeView<string>): void {
         this.treeView = treeView
         treeView.onDidChangeVisibility(event => {

src/queries/gitReferencesQuery.ts

@@ -0,0 +1,63 @@
+import * as vscode from 'vscode'
+import { graphqlQuery } from './graphqlQuery'
+import gql from 'tagged-template-noop'
+
+export interface GitReference {
+    displayName: string
+    name: string
+    url: string
+    type: string
+}
+
+export async function gitReferencesQuery(
+    parameters: GitReferencesParameters,
+    token: vscode.CancellationToken
+): Promise<GitReference[]> {
+    const result = await graphqlQuery<GitReferencesParameters, GitReferencesResult>(
+        gql`
+            query RepositoryGitRefs($repositoryId: ID!, $query: String) {
+                node(id: $repositoryId) {
+                    __typename
+                    ... on Repository {
+                        gitRefs(first: 100, query: $query, orderBy: AUTHORED_OR_COMMITTED_AT) {
+                            __typename
+                            ...GitRefConnectionFields
+                        }
+                        __typename
+                    }
+                }
+            }
+
+            fragment GitRefConnectionFields on GitRefConnection {
+                nodes {
+                    __typename
+                    ...GitRefFields
+                }
+            }
+
+            fragment GitRefFields on GitRef {
+                displayName
+                name
+                url
+                type
+            }
+        `,
+        parameters,
+        token
+    )
+    return result?.data?.node?.gitRefs?.nodes || []
+}
+
+interface GitReferencesParameters {
+    repositoryId: string
+    query: string
+}
+interface GitReferencesResult {
+    data?: {
+        node?: {
+            gitRefs?: {
+                nodes?: GitReference[]
+            }
+        }
+    }
+}