Merge pull request #6 from alexisvisco/1-remove-driver-dependencies

1-remove-driver-dependencies
alexisvisco · May 26, 2024 · 0ee9811 · 0ee9811
2 parents 8790c85 + 9358526
commit 0ee9811
Show file tree

Hide file tree

Showing 4 changed files with 24 additions and 11 deletions.
diff --git a/example/pg/.amigo/config.yml b/example/pg/.amigo/config.yml
@@ -7,3 +7,4 @@ package: migrations
 schema-version-table: public.mig_schema_versions
 shell-path: /bin/bash
 sql: false
+sql-syntax-highlighting: true
diff --git a/pkg/amigo/execute_main.go b/pkg/amigo/execute_main.go
@@ -58,6 +58,10 @@ func (a Amigo) ExecuteMain(arg MainArg) error {
 		args = append(args, "-sql")
 	}
 
+	if a.ctx.ShowSQLSyntaxHighlighting {
+		args = append(args, "-sql-syntax-highlighting")
+	}
+
 	if a.ctx.Debug {
 		args = append(args, "-debug")
 	}

diff --git a/pkg/entrypoint/main.go b/pkg/entrypoint/main.go
@@ -58,6 +58,8 @@ func AmigoContextFromFlags() (*amigoctx.Context, amigo.MainArg) {
 	continueOnErrorFlag := flag.Bool("continue-on-error", false,
 		"Continue on error will not rollback the migration if an error occurs")
 	stepsFlag := flag.Int("steps", 1, "Number of steps to rollback")
+	showSQLSyntaxHighlightingFlag := flag.Bool("sql-syntax-highlighting", false,
+		"Print SQL statements with syntax highlighting")
 
 	// Parse flags
 	flag.Parse()
@@ -75,16 +77,17 @@ func AmigoContextFromFlags() (*amigoctx.Context, amigo.MainArg) {
 
 	a := &amigoctx.Context{
 		Root: &amigoctx.Root{
-			AmigoFolderPath:    "",
-			DSN:                *dsnFlag,
-			JSON:               *jsonFlag,
-			ShowSQL:            *showSQLFlag,
-			MigrationFolder:    "",
-			PackagePath:        "",
-			SchemaVersionTable: *schemaVersionTableFlag,
-			ShellPath:          "",
-			PGDumpPath:         "",
-			Debug:              *debugFlag,
+			AmigoFolderPath:           "",
+			DSN:                       *dsnFlag,
+			JSON:                      *jsonFlag,
+			ShowSQL:                   *showSQLFlag,
+			MigrationFolder:           "",
+			PackagePath:               "",
+			SchemaVersionTable:        *schemaVersionTableFlag,
+			ShellPath:                 "",
+			PGDumpPath:                "",
+			Debug:                     *debugFlag,
+			ShowSQLSyntaxHighlighting: *showSQLSyntaxHighlightingFlag,
 		},
 	}
 

diff --git a/pkg/types/types.go b/pkg/types/types.go
@@ -70,7 +70,12 @@ func (d Driver) PackageName() string {
 }
 
 func (d Driver) String() string {
-	return string(d)
+	switch d {
+	case DriverPostgres:
+		return "pgx"
+	default:
+		return ""
+	}
 }
 
 func (d Driver) IsValid() bool {