Adding an option for custom branch name for backport (#1)

Signed-off-by: Vacha Shah <vachshah@amazon.com>
VachaShah · Jan 5, 2022 · 727f307 · 727f307
1 parent 7eb9608
commit 727f307
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 2 deletions.
diff --git a/src/backport.ts b/src/backport.ts
@@ -5,6 +5,10 @@ import { GitHub } from "@actions/github/lib/utils";
 import { EventPayloads } from "@octokit/webhooks";
 import escapeRegExp from "lodash/escapeRegExp";
 
+export interface Inputs {
+  branchName: string
+}
+
 const labelRegExp = /^backport ([^ ]+)(?: ([^ ]+))?$/;
 
 const getLabelNames = ({
@@ -31,11 +35,13 @@ const getBackportBaseToHead = ({
   label,
   labels,
   pullRequestNumber,
+  branchName,
 }: {
   action: EventPayloads.WebhookPayloadPullRequest["action"];
   label: { name: string };
   labels: EventPayloads.WebhookPayloadPullRequest["pull_request"]["labels"];
   pullRequestNumber: number;
+  branchName: string;
 }): Record<string, string> => {
   const baseToHead: Record<string, string> = {};
 
@@ -46,7 +52,7 @@ const getBackportBaseToHead = ({
       const [
         ,
         base,
-        head = `backport-${pullRequestNumber}-to-${base}`,
+        head = (branchName != null) ? branchName : `backport-${pullRequestNumber}-to-${base}`,
       ] = matches;
       baseToHead[base] = head;
     }
@@ -193,11 +199,13 @@ const backport = async ({
   },
   titleTemplate,
   token,
+  inputs,
 }: {
   labelsToAdd: string[];
   payload: EventPayloads.WebhookPayloadPullRequest;
   titleTemplate: string;
   token: string;
+  inputs: Inputs;
 }) => {
   if (merged !== true) {
     return;
@@ -209,6 +217,7 @@ const backport = async ({
     label: label!,
     labels,
     pullRequestNumber,
+    inputs.branchName,
   });
 
   if (Object.keys(backportBaseToHead).length === 0) {

diff --git a/src/index.ts b/src/index.ts
@@ -2,11 +2,14 @@ import { debug, error as logError, getInput, setFailed } from "@actions/core";
 import { context } from "@actions/github";
 import { EventPayloads } from "@octokit/webhooks";
 
-import { backport } from "./backport";
+import { Inputs, backport } from "./backport";
 import { getLabelsToAdd } from "./get-labels-to-add";
 
 const run = async () => {
   try {
+    const inputs: Inputs = {
+      branchName: getInput('branch-name'),
+    }
     const token = getInput("github_token", { required: true });
     const titleTemplate = getInput("title_template");
     debug(JSON.stringify(context, undefined, 2));
@@ -17,6 +20,7 @@ const run = async () => {
       payload: context.payload as EventPayloads.WebhookPayloadPullRequest,
       titleTemplate,
       token,
+      inputs
     });
   } catch (error: unknown) {
     if (typeof error !== "string" && !(error instanceof Error)) {