Combine mutually exclusive actions

Author: jablko

src/compute-pr-actions.ts

@@ -52,8 +52,7 @@ export interface Actions {
     projectColumn?: ColumnName | "REMOVE";
     labels: LabelName[];
     responseComments: Comments.Comment[];
-    shouldClose: boolean;
-    shouldMerge: boolean;
+    state?: "close" | "merge";
     shouldUpdateLabels: boolean;
 }
 
@@ -62,8 +61,6 @@ function createDefaultActions(): Actions {
         projectColumn: "Other",
         labels: [],
         responseComments: [],
-        shouldClose: false,
-        shouldMerge: false,
         shouldUpdateLabels: true,
     };
 }
@@ -72,8 +69,6 @@ function createEmptyActions(): Actions {
     return {
         labels: [],
         responseComments: [],
-        shouldClose: false,
-        shouldMerge: false,
         shouldUpdateLabels: false,
     };
 }
@@ -320,7 +315,7 @@ export function process(prInfo: BotResult,
                                          (info.tooManyOwners || info.hasMultiplePackages) ? [] : info.otherOwners,
                                          headCommitAbbrOid));
             if (info.hasValidMergeRequest) {
-                actions.shouldMerge = true;
+                actions.state = "merge";
                 actions.projectColumn = "Recently Merged";
             } else {
                 actions.projectColumn = "Waiting for Author to Merge";
@@ -334,7 +329,7 @@ export function process(prInfo: BotResult,
         }
     }
 
-    if (!actions.shouldMerge) {
+    if (!actions.state) {
         if (info.mergeRequestUser) {
             post(Comments.WaitUntilMergeIsOK(info.mergeRequestUser, headCommitAbbrOid, urls.workflow));
         }
@@ -372,7 +367,7 @@ function makeStaleness(now: Date, author: string, otherOwners: string[]) { // cu
             }
             if (state === "done") {
                 if (doneColumn === "CLOSE") {
-                    actions.shouldClose = true;
+                    actions.state = "close";
                     actions.projectColumn = "REMOVE";
                 } else {
                     actions.projectColumn = doneColumn;

src/execute-pr-actions.ts

@@ -101,20 +101,20 @@ function getMutationsForCommentRemovals(actions: Actions, botComments: ParsedCom
     });
 }
 
-function getMutationsForChangingPRState(actions: Actions, pr: PR_repository_pullRequest) {
-    return [
-        actions.shouldMerge
-            ? createMutation<schema.MergePullRequestInput>("mergePullRequest", {
+function* getMutationsForChangingPRState(actions: Actions, pr: PR_repository_pullRequest) {
+    switch (actions.state) {
+        case "close":
+            yield createMutation<schema.ClosePullRequestInput>("closePullRequest", { pullRequestId: pr.id });
+            break;
+        case "merge":
+            yield createMutation<schema.MergePullRequestInput>("mergePullRequest", {
                 commitHeadline: `🤖 Merge PR #${pr.number} ${pr.title} by @${pr.author?.login ?? "(ghost)"}`,
                 expectedHeadOid: pr.headRefOid,
                 mergeMethod: "SQUASH",
                 pullRequestId: pr.id,
-            })
-            : null,
-        actions.shouldClose
-            ? createMutation<schema.ClosePullRequestInput>("closePullRequest", { pullRequestId: pr.id })
-            : null
-    ];
+            });
+            break;
+    }
 }
 
 async function getProjectBoardColumnIdByName(name: string): Promise<string> {