๐Ÿ“ฆ apache / superset

๐Ÿ“„ ephemeral-env.yml ยท 345 lines
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
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345name: Ephemeral env workflow [DEPRECATED]

# โš ๏ธ  DEPRECATION NOTICE โš ๏ธ
# This workflow is deprecated and will be removed in a future version.
# Please use the new Superset Showtime workflow instead:
# - Use label "๐ŸŽช trigger-start" instead of "testenv-up"
# - Showtime provides better reliability and easier management
# - See .github/workflows/showtime.yml for the replacement
# - Migration guide: https://github.com/mistercrunch/superset-showtime

# Example manual trigger:
# gh workflow run ephemeral-env.yml --ref fix_ephemerals --field label_name="testenv-up" --field issue_number=666

on:
  pull_request_target:
    types:
      - labeled
  workflow_dispatch:
    inputs:
      label_name:
        description: 'Label name to simulate label-based /testenv trigger'
        required: true
        default: 'testenv-up'
      issue_number:
        description: 'Issue or PR number'
        required: true

jobs:
  ephemeral-env-label:
    concurrency:
      group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.run_id }}-label
      cancel-in-progress: true
    name: Evaluate ephemeral env label trigger
    runs-on: ubuntu-24.04
    permissions:
      pull-requests: write
    outputs:
      slash-command: ${{ steps.eval-label.outputs.result }}
      feature-flags: ${{ steps.eval-feature-flags.outputs.result }}
      sha: ${{ steps.get-sha.outputs.sha }}
    env:
      DOCKERHUB_USER: ${{ secrets.DOCKERHUB_USER }}
      DOCKERHUB_TOKEN: ${{ secrets.DOCKERHUB_TOKEN }}

    steps:
      - name: Check for the "testenv-up" label
        id: eval-label
        run: |
          if [[ "${{ github.event_name }}" == "workflow_dispatch" ]]; then
            LABEL_NAME="${{ github.event.inputs.label_name }}"
          else
            LABEL_NAME="${{ github.event.label.name }}"
          fi

          echo "Evaluating label: $LABEL_NAME"

          if [[ "$LABEL_NAME" == "testenv-up" ]]; then
            echo "result=up" >> $GITHUB_OUTPUT
          else
            echo "result=noop" >> $GITHUB_OUTPUT
          fi

      - name: Get event SHA
        id: get-sha
        if: steps.eval-label.outputs.result == 'up'
        uses: actions/github-script@v8
        with:
          github-token: ${{ secrets.GITHUB_TOKEN }}
          script: |
            let prSha;

            // If event is workflow_dispatch, use the issue_number from inputs
            if (context.eventName === "workflow_dispatch") {
              const prNumber = "${{ github.event.inputs.issue_number }}";
              if (!prNumber) {
                console.log("No PR number found.");
                return;
              }

              // Fetch PR details using the provided issue_number
              const { data: pr } = await github.rest.pulls.get({
                owner: context.repo.owner,
                repo: context.repo.repo,
                pull_number: prNumber
              });

              prSha = pr.head.sha;
            } else {
              // If it's not workflow_dispatch, use the PR head sha from the event
              prSha = context.payload.pull_request.head.sha;
            }

            console.log(`PR SHA: ${prSha}`);
            core.setOutput("sha", prSha);

      - name: Looking for feature flags in PR description
        uses: actions/github-script@v8
        id: eval-feature-flags
        if: steps.eval-label.outputs.result == 'up'
        with:
          script: |
            const description = context.payload.pull_request
              ? context.payload.pull_request.body || ''
              : context.payload.inputs.pr_description || '';

            const pattern = /FEATURE_(\w+)=(\w+)/g;
            let results = [];
            [...description.matchAll(pattern)].forEach(match => {
              const config = {
                name: `SUPERSET_FEATURE_${match[1]}`,
                value: match[2],
              };
              results.push(config);
            });

            return results;

      - name: Reply with confirmation comment
        uses: actions/github-script@v8
        if: steps.eval-label.outputs.result == 'up'
        with:
          github-token: ${{ secrets.GITHUB_TOKEN }}
          script: |
            const action = '${{ steps.eval-label.outputs.result }}';
            const user = context.actor;
            const runId = context.runId;
            const workflowUrl = `${context.serverUrl}/${context.repo.owner}/${context.repo.repo}/actions/runs/${runId}`;

            const issueNumber = context.payload.pull_request
              ? context.payload.pull_request.number
              : context.payload.inputs.issue_number;

            if (!issueNumber) {
              throw new Error("Issue number is not available.");
            }

            const body = `โš ๏ธ **DEPRECATED WORKFLOW** โš ๏ธ\n\n@${user} This workflow is deprecated! Please use the new **Superset Showtime** system instead:\n\n` +
              `- Replace "testenv-up" label with "๐ŸŽช trigger-start"\n` +
              `- Better reliability and easier management\n` +
              `- See https://github.com/mistercrunch/superset-showtime for details\n\n` +
              `Processing your ephemeral environment request [here](${workflowUrl}). Action: **${action}**.` +
              ` More information on [how to use or configure ephemeral environments]` +
              `(https://superset.apache.org/docs/contributing/howtos/#github-ephemeral-environments)`;


            await github.rest.issues.createComment({
              owner: context.repo.owner,
              repo: context.repo.repo,
              issue_number: issueNumber,
              body,
            });

  ephemeral-docker-build:
    concurrency:
      group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.run_id }}-build
      cancel-in-progress: true
    needs: ephemeral-env-label
    if: needs.ephemeral-env-label.outputs.slash-command == 'up'
    name: ephemeral-docker-build
    runs-on: ubuntu-24.04
    steps:
      - name: "Checkout ${{ github.ref }} ( ${{ needs.ephemeral-env-label.outputs.sha }} : ${{steps.get-sha.outputs.sha}} )"
        uses: actions/checkout@v6
        with:
          ref: ${{ needs.ephemeral-env-label.outputs.sha }}
          persist-credentials: false

      - name: Setup Docker Environment
        uses: ./.github/actions/setup-docker
        with:
          dockerhub-user: ${{ secrets.DOCKERHUB_USER }}
          dockerhub-token: ${{ secrets.DOCKERHUB_TOKEN }}
          build: "true"
          install-docker-compose: "false"

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

      - name: Build ephemeral env image
        env:
          GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
        run: |
          supersetbot docker \
            --push \
            --load \
            --preset ci \
            --platform  linux/amd64 \
            --context-ref "$RELEASE" \
            --extra-flags "--build-arg INCLUDE_CHROMIUM=false"

      - name: Configure AWS credentials
        uses: aws-actions/configure-aws-credentials@v5
        with:
          aws-access-key-id: ${{ secrets.AWS_ACCESS_KEY_ID }}
          aws-secret-access-key: ${{ secrets.AWS_SECRET_ACCESS_KEY }}
          aws-region: us-west-2

      - name: Login to Amazon ECR
        id: login-ecr
        uses: aws-actions/amazon-ecr-login@v2

      - name: Load, tag and push image to ECR
        id: push-image
        env:
          ECR_REGISTRY: ${{ steps.login-ecr.outputs.registry }}
          ECR_REPOSITORY: superset-ci
          IMAGE_TAG: apache/superset:${{ needs.ephemeral-env-label.outputs.sha }}-ci
          PR_NUMBER: ${{ github.event.inputs.issue_number || github.event.pull_request.number }}
        run: |
          docker tag $IMAGE_TAG $ECR_REGISTRY/$ECR_REPOSITORY:pr-$PR_NUMBER-ci
          docker push -a $ECR_REGISTRY/$ECR_REPOSITORY

  ephemeral-env-up:
    needs: [ephemeral-env-label, ephemeral-docker-build]
    if: needs.ephemeral-env-label.outputs.slash-command == 'up'
    name: Spin up an ephemeral environment
    runs-on: ubuntu-24.04
    permissions:
      contents: read
      pull-requests: write

    steps:
      - uses: actions/checkout@v6
        with:
          persist-credentials: false

      - name: Configure AWS credentials
        uses: aws-actions/configure-aws-credentials@v5
        with:
          aws-access-key-id: ${{ secrets.AWS_ACCESS_KEY_ID }}
          aws-secret-access-key: ${{ secrets.AWS_SECRET_ACCESS_KEY }}
          aws-region: us-west-2

      - name: Login to Amazon ECR
        id: login-ecr
        uses: aws-actions/amazon-ecr-login@v2

      - name: Check target image exists in ECR
        id: check-image
        continue-on-error: true
        env:
          PR_NUMBER: ${{ github.event.inputs.issue_number || github.event.pull_request.number }}
        run: |
          aws ecr describe-images \
          --registry-id $(echo "${{ steps.login-ecr.outputs.registry }}" | grep -Eo "^[0-9]+") \
          --repository-name superset-ci \
          --image-ids imageTag=pr-$PR_NUMBER-ci

      - name: Fail on missing container image
        if: steps.check-image.outcome == 'failure'
        uses: actions/github-script@v8
        with:
          github-token: ${{ github.token }}
          script: |
            const errMsg = '@${{ github.event.comment.user.login }} Container image not yet published for this PR. Please try again when build is complete.';
            github.rest.issues.createComment({
              issue_number: ${{ github.event.inputs.issue_number || github.event.pull_request.number }},
              owner: context.repo.owner,
              repo: context.repo.repo,
              body: errMsg
            });
            core.setFailed(errMsg);

      - name: Fill in the new image ID in the Amazon ECS task definition
        id: task-def
        uses: aws-actions/amazon-ecs-render-task-definition@v1
        with:
          task-definition: .github/workflows/ecs-task-definition.json
          container-name: superset-ci
          image: ${{ steps.login-ecr.outputs.registry }}/superset-ci:pr-${{ github.event.inputs.issue_number || github.event.pull_request.number }}-ci

      - name: Update env vars in the Amazon ECS task definition
        run: |
          cat <<< "$(jq '.containerDefinitions[0].environment += ${{ needs.ephemeral-env-label.outputs.feature-flags }}' < ${{ steps.task-def.outputs.task-definition }})" > ${{ steps.task-def.outputs.task-definition }}

      - name: Describe ECS service
        id: describe-services
        run: |
          echo "active=$(aws ecs describe-services --cluster superset-ci --services pr-${{ github.event.inputs.issue_number || github.event.pull_request.number }}-service | jq '.services[] | select(.status == "ACTIVE") | any')" >> $GITHUB_OUTPUT
      - name: Create ECS service
        id: create-service
        if: steps.describe-services.outputs.active != 'true'
        env:
          ECR_SUBNETS: subnet-0e15a5034b4121710,subnet-0e8efef4a72224974
          ECR_SECURITY_GROUP: sg-092ff3a6ae0574d91
          PR_NUMBER: ${{ github.event.inputs.issue_number || github.event.pull_request.number }}
        run: |
          aws ecs create-service \
          --cluster superset-ci \
          --service-name pr-$PR_NUMBER-service \
          --task-definition superset-ci \
          --launch-type FARGATE \
          --desired-count 1 \
          --platform-version LATEST \
          --network-configuration "awsvpcConfiguration={subnets=[$ECR_SUBNETS],securityGroups=[$ECR_SECURITY_GROUP],assignPublicIp=ENABLED}" \
          --tags key=pr,value=$PR_NUMBER key=github_user,value=${{ github.actor }}
      - name: Deploy Amazon ECS task definition
        id: deploy-task
        uses: aws-actions/amazon-ecs-deploy-task-definition@v2
        with:
          task-definition: ${{ steps.task-def.outputs.task-definition }}
          service: pr-${{ github.event.inputs.issue_number || github.event.pull_request.number }}-service
          cluster: superset-ci
          wait-for-service-stability: true
          wait-for-minutes: 10

      - name: List tasks
        id: list-tasks
        run: |
          echo "task=$(aws ecs list-tasks --cluster superset-ci --service-name pr-${{ github.event.inputs.issue_number || github.event.pull_request.number }}-service | jq '.taskArns | first')" >> $GITHUB_OUTPUT
      - name: Get network interface
        id: get-eni
        run: |
          echo "eni=$(aws ecs describe-tasks --cluster superset-ci --tasks ${{ steps.list-tasks.outputs.task }} | jq '.tasks[0].attachments[0].details | map(select(.name=="networkInterfaceId"))[0].value')" >> $GITHUB_OUTPUT
      - name: Get public IP
        id: get-ip
        run: |
          echo "ip=$(aws ec2 describe-network-interfaces --network-interface-ids ${{ steps.get-eni.outputs.eni }} | jq -r '.NetworkInterfaces | first | .Association.PublicIp')" >> $GITHUB_OUTPUT
      - name: Comment (success)
        if: ${{ success() }}
        uses: actions/github-script@v8
        with:
          github-token: ${{github.token}}
          script: |
            const issue_number = context.payload.inputs?.issue_number || context.issue.number;
            github.rest.issues.createComment({
              issue_number: issue_number,
              owner: context.repo.owner,
              repo: context.repo.repo,
              body: `@${{ github.actor }} Ephemeral environment spinning up at http://${{ steps.get-ip.outputs.ip }}:8080. Credentials are 'admin'/'admin'. Please allow several minutes for bootstrapping and startup.`
            });
      - name: Comment (failure)
        if: ${{ failure() }}
        uses: actions/github-script@v8
        with:
          github-token: ${{github.token}}
          script: |
            const issue_number = context.payload.inputs?.issue_number || context.issue.number;
            github.rest.issues.createComment({
              issue_number: issue_number,
              owner: context.repo.owner,
              repo: context.repo.repo,
              body: '@${{ github.event.inputs.user_login || github.event.comment.user.login }} Ephemeral environment creation failed. Please check the Actions logs for details.'
            })