implementing intersystems-community#974

Author: daimor

package.json

@@ -1249,6 +1249,11 @@
           "default": true,
           "markdownDescription": "Show inline `Debug this method` CodeLens action for ClassMethods."
         },
+        "objectscript.debug.runThisMethod": {
+          "type": "boolean",
+          "default": true,
+          "markdownDescription": "Show inline `Run this method in terminal` CodeLens action for ClassMethods."
+        },
         "objectscript.studioActionDebugOutput": {
           "type": "boolean",
           "default": false,

src/extension.ts

@@ -58,7 +58,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 { ObjectScriptCodeLensProvider } from "./providers/ObjectScriptCodeLensProvider";
 import { XmlContentProvider } from "./providers/XmlContentProvider";
 
 import { AtelierAPI } from "./api";
@@ -705,9 +705,15 @@ export async function activate(context: vscode.ExtensionContext): Promise<any> {
       Promise.all(files.map((file) => importFileOrFolder(file, true)))
     ),
     vscode.commands.registerCommand("vscode-objectscript.export", exportAll),
+    vscode.commands.registerCommand("vscode-objectscript.runInTerminal", (command: string) => {
+      if (vscode.window.activeTerminal) {
+        vscode.window.activeTerminal.sendText(command, false);
+        vscode.window.activeTerminal.show();
+      }
+    }),
     vscode.commands.registerCommand("vscode-objectscript.debug", (program: string, askArgs: boolean) => {
       const startDebugging = (args) => {
-        const programWithArgs = program + `(${args})`;
+        const programWithArgs = program + (program.includes("##class") || args.length ? `(${args})` : "");
         vscode.debug.startDebugging(undefined, {
           type: "objectscript",
           request: "launch",
@@ -848,8 +854,8 @@ export async function activate(context: vscode.ExtensionContext): Promise<any> {
     vscode.debug.registerDebugAdapterDescriptorFactory("objectscript", debugAdapterFactory),
     debugAdapterFactory,
     vscode.languages.registerCodeLensProvider(
-      documentSelector("objectscript-class"),
-      new ObjectScriptClassCodeLensProvider()
+      documentSelector("objectscript-class", "objectscript"),
+      new ObjectScriptCodeLensProvider()
     ),
     vscode.commands.registerCommand("vscode-objectscript.compileOnly", () => compileOnly(false)),
     vscode.commands.registerCommand("vscode-objectscript.compileOnlyWithFlags", () => compileOnly(true)),
@@ -1013,6 +1019,9 @@ export async function activate(context: vscode.ExtensionContext): Promise<any> {
         }
       }
     }),
+    vscode.window.onDidOpenTerminal((t) => {
+      console.log("openTerminal", t);
+    }),
     vscode.window.onDidCloseTerminal((t) => {
       const terminalIndex = terminals.findIndex((terminal) => terminal.name == t.name);
       if (terminalIndex > -1) {

src/providers/ObjectScriptClassCodeLensProvider.ts

@@ -0,0 +1,122 @@
+import * as vscode from "vscode";
+import { config } from "../extension";
+import { currentFile } from "../utils";
+
+export class ObjectScriptCodeLensProvider implements vscode.CodeLensProvider {
+  public provideCodeLenses(
+    document: vscode.TextDocument,
+    token: vscode.CancellationToken
+  ): vscode.ProviderResult<vscode.CodeLens[]> {
+    const result = new Array<vscode.CodeLens>();
+
+    if (document.fileName.toLowerCase().endsWith(".cls")) {
+      result.push(...this.classMethods(document));
+    }
+    if (document.fileName.toLowerCase().endsWith(".mac")) {
+      result.push(...this.routineLabels(document));
+    }
+    return result;
+  }
+
+  private classMethods(document: vscode.TextDocument): vscode.CodeLens[] {
+    const file = currentFile(document);
+    const result = new Array<vscode.CodeLens>();
+
+    if (!file.name.match(/\.cls$/i)) {
+      return result;
+    }
+    const className = file.name.split(".").slice(0, -1).join(".");
+
+    const { debugThisMethod, runThisMethod } = config("debug");
+    let inComment = false;
+    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;
+      }
+
+      const methodMatch = text.match(/(?<=^ClassMethod\s)([^(]+)(\(.)/i);
+      if (methodMatch) {
+        const [, name, parens] = methodMatch;
+
+        debugThisMethod && result.push(this.addDebugThisMethod(i, [`##class(${className}).${name}`, parens !== "()"]));
+        runThisMethod && result.push(this.addRunThisMethod(i, [`Do ##class(${className}).${name}()`]));
+      }
+    }
+    return result;
+  }
+
+  private routineLabels(document: vscode.TextDocument): vscode.CodeLens[] {
+    const file = currentFile(document);
+    const result = new Array<vscode.CodeLens>();
+
+    if (!file.name.match(/\.mac$/i)) {
+      return result;
+    }
+    const routineName = file.name.split(".").slice(0, -1).join(".");
+
+    const { debugThisMethod, runThisMethod } = config("debug");
+
+    debugThisMethod && result.push(this.addDebugThisMethod(0, [`^${routineName}`, false]));
+    runThisMethod && result.push(this.addRunThisMethod(0, [`Do ^${routineName}`]));
+
+    let inComment = false;
+    for (let i = 1; 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;
+      }
+
+      const labelMatch = text.match(/^(\w[^(\n\s]+)(?:\(([^)]*)\))?/i);
+      if (labelMatch) {
+        const [, name, parens] = labelMatch;
+
+        debugThisMethod && result.push(this.addDebugThisMethod(i, [`${name}^${routineName}`, parens !== "()"]));
+        runThisMethod && result.push(this.addRunThisMethod(i, [`Do ${name}^${routineName}`]));
+      }
+    }
+
+    return result;
+  }
+
+  private addDebugThisMethod(line: number, args: any[]) {
+    return new vscode.CodeLens(new vscode.Range(line, 0, line, 80), {
+      title: `Debug this method`,
+      command: "vscode-objectscript.debug",
+      arguments: args,
+    });
+  }
+
+  private addRunThisMethod(line: number, args: any[]) {
+    return new vscode.CodeLens(new vscode.Range(line, 0, line, 80), {
+      title: `Run this method in terminal`,
+      command: "vscode-objectscript.runInTerminal",
+      arguments: args,
+    });
+  }
+
+  private stripLineComments(text: string) {
+    text = text.replace(/\/\/.*$/, "");
+    text = text.replace(/#+;.*$/, "");
+    text = text.replace(/;.*$/, "");
+    return text;
+  }
+}