From 583bd6937fdd212aad41d7b7c2ce038ed8caa959 Mon Sep 17 00:00:00 2001
From: Morgan Hein <morgan@telegraph.io>
Date: Wed, 4 Sep 2024 10:23:22 -0700
Subject: [PATCH] wip: added --no-schema option scaffolding, needs tests

---
 boilingcore/templates.go              |  1 +
 drivers/config.go                     | 15 +++++++++++++++
 drivers/interface.go                  |  1 +
 drivers/sqlboiler-psql/driver/psql.go | 13 +++++++++++--
 main.go                               | 12 +++++++-----
 5 files changed, 35 insertions(+), 7 deletions(-)

diff --git a/boilingcore/templates.go b/boilingcore/templates.go
index 209f8efa3..82c2cb9a9 100644
--- a/boilingcore/templates.go
+++ b/boilingcore/templates.go
@@ -39,6 +39,7 @@ type templateData struct {
 	RQ string
 
 	// Control various generation features
+	// NoOutputSchema, which is also used for generation, is defined in the drivers configuration
 	AddGlobal             bool
 	AddPanic              bool
 	AddSoftDeletes        bool
diff --git a/drivers/config.go b/drivers/config.go
index 9731a6ca6..397f9e595 100644
--- a/drivers/config.go
+++ b/drivers/config.go
@@ -93,6 +93,21 @@ func (c Config) DefaultString(key, def string) string {
 	return str
 }
 
+// DefaultBool retrieves a non-empty bool or the default value provided.
+func (c Config) DefaultBool(key string, def bool) bool {
+	b, ok := c[key]
+	if !ok {
+		return def
+	}
+
+	bul, ok := b.(bool)
+	if !ok {
+		return def
+	}
+
+	return bul
+}
+
 // Int retrieves an int, the bool says if it exists, is of the appropriate type,
 // and is non-zero. Coerces float64 to int because JSON and Javascript kinda suck.
 func (c Config) Int(key string) (int, bool) {
diff --git a/drivers/interface.go b/drivers/interface.go
index 92fec3f5f..e74cb4131 100644
--- a/drivers/interface.go
+++ b/drivers/interface.go
@@ -17,6 +17,7 @@ const (
 	ConfigBlacklist      = "blacklist"
 	ConfigWhitelist      = "whitelist"
 	ConfigSchema         = "schema"
+	ConfigNoOutputSchema = "no-output-schema" // Determine if templates/output include a schema, even if we used one to gather info
 	ConfigAddEnumTypes   = "add-enum-types"
 	ConfigEnumNullPrefix = "enum-null-prefix"
 	ConfigConcurrency    = "concurrency"
diff --git a/drivers/sqlboiler-psql/driver/psql.go b/drivers/sqlboiler-psql/driver/psql.go
index 73788f96a..ad5e2cab1 100644
--- a/drivers/sqlboiler-psql/driver/psql.go
+++ b/drivers/sqlboiler-psql/driver/psql.go
@@ -97,11 +97,20 @@ func (p *PostgresDriver) Assemble(config drivers.Config) (dbinfo *drivers.DBInfo
 	port := config.DefaultInt(drivers.ConfigPort, 5432)
 	sslmode := config.DefaultString(drivers.ConfigSSLMode, "require")
 	schema := config.DefaultString(drivers.ConfigSchema, "public")
+	noOutputSchema := config.DefaultBool(drivers.ConfigNoOutputSchema, false)
 	whitelist, _ := config.StringSlice(drivers.ConfigWhitelist)
 	blacklist, _ := config.StringSlice(drivers.ConfigBlacklist)
 	concurrency := config.DefaultInt(drivers.ConfigConcurrency, drivers.DefaultConcurrency)
 
-	useSchema := schema != "public"
+	switch {
+	case noOutputSchema:
+		break
+	case schema == "public":
+		noOutputSchema = true
+	default:
+		// just to be explicit, even though it's the default in the getter
+		noOutputSchema = false
+	}
 
 	p.addEnumTypes, _ = config[drivers.ConfigAddEnumTypes].(bool)
 	p.enumNullPrefix = strmangle.TitleCase(config.DefaultString(drivers.ConfigEnumNullPrefix, "Null"))
@@ -135,7 +144,7 @@ func (p *PostgresDriver) Assemble(config drivers.Config) (dbinfo *drivers.DBInfo
 			RQ: '"',
 
 			UseIndexPlaceholders: true,
-			UseSchema:            useSchema,
+			UseSchema:            !noOutputSchema,
 			UseDefaultKeyword:    true,
 		},
 	}
diff --git a/main.go b/main.go
index 31698e682..cb081bda3 100644
--- a/main.go
+++ b/main.go
@@ -101,6 +101,7 @@ func main() {
 	rootCmd.PersistentFlags().BoolP("no-auto-timestamps", "", false, "Disable automatic timestamps for created_at/updated_at")
 	rootCmd.PersistentFlags().BoolP("no-driver-templates", "", false, "Disable parsing of templates defined by the database driver")
 	rootCmd.PersistentFlags().BoolP("no-back-referencing", "", false, "Disable back referencing in the loaded relationship structs")
+	rootCmd.PersistentFlags().BoolP("no-schema", "", false, "Disable generating a schema in the output")
 	rootCmd.PersistentFlags().BoolP("always-wrap-errors", "", false, "Wrap all returned errors with stacktraces, also sql.ErrNoRows")
 	rootCmd.PersistentFlags().BoolP("add-global-variants", "", false, "Enable generation for global variants")
 	rootCmd.PersistentFlags().BoolP("add-panic-variants", "", false, "Enable generation for panic variants")
@@ -213,11 +214,12 @@ func preRun(cmd *cobra.Command, args []string) error {
 
 	// Configure the driver
 	cmdConfig.DriverConfig = map[string]interface{}{
-		"whitelist":        viper.GetStringSlice(driverName + ".whitelist"),
-		"blacklist":        viper.GetStringSlice(driverName + ".blacklist"),
-		"add-enum-types":   cmdConfig.AddEnumTypes,
-		"enum-null-prefix": cmdConfig.EnumNullPrefix,
-		"foreign-keys":     cmdConfig.ForeignKeys,
+		"whitelist":                  viper.GetStringSlice(driverName + ".whitelist"),
+		"blacklist":                  viper.GetStringSlice(driverName + ".blacklist"),
+		drivers.ConfigNoOutputSchema: viper.GetBool("no-schema"),
+		"add-enum-types":             cmdConfig.AddEnumTypes,
+		"enum-null-prefix":           cmdConfig.EnumNullPrefix,
+		"foreign-keys":               cmdConfig.ForeignKeys,
 	}
 
 	keys := allKeys(driverName)