Add improved regex

Author: marcnjaramillo

extensions/ql-vscode/src/CodeLensProvider.ts

@@ -15,13 +15,13 @@ class MyCodeLensProvider implements CodeLensProvider {
 
     for (let index = 0; index < document.lineCount; index++) {
       const textLine = document.lineAt(index);
-      if (textLine.text.includes('predicate')) {
+      const regex = new RegExp(/\w+\(\s*.*(?:,\s*\S+)*\s*\)/g);
+      if (textLine.text.match(regex)) {
         const uri = document.uri;
-        const regex = new RegExp(/predicate/);
         const startIndex = textLine.text.search(regex);
         const range: Range = new Range(
-          textLine.range.start.line, startIndex + 10,
-          textLine.range.end.line, startIndex + 11
+          textLine.range.start.line, startIndex,
+          textLine.range.end.line, startIndex + 1
         );
 
         const command: Command = {