atlasexec: add --directive to schema plan

atlasexec/atlas_models.go

@@ -1,6 +1,7 @@
 package atlasexec
 
 import (
+	"errors"
 	"time"
 
 	"ariga.io/atlas/sql/sqlcheck"
@@ -129,3 +130,14 @@ func (r *SummaryReport) DiagnosticsCount() int {
 	}
 	return n
 }
+
+// Errors returns the errors in the summary report, if exists.
+func (r *SummaryReport) Errors() []error {
+	var errs []error
+	for _, f := range r.Files {
+		if f.Error != "" {
+			errs = append(errs, errors.New(f.Error))
+		}
+	}
+	return errs
+}

atlasexec/atlas_schema.go

@@ -98,9 +98,10 @@ type (
 		Context   *RunContext
 		DevURL    string
 
-		From, To []string
-		Repo     string
-		Name     string
+		From, To   []string
+		Repo       string
+		Name       string
+		Directives []string
 		// The below are mutually exclusive and can be replaced
 		// with the 'schema plan' sub-commands instead.
 		DryRun     bool // If false, --auto-approve is set.
@@ -418,6 +419,9 @@ func (c *Client) SchemaPlan(ctx context.Context, params *SchemaPlanParams) (*Sch
 	} else {
 		args = append(args, "--auto-approve")
 	}
+	for _, d := range params.Directives {
+		args = append(args, "--directive", d)
+	}
 	// NOTE: This command only support one result.
 	return firstResult(jsonDecode[SchemaPlan](c.runCommand(ctx, args)))
 }

atlasexec/atlas_schema_test.go

@@ -249,6 +249,15 @@ func TestSchema_Plan(t *testing.T) {
 			},
 			args: `schema plan --format {{ json . }} --from 1,2 --to 2,3 --auto-approve`,
 		},
+		{
+			name: "with from to and directives",
+			params: &atlasexec.SchemaPlanParams{
+				From:       []string{"1", "2"},
+				To:         []string{"2", "3"},
+				Directives: []string{"atlas:nolint", "\"atlas:txmode none\""},
+			},
+			args: `schema plan --format {{ json . }} --from 1,2 --to 2,3 --auto-approve --directive atlas:nolint --directive "atlas:txmode none"`,
+		},
 		{
 			name: "with config",
 			params: &atlasexec.SchemaPlanParams{