build: update upstream docker image build py script to support config…

…urable main branch
Ortege-xyz · Jun 6, 2024 · ed5e21c · ed5e21c
1 parent db48bcf
commit ed5e21c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/.github/workflows/image.yml b/.github/workflows/image.yml
@@ -46,6 +46,7 @@ jobs:
         env:
           DOCKERHUB_USER: ${{ secrets.CR_USERNAME }}
           DOCKERHUB_REPO: ${{ secrets.CR_ENDPOINT }}/ortege/studio
+          MAIN_BRANCH: ortege
         run: |
           pip install click
           ./scripts/build_docker.py ${{ matrix.build_preset }} ${{ github.event_name }} --platform ${{ matrix.platform }}
diff --git a/scripts/build_docker.py b/scripts/build_docker.py
@@ -27,6 +27,7 @@
 CACHE_REPO = f"{REPO}-cache"
 BASE_PY_IMAGE = "3.9-slim-bookworm"
 
+MAIN_BRANCH = os.getenv("MAIN_BRANCH", "master")
 
 def run_cmd(command: str, raise_on_failure: bool = True) -> str:
     process = subprocess.Popen(
@@ -124,8 +125,8 @@ def get_docker_tags(
         if is_latest or force_latest:
             # add a latest tag
             tags.add(make_docker_tag(["latest"] + tag_chunks))
-    elif build_context == "push" and build_context_ref == "master":
-        tags.add(make_docker_tag(["master"] + tag_chunks))
+    elif build_context == "push" and build_context_ref == MAIN_BRANCH:
+        tags.add(make_docker_tag([MAIN_BRANCH] + tag_chunks))
     elif build_context == "pull_request":
         tags.add(make_docker_tag([f"pr-{build_context_ref}"] + tag_chunks))
     return tags