From 2e2a9c6d5862d22e0f8cfb59d46bc47bf8eb1fe0 Mon Sep 17 00:00:00 2001
From: Tom Hu <88201630+thomasrockhu-codecov@users.noreply.github.com>
Date: Fri, 15 Nov 2024 07:50:22 -0500
Subject: [PATCH] fix: update tokenless branch logic (#1650)

* fix: update tokenless branch logic

* fix: use the github context

* fix: use double quotes

* fix: lots of quotes

* fix: ai fixes

* fix: just shove it in

* fix: more ai fixes

* fix: missing double quote

* fix: more clenaup

* fix: add CC_BRANCH setting for not forks

* fix: cleanup

* fix: more cleanup
---
 action.yml | 21 ++++++++++++++++++++-
 1 file changed, 20 insertions(+), 1 deletion(-)

diff --git a/action.yml b/action.yml
index bcbc9b942..a2959d073 100644
--- a/action.yml
+++ b/action.yml
@@ -175,13 +175,32 @@ runs:
             echo "CODECOV_TOKEN=$CODECOV_TOKEN" >> $GITHUB_ENV
           fi
         fi
+
+    - name: Override branch for forks
+      shell: bash
+      run: |
+        if [ -z "$CC_BRANCH" ] && [ -z "$CODECOV_TOKEN" ] && [ "${GITHUB_EVENT_PULL_REQUEST_REPO_FULL_NAME}" != "$GITHUB_REPOSITORY" ];
+        then
+          echo -e "\033[0;32m==>\033[0m Fork detected, tokenless uploading used"
+          TOKENLESS="$GITHUB_EVENT_PULL_REQUEST_HEAD_LABEL"
+          CC_BRANCH="$GITHUB_EVENT_PULL_REQUEST_HEAD_LABEL"
+          echo "TOKENLESS=$TOKENLESS" >> $GITHUB_ENV
+        fi
+
+        echo "CC_BRANCH=$CC_BRANCH" >> $GITHUB_ENV
+      env:
+        CC_BRANCH: ${{ inputs.override_branch }}
+        GITHUB_EVENT_PULL_REQUEST_HEAD_LABEL: ${{ github.event.pull_request.head.label }}
+        GITHUB_EVENT_PULL_REQUEST_REPO_FULL_NAME: ${{ github.event.pull_request.repo.full_name }}
+        GITHUB_REPOSITORY: ${{ github.repository }}
+
     - name: Upload coverage to Codecov
       run: ${GITHUB_ACTION_PATH}/dist/codecov.sh
       shell: bash
       working-directory: ${{ inputs.working-directory }}
       env:
         CC_BINARY: ${{ inputs.binary }}
-        CC_BRANCH: ${{ inputs.override_branch }}
+        CC_BRANCH: $CC_BRANCH
         CC_BUILD: ${{ inputs.override_build }}
         CC_BUILD_URL: ${{ inputs.override_build_url }}
         CC_CODE: ${{ inputs.report_code }}