quickly debug classmethod by action from code

Author: daimor

package.json

@@ -580,7 +580,7 @@
             "properties": {
               "program": {
                 "type": "string",
-                "description": "Absolute path to the program."
+                "description": "Class or routine to debug."
               }
             }
           },

src/extension.ts

@@ -26,6 +26,7 @@ import { ObjectScriptDefinitionProvider } from "./providers/ObjectScriptDefiniti
 import { ObjectScriptFoldingRangeProvider } from "./providers/ObjectScriptFoldingRangeProvider";
 import { ObjectScriptHoverProvider } from "./providers/ObjectScriptHoverProvider";
 import { ObjectScriptRoutineSymbolProvider } from "./providers/ObjectScriptRoutineSymbolProvider";
+import { ObjectScriptClassCodeLensProvider } from "./providers/ObjectScriptClassCodeLensProvider";
 import { XmlContentProvider } from "./providers/XmlContentProvider";
 
 import { StatusCodeError } from "request-promise/errors";
@@ -184,6 +185,28 @@ export async function activate(context: vscode.ExtensionContext): Promise<void>
     vscode.commands.registerCommand("vscode-objectscript.compileAllWithFlags", () => namespaceCompile(true)),
     vscode.commands.registerCommand("vscode-objectscript.compileFolder", importFileOrFolder),
     vscode.commands.registerCommand("vscode-objectscript.export", exportAll),
+    vscode.commands.registerCommand("vscode-objectscript.debug", (program: string, askArgs: boolean) => {
+      const startDebugging = args => {
+        const programWithArgs = program + `(${args})`;
+        vscode.debug.startDebugging(undefined, {
+          type: "objectscript",
+          request: "launch",
+          name: `Debug ${program}`,
+          program: programWithArgs,
+        });
+      };
+      if (!askArgs) {
+        startDebugging("");
+        return;
+      }
+      return vscode.window
+        .showInputBox({
+          placeHolder: "Please enter comma delimited arguments list",
+        })
+        .then(args => {
+          startDebugging(args);
+        });
+    }),
     vscode.commands.registerCommand("vscode-objectscript.pickProcess", async config => {
       const system = config.system;
       const api = new AtelierAPI();
@@ -282,6 +305,10 @@ export async function activate(context: vscode.ExtensionContext): Promise<void>
     vscode.debug.registerDebugConfigurationProvider("objectscript", new ObjectScriptConfigurationProvider()),
     vscode.debug.registerDebugAdapterDescriptorFactory("objectscript", debugAdapterFactory),
     debugAdapterFactory,
+    vscode.languages.registerCodeLensProvider(
+      documentSelector("objectscript-class"),
+      new ObjectScriptClassCodeLensProvider()
+    ),
 
     /* from proposed api */
     vscode.workspace.registerFileSearchProvider(FILESYSTEM_SCHEMA, new FileSearchProvider()),

src/providers/ObjectScriptClassCodeLensProvider.ts

@@ -0,0 +1,72 @@
+import * as vscode from "vscode";
+
+export class ObjectScriptClassCodeLensProvider implements vscode.CodeLensProvider {
+  public provideCodeLenses(
+    document: vscode.TextDocument,
+    token: vscode.CancellationToken
+  ): vscode.ProviderResult<vscode.CodeLens[]> {
+    const result = new Array<vscode.CodeLens>();
+
+    const isClass = document.fileName.toLowerCase().endsWith(".cls");
+
+    if (isClass) {
+      result.push(...this.classMethods(document));
+    }
+    return result;
+  }
+
+  private classMethods(document: vscode.TextDocument): vscode.CodeLens[] {
+    const result = new Array<vscode.CodeLens>();
+
+    let inComment = false;
+    let className = "";
+    for (let i = 0; i < document.lineCount; i++) {
+      const line = document.lineAt(i);
+      const text = this.stripLineComments(line.text);
+
+      if (text.match(/\/\*/)) {
+        inComment = true;
+      }
+
+      if (inComment) {
+        if (text.match(/\*\//)) {
+          inComment = false;
+        }
+        continue;
+      }
+      if (!className.length) {
+        const classNameMatch = text.match(/(?<=^Class\s)[^ ]+/i);
+        if (classNameMatch) {
+          [className] = classNameMatch;
+        }
+      }
+      const methodMatch = text.match(/(?<=^ClassMethod\s)([^(]+)(\(.)/i);
+      if (methodMatch) {
+        const [, name, parens] = methodMatch;
+        const program = `##class(${className}).${name}`;
+        const askArgs = parens !== "()";
+        result.push(
+          new vscode.CodeLens(
+            new vscode.Range(
+              new vscode.Position(i, methodMatch.index),
+              new vscode.Position(i, methodMatch.index + name.length)
+            ),
+            {
+              title: `Debug this method`,
+              command: "vscode-objectscript.debug",
+              arguments: [program, askArgs],
+            }
+          )
+        );
+      }
+    }
+    return result;
+  }
+
+  private stripLineComments(text: string) {
+    text = text.replace(/\/\/.*$/, "");
+    text = text.replace(/#+;.*$/, "");
+    text = text.replace(/;.*$/, "");
+    return text;
+  }
+}