feat(template): add schema push & schema apply

ariga · Sep 17, 2024 · 9e94d71 · 9e94d71
1 parent be05417
commit 9e94d71
Show file tree

Hide file tree

Showing 6 changed files with 149 additions and 9 deletions.
diff --git a/gen/atlas.tmpl b/gen/atlas.tmpl
@@ -10,8 +10,6 @@ permissions:
 {{- end }}
 
 {{- define "job-setup" }}
-jobs:
-  atlas:
     {{- if .CreateDevURL }}
     {{- template "services" . }}
     {{- end }}

diff --git a/gen/declarative.tmpl b/gen/declarative.tmpl
@@ -6,8 +6,11 @@ on:
     branches: [ {{ .DefaultBranch }} ]
   workflow_dispatch:
 {{- template "permissions" . }}
-{{- template "job-setup" . }}
-      - uses: ariga/atlas-action/schema/plan@v1
+jobs:
+  plan:
+    if: {{`${{ github.event_name == 'pull_request' }}`}}
+    {{- template "job-setup" . }}
+      - uses: ariga/atlas-action/schema/plan@master
         with:
           {{- if .From }}
           from: {{ .From }}
@@ -24,5 +27,56 @@ on:
           {{- with .Env }}
           env: '{{ . }}'
           {{- end }}
+        env:
+          GITHUB_TOKEN: {{`${{ github.token }}`}}
+  apply:
+    if: {{`${{ github.event_name == 'push' && github.ref == 'refs/heads/master' }}`}}
+    {{- template "job-setup" . }}
+      - uses: ariga/atlas-action/schema/push@master
+        with:
+          {{- if .CreateDevURL }}
+          {{ template "UseServices" . }}
+          {{- end -}}
+          {{- with .ConfigPath }}
+          config: 'file://{{ . }}'
+          {{- end }}
+          {{- with .Env }}
+          env: '{{ . }}'
+          {{- end }}
+        env:
+          GITHUB_TOKEN: {{`${{ github.token }}`}}
+      - uses: ariga/atlas-action/schema/plan@master
+        id: schema-plan
+        with:
+          {{- if .From }}
+          from: {{ .From }}
+          {{- end }}
+          {{- if .To }}
+          to: {{ .To }}
+          {{- end }}
+          {{- if .CreateDevURL }}
+          {{ template "UseServices" . }}
+          {{- end -}}
+          {{- with .ConfigPath }}
+          config: 'file://{{ . }}'
+          {{- end }}
+          {{- with .Env }}
+          env: '{{ . }}'
+          {{- end }}
+        env:
+          GITHUB_TOKEN: {{`${{ github.token }}`}}
+      - uses: ariga/atlas-action/schema/apply@master
+        if: {{`${{ steps.schema-plan.outputs.status == 'APPROVED' }}`}}
+        with:
+          {{- if .CreateDevURL }}
+          {{ template "UseServices" . }}
+          {{- end -}}
+          {{- with .ConfigPath }}
+          config: 'file://{{ . }}'
+          {{- end }}
+          {{- with .Env }}
+          env: '{{ . }}'
+          {{- end }}
+          plan: {{`${{ steps.schema-plan.outputs.plan }}`}}
         env:
           GITHUB_TOKEN: {{`${{ github.token }}`}}
diff --git a/gen/gen_test.go b/gen/gen_test.go
@@ -89,6 +89,7 @@ func TestDeclarativeFlowGen(t *testing.T) {
 			expected, err := os.ReadFile("testdata/declarative/" + tc.filename)
 			require.NoError(t, err)
 			actual, err := Generate(tc.config)
+			os.WriteFile("testdata/declarative1/"+tc.filename, actual, 0644)
 			require.NoError(t, err)
 			require.Equal(t, strings.TrimSpace(string(expected)), strings.TrimSpace(string(actual)))
 		})

diff --git a/gen/testdata/declarative/plan_full.yml b/gen/testdata/declarative/plan_full.yml
@@ -10,7 +10,8 @@ permissions:
   contents: read
   pull-requests: write
 jobs:
-  atlas:
+  plan:
+    if: ${{ github.event_name == 'pull_request' }}
     services:
       # Spin up a mysql:8 container to be used as the dev-database for analysis.
       mysql:
@@ -34,12 +35,63 @@ jobs:
       - uses: ariga/setup-atlas@v0
         with:
           cloud-token: ${{ secrets.ATLAS_CLOUD_TOKEN_X1 }}
-      - uses: ariga/atlas-action/schema/plan@v1
+      - uses: ariga/atlas-action/schema/plan@master
         with:
           from: atlas://myrepo:v1
           to: atlas://myrepo:v2
           dev-url: 'mysql://root:pass@localhost:3306'
           config: 'file://atlas.hcl'
           env: 'GORM_MIGRATION_SOURCE'
