reset-fork.yml

name: Reset Current Branch to Upstream After Squash Merge

on:
  workflow_dispatch:
    inputs:
      branch:
        description: 'Branch to reset (leave blank for current branch)'
        required: false
        default: ''

permissions:
  contents: write

jobs:
  reset-branch:
    runs-on: ubuntu-latest

    steps:
      - name: Checkout Repository
        uses: actions/checkout@34e114876b0b11c390a56381ad16ebd13914f8d5 # v4.3.1
        with:
          fetch-depth: 0

      - name: Detect Current Branch
        if: ${{ inputs.branch == '' }}
        run: echo "branch=$(git rev-parse --abbrev-ref HEAD)" >> $GITHUB_ENV

      - name: Use Input Branch
        if: ${{ inputs.branch != '' }}
        env:
          BRANCH_INPUT: ${{ inputs.branch }}
        run: echo "branch=${BRANCH_INPUT}" >> $GITHUB_ENV

      - name: Add Upstream Remote
        run: |
          git remote add upstream https://github.com/mlcommons/mlcflow.git
          git fetch upstream
      - name: Reset Branch to Upstream
        run: |
          git checkout ${{ env.branch }}
          git reset --hard upstream/${{ env.branch }}
        if: success()

      - name: Force Push to Origin
        run: |
          git push origin ${{ env.branch }} --force-with-lease
        if: success()