Assigns labels based on branch names

__tests__/fixtures/branches.yml

@@ -0,0 +1,11 @@
+test-branch:
+  - branch: "test/**"
+
+feature-branch:
+  - branch: "*/feature/*"
+
+bug-branch:
+  - branch: "{bug,fix}/*"
+
+array-branch:
+  - branch: ["array/*"]

__tests__/main.test.ts

@@ -15,6 +15,7 @@ const paginateMock = jest.spyOn(gh, "paginate");
 const getPullMock = jest.spyOn(gh.rest.pulls, "get");
 
 const yamlFixtures = {
+  "branches.yml": fs.readFileSync("__tests__/fixtures/branches.yml"),
   "only_pdfs.yml": fs.readFileSync("__tests__/fixtures/only_pdfs.yml"),
 };
 
@@ -102,6 +103,64 @@ describe("run", () => {
     expect(addLabelsMock).toHaveBeenCalledTimes(0);
     expect(removeLabelMock).toHaveBeenCalledTimes(0);
   });
+
+  it("adds labels based on the branch names that match the glob pattern", async () => {
+    github.context.payload.pull_request!.head = { ref: "test/testing-time" };
+    usingLabelerConfigYaml("branches.yml");
+    await run();
+
+    expect(addLabelsMock).toHaveBeenCalledTimes(1);
+    expect(addLabelsMock).toHaveBeenCalledWith({
+      owner: "monalisa",
+      repo: "helloworld",
+      issue_number: 123,
+      labels: ["test-branch"],
+    });
+  });
+
+  it("adds multiple labels based on branch names that match different glob patterns", async () => {
+    github.context.payload.pull_request!.head = {
+      ref: "test/feature/123",
+    };
+    usingLabelerConfigYaml("branches.yml");
+    await run();
+
+    expect(addLabelsMock).toHaveBeenCalledTimes(1);
+    expect(addLabelsMock).toHaveBeenCalledWith({
+      owner: "monalisa",
+      repo: "helloworld",
+      issue_number: 123,
+      labels: ["test-branch", "feature-branch"],
+    });
+  });
+
+  it("it can support multiple branches by batching", async () => {
+    github.context.payload.pull_request!.head = { ref: "fix/123" };
+    usingLabelerConfigYaml("branches.yml");
+    await run();
+
+    expect(addLabelsMock).toHaveBeenCalledTimes(1);
+    expect(addLabelsMock).toHaveBeenCalledWith({
+      owner: "monalisa",
+      repo: "helloworld",
+      issue_number: 123,
+      labels: ["bug-branch"],
+    });
+  });
+
+  it("it can support multiple branches by providing an array", async () => {
+    github.context.payload.pull_request!.head = { ref: "array/123" };
+    usingLabelerConfigYaml("branches.yml");
+    await run();
+
+    expect(addLabelsMock).toHaveBeenCalledTimes(1);
+    expect(addLabelsMock).toHaveBeenCalledWith({
+      owner: "monalisa",
+      repo: "helloworld",
+      issue_number: 123,
+      labels: ["array-branch"],
+    });
+  });
 });
 
 function usingLabelerConfigYaml(fixtureName: keyof typeof yamlFixtures): void {

dist/index.js

@@ -92,6 +92,14 @@ function getPrNumber() {
     }
     return pullRequest.number;
 }
