handle converting pr to/from draft (#13)

* handle converting pr to/from draft

* extend WIP regex to other common cases

Author: matt-fidd

src/handlers/checkSuite.ts

@@ -12,7 +12,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({

src/handlers/pullRequest.ts

@@ -1,4 +1,6 @@
 import { Probot } from 'probot';
+
+import { labels } from '../labels.js';
 import PullRequest from '../classes/PullRequest.js';
 
 export default (app: Probot) => {
@@ -7,7 +9,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 +38,18 @@ export default (app: Probot) => {
       await pr.clearLabels();
     }
   });
+
+  app.on(['pull_request.converted_to_draft'], async context => {
+    const pr = new PullRequest(context);
+    await pr.addLabel('wip');
+  });
+
+  app.on(['pull_request.ready_for_review'], async context => {
+    const pr = new PullRequest(context);
+
+    const title = pr.data.title.replace(labels.wip.regex ?? '', '');
+    await pr.setTitle(title);
+
+    await pr.addLabel('readyForReview');
+  });
 };

src/labels.ts

@@ -7,7 +7,7 @@ type Label = {
 
 export const labels: Record<string, Label> = {
   wip: {
-    regex: /^\[WIP\]\s/i,
+    regex: /^\s*(\[WIP\]\s*|WIP:\s*|WIP\s+)+/i,
     name: ':construction: WIP',
     color: '#FBCA04',
     description: 'Still under development, not yet ready for review',