feat: Add protect & unprotect workunit to WU context menu

Signed-off-by: David de Hilster <[email protected]>

Author: dehilsterlexis

package.json

@@ -402,64 +402,16 @@
         "enablement": "viewItem =~ /ECLWUNodeComplete/"
       },
       {
-        "command": "hpccPlatform.setStateCompiled",
+        "command": "hpccPlatform.protectWU",
         "category": "ECL",
-        "title": "%Compiled%"
-      },
-      {
-        "command": "hpccPlatform.setStateRunning",
-        "category": "ECL",
-        "title": "%Running%"
-      },
-      {
-        "command": "hpccPlatform.setStateCompleted",
-        "category": "ECL",
-        "title": "%Completed%"
-      },
-      {
-        "command": "hpccPlatform.setStateFailed",
-        "category": "ECL",
-        "title": "%Failed%"
-      },
-      {
-        "command": "hpccPlatform.setStateArchived",
-        "category": "ECL",
-        "title": "%Archived%"
-      },
-      {
-        "command": "hpccPlatform.setStateAborting",
-        "category": "ECL",
-        "title": "%Aborting%"
-      },
-      {
-        "command": "hpccPlatform.setStateAborted",
-        "category": "ECL",
-        "title": "%Aborted%"
-      },
-      {
-        "command": "hpccPlatform.setStateBlocked",
-        "category": "ECL",
-        "title": "%Blocked%"
-      },
-      {
-        "command": "hpccPlatform.setStateSubmitted",
-        "category": "ECL",
-        "title": "%Submitted%"
-      },
-      {
-        "command": "hpccPlatform.setStateScheduled",
-        "category": "ECL",
-        "title": "%Scheduled%"
-      },
-      {
-        "command": "hpccPlatform.setStateCompiling",
-        "category": "ECL",
-        "title": "%Compiling%"
+        "title": "%Protect Workunit%",
+        "enablement": "viewItem =~ /ECLWUNodeComplete/"
       },
       {
-        "command": "hpccPlatform.setStateWaiting",
+        "command": "hpccPlatform.unprotectWU",
         "category": "ECL",
-        "title": "%Waiting%"
+        "title": "%Unprotect Workunit%",
+        "enablement": "viewItem =~ /ECLWUNodeComplete/"
       },
       {
         "command": "hpccPlatform.refresh",
@@ -712,63 +664,7 @@
         }
       }
     ],
