feat: support branches only active in the current repository (#193)

elastic · Dec 11, 2024 · 0d55f7a · 0d55f7a
1 parent 67e2bd4
commit 0d55f7a
Show file tree

Hide file tree

Showing 3 changed files with 40 additions and 4 deletions.
diff --git a/elastic/active-branches/README.md b/elastic/active-branches/README.md
@@ -9,9 +9,11 @@ Fetch the current list of active branches in Elastic (the ones based on the Unif
 
 ## Inputs
 <!--inputs-->
-| Name               | Description                      | Required | Default |
-|--------------------|----------------------------------|----------|---------|
-| `exclude-branches` | Exclude branches comma separated | `false`  | ` `     |
+| Name               | Description                                                                     | Required | Default |
+|--------------------|---------------------------------------------------------------------------------|----------|---------|
+| `exclude-branches` | Exclude branches comma separated                                                | `false`  | ` `     |
+| `filter-branches`  | Whether to fitler those branches that only exist in the {{ github.repository }} | `false`  | `false` |
+| `github-token`     | The GitHub access token.                                                        | `false`  | ` `     |
 <!--/inputs-->
 
 ## Outputs
@@ -49,7 +51,6 @@ jobs:
         with:
           ref: "${{ matrix.branch }}"
 
-
       # ...
 ```
 <!--/usage-->
diff --git a/elastic/active-branches/action.yml b/elastic/active-branches/action.yml
@@ -7,6 +7,15 @@ inputs:
     required: false
     type: string
     default: ''
+  filter-branches:
+    description: "Whether to fitler those branches that only exist in the {{ github.repository }}"
+    default: false
+    type: boolean
+  github-token:
+    description: 'The GitHub access token.'
+    required: false
+    type: string
+    default: ''
 outputs:
   matrix:
     description: "Processed matrix with the branches (using the include format)"
@@ -22,6 +31,9 @@ runs:
       run: python ${{ github.action_path }}/script.py
       env:
         EXCLUDE_BRANCHES: ${{ inputs.exclude-branches }}
+        FILTER: ${{ inputs.filter-branches }}
+        REPOSITORY: ${{ github.repository }}
+        GITHUB_TOKEN: ${{ inputs.github-token }}
     - id: debug
       shell: bash
       run: |

diff --git a/elastic/active-branches/script.py b/elastic/active-branches/script.py
@@ -26,6 +26,29 @@ def fails(msg):
 if exclude_branches:
     branches = list(filter(lambda branch: branch not in exclude_branches, branches))
 
+# Filter branches based on the existence in the GitHub repository
+filter_str = os.environ.get('FILTER', 'false')
+filter = filter_str.lower() in ('true', '1', 't', 'y', 'yes')
+repository = os.environ.get('REPOSITORY', '')
+github_token = os.environ.get('GITHUB_TOKEN', '')
+if filter and repository and github_token:
+    # Check if branches exist in the GitHub repository
+    headers = {
+        'Authorization': f'token {github_token}',
+        'Accept': 'application/vnd.github.v3+json'
+    }
+
+    existing_branches = []
+    for branch in branches:
+        branch_url = f'https://api.github.com/repos/{repository}/branches/{branch}'
+        response = requests.get(branch_url, headers=headers)
+        if response.status_code == 200:
+            existing_branches.append(branch)
+        else:
+            print(f"Branch {branch} does not exist in the repository")
+
+    branches = existing_branches
+
 include_branches = list(map(lambda branch: {"branch": branch}, branches))
 matrix = {'include': include_branches}