fix: tern code snapshot uses TERN_MIGRATIONS envvar

jackc · Jun 13, 2024 · ae53fa8 · ae53fa8
1 parent 84145e7
commit ae53fa8
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/main.go b/main.go
@@ -794,6 +794,12 @@ func SnapshotCode(cmd *cobra.Command, args []string) {
 	}
 
 	migrationsPath := cliOptions.migrationsPath
+
+	// If no migrations path was set in CLI argument look in environment.
+	if migrationsPath == "" {
+		migrationsPath = os.Getenv("TERN_MIGRATIONS")
+	}
+
 	migrations, err := migrate.FindMigrations(os.DirFS(migrationsPath))
 	if err != nil {
 		fmt.Fprintf(os.Stderr, "Error loading migrations:\n  %v\n", err)