name: Benchmark

on:
  workflow_dispatch:
    inputs:
      ref_before:
        description: "Branch, tag, or SHA before changes"
        default: "main"
        type: string
        required: true
      ref_after:
        description: "Branch, tag, or SHA after changes"
        type: string
        required: true
      pr_number:
        description: "PR number (for posting the result as a comment)"
        type: number
        required: false

# Cancels in-progress runs for runs comparing the same refs if the workflow is re-triggered.
concurrency:
  group: ${{ github.workflow }}-${{ inputs.ref_before }}-${{ inputs.ref_after }}
  cancel-in-progress: true

jobs:
  benchmark:
    runs-on: parity-large
    permissions:
      pull-requests: write
    steps:
      - name: Checkout Commit Before Changes
        uses: actions/checkout@v4
        with:
          ref: ${{ inputs.ref_before }}

      - name: Set Up Rust Toolchain
        uses: actions-rust-lang/setup-rust-toolchain@v1
        with:
          # Without this it will override our rust flags.
          rustflags: ""

      - name: Install Node.js
        uses: actions/setup-node@v6
        with:
          node-version: "lts/*"

      - name: Install Benchmarking Tools
        run: |
          sudo apt-get update
          sudo apt-get install -y build-essential
          cargo install critcmp

      - name: Install Solc
        uses: ./.github/actions/get-solc

      - name: Download LLVM
        uses: ./.github/actions/get-llvm
        with:
          target: x86_64-unknown-linux-gnu

      - name: Set LLVM Environment Variables
        run: |
          echo "LLVM_SYS_211_PREFIX=$(pwd)/llvm-x86_64-unknown-linux-gnu" >> $GITHUB_ENV

      - name: Build resolc
        run: |
          make install

      - name: Run Benchmarks on Commit Before Changes
        run: |
          cargo bench --package resolc --bench compile -- --save-baseline resolc_before
          cargo bench --package revive-benchmarks --bench parse -- --save-baseline yul_parse_before
          cargo bench --package revive-benchmarks --bench lower -- --save-baseline yul_lower_before
        timeout-minutes: 40

      - name: Checkout Commit After Changes
        run: |
          git fetch origin ${{ inputs.ref_after }}
          git checkout -f ${{ inputs.ref_after }}

      - name: Install Solc
        uses: ./.github/actions/get-solc

      - name: Download LLVM
        uses: ./.github/actions/get-llvm
        with:
          target: x86_64-unknown-linux-gnu

      - name: Build resolc
        run: |
          make install

      - name: Run Benchmarks on Commit After Changes
        run: |
          cargo bench --package resolc --bench compile -- --save-baseline resolc_changes
          cargo bench --package revive-benchmarks --bench parse -- --save-baseline yul_parse_changes
          cargo bench --package revive-benchmarks --bench lower -- --save-baseline yul_lower_changes
        timeout-minutes: 40

      - name: Compare Benchmarks
        run: |
          critcmp resolc_before resolc_changes > benchmarks_resolc
          critcmp yul_parse_before yul_parse_changes > benchmarks_yul_parse
          critcmp yul_lower_before yul_lower_changes > benchmarks_yul_lower

      - name: Create Report
        env:
          REF_BEFORE: ${{ inputs.ref_before }}
          REF_AFTER: ${{ inputs.ref_after }}
        run: |
          cat > BENCHMARK_REPORT.md << EOF
          # Benchmarks

          * **Ref Before:** \`$REF_BEFORE\`
          * **Ref After:** \`$REF_AFTER\`

          ## Compile E2E

          <details>
              <summary>Results</summary>

          \`\`\`
          $(cat benchmarks_resolc)
          \`\`\`

          </details>

          ## Parse Yul

          <details>
              <summary>Results</summary>

          \`\`\`
          $(cat benchmarks_yul_parse)
          \`\`\`

          </details>

          ## Lower Yul

          <details>
              <summary>Results</summary>

          \`\`\`
          $(cat benchmarks_yul_lower)
          \`\`\`

          </details>
          EOF

      - name: Post PR Comment with Benchmark Report
        if: ${{ inputs.pr_number }}
        uses: marocchino/sticky-pull-request-comment@v2
        with:
          header: benchmark
          number: ${{ inputs.pr_number }}
          path: BENCHMARK_REPORT.md

      - name: Upload Benchmark Results
        if: ${{ !inputs.pr_number }}
        uses: actions/upload-artifact@v4
        with:
          name: benchmark-report
          path: BENCHMARK_REPORT.md
          retention-days: 7