cmd/main: set env var with action name

ariga · Dec 25, 2023 · e056977 · e056977
1 parent 3347eaa
commit e056977
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/cmd/atlas-action/main.go b/cmd/atlas-action/main.go
@@ -8,6 +8,7 @@ import (
 	"context"
 	"errors"
 	"fmt"
+	"os"
 
 	"ariga.io/atlas-action/atlasaction"
 	"ariga.io/atlas-go-sdk/atlasexec"
@@ -61,6 +62,10 @@ type RunAction struct {
 }
 
 func (r *RunAction) Run(ctx context.Context, client *atlasexec.Client, action *githubactions.Action) error {
+	_ = os.Setenv("ATLAS_ACTION_COMMAND", r.Action)
+	defer func() {
+		_ = os.Unsetenv("ATLAS_ACTION_COMMAND")
+	}()
 	switch r.Action {
 	case CmdMigrateApply:
 		return atlasaction.MigrateApply(ctx, client, action)