Create branch for Code Lens feature

Author: marcnjaramillo

extensions/ql-vscode/package.json

@@ -128,6 +128,10 @@
       "type": "object",
       "title": "CodeQL",
       "properties": {
+        "codelens-sample.enableCodeLens": {
+          "type": "boolean",
+          "default": true
+        },
         "codeQL.cli.executablePath": {
           "scope": "window",
           "type": "string",
@@ -267,6 +271,16 @@
       }
     },
     "commands": [
+      {
+        "title": "Enable CodeLens",
+        "command": "codelens-sample.enableCodeLens",
+        "category": "CodeLens Sample"
+      },
+      {
+          "title": "Disable Codelens",
+          "command": "codelens-sample.disableCodeLens",
+          "category": "CodeLens Sample"
+      },
       {
         "command": "codeQL.authenticateToGitHub",
         "title": "CodeQL: Authenticate to GitHub"

extensions/ql-vscode/src/CodeLensProvider.ts

@@ -0,0 +1,28 @@
+import {
+  CodeLensProvider,
+  TextDocument,
+  CodeLens,
+  Range,
+  Command
+} from 'vscode';
+
+class MyCodeLensProvider implements CodeLensProvider {
+  // Each provider requires a provideCodeLenses function which will give the various documents
+  // the code lenses
+  async provideCodeLenses(_document: TextDocument): Promise<CodeLens[]> {
+    // Define where the CodeLens will exist
+    const topOfDocument = new Range(0, 0, 0, 0);
+
+    // Define what command we want to trigger when activating the CodeLens
+    const command: Command = {
+      command: 'codeQL.quickEval',
+      title: 'CodeQL: Quick Evaluation'
+    };
+
+    const codeLens = new CodeLens(topOfDocument, command);
+
+    return [codeLens];
+  }
+}
+
+export default MyCodeLensProvider;

extensions/ql-vscode/src/extension.ts

@@ -8,6 +8,7 @@ import {
   ProgressLocation,
   ProgressOptions,
   Uri,
+  // workspace,
   window as Window,
   env,
   window,
@@ -20,6 +21,7 @@ import { testExplorerExtensionId, TestHub } from 'vscode-test-adapter-api';
 
 import { AstViewer } from './astViewer';
 import * as archiveFilesystemProvider from './archive-filesystem-provider';
+import MyCodeLensProvider from './CodelensProvider';
 import { CodeQLCliServer, CliVersionConstraint } from './cli';
 import {
   CliConfigListener,
@@ -154,6 +156,23 @@ export interface CodeQLExtensionInterface {
  * @returns CodeQLExtensionInterface
  */
 export async function activate(ctx: ExtensionContext): Promise<CodeQLExtensionInterface | Record<string, never>> {
+
+  const codelensProvider = new MyCodeLensProvider();
+
+  languages.registerCodeLensProvider('*', codelensProvider);
+
+  // commands.registerCommand('codelens-sample.enableCodeLens', () => {
+  //     void workspace.getConfiguration('codelens-sample').update('enableCodeLens', true, true);
+  // });
+
+  // commands.registerCommand('codelens-sample.disableCodeLens', () => {
+  //     void workspace.getConfiguration('codelens-sample').update('enableCodeLens', false, true);
+  // });
+
+  // commands.registerCommand('codelens-sample.codelensAction', (args: any) => {
+  //     void window.showInformationMessage(`CodeLens action clicked with args=${args}`);
+  // });
+  
   void logger.log(`Starting ${extensionId} extension`);
   if (extension === undefined) {
     throw new Error(`Can't find extension ${extensionId}`);
@@ -461,6 +480,7 @@ async function activateWithInstalledDistribution(
     forceReveal: WebviewReveal
   ): Promise<void> {
     await intm.showResults(query, forceReveal, false);
+    console.log(query.query.quickEvalPosition);
   }
 
   async function compileAndRunQuery(