Rename WithForeignKeyConstraintDisabled to WithForeignKeys

README.md

@@ -125,7 +125,7 @@ data "external_schema" "gorm" {
     "load",
     "--path", "./path/to/models",
     "--dialect", "mysql", // | postgres | sqlite
-    "--disable-migration-foreign-key-constraint",
+    "--disable-foreign-keys",
   ]
 }
 ```
@@ -144,7 +144,7 @@ import (
 )
 
 func main() {
-  stmts, err := gormschema.New("mysql" gormschema.WithForeignKeyConstraintDisabled()).Load(&models.User{}, &models.Pet{})
+  stmts, err := gormschema.New("mysql" gormschema.WithForeignKeys(false)).Load(&models.User{}, &models.Pet{})
   if err != nil {
     fmt.Fprintf(os.Stderr, "failed to load gorm schema: %v\n", err)
     os.Exit(1)

gormschema/gorm.go

@@ -55,9 +55,9 @@ func (l *Loader) Load(models ...interface{}) (string, error) {
 	return s.Stmts(), nil
 }
 
-func WithForeignKeyConstraintDisabled() func(*gorm.Config) {
+func WithForeignKeys(b bool) func(*gorm.Config) {
 	return func(c *gorm.Config) {
-		c.DisableForeignKeyConstraintWhenMigrating = true
+		c.DisableForeignKeyConstraintWhenMigrating = !b
 	}
 }
 

loader.tmpl

@@ -33,8 +33,8 @@ func handleError(err error, message string) {
 }
 
 func constructGormConfig() (config []func(*gorm.Config)) {
-	if {{ .DisableMigrationForeignKeyConstraint }} {
-		config = append(config, gormschema.WithForeignKeyConstraintDisabled())
+	if {{ .DisableForeignKeys }} {
+		config = append(config, gormschema.WithForeignKeys(false))
 	}
 
 	return

main.go

@@ -39,11 +39,11 @@ func main() {
 
 // LoadCmd is a command to load models
 type LoadCmd struct {
-	Path                                 string   `help:"path to schema package" required:""`
-	Models                               []string `help:"Models to load"`
-	Dialect                              string   `help:"dialect to use" enum:"mysql,sqlite,postgres" required:""`
-	DisableMigrationForeignKeyConstraint bool     `help:"disables foreign key constraint when migrating"`
-	out                                  io.Writer
+	Path               string   `help:"path to schema package" required:""`
+	Models             []string `help:"Models to load"`
+	Dialect            string   `help:"dialect to use" enum:"mysql,sqlite,postgres" required:""`
+	DisableForeignKeys bool     `help:"disables foreign key constraint"`
+	out                io.Writer
 }
 
 func (c *LoadCmd) Run() error {
@@ -54,9 +54,9 @@ func (c *LoadCmd) Run() error {
 	}
 	models := gatherModels(pkgs)
 	p := Payload{
-		Models:                               models,
-		Dialect:                              c.Dialect,
-		DisableMigrationForeignKeyConstraint: c.DisableMigrationForeignKeyConstraint,
+		Models:             models,
+		Dialect:            c.Dialect,
+		DisableForeignKeys: c.DisableForeignKeys,
 	}
 	var buf bytes.Buffer
 	if err := loaderTmpl.Execute(&buf, p); err != nil {
@@ -119,9 +119,9 @@ func filename(pkg string) string {
 }
 
 type Payload struct {
-	Models                               []model
-	Dialect                              string
-	DisableMigrationForeignKeyConstraint bool
+	Models             []model
+	Dialect            string
+	DisableForeignKeys bool
 }
 
 func (p Payload) Imports() []string {

main_test.go

@@ -9,41 +9,41 @@ import (
 
 func TestLoad(t *testing.T) {
 	testCases := []struct {
-		dialect                              string
-		disableMigrationForeignKeyConstraint bool
+		dialect            string
+		disableForeignKeys bool
 	}{
-		{dialect: "mysql", disableMigrationForeignKeyConstraint: false},
-		{dialect: "sqlite", disableMigrationForeignKeyConstraint: false},
-		{dialect: "postgres", disableMigrationForeignKeyConstraint: false},
-		{dialect: "mysql", disableMigrationForeignKeyConstraint: true},
-		{dialect: "sqlite", disableMigrationForeignKeyConstraint: true},
-		{dialect: "postgres", disableMigrationForeignKeyConstraint: true},
+		{dialect: "mysql", disableForeignKeys: false},
+		{dialect: "sqlite", disableForeignKeys: false},
+		{dialect: "postgres", disableForeignKeys: false},
+		{dialect: "mysql", disableForeignKeys: true},
+		{dialect: "sqlite", disableForeignKeys: true},
+		{dialect: "postgres", disableForeignKeys: true},
 	}
 
 	for _, tc := range testCases {
 		t.Run(tc.dialect, func(t *testing.T) {
 			var buf bytes.Buffer
 			cmd := &LoadCmd{
-				Path:                                 "./internal/testdata/models",
-				Dialect:                              tc.dialect,
-				out:                                  &buf,
-				DisableMigrationForeignKeyConstraint: tc.disableMigrationForeignKeyConstraint,
+				Path:               "./internal/testdata/models",
+				Dialect:            tc.dialect,
+				out:                &buf,
+				DisableForeignKeys: tc.disableForeignKeys,
 			}
 			err := cmd.Run()
 			require.NoError(t, err)
 
-			assertLoadOutput(t, buf.String(), tc.disableMigrationForeignKeyConstraint)
+			assertLoadOutput(t, buf.String(), tc.disableForeignKeys)
 		})
 	}
 }
 
-func assertLoadOutput(t *testing.T, output string, disableMigrationForeignKeyConstraint bool) {
+func assertLoadOutput(t *testing.T, output string, disableForeignKeys bool) {
 	require.Contains(t, output, "CREATE TABLE")
 	require.Contains(t, output, "pets")
 	require.Contains(t, output, "users")
 	require.NotContains(t, output, "toys") // Struct without GORM annotations
 
-	if disableMigrationForeignKeyConstraint {
+	if disableForeignKeys {
 		require.NotContains(t, output, "CONSTRAINT")
 		require.NotContains(t, output, "FOREIGN KEY")
 	} else {