From 139be3d7ba6366e9d7da49a14059367dce65609e Mon Sep 17 00:00:00 2001
From: Sofia Papagiannaki <papagian@users.noreply.github.com>
Date: Tue, 26 May 2020 17:26:16 +0300
Subject: [PATCH] grafana/toolkit: Modify close milestone task to remove label
 from more than 100 pull requests and add dry run option (#25108)

* Fix close milestone to remove label from all the required pull requests

There used to be a limit to 100 pull requests.

* Add dry run option in close-milestone task
---
 packages/grafana-toolkit/src/cli/index.ts     |  2 +
 .../src/cli/tasks/closeMilestone.ts           | 89 +++++++++++--------
 2 files changed, 56 insertions(+), 35 deletions(-)

diff --git a/packages/grafana-toolkit/src/cli/index.ts b/packages/grafana-toolkit/src/cli/index.ts
index 977b3293395..4b28ee9ba21 100644
--- a/packages/grafana-toolkit/src/cli/index.ts
+++ b/packages/grafana-toolkit/src/cli/index.ts
@@ -104,6 +104,7 @@ export const run = (includeInternalScripts = false) => {
     program
       .command('close-milestone')
       .option('-m, --milestone <milestone>', 'Specify milestone')
+      .option('--dryRun', 'Only simulate actions')
       .description('Helps ends a milestone by removing the cherry-pick label and closing it')
       .action(async cmd => {
         if (!cmd.milestone) {
@@ -113,6 +114,7 @@ export const run = (includeInternalScripts = false) => {
 
         await execTask(closeMilestoneTask)({
           milestone: cmd.milestone,
+          dryRun: !!cmd.dryRun,
         });
       });
 
diff --git a/packages/grafana-toolkit/src/cli/tasks/closeMilestone.ts b/packages/grafana-toolkit/src/cli/tasks/closeMilestone.ts
index 3cc10b66538..0ac396a36fe 100644
--- a/packages/grafana-toolkit/src/cli/tasks/closeMilestone.ts
+++ b/packages/grafana-toolkit/src/cli/tasks/closeMilestone.ts
@@ -3,9 +3,10 @@ import GithubClient from '../utils/githubClient';
 
 interface CloseMilestoneOptions {
   milestone: string;
+  dryRun: boolean;
 }
 
-const closeMilestoneTaskRunner: TaskRunner<CloseMilestoneOptions> = async ({ milestone }) => {
+const closeMilestoneTaskRunner: TaskRunner<CloseMilestoneOptions> = async ({ milestone, dryRun }) => {
   const githubClient = new GithubClient({ required: true });
 
   const cherryPickLabel = 'cherry-pick needed';
@@ -16,6 +17,10 @@ const closeMilestoneTaskRunner: TaskRunner<CloseMilestoneOptions> = async ({ mil
     return;
   }
 
+  if (dryRun) {
+    console.log('dry run is enabled');
+  }
+
   const milestoneRes = await client.get(`/milestones/${milestone}`, {});
 
   const milestoneState = milestoneRes.data.state;
@@ -27,46 +32,60 @@ const closeMilestoneTaskRunner: TaskRunner<CloseMilestoneOptions> = async ({ mil
 
   console.log('fetching issues/PRs of the milestone ⏬');
 
-  // Get all the issues/PRs with the label cherry-pick
-  // Every pull request is actually an issue
-  const issuesRes = await client.get('/issues', {
-    params: {
-      state: 'closed',
-      labels: cherryPickLabel,
-      per_page: 100,
-      milestone: milestone,
-    },
-  });
+  let totalIssues = 0;
 
-  if (issuesRes.data.length < 1) {
-    console.log('no issues to remove label from');
-  } else {
-    console.log(`found ${issuesRes.data.length} issues to remove the cherry-pick label from 🔎`);
-  }
+  while (true) {
+    // Get first 100 issues/PRs with the label cherry-pick
+    // Every pull request is actually an issue
+    const issuesRes = await client.get('/issues', {
+      params: {
+        state: 'closed',
+        labels: cherryPickLabel,
+        per_page: 100,
+        milestone: milestone,
+      },
+    });
 
-  for (const issue of issuesRes.data) {
-    // the reason for using stdout.write is for achieving 'action -> result' on
-    // the same line
-    process.stdout.write(`🔧removing label from issue #${issue.number} 🗑...`);
-    const resDelete = await client.delete(`/issues/${issue.number}/labels/${cherryPickLabel}`, {});
-    if (resDelete.status === 200) {
-      process.stdout.write('done ✅\n');
-    } else {
-      console.log('failed ❌');
+    if (issuesRes.data.length < 1) {
+      break;
+    }
+
+    const comparativeStr = totalIssues === 0 ? ' ' : ' more ';
+    console.log(`found ${issuesRes.data.length}${comparativeStr}issues to remove the cherry-pick label from 🔎`);
+    totalIssues += issuesRes.data.length;
+
+    for (const issue of issuesRes.data) {
+      // the reason for using stdout.write is for achieving 'action -> result' on
+      // the same line
+      process.stdout.write(`🔧removing label from issue #${issue.number} 🗑...`);
+      if (!dryRun) {
+        const resDelete = await client.delete(`/issues/${issue.number}/labels/${cherryPickLabel}`, {});
+        if (resDelete.status === 200) {
+          process.stdout.write('done ✅\n');
+        } else {
+          console.log('failed ❌');
+        }
+      }
     }
   }
 
-  console.log(`cleaned up ${issuesRes.data.length} issues/prs ⚡️`);
-
-  const resClose = await client.patch(`/milestones/${milestone}`, {
-    state: 'closed',
-  });
-
-  if (resClose.status === 200) {
-    console.log('milestone closed 🙌');
+  if (totalIssues === 0) {
+    console.log('no issues to remove label from');
   } else {
-    console.log('failed to close the milestone, response:');
-    console.log(resClose);
+    console.log(`cleaned up ${totalIssues} issues/prs ⚡️`);
+  }
+
+  if (!dryRun) {
+    const resClose = await client.patch(`/milestones/${milestone}`, {
+      state: 'closed',
+    });
+
+    if (resClose.status === 200) {
+      console.log('milestone closed 🙌');
+    } else {
+      console.log('failed to close the milestone, response:');
+      console.log(resClose);
+    }
   }
 };