add: parameterize branch name

Author: shqear93

README.md

@@ -32,11 +32,11 @@ jobs:
 
 ### Action inputs
 
-| Name           | Description                                  | Required | Default |
-|----------------|----------------------------------------------|----------|---------|
-| `github-token` | Token that is used to create comments        | ✅        |         |
-| `check-names`  | Comma-separated list of check names to rerun | ✅        |         |
-
+| Name            | Description                                  | Required | Default                                          |
+|-----------------|----------------------------------------------|----------|--------------------------------------------------|
+| `github-token`  | Token that is used to create comments        | ✅        |                                                  |
+| `check-names`   | Comma-separated list of check names to rerun | ✅        |                                                  |
+| `target-branch` | Branch for which checks should be rerun      | ❌        | the head ref of the pull request, default branch |
 ## Permissions
 
 Depending on the permissions granted to your token, you may lack some rights.

action.yml

@@ -12,6 +12,12 @@ inputs:
   github-token:
     description: 'GitHub token'
     required: true
+  target-branch:
+    description: |
+      Branch for which checks should be rerun.
+      If not provided, the branch of the pull request that triggered
+      the workflow is used
+    required: false
 
 runs:
   using: 'node20'

dist/index.js

@@ -33746,7 +33746,16 @@ async function run() {
   try {
     const checkNames = core.getInput('check-names').split(', ');
     const { owner, repo } = github.context.repo;
-    const branch = github.context.payload.pull_request.head.ref;
+    let branch = null;
+
+    if (core.getInput('target-branch')) {
+      branch = core.getInput('target-branch');
+    } else if (github.context.payload.pull_request) {
+      branch = github.context.payload.pull_request.head.ref;
+    } else {
+      const repoInfo = await octokit.rest.repos.get({ owner, repo });
+      branch = repoInfo.data.default_branch; // default branch
+    }
 
     const token = core.getInput('github-token');
     const octokit = github.getOctokit(token);

src/index.js

@@ -5,7 +5,16 @@ async function run() {
   try {
     const checkNames = core.getInput('check-names').split(', ');
     const { owner, repo } = github.context.repo;
-    const branch = github.context.payload.pull_request.head.ref;
+    let branch = null;
+
+    if (core.getInput('target-branch')) {
+      branch = core.getInput('target-branch');
+    } else if (github.context.payload.pull_request) {
+      branch = github.context.payload.pull_request.head.ref;
+    } else {
+      const repoInfo = await octokit.rest.repos.get({ owner, repo });
+      branch = repoInfo.data.default_branch; // default branch
+    }
 
     const token = core.getInput('github-token');
     const octokit = github.getOctokit(token);