Implement #24

Author: lramos15

api/api.ts

@@ -33,6 +33,11 @@ export interface GitHubIssue extends GitHub {
 	unlockIssue(): Promise<void>;
 
 	setMilestone(milestoneId: number): Promise<void>;
+	/**
+	 * Returns what we think the current milestone for the repo is based on the due on date.
+	 * @returns The milestone id if one is found, else undefined
+	 */
+	getCurrentRepoMilestone(): Promise<number | undefined>;
 
 	addLabel(label: string): Promise<void>;
 	removeLabel(label: string): Promise<void>;

api/octokit.js

@@ -313,6 +313,29 @@ export class OctoKitIssue extends OctoKit implements GitHubIssue {
 			});
 	}
 
+	async getCurrentRepoMilestone(): Promise<number | undefined> {
+		safeLog(`Getting repo milestone for ${this.issueData.number}`);
+		// Fetch all milestones open for this repo
+		const allMilestones = (
+			await this.octokit.issues.listMilestonesForRepo({
+				owner: this.params.owner,
+				repo: this.params.repo,
+				state: 'open',
+				sort: 'due_on',
+				direction: 'asc',
+			})
+		).data;
+		const currentDate = new Date();
+		const possibleMilestones = allMilestones.filter(
+			(milestone) =>
+				new Date(milestone.due_on) > currentDate && currentDate > new Date(milestone.created_at),
+		);
+		if (possibleMilestones.length === 0) {
+			return undefined;
+		}
+		return possibleMilestones[0].id;
+	}
+
 	async *getComments(last?: boolean): AsyncIterableIterator<Comment[]> {
 		safeLog('Fetching comments for ' + this.issueData.number);
 

api/octokit.ts

@@ -147,6 +147,13 @@ export class TestbedIssue extends Testbed implements GitHubIssue {
 		}
 	}
 
+	async getCurrentRepoMilestone(): Promise<number | undefined> {
+		if (this.issueConfig.issue.milestone) {
+			return this.issueConfig.issue.milestone.milestoneId;
+		}
+		return undefined;
+	}
+
 	async getIssue(): Promise<Issue> {
 		const labels = [...this.issueConfig.labels];
 		return { ...this.issueConfig.issue, labels };

api/testbed.js

@@ -84,6 +84,13 @@ export const enrollIssue = async (issue: GitHubIssue, notYetReleasedLabel: strin
 	const closingHash = (await issue.getClosingInfo())?.hash;
 	if (closingHash) {
 		await issue.addLabel(notYetReleasedLabel);
+		// Get the milestone linked to the current release and set it if the issue doesn't have one
+		const releaseMilestone = (await issue.getIssue()).milestone
+			? undefined
+			: await issue.getCurrentRepoMilestone();
+		if (releaseMilestone !== undefined) {
+			await issue.setMilestone(releaseMilestone);
+		}
 		await trackEvent(issue, 'insiders-released:unreleased');
 	} else {
 		await trackEvent(issue, 'insiders-released:skipped');

api/testbed.ts

@@ -33,4 +33,4 @@ class ReleasePipelineAction extends Action {
 	}
 }
 
-new ReleasePipelineAction().run() // eslint-disable-line
+new ReleasePipelineAction().run(); // eslint-disable-line