Make generated output files optional

Author: GeertJohan

cmd/generate.go

@@ -461,7 +461,7 @@ var (
 	modelContent = ""
 )
 
-//nolint:cyclop
+//nolint:cyclop,gocognit
 func generateMigrationStatements(
 	ctx context.Context,
 	config *configuration.Config,
@@ -475,25 +475,40 @@ func generateMigrationStatements(
 ) (string, error) {
 	log.Println("Generating migration statements")
 
-	err := internal.PgModelerExportToFile(
-		ctx,
-		filepath.Join(wd, fmt.Sprintf("%s.dbm", config.ModelName)),
-		filepath.Join(wd, fmt.Sprintf("%s.sql", config.ModelName)),
-	)
+	dbmPath := filepath.Join(wd, fmt.Sprintf("%s.dbm", config.ModelName))
+	// Generate SQL file in tmpDir for internal use during migration generation
+	tmpSQLPath := filepath.Join(tmpDir, fmt.Sprintf("%s.sql", config.ModelName))
+
+	err := internal.PgmodelerExportSQL(ctx, dbmPath, tmpSQLPath)
 	if err != nil {
 		return "", fmt.Errorf("failed to export model: %w", err)
 	}
 
-	go func() {
-		err = internal.PgModelerExportToPng(
-			ctx,
-			filepath.Join(wd, fmt.Sprintf("%s.dbm", config.ModelName)),
-			filepath.Join(wd, fmt.Sprintf("%s.png", config.ModelName)),
-		)
-		if err != nil {
-			log.Printf("Failed to export png: %v\n", err)
-		}
-	}()
+	if pngPath := config.GetOutputPath("png"); pngPath != "" {
+		go func() {
+			pngErr := internal.PgmodelerExportPNG(
+				ctx,
+				dbmPath,
+				filepath.Join(wd, pngPath),
+			)
+			if pngErr != nil {
+				log.Printf("Failed to export png: %v\n", pngErr)
+			}
+		}()
+	}
+
+	if svgPath := config.GetOutputPath("svg"); svgPath != "" {
+		go func() {
+			svgErr := internal.PgmodelerExportSVG(
+				ctx,
+				dbmPath,
+				filepath.Join(wd, svgPath),
+			)
+			if svgErr != nil {
+				log.Printf("Failed to export svg: %v\n", svgErr)
+			}
+		}()
+	}
 
 	for _, role := range config.Roles {
 		_, err = targetConn.Exec(ctx, fmt.Sprintf("CREATE ROLE %q WITH LOGIN;", role.Name))
@@ -502,55 +517,69 @@ func generateMigrationStatements(
 		}
 	}
 
-	err = executeTargetSQL(ctx, config, wd, targetConn)
+	err = executeTargetSQL(ctx, tmpSQLPath, targetConn)
 	if err != nil {
 		return "", fmt.Errorf("failed to execute target sql: %w", err)
 	}
 
+	var output string
+
 	if initial {
 		// If we are developing the schema initially, there will be no diffs,
 		// and we want to copy over the schema file to the initial migration file
 		var input []byte
-		input, err = os.ReadFile(filepath.Join(wd, fmt.Sprintf("%s.sql", config.ModelName)))
+		input, err = os.ReadFile(tmpSQLPath)
 		if err != nil {
 			return "", fmt.Errorf("failed to read sql file: %w", err)
 		}
+		output = string(input)
+	} else {
+		err = executeMigrateSQL(migrationsDir, migrateConn)
+		if err != nil {
+			return "", fmt.Errorf("failed to execute migrate sql: %w", err)
+		}
 
-		return string(input), nil
-	}
-
-	err = executeMigrateSQL(migrationsDir, migrateConn)
-	if err != nil {
-		return "", fmt.Errorf("failed to execute migrate sql: %w", err)
-	}
+		var statements string
+		statements, err = internal.Diff(
+			ctx,
+			postgresConn,
+			migrateConn,
+			targetConn,
+		)
+		if err != nil {
+			return "", fmt.Errorf("failed to diff: %w", err)
+		}
 
-	statements, err := internal.Diff(
-		ctx,
-		postgresConn,
-		migrateConn,
-		targetConn,
-	)
-	if err != nil {
-		return "", fmt.Errorf("failed to diff: %w", err)
-	}
+		var extraStatements string
+		extraStatements, err = generateMissingPermissionStatements(ctx, tmpDir, statements, targetConn, migrateConn)
+		if err != nil {
+			return "", fmt.Errorf("failed to generate missing permission statements: %w", err)
+		}
 
-	extraStatements, err := generateMissingPermissionStatements(ctx, tmpDir, statements, targetConn, migrateConn)
-	if err != nil {
-		return "", fmt.Errorf("failed to generate missing permission statements: %w", err)
+		if statements != "" {
+			output += statements
+		}
+		if statements != "" && extraStatements != "" {
+			output += "\n\n"
+		}
+		if extraStatements != "" {
+			output += "-- Statements generated automatically, please review:\n" + extraStatements
+		}
+		if output != "" {
+			output += "\n"
+		}
 	}
 
-	var output string
-	if statements != "" {
-		output += statements
-	}
-	if statements != "" && extraStatements != "" {
-		output += "\n\n"
-	}
-	if extraStatements != "" {
-		output += "-- Statements generated automatically, please review:\n" + extraStatements
-	}
-	if output != "" {
-		output += "\n"
+	// Copy SQL to output path if enabled
+	if sqlPath := config.GetOutputPath("sql"); sqlPath != "" {
+		sqlContent, readErr := os.ReadFile(tmpSQLPath)
+		if readErr != nil {
+			return "", fmt.Errorf("failed to read sql file: %w", readErr)
+		}
+		err = os.WriteFile(filepath.Join(wd, sqlPath), sqlContent, 0o644) //nolint:gosec
+		if err != nil {
+			return "", fmt.Errorf("failed to write sql output file: %w", err)
+		}
 	}
 
 	return output, nil
@@ -569,8 +598,8 @@ func executeMigrateSQL(migrationsDir string, migrateConn *pgx.Conn) error {
 	return nil
 }
 
-func executeTargetSQL(ctx context.Context, config *configuration.Config, wd string, targetConn *pgx.Conn) error {
-	targetSQL, err := os.ReadFile(filepath.Join(wd, fmt.Sprintf("%s.sql", config.ModelName)))
+func executeTargetSQL(ctx context.Context, sqlPath string, targetConn *pgx.Conn) error {
+	targetSQL, err := os.ReadFile(sqlPath)
 	if err != nil {
 		return fmt.Errorf("failed to read target sql: %w", err)
 	}

example/foo.sql example/foo.gen.sqlexample/foo.sql renamed to example/foo.gen.sql

@@ -3,3 +3,6 @@ db_name: bar
 roles:
   - name: alice
   - name: bob
+output:
+  sql: {}
+  svg: {}

example/foo.gen.svg

@@ -24,7 +24,9 @@ type Config struct {
 	//nolint:tagliatelle
 	Roles     []Role     `yaml:"roles"`
 	Templates []Template `yaml:"templates"`
+	Output    *Output    `yaml:"output"`
 }
+
 type Role struct {
 	Name string `yaml:"name"`
 }
@@ -34,6 +36,16 @@ type Template struct {
 	Content string `yaml:"content"`
 }
 
+type OutputFile struct {
+	Path string `yaml:"path"`
+}
+
+type Output struct {
+	SQL *OutputFile `yaml:"sql"`
+	PNG *OutputFile `yaml:"png"`
+	SVG *OutputFile `yaml:"svg"`
+}
+
 func ReadConfig(wd string) (*Config, error) {
 	var config *Config
 	file, err := os.ReadFile(filepath.Join(wd, "trek.yaml"))
@@ -85,6 +97,38 @@ func (c *Config) validate() (problems []string) {
 	return problems
 }
 
+// GetOutputPath returns the output path for the given type if enabled, or empty string if not.
+// The outputType must be one of: "sql", "png", "svg".
+func (c *Config) GetOutputPath(outputType string) string {
+	if c.Output == nil {
+		return ""
+	}
+
+	var outputFile *OutputFile
+
+	switch outputType {
+	case "sql":
+		outputFile = c.Output.SQL
+	case "png":
+		outputFile = c.Output.PNG
+	case "svg":
+		outputFile = c.Output.SVG
+	default:
+		return ""
+	}
+
+	if outputFile == nil {
+		return ""
+	}
+
+	if outputFile.Path != "" {
+		return outputFile.Path
+	}
+
+	// Default path: {model_name}.gen.{ext}
+	return fmt.Sprintf("%s.gen.%s", c.ModelName, outputType)
+}
+
 func ValidateIdentifier(identifier string) bool {
 	return regexpValidIdentifier.MatchString(identifier)
 }

example/foo.png

@@ -7,7 +7,7 @@ import (
 	"os/exec"
 )
 
-func PgModelerExportToFile(ctx context.Context, input, output string) error {
+func PgmodelerExportSQL(ctx context.Context, input, output string) error {
 	//nolint:gosec
 	err := os.WriteFile(output, []byte{}, 0o644)
 	if err != nil {
@@ -35,7 +35,7 @@ func PgModelerExportToFile(ctx context.Context, input, output string) error {
 	return nil
 }
 
-func PgModelerExportToPng(ctx context.Context, input, output string) error {
+func PgmodelerExportPNG(ctx context.Context, input, output string) error {
 	//nolint:gosec
 	err := os.WriteFile(output, []byte{}, 0o644)
 	if err != nil {
@@ -60,3 +60,29 @@ func PgModelerExportToPng(ctx context.Context, input, output string) error {
 
 	return nil
 }
+
+func PgmodelerExportSVG(ctx context.Context, input, output string) error {
+	//nolint:gosec
+	err := os.WriteFile(output, []byte{}, 0o644)
+	if err != nil {
+		return fmt.Errorf("failed to create output svg: %w", err)
+	}
+	//nolint:gosec
+	cmdPgModeler := exec.CommandContext(
+		ctx,
+		"pgmodeler-cli",
+		"--input",
+		input,
+		"--export-to-svg",
+		"--output",
+		output,
+	)
+	cmdPgModeler.Stderr = os.Stderr
+
+	out, err := cmdPgModeler.Output()
+	if err != nil {
+		return fmt.Errorf("failed to run pgmodeler: %w %s", err, string(out))
+	}
+
+	return nil
+}

example/trek.yaml

@@ -2,3 +2,6 @@ model_name: {{.model_name}}
 db_name: {{.db_name}}
 roles:{{range .roleNames}}
   - name: {{.}}{{end}}
+output:
+  sql: {}
+  svg: {}