Move run context value logic to run model

joshmgross · joshmgross · commit 28022339a0e7 · 2023-03-10T12:15:46.000-05:00
diff --git a/src/store/workflowRun.ts b/src/store/workflowRun.ts
@@ -1,4 +1,5 @@
 import {GitHubRepoContext} from "../git/repository";
+import {RepositoryPermission, hasWritePermission} from "../git/repository-permissions";
 import {log, logDebug} from "../log";
 import * as model from "../model";
 import {WorkflowJob} from "./WorkflowJob";
@@ -50,6 +51,18 @@ abstract class WorkflowRunBase {
     return this._jobs;
   }
 
+  contextValue(permission: RepositoryPermission): string {
+    const contextValues = ["run"];
+    const completed = this._run.status === "completed";
+    if (hasWritePermission(permission)) {
+      contextValues.push(completed ? "rerunnable" : "cancelable");
+    }
+    if (completed) {
+      contextValues.push("completed");
+    }
+    return contextValues.join(" ");
+  }
+
   protected abstract fetchJobs(): Promise<WorkflowJob[]>;
 }
 
diff --git a/src/treeViews/shared/workflowRunNode.ts b/src/treeViews/shared/workflowRunNode.ts
@@ -28,15 +28,7 @@ export class WorkflowRunNode extends vscode.TreeItem {
     this.run = run;
     this.label = WorkflowRunNode._getLabel(run, this.workflowName);
 
-    const contextValues = ["run"];
-    const completed = this.run.run.status === "completed";
-    if (hasWritePermission(this.gitHubRepoContext.permissionLevel)) {
-      contextValues.push(completed ? "rerunnable" : "cancelable");
-    }
-    if (completed) {
-      contextValues.push("completed");
-    }
-    this.contextValue = contextValues.join(" ");
+    this.contextValue = this.run.contextValue(this.gitHubRepoContext.permissionLevel);
 
     this.iconPath = getIconForWorkflowRun(this.run.run);
     this.tooltip = this.getTooltip();