+function getBranchName() {
+    var _a;
+    const pullRequest = github.context.payload.pull_request;
+    if (!pullRequest) {
+        return undefined;
+    }
+    return (_a = pullRequest.head) === null || _a === void 0 ? void 0 : _a.ref;
+}
 function getChangedFiles(client, prNumber) {
     return __awaiter(this, void 0, void 0, function* () {
         const listFilesOptions = client.rest.pulls.listFiles.endpoint.merge({
@@ -203,6 +211,38 @@ function checkAll(changedFiles, globs) {
     core.debug(`  "all" patterns matched all files`);
     return true;
 }
+function matchBranchPattern(matcher, branchName) {
+    core.debug(`  - ${printPattern(matcher)}`);
+    if (matcher.match(branchName)) {
+        core.debug(`   "branch" pattern matched`);
+        return true;
+    }
+    core.debug(`   ${printPattern(matcher)} did not match`);
+    return false;
+}
+function checkBranch(glob) {
+    const branchName = getBranchName();
+    if (!branchName) {
+        core.debug(` no branch name`);
+        return false;
+    }
+    core.debug(` checking "branch" pattern against ${branchName}`);
+    if (Array.isArray(glob)) {
+        const matchers = glob.map((g) => new minimatch_1.Minimatch(g));
+        for (const matcher of matchers) {
+            if (matchBranchPattern(matcher, branchName)) {
+                core.debug(`  "branch" patterns matched against ${branchName}`);
+                return true;
+            }
+        }
+        core.debug(`  "branch" patterns did not match against ${branchName}`);
+        return false;
+    }
+    else {
+        const matcher = new minimatch_1.Minimatch(glob);
+        return matchBranchPattern(matcher, branchName);
+    }
+}
 function checkMatch(changedFiles, matchConfig) {
     if (matchConfig.all !== undefined) {
         if (!checkAll(changedFiles, matchConfig.all)) {
@@ -214,6 +254,11 @@ function checkMatch(changedFiles, matchConfig) {
             return false;
         }
     }
+    if (matchConfig.branch !== undefined) {
+        if (!checkBranch(matchConfig.branch)) {
+            return false;
+        }
+    }
     return true;
 }
 function addLabels(client, prNumber, labels) {
@@ -11709,103 +11754,103 @@ module.exports = eval("require")("encoding");
 /***/ ((module) => {
 
 "use strict";
-module.exports = require("assert");;
+module.exports = require("assert");
 
 /***/ }),
 
 /***/ 8614:
 /***/ ((module) => {
 
 "use strict";
-module.exports = require("events");;
+module.exports = require("events");
 
 /***/ }),
 
 /***/ 5747:
 /***/ ((module) => {
 
 "use strict";
-module.exports = require("fs");;
+module.exports = require("fs");
 
 /***/ }),
 
 /***/ 8605:
 /***/ ((module) => {
 
 "use strict";
-module.exports = require("http");;
+module.exports = require("http");
 
 /***/ }),
 
 /***/ 7211:
 /***/ ((module) => {
 
 "use strict";
-module.exports = require("https");;
+module.exports = require("https");
 
 /***/ }),
 
 /***/ 1631:
 /***/ ((module) => {
 
 "use strict";
-module.exports = require("net");;
+module.exports = require("net");
 
 /***/ }),
 
 /***/ 2087:
 /***/ ((module) => {
 
 "use strict";
-module.exports = require("os");;
+module.exports = require("os");
 
 /***/ }),
 
 /***/ 5622:
 /***/ ((module) => {
 
 "use strict";
-module.exports = require("path");;
+module.exports = require("path");
 
 /***/ }),
 
 /***/ 2413:
 /***/ ((module) => {
 
 "use strict";
-module.exports = require("stream");;
+module.exports = require("stream");
 
 /***/ }),
 
 /***/ 4016:
 /***/ ((module) => {
 
 "use strict";
-module.exports = require("tls");;
+module.exports = require("tls");
 
 /***/ }),
 
 /***/ 8835:
 /***/ ((module) => {
 
 "use strict";
-module.exports = require("url");;
+module.exports = require("url");
 
 /***/ }),
 
 /***/ 1669:
 /***/ ((module) => {
 
 "use strict";
-module.exports = require("util");;
+module.exports = require("util");
 
 /***/ }),
 
 /***/ 8761:
 /***/ ((module) => {
 
 "use strict";
-module.exports = require("zlib");;
+module.exports = require("zlib");
 
 /***/ })
 
@@ -11844,7 +11889,9 @@ module.exports = require("zlib");;
 /************************************************************************/
 /******/ 	/* webpack/runtime/compat */
 /******/ 	
-/******/ 	if (typeof __nccwpck_require__ !== 'undefined') __nccwpck_require__.ab = __dirname + "/";/************************************************************************/
+/******/ 	if (typeof __nccwpck_require__ !== 'undefined') __nccwpck_require__.ab = __dirname + "/";
+/******/ 	
+/************************************************************************/
 var __webpack_exports__ = {};
 // This entry need to be wrapped in an IIFE because it need to be in strict mode.
 (() => {

src/labeler.ts

@@ -6,6 +6,7 @@ import { Minimatch, IMinimatch } from "minimatch";
 interface MatchConfig {
   all?: string[];
   any?: string[];
+  branch?: string | string[];
 }
 
 type StringOrMatchConfig = string | MatchConfig;
@@ -71,6 +72,15 @@ function getPrNumber(): number | undefined {
   return pullRequest.number;
 }
 
+function getBranchName(): string | undefined {
+  const pullRequest = github.context.payload.pull_request;
+  if (!pullRequest) {
+    return undefined;
+  }
+
+  return pullRequest.head?.ref;
+}
+
 async function getChangedFiles(
   client: ClientType,
   prNumber: number
@@ -213,6 +223,42 @@ function checkAll(changedFiles: string[], globs: string[]): boolean {
   return true;
 }
 
+function matchBranchPattern(matcher: IMinimatch, branchName: string): boolean {
+  core.debug(`  - ${printPattern(matcher)}`);
+  if (matcher.match(branchName)) {
+    core.debug(`   "branch" pattern matched`);
+    return true;
+  }
+
+  core.debug(`   ${printPattern(matcher)} did not match`);
+  return false;
+}
+
+function checkBranch(glob: string | string[]): boolean {
+  const branchName = getBranchName();
+  if (!branchName) {
+    core.debug(` no branch name`);
+    return false;
+  }
+
+  core.debug(` checking "branch" pattern against ${branchName}`);
+  if (Array.isArray(glob)) {
+    const matchers = glob.map((g) => new Minimatch(g));
+    for (const matcher of matchers) {
+      if (matchBranchPattern(matcher, branchName)) {
+        core.debug(`  "branch" patterns matched against ${branchName}`);
+        return true;
+      }
+    }
+
+    core.debug(`  "branch" patterns did not match against ${branchName}`);
+    return false;
+  } else {
+    const matcher = new Minimatch(glob);
+    return matchBranchPattern(matcher, branchName);
+  }
+}
+
 function checkMatch(changedFiles: string[], matchConfig: MatchConfig): boolean {
   if (matchConfig.all !== undefined) {
     if (!checkAll(changedFiles, matchConfig.all)) {
@@ -226,6 +272,12 @@ function checkMatch(changedFiles: string[], matchConfig: MatchConfig): boolean {
     }
   }
 
+  if (matchConfig.branch !== undefined) {
+    if (!checkBranch(matchConfig.branch)) {
+      return false;
+    }
+  }
+
   return true;
 }