Make changes per review

Author: marcnjaramillo

extensions/ql-vscode/CHANGELOG.md

@@ -2,9 +2,10 @@
 
 ## [UNRELEASED]
 
+- Add CodeLens feature to make the CodeQL quick evaluation command more accessible. [#1035](https://github.com/github/vscode-codeql/pull/1035)
+  
 ## 1.5.8 - 2 December 2021
 
-- Add CodeLens feature to make the CodeQL quick evaluation command more accessible. [#1035](https://github.com/github/vscode-codeql/pull/1035)
 - Emit a more explicit error message when a user tries to add a database with an unzipped source folder to the workspace. [#1021](https://github.com/github/vscode-codeql/pull/1021)
 - Ensure `src.zip` archives are used as the canonical source instead of `src` folders when importing databases. [#1025](https://github.com/github/vscode-codeql/pull/1025)
 

extensions/ql-vscode/src/CodeLensProvider.ts

@@ -16,23 +16,24 @@ class QuickEvalCodeLensProvider implements CodeLensProvider {
     for (let index = 0; index < document.lineCount; index++) {
       const textLine = document.lineAt(index);
       // Match a predicate signature, including predicate name, parameter list, and opening brace.
-      const regex = new RegExp(/\w+\(\s*.*(?:,\s*)*\)\s*\{/);
-        const startIndex = textLine.text.search(regex);
-        if (startIndex !== -1) {
-      const range: Range = new Range(
-        textLine.range.start.line, startIndex,
-        textLine.range.end.line, startIndex + 1
-      );
+      const regex = new RegExp(/(\w+)\s*\(\s*.*(?:,\s*)*\)\s*\{/);
+      const matches = textLine.text.match(regex);
+      const startIndex = textLine.text.search(regex);
+      if (startIndex !== -1) {
+        const range: Range = new Range(
+          textLine.range.start.line, startIndex,
+          textLine.range.end.line, startIndex + 1
+        );
 
-      const command: Command = {
-        command: 'codeQL.codeLensQuickEval',
-        title: 'CodeQL: Quick Evaluation',
-        arguments: [document.uri, range]
-      };
-      const codeLens = new CodeLens(range, command);
-      codeLenses.push(codeLens);
+        const command: Command = {
+          command: 'codeQL.codeLensQuickEval',
+          title: `Quick Evaluation: ${matches![1]}`,
+          arguments: [document.uri, range]
+        };
+        const codeLens = new CodeLens(range, command);
+        codeLenses.push(codeLens);
+      }
     }
-  }
     return codeLenses;
   }
 }