+        env:
+          GITHUB_TOKEN: ${{ github.token }}
+  apply:
+    if: ${{ github.event_name == 'push' && github.ref == 'refs/heads/master' }}
+    services:
+      # Spin up a mysql:8 container to be used as the dev-database for analysis.
+      mysql:
+        image: mysql:8
+        env:
+          MYSQL_DATABASE: dev
+          MYSQL_ROOT_PASSWORD: pass
+        ports:
+          - 3306:3306
+        options: >-
+          --health-cmd "mysqladmin ping -ppass"
+          --health-interval 10s
+          --health-start-period 10s
+          --health-timeout 5s
+          --health-retries 10
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v3
+        with:
+          fetch-depth: 0
+      - uses: ariga/setup-atlas@v0
+        with:
+          cloud-token: ${{ secrets.ATLAS_CLOUD_TOKEN_X1 }}
+      - uses: ariga/atlas-action/schema/push@master
+        with:
+          dev-url: 'mysql://root:pass@localhost:3306'
+          config: 'file://atlas.hcl'
+          env: 'GORM_MIGRATION_SOURCE'
+        env:
+          GITHUB_TOKEN: ${{ github.token }}
+      - uses: ariga/atlas-action/schema/plan@master
+        id: schema-plan
+        with:
+          from: atlas://myrepo:v1
+          to: atlas://myrepo:v2
+          dev-url: 'mysql://root:pass@localhost:3306'
+          config: 'file://atlas.hcl'
+          env: 'GORM_MIGRATION_SOURCE'
+        env:
+          GITHUB_TOKEN: ${{ github.token }}
+      - uses: ariga/atlas-action/schema/apply@master
+        if: ${{ steps.schema-plan.outputs.status == 'APPROVED' }}
+        with:
+          dev-url: 'mysql://root:pass@localhost:3306'
+          config: 'file://atlas.hcl'
+          env: 'GORM_MIGRATION_SOURCE'
+          plan: ${{ steps.schema-plan.outputs.plan }}
         env:
           GITHUB_TOKEN: ${{ github.token }}
diff --git a/gen/testdata/declarative/plan_only_from.yml b/gen/testdata/declarative/plan_only_from.yml
@@ -10,7 +10,8 @@ permissions:
   contents: read
   pull-requests: write
 jobs:
-  atlas:
+  plan:
+    if: ${{ github.event_name == 'pull_request' }}
     runs-on: ubuntu-latest
     steps:
       - uses: actions/checkout@v3
@@ -19,10 +20,42 @@ jobs:
       - uses: ariga/setup-atlas@v0
         with:
           cloud-token: ${{ secrets.ATLAS_CLOUD_TOKEN_X3 }}
-      - uses: ariga/atlas-action/schema/plan@v1
+      - uses: ariga/atlas-action/schema/plan@master
         with:
           from: atlas://myrepo:latest
           config: 'file://atlas.hcl'
           env: 'GORM_MIGRATION_SOURCE'
+        env:
+          GITHUB_TOKEN: ${{ github.token }}
+  apply:
+    if: ${{ github.event_name == 'push' && github.ref == 'refs/heads/master' }}
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v3
+        with:
+          fetch-depth: 0
+      - uses: ariga/setup-atlas@v0
+        with:
+          cloud-token: ${{ secrets.ATLAS_CLOUD_TOKEN_X3 }}
+      - uses: ariga/atlas-action/schema/push@master
+        with:
+          config: 'file://atlas.hcl'
+          env: 'GORM_MIGRATION_SOURCE'
+        env:
+          GITHUB_TOKEN: ${{ github.token }}
+      - uses: ariga/atlas-action/schema/plan@master
+        id: schema-plan
+        with:
+          from: atlas://myrepo:latest
+          config: 'file://atlas.hcl'
+          env: 'GORM_MIGRATION_SOURCE'
+        env:
+          GITHUB_TOKEN: ${{ github.token }}
+      - uses: ariga/atlas-action/schema/apply@master
+        if: ${{ steps.schema-plan.outputs.status == 'APPROVED' }}
+        with:
+          config: 'file://atlas.hcl'
+          env: 'GORM_MIGRATION_SOURCE'
+          plan: ${{ steps.schema-plan.outputs.plan }}
         env:
           GITHUB_TOKEN: ${{ github.token }}
diff --git a/gen/versioned.tmpl b/gen/versioned.tmpl
@@ -10,7 +10,9 @@ on:
     paths:
       - '{{ .Path }}/*'
 {{- template "permissions" . }}
-{{- template "job-setup" . }}
+jobs:
+  atlas:
+      {{- template "job-setup" . }}
       - uses: ariga/atlas-action/migrate/lint@v1
         with:
           dir: 'file://{{ .Path }}'