feat: add task for determining target repo based on flavor (#188)

Author: Marshall Cottrell

tasks/publish.yaml

@@ -1,3 +1,10 @@
+includes:
+  - utils: https://raw.githubusercontent.com/defenseunicorns/uds-common/7d5f445a50b19e97b9a8bed0d2445654ae05bab6/tasks/utils.yaml
+
+variables:
+  - name: FLAVOR
+    default: upstream
+
 tasks:
   - name: package
     description: Publish package for the supplied architecture
@@ -11,13 +18,16 @@ tasks:
       architecture:
         description: The architecture of the package to publish
         default: ${UDS_ARCH}
-      target_repo:
-        description: The repository to publish into
-        default: ghcr.io/defenseunicorns/packages/uds
+      team:
+        description: The team publishing the package
+        default: uds
     actions:
+      - task: utils:determine-repo
+        with:
+          team: ${{.inputs.team}}
       - description: Publish package for the supplied architecture
         cmd: |
-          ./uds zarf package publish ${{ .inputs.path }}/zarf-package-*-${{ .inputs.architecture }}-${{ .inputs.version }}.tar.zst oci://${{ .inputs.target_repo }}
+          ./uds zarf package publish ${{ .inputs.path }}/zarf-package-*-${{ .inputs.architecture }}-${{ .inputs.version }}.tar.zst oci://${TARGET_REPO}
 
   - name: test-bundle
     description: Publish bundle for the supplied architecture
@@ -31,10 +41,8 @@ tasks:
       architecture:
         description: The architecture of the bundle to publish
         default: ${UDS_ARCH}
-      target_repo:
-        description: The repository to publish into
-        default: ghcr.io/defenseunicorns/packages/uds/bundles
     actions:
+      - task: utils:determine-repo
       - description: Publish bundle for the supplied architecture
         cmd: |
-          ./uds publish ${{ .inputs.path }}/uds-bundle-*-${{ .inputs.architecture }}-${{ .inputs.version }}.tar.zst oci://${{ .inputs.target_repo }} --no-progress
+          ./uds publish ${{ .inputs.path }}/uds-bundle-*-${{ .inputs.architecture }}-${{ .inputs.version }}.tar.zst oci://${TARGET_REPO}/bundles --no-progress

tasks/pull.yaml

@@ -1,8 +1,9 @@
+includes:
+  - utils: https://raw.githubusercontent.com/defenseunicorns/uds-common/7d5f445a50b19e97b9a8bed0d2445654ae05bab6/tasks/utils.yaml
+
 variables:
   - name: FLAVOR
     default: upstream
-  - name: TARGET_REPO
-    default: ghcr.io/defenseunicorns/packages/uds
 
 tasks:
   - name: latest-package-release
@@ -13,20 +14,23 @@ tasks:
       spoof_release:
         description: Whether to spoof the pulled package version to the current repo version
         default: "false"
-      target_repo:
-        description: The repository to pull from
-        default: ghcr.io/defenseunicorns/packages/uds
+      team:
+        description: The team you want to pull the package from
+        default: uds
     actions:
+      - task: utils:determine-repo
+        with:
+          team: ${{.inputs.team}}
       - description: Get the current Zarf package name
         cmd: cat ${{ .inputs.path }}/zarf.yaml | yq .metadata.name
         setVariables:
           - name: PACKAGE_NAME
       - description: Get latest tag version from OCI
-        cmd: ./uds zarf tools registry ls ${{ .inputs.target_repo }}/${PACKAGE_NAME} | grep ${FLAVOR} | sort -V | tail -1
+        cmd: ./uds zarf tools registry ls ${TARGET_REPO}/${PACKAGE_NAME} | grep ${FLAVOR} | sort -V | tail -1
         setVariables:
           - name: LATEST_VERSION
       - description: Pull the latest package release
-        cmd: ./uds zarf package pull oci://${{ .inputs.target_repo }}/${PACKAGE_NAME}:${LATEST_VERSION} --no-progress -o ${{ .inputs.path }}
+        cmd: ./uds zarf package pull oci://${TARGET_REPO}/${PACKAGE_NAME}:${LATEST_VERSION} --no-progress -o ${{ .inputs.path }}
       # TODO (@WSTARR): This is currently needed to get around the chicken+egg condition when release please updates the version in GH
       - description: Spoof the latest release to the current version
         cmd: |
@@ -47,16 +51,17 @@ tasks:
         description: The repository to pull from
         default: ghcr.io/defenseunicorns/packages/uds/bundles
     actions:
+      - task: utils:determine-repo
       - description: Get the current uds bundle name
         cmd: cat ${{ .inputs.path }}/uds-bundle.yaml | yq .metadata.name
         setVariables:
           - name: BUNDLE_NAME
       - description: Get latest tag version from OCI
-        cmd: ./uds zarf tools registry ls ${{ .inputs.target_repo }}/${BUNDLE_NAME} | sort -V | tail -1
+        cmd: ./uds zarf tools registry ls ${TARGET_REPO}/bundles/${BUNDLE_NAME} | sort -V | tail -1
         setVariables:
           - name: LATEST_VERSION
       - description: Pull the latest bundle release
-        cmd: ./uds pull oci://${{ .inputs.target_repo }}/${BUNDLE_NAME}:${LATEST_VERSION} --no-progress -o ${{ .inputs.path }}
+        cmd: ./uds pull oci://${TARGET_REPO}/bundles/${BUNDLE_NAME}:${LATEST_VERSION} --no-progress -o ${{ .inputs.path }}
       # TODO (@ZMILLER): This is currently needed to get around the chicken+egg condition when release please updates the version in GH
       - description: Spoof the latest release to the current version
         cmd: |

tasks/utils.yaml

@@ -0,0 +1,36 @@
+variables:
+  - name: FLAVOR
+    default: upstream
+
+tasks:
+  - name: determine-repo
+    inputs:
+      base_repo:
+        description: The base repository to publish into
+        default: ghcr.io/defenseunicorns/packages
+      team:
+        description: The team publishing the package
+        default: uds
+      flavor:
+        description: The flavor of the package to publish
+        default: ${FLAVOR}
+      snapshot:
+        description: Whether this is a snapshot release
+        default: "false"
+    actions:
+      - description: Determine repository for the given flavor/type of release
+        cmd: |
+          repo="${{.inputs.base_repo}}"
+          # unicorn flavor = private repository
+          if [ "${{.inputs.flavor}}" = "unicorn" ]; then
+            repo="${repo}/private"
+          fi
+          repo="${repo}/${{.inputs.team}}"
+          # snapshots = snapshot repository
+          if [ "${{.inputs.snapshot}}" = "true" ]; then
+            repo="${repo}/snapshots"
+          fi
+          echo "${repo}"
+        mute: true
+        setVariables:
+          - name: TARGET_REPO