Added studio actions

Tyler Deemer · Tyler Deemer · commit 7429c7423377 · 2020-06-10T10:05:43.000-04:00
diff --git a/package.json b/package.json
@@ -156,6 +156,10 @@
           "command": "vscode-objectscript.explorer.compile",
           "when": "view == ObjectScriptExplorer && viewItem =~ /^dataNode:/"
         },
+        {
+          "command": "vscode-objectscript.studio.contextActions",
+          "when": "view == ObjectScriptExplorer && viewItem =~ /^dataNode:/"
+        },
         {
           "command": "vscode-objectscript.explorer.otherNamespace",
           "when": "view == ObjectScriptExplorer && viewItem =~ /^serverNode((?!:extra:).)*$/",
@@ -410,6 +414,11 @@
         "category": "ObjectScript",
         "command": "vscode-objectscript.studio.actions",
         "title": "Studio actions"
+      },
+      {
+        "category": "ObjectScript",
+        "command": "vscode-objectscript.studio.contextActions",
+        "title": "Studio context actions"
       }
     ],
     "keybindings": [
diff --git a/src/commands/studio.ts b/src/commands/studio.ts
@@ -2,6 +2,11 @@ import * as vscode from "vscode";
 import { AtelierAPI } from "../api";
 import { config, FILESYSTEM_SCHEMA } from "../extension";
 import { outputChannel } from "../utils";
+import { DocumentContentProvider } from "../providers/DocumentContentProvider";
+import { ClassNode } from "../explorer/models/classesNode";
+import { PackageNode } from "../explorer/models/packageNode";
+import { RoutineNode } from "../explorer/models/routineNode";
+import { NodeBase } from "../explorer/models/nodeBase";
 
 interface StudioAction extends vscode.QuickPickItem {
   name: string;
@@ -13,10 +18,19 @@ class StudioActions {
   private api: AtelierAPI;
   private name: string;
 
-  public constructor(uri: vscode.Uri) {
-    this.uri = uri;
-    this.name = this.uri.path.slice(1).replace(/\//g, ".");
-    this.api = new AtelierAPI(uri.authority);
+  public constructor(uriOrNode: vscode.Uri | PackageNode | ClassNode | RoutineNode) {
+    if(uriOrNode instanceof vscode.Uri) {
+      const uri: vscode.Uri = uriOrNode;
+      this.uri = uri;
+      this.name = this.uri.path.slice(1).replace(/\//g, ".");
+      this.api = new AtelierAPI(uri.authority);
+    } else {
+      const node: NodeBase = uriOrNode;
+      this.api = new AtelierAPI();
+      this.name = (node instanceof PackageNode)
+        ? node.fullName + ".PKG"
+        : node.fullName;
+    }
   }
 
   public processUserAction(userAction): Thenable<any> {
@@ -36,79 +50,133 @@ class StudioActions {
           .showWarningMessage(target, { modal: true }, "Yes", "No")
           .then(answer => (answer === "Yes" ? "1" : answer === "No" ? "0" : "2"));
       case 2: // Run a CSP page/Template. The Target is the full url to the CSP page/Template
-        // Open the target URL in a webview
-        const conn = config().conn;
-        const column = vscode.window.activeTextEditor
-          ? vscode.window.activeTextEditor.viewColumn
-          : undefined;
-        const panel = vscode.window.createWebviewPanel(
-          'studioactionwebview',
-          'CSP Page',
-          column || vscode.ViewColumn.One,
-          {
-            enableScripts: true,
-          }
-        );
-        panel.webview.html = `
-          <!DOCTYPE html>
-          <html lang="en">
-          <head>
-            <style type="text/css">
-              body, html
-              {
-                margin: 0; padding: 0; height: 100%; overflow: hidden;
-              }
-              #content
-              {
-                position:absolute; left: 0; right: 0; bottom: 0; top: 0px;
-              }
-            </style>
-          </head>
-          <body>
-            <div id="content">
-              <iframe src="http://${conn.host}:${conn.port}${target}" onLoad="checkForCancelState()" width="100%" height="100%" frameborder="0"></iframe>
-            </div>
-            <script>
-              function checkForCancelState() {
-                var x = document.getElementsByTagName("BODY")[0];
-                console.log(x);
-              }
-            </script>
-          </body>
-          </html>
-          `;
-        panel.onDidDispose(
-          () => {
-            // fire a cancel answer if the user closes the webview
-            return "2";
-          }
-        );
-        // TODO: use panel.dispose() when the cancel text is sent back in the iframe
-        break;
-        // throw new Error("Not suppoorted");
+        return new Promise((resolve) => {
+          let answer = "2";
+          const conn = config().conn;
+          const column = vscode.window.activeTextEditor
+            ? vscode.window.activeTextEditor.viewColumn
+            : undefined;
+          const panel = vscode.window.createWebviewPanel(
+            "studioactionwebview",
+            "Studio Extension Page",
+            column || vscode.ViewColumn.One,
+            {
+              enableScripts: true,
+            }
+          );
+          panel.webview.onDidReceiveMessage(message => {
+            if(message.result && message.result === "done") {
+              answer = "1";
+              panel.dispose();
+            }
+          });
+          panel.onDidDispose(() => resolve(answer));
+
+          const url = new URL(`http://${conn.host}:${conn.port}${target}`);
+          const api = new AtelierAPI();
+          api.actionQuery("select %Atelier_v1_Utils.General_GetCSPToken(?) token", [url.toString()]).then(tokenObj => {
+            const csptoken = tokenObj.result.content[0].token;
+            url.searchParams.set('CSPCHD', csptoken);
+            url.searchParams.set('Namespace', conn.ns);
+            panel.webview.html = `
+              <!DOCTYPE html>
+              <html lang="en">
+              <head>
+                <style type="text/css">
+                  body, html {
+                    margin: 0; padding: 0; height: 100%; overflow: hidden;
+                    background-color: white;
+                  }
+                  #content {
+                    position:absolute; left: 0; right: 0; bottom: 0; top: 0px;
+                  }
+                </style>
+              </head>
+              <body>
+                <div id="content">
+                  <iframe src="${url.toString()}" width="100%" height="100%" frameborder="0"></iframe>
+                </div>
+                <script>
+                  const vscode = acquireVsCodeApi();
+                  window.addEventListener("message", receiveMessage, false);
+                  function receiveMessage(event) {
+                    vscode.postMessage(event.data);
+                  }
+                </script>
+              </body>
+              </html>
+              `;
+          });
+        });
       case 3: // Run an EXE on the client.
         throw new Error("Not suppoorted");
       case 4: // Insert the text in Target in the current document at the current selection point
-        throw new Error("Not suppoorted");
+        const editor = vscode.window.activeTextEditor;
+        if(editor) {
+          editor.edit(editBuilder => {
+            editBuilder.replace(editor.selection, target);
+          });
+        }
+        return;
       case 5: // Studio will open the documents listed in Target
-        throw new Error("Not suppoorted");
+        target.split(",").forEach(element => {
+          let classname = element;
+          let method: string;
+          let offset = 0;
+          if(element.includes(":")) {
+            [classname, method] = element.split(":");
+            if(method.includes("+")) {
+              offset = +method.split("+")[1];
+              method = method.split("+")[0];
+            }
+          }
+
+          const splitClassname = classname.split(".");
+          const filetype = splitClassname[splitClassname.length - 1];
+          const isCorrectMethod = (text: string) => (filetype === "cls")
+            ? text.match("Method " + method)
+            : text.startsWith(method)
+
+          const uri = DocumentContentProvider.getUri(classname);
+          vscode.window.showTextDocument(uri, {"preview": false}).then(newEditor => {
+            if(method) {
+              const document = newEditor.document;
+              for(let i = 0; i < document.lineCount; i++) {
+                const line = document.lineAt(i);
+                if(isCorrectMethod(line.text)) {
+                  if(!line.text.endsWith("{")) offset++;
+                  const cursor = newEditor.selection.active;
+                  const newPosition = cursor.with(i + offset, 0);
+                  newEditor.selection = new vscode.Selection(newPosition, newPosition);
+                  break;
+                }
+              }
+            }
+          });
+        });
+        return;
       case 6: // Display an alert dialog in Studio with the text from the Target variable.
         return vscode.window.showWarningMessage(target, { modal: true });
       case 7: // Display a dialog with a textbox and Yes/No/Cancel buttons.
         return vscode.window.showInputBox({
           prompt: target,
+        }).then(msg => {
+          return {
+            "msg": (msg ? msg : ""),
+            "answer": (msg ? 1 : 2)
+          }
         });
       default:
         throw new Error("Not suppoorted");
     }
   }
 
-  private userAction(action, afterUserAction = false, answer = "", msg = ""): Thenable<void> {
+  private userAction(action, afterUserAction = false, answer = "", msg = "", type = 0): Thenable<void> {
     if (!action) {
       return;
     }
-    const func = afterUserAction ? "AfterUserAction" : "UserAction";
-    const query = `select * from %Atelier_v1_Utils.Extension_${func}(?, ?, ?, ?)`;
+    const func = afterUserAction ? "AfterUserAction(?, ?, ?, ?, ?)" : "UserAction(?, ?, ?, ?)";
+    const query = `select * from %Atelier_v1_Utils.Extension_${func}`;
     let selectedText = "";
     const editor = vscode.window.activeTextEditor;
     if (!editor) {
@@ -118,8 +186,9 @@ class StudioActions {
     selectedText = editor.document.getText(selection);
 
     const parameters = afterUserAction
-      ? ["0", action.id, this.name, answer]
-      : ["0", action.id, this.name, selectedText];
+      ? [type.toString(), action.id, this.name, answer, msg]
+      : [type.toString(), action.id, this.name, selectedText];
+
     return vscode.window.withProgress(
       {
         cancellable: false,
@@ -129,34 +198,44 @@ class StudioActions {
       () =>
         this.api
           .actionQuery(query, parameters)
-          .then(data => data.result.content.pop())
+          .then(data => {
+            const actionInfo = data.result.content.pop();
+            actionInfo.save = action.save;
+            return actionInfo;
+          })
+          .then(this.processSaveFlag)
           .then(this.processUserAction)
           .then(answer => {
             if (answer) {
-              return this.userAction(action, true, answer);
+              return (answer.msg || answer.msg === "")
+                ? this.userAction(action, true, answer.answer, answer.msg, type)
+                : this.userAction(action, true, answer, "", type);
             }
           })
           .catch(err => {
+            console.log(err);
             outputChannel.appendLine(`Studio Action "${action.label}" not supported`);
             outputChannel.show();
           })
     );
   }
 
-  private constructMenu(menu): any[] {
+  private constructMenu(menu, contextOnly = false): any[] {
     return menu
+      .filter(menuGroup => !(contextOnly && menuGroup.type === "main"))
       .reduce(
         (list, sub) =>
           list.concat(
             sub.items
               .filter(el => el.id !== "" && el.separator == 0)
-              // .filter(el => el.enabled == 1)
+              .filter(el => el.enabled == 1)
               .map(el => ({
                 ...el,
                 id: `${sub.id},${el.id}`,
                 label: el.name.replace("&", ""),
                 itemId: el.id,
                 type: sub.type,
+                description: sub.name.replace("&", ""),
               }))
           ),
         []
@@ -170,19 +249,57 @@ class StudioActions {
       });
   }
 
-  public getMenu(menuType: string): Thenable<any> {
+  public getMenu(menuType: string, contextOnly = false): Thenable<any> {
+    let selectedText = "";
+    const editor = vscode.window.activeTextEditor;
+    if(this.uri && editor) {
+      const selection = editor.selection;
+      selectedText = editor.document.getText(selection);
+    }
+
     const query = "select * from %Atelier_v1_Utils.Extension_GetMenus(?,?,?)";
-    const parameters = [menuType, this.name, ""];
+    const parameters = [menuType, this.name, selectedText];
 
     return this.api
       .actionQuery(query, parameters)
       .then(data => data.result.content)
-      .then(this.constructMenu)
+      .then(menu => this.constructMenu(menu, contextOnly))
       .then(menuItems => {
         return vscode.window.showQuickPick<StudioAction>(menuItems, { canPickMany: false });
       })
       .then(action => this.userAction(action));
   }
+
+  public attemptedEdit() {
+    const query = "select * from %Atelier_v1_Utils.Extension_GetStatus(?)";
+    this.api.actionQuery(query, [this.name]).then(statusObj => {
+      const docStatus = statusObj.result.content.pop();
+      // if(!docStatus.editable && docStatus.inSourceControl && !docStatus.isCheckedOut) {
+      if(!docStatus.editable) {
+        const attemptedEditAction = {
+          id: "0",
+          label: "Attempted Edit"
+        };
+        vscode.commands.executeCommand('undo');
+        this.userAction(attemptedEditAction, false, "", "", 1);
+      } // else if(!docStatus.editable && docStatus.in)
+    });
+  }
+
+  private async processSaveFlag(userAction) {
+    if(userAction.save) {
+      const bitString = userAction.save.toString().padStart(3, "0");
+      // Save the current document
+      if(bitString.charAt(0) === "1") {
+        await vscode.window.activeTextEditor.document.save();
+      }
+      // Save all documents
+      if(bitString.charAt(2) === "1") {
+        await vscode.workspace.saveAll();
+      }
+    }
+    return userAction;
+  }
 }
 
 // export function contextMenu(uri: vscode.Uri): Promise<void> {
@@ -197,3 +314,20 @@ export async function mainMenu(uri: vscode.Uri) {
   const studioActions = new StudioActions(uri);
   return studioActions && studioActions.getMenu("");
 }
+
+export async function fireAttemptedEdit(uri: vscode.Uri) {
+  if(!uri || uri.scheme !== FILESYSTEM_SCHEMA) {
+    return;
+  }
+  const studioActions = new StudioActions(uri);
+  studioActions.attemptedEdit();
+}
+
+export async function contextMenu(node: PackageNode | ClassNode | RoutineNode): Promise<any> {
+  const nodeOrUri = node || vscode.window.activeTextEditor.document.uri;
+  if(!nodeOrUri || (nodeOrUri instanceof vscode.Uri && nodeOrUri.scheme !== FILESYSTEM_SCHEMA)) {
+    return;
+  }
+  const studioActions = new StudioActions(nodeOrUri);
+  return studioActions && studioActions.getMenu("", true);
+}
diff --git a/src/extension.ts b/src/extension.ts