-    "submenus": [
-      {
-        "id": "setState",
-        "label": "Set State"
-      }
-    ],
     "menus": {
-      "setState": [
-        {
-            "command": "hpccPlatform.setStateCompiled",
-            "group": "set@01"
-        },
-        {
-            "command": "hpccPlatform.setStateRunning",
-            "group": "set@02"
-        },
-        {
-            "command": "hpccPlatform.setStateCompleted",
-            "group": "set@03"
-        },
-        {
-            "command": "hpccPlatform.setStateFailed",
-            "group": "set@04"
-        },
-        {
-            "command": "hpccPlatform.setStateArchived",
-            "group": "set@05"
-        },
-        {
-            "command": "hpccPlatform.setStateAborting",
-            "group": "set@06"
-        },
-        {
-            "command": "hpccPlatform.setStateAborted",
-            "group": "set@07"
-        },
-        {
-            "command": "hpccPlatform.setStateBlocked",
-            "group": "set@08"
-        },
-        {
-            "command": "hpccPlatform.setStateSubmitted",
-            "group": "set@09"
-        },
-        {
-            "command": "hpccPlatform.setStateScheduled",
-            "group": "set@10"
-        },
-        {
-            "command": "hpccPlatform.setStateCompiling",
-            "group": "set@11"
-        },
-        {
-            "command": "hpccPlatform.setStateWaiting",
-            "group": "set@12"
-        }
-      ],
       "explorer/context": [
         {
           "when": "resourceLangId == ecl && resourceExtname == .ecl",
@@ -971,8 +867,14 @@
           "group": "3action@920"
         },
         {
-          "submenu": "setState",
-          "group": "3action@950"
+          "when": "view == hpccPlatform && viewItem =~ /ECLWUNodeUnprotected/",
+          "command": "hpccPlatform.protectWU",
+          "group": "3action@930"
+        },
+        {
+          "when": "view == hpccPlatform && viewItem =~ /ECLWUNodeProtected/",
+          "command": "hpccPlatform.unprotectWU",
+          "group": "3action@940"
         },
         {
           "command": "hpccResources.bundles.homepage",

package.nls.json

@@ -79,6 +79,8 @@
     "Ping interval (secs, -1 to disable)": "Ping interval (secs, -1 to disable)",
     "Pinned launch configurations": "Pinned launch configurations",
     "Private": "Private",
+    "Protect": "Protect",
+    "Protect Workunit": "Protect Workunit",
     "Public": "Public",
     "Refresh": "Refresh",
     "Refresh Tree": "Refresh Tree",
@@ -123,6 +125,7 @@
     "Terminal": "Terminal",
     "Uninstall": "Uninstall",
     "Uninstall Bundle": "Uninstall Bundle",
+    "Unprotect Workunit": "Unprotect Workunit",
     "User ID": "User ID",
     "User password": "User password",
     "Verify ECL Digital Signature": "Verify ECL Digital Signature",

src/ecl/eclWatchTree.ts

@@ -93,52 +93,12 @@ export class ECLWatchTree extends Tree {
             wuNode.delete();
         });
 
-        vscode.commands.registerCommand("hpccPlatform.setStateCompiled", (wuNode: ECLWUNode) => {
-            wuNode.setStateCompiled();
+        vscode.commands.registerCommand("hpccPlatform.protectWU", (wuNode: ECLWUNode) => {
+            wuNode.protect();
         });
 
-        vscode.commands.registerCommand("hpccPlatform.setStateRunning", (wuNode: ECLWUNode) => {
-            wuNode.setStateRunning();
-        });
-        
-        vscode.commands.registerCommand("hpccPlatform.setStateCompleted", (wuNode: ECLWUNode) => {
-            wuNode.setStateCompleted();
-        });
-        
-        vscode.commands.registerCommand("hpccPlatform.setStateFailed", (wuNode: ECLWUNode) => {
-            wuNode.setStateFailed();
-        });
-        
-        vscode.commands.registerCommand("hpccPlatform.setStateArchived", (wuNode: ECLWUNode) => {
-            wuNode.setStateArchived();
-        });
-        
-        vscode.commands.registerCommand("hpccPlatform.setStateAborting", (wuNode: ECLWUNode) => {
-            wuNode.setStateAborting();
-        });
-        
-        vscode.commands.registerCommand("hpccPlatform.setStateAborted", (wuNode: ECLWUNode) => {
-            wuNode.setStateAborted();
-        });
-        
-        vscode.commands.registerCommand("hpccPlatform.setStateBlocked", (wuNode: ECLWUNode) => {
-            wuNode.setStateBlocked();
-        });
-        
-        vscode.commands.registerCommand("hpccPlatform.setStateSubmitted", (wuNode: ECLWUNode) => {
-            wuNode.setStateSubmitted();
-        });
-        
-        vscode.commands.registerCommand("hpccPlatform.setStateScheduled", (wuNode: ECLWUNode) => {
-            wuNode.setStateScheduled();
-        });
-        
-        vscode.commands.registerCommand("hpccPlatform.setStateCompiling", (wuNode: ECLWUNode) => {
-            wuNode.setStateCompiling();
-        });
-        
-        vscode.commands.registerCommand("hpccPlatform.setStateWait", (wuNode: ECLWUNode) => {
-            wuNode.setStateWait();
+        vscode.commands.registerCommand("hpccPlatform.unprotectWU", (wuNode: ECLWUNode) => {
+            wuNode.unprotect();
         });
 
     }
@@ -448,60 +408,12 @@ export class ECLWUNode extends Item<ECLWatchTree> {
         this._wu.abort().then(() => this._tree.refresh(this));
     }
 
-    setState(stateID: WUStateID) {
-        const service = new WorkunitsService({ baseUrl: this._wu.BaseUrl });
-        return service.WUUpdate({
-            Wuid: this._wu.Wuid,
-            State: stateID as unknown as string
-        }).then(() => this._tree.refresh());
-    }
-
-    setStateCompiled() {
-        this.setState(WUStateID.Compiled);
-    }
-
-    setStateRunning() {
-        this.setState(WUStateID.Running);
-    }
-
-    setStateCompleted() {
-        this.setState(WUStateID.Completed);
-    }
-
-    setStateFailed() {
-        this.setState(WUStateID.Failed);
-    }
-
-    setStateArchived() {
-        this.setState(WUStateID.Archived);
-    }
-
-    setStateAborting() {
-        this.setState(WUStateID.Aborting);
-    }
-
-    setStateAborted() {
-        this.setState(WUStateID.Aborted);
-    }
-
-    setStateBlocked() {
-        this.setState(WUStateID.Blocked);
-    }
-
-    setStateSubmitted() {
-        this.setState(WUStateID.Submitted);
-    }
-
-    setStateScheduled() {
-        this.setState(WUStateID.Scheduled);
-    }
-
-    setStateCompiling() {
-        this.setState(WUStateID.Compiling);
+    protect() {
+        this._wu.protect().then(() => this._tree.refresh());
     }
 
-    setStateWait() {
-        this.setState(WUStateID.Wait);
+    unprotect() {
+        this._wu.unprotect().then(() => this._tree.refresh());
     }
 
     delete() {
@@ -525,7 +437,8 @@ export class ECLWUNode extends Item<ECLWatchTree> {
     }
 
     contextValue(): string {
-        return this._wu.isComplete() ? "ECLWUNodeComplete" : "ECLWUNode";
+        const prot = this._wu.Protected ? ",ECLWUNodeProtected" : ",ECLWUNodeUnprotected";
+        return this._wu.isComplete() ? `ECLWUNodeComplete${prot}` : `ECLWUNode${prot}`;
     }
 }