ComfyUI_frontend/.github/workflows/pr-size-report.yaml at main · Comfy-Org/ComfyUI_frontend · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
name: 'PR: Size Report'

on:
  workflow_run:
    workflows: ['CI: Size Data']
    types:
      - completed
  workflow_dispatch:
    inputs:
      pr_number:
        description: 'PR number to report on'
        required: true
        type: number
      run_id:
        description: 'Size data workflow run ID'
        required: true
        type: string

permissions:
  contents: read
  pull-requests: write
  issues: write

jobs:
  comment:
    runs-on: ubuntu-latest
    if: >
      github.repository == 'Comfy-Org/ComfyUI_frontend' &&
      (
        (github.event_name == 'workflow_run' &&
         github.event.workflow_run.event == 'pull_request' &&
         github.event.workflow_run.conclusion == 'success') ||
        github.event_name == 'workflow_dispatch'
      )
    steps:
      - uses: actions/checkout@v6

      - name: Setup frontend
        uses: ./.github/actions/setup-frontend

      - name: Download size data
        uses: dawidd6/action-download-artifact@0bd50d53a6d7fb5cb921e607957e9cc12b4ce392 # v12
        with:
          name: size-data
          run_id: ${{ github.event_name == 'workflow_dispatch' && inputs.run_id || github.event.workflow_run.id }}
          path: temp/size

      - name: Resolve and validate PR metadata
        id: pr-meta
        uses: actions/github-script@v8
        with:
          script: |
            const fs = require('fs');

            // workflow_dispatch: validate artifact metadata against API-resolved PR
            if (context.eventName === 'workflow_dispatch') {
              const pullNumber = Number('${{ inputs.pr_number }}');
              const { data: dispatchPr } = await github.rest.pulls.get({
                owner: context.repo.owner,
                repo: context.repo.repo,
                pull_number: pullNumber,
              });

              const artifactPr = Number(fs.readFileSync('temp/size/number.txt', 'utf8').trim());
              const artifactBase = fs.readFileSync('temp/size/base.txt', 'utf8').trim();

              if (artifactPr !== dispatchPr.number) {
                core.setFailed(`Artifact PR number (${artifactPr}) does not match dispatch PR (${dispatchPr.number}).`);
                return;
              }
              if (artifactBase !== dispatchPr.base.ref) {
                core.setFailed(`Artifact base (${artifactBase}) does not match dispatch PR base (${dispatchPr.base.ref}).`);
                return;
              }

              core.setOutput('number', String(dispatchPr.number));
              core.setOutput('base', dispatchPr.base.ref);
              return;
            }

            // workflow_run: validate artifact metadata against trusted context
            const artifactPr = Number(fs.readFileSync('temp/size/number.txt', 'utf8').trim());
            const artifactBase = fs.readFileSync('temp/size/base.txt', 'utf8').trim();

            let pr = context.payload.workflow_run.pull_requests?.[0];
            if (!pr) {
              const { data: prs } = await github.rest.repos.listPullRequestsAssociatedWithCommit({
                owner: context.repo.owner,
                repo: context.repo.repo,
                commit_sha: context.payload.workflow_run.head_sha,
              });
              pr = prs.find(p => p.state === 'open');
            }

            if (!pr) {
              core.setFailed('Unable to resolve PR from workflow_run context.');
              return;
            }

            if (Number(pr.number) !== artifactPr) {
              core.setFailed(`Artifact PR number (${artifactPr}) does not match trusted context (${pr.number}).`);
              return;
            }

            const trustedBase = pr.base?.ref;
            if (!trustedBase || artifactBase !== trustedBase) {
              core.setFailed(`Artifact base (${artifactBase}) does not match trusted context (${trustedBase}).`);
              return;
            }

            core.setOutput('number', String(pr.number));
            core.setOutput('base', trustedBase);

      - name: Download previous size data
        uses: dawidd6/action-download-artifact@0bd50d53a6d7fb5cb921e607957e9cc12b4ce392 # v12
        with:
          branch: ${{ steps.pr-meta.outputs.base }}
          workflow: ci-size-data.yaml
          event: push
          name: size-data
          path: temp/size-prev
          if_no_artifact_found: warn

      - name: Generate size report
        run: node scripts/size-report.js > size-report.md

      - name: Post PR comment
        uses: ./.github/actions/post-pr-report-comment
        with:
          pr-number: ${{ steps.pr-meta.outputs.number }}
          report-file: ./size-report.md
          comment-marker: '<!-- COMFYUI_FRONTEND_SIZE -->'
          token: ${{ secrets.GITHUB_TOKEN }}