feat: Add protect & unprotect workunit to WU context menu

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

Author: dehilsterlexis

package.json

@@ -401,6 +401,18 @@
         "title": "%Delete Workunit%",
         "enablement": "viewItem =~ /ECLWUNodeComplete/"
       },
+      {
+        "command": "hpccPlatform.protectWU",
+        "category": "ECL",
+        "title": "%Protect Workunit%",
+        "enablement": "viewItem =~ /ECLWUNodeComplete/"
+      },
+      {
+        "command": "hpccPlatform.unprotectWU",
+        "category": "ECL",
+        "title": "%Unprotect Workunit%",
+        "enablement": "viewItem =~ /ECLWUNodeComplete/"
+      },
       {
         "command": "hpccPlatform.setStateCompiled",
         "category": "ECL",
@@ -970,6 +982,16 @@
           "when": "view == hpccPlatform && viewItem =~ /^ECLWUNode/",
           "group": "3action@920"
         },
+        {
+          "when": "view == hpccPlatform && viewItem =~ /ECLWUNodeUnprotected/",
+          "command": "hpccPlatform.protectWU",
+          "group": "3action@930"
+        },
+        {
+          "when": "view == hpccPlatform && viewItem =~ /ECLWUNodeProtected/",
+          "command": "hpccPlatform.unprotectWU",
+          "group": "3action@940"
+        },
         {
           "submenu": "setState",
           "group": "3action@950"

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,14 +93,22 @@ export class ECLWatchTree extends Tree {
             wuNode.delete();
         });
 
+        vscode.commands.registerCommand("hpccPlatform.protectWU", (wuNode: ECLWUNode) => {
+            wuNode.protect();
+        });
+
+        vscode.commands.registerCommand("hpccPlatform.unprotectWU", (wuNode: ECLWUNode) => {
+            wuNode.unprotect();
+        });
+        
         vscode.commands.registerCommand("hpccPlatform.setStateCompiled", (wuNode: ECLWUNode) => {
             wuNode.setStateCompiled();
         });
 
         vscode.commands.registerCommand("hpccPlatform.setStateRunning", (wuNode: ECLWUNode) => {
             wuNode.setStateRunning();
         });
-        
+
         vscode.commands.registerCommand("hpccPlatform.setStateCompleted", (wuNode: ECLWUNode) => {
             wuNode.setStateCompleted();
         });
@@ -448,6 +456,14 @@ export class ECLWUNode extends Item<ECLWatchTree> {
         this._wu.abort().then(() => this._tree.refresh(this));
     }
 
+    protect() {
+        this._wu.protect().then(() => this._tree.refresh());
+    }
+
+    unprotect() {
+        this._wu.unprotect().then(() => this._tree.refresh());
+    }
+
     setState(stateID: WUStateID) {
         const service = new WorkunitsService({ baseUrl: this._wu.BaseUrl });
         return service.WUUpdate({
@@ -525,7 +541,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}`;
     }
 }