handle converting pr to/from draft

actualbudget · Jul 7, 2024 · 7b41e96 · 7b41e96
1 parent 56e0a28
commit 7b41e96
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 2 deletions.
diff --git a/src/handlers/checkSuite.ts b/src/handlers/checkSuite.ts
@@ -9,7 +9,7 @@ export default (app: Probot) => {
 
     const pr = await PullRequest.getFromNumber(context, prNum);
 
-    if (pr.wip) return;
+    if (pr.wip || pr.data.draft) return;
 
     const suites = (
       await context.octokit.checks.listSuitesForRef({

diff --git a/src/handlers/pullRequest.ts b/src/handlers/pullRequest.ts
@@ -7,7 +7,7 @@ export default (app: Probot) => {
 
     let title = pr.data.title;
 
-    if (!pr.wip) {
+    if (!pr.wip && !pr.data.draft) {
       title = `[WIP] ${title}`;
       await pr.setTitle(title);
     }
@@ -36,4 +36,28 @@ export default (app: Probot) => {
       await pr.clearLabels();
     }
   });
+
+  app.on(['pull_request.converted_to_draft'], async context => {
+    const pr = new PullRequest(context);
+
+    let title = pr.data.title;
+    if (pr.wip) {
+      title = title.substring(6);
+      await pr.setTitle(title);
+    }
+
+    await pr.addLabel('wip');
+  });
+
+  app.on(['pull_request.ready_for_review'], async context => {
+    const pr = new PullRequest(context);
+
+    let title = pr.data.title;
+    if (pr.wip) {
+      title = title.substring(6);
+      await pr.setTitle(title);
+    }
+
+    await pr.addLabel('readyForReview');
+  });
 };