wip

sqlgen/main.go

@@ -5,14 +5,19 @@ import (
 	"os"
 )
 
+const (
+	DRIVER_SQLITE   = "sqlite"
+	DRIVER_POSTGRES = "postgres"
+)
+
 func Main() {
 	usage := func() {
 		fmt.Fprintf(os.Stderr, `
 %s DRIVER DEFS_PATH OUTPUT_PATH
 
-Drivers are one of: sqlite, postgres
+Drivers are one of: %s %s
 `,
-			os.Args[0])
+			os.Args[0], DRIVER_SQLITE, DRIVER_POSTGRES)
 		os.Exit(1)
 	}
 
@@ -21,21 +26,20 @@ Drivers are one of: sqlite, postgres
 	}
 
 	var (
-		template   string
 		driver     = os.Args[1]
 		defsPath   = os.Args[2]
 		outputPath = os.Args[3]
 	)
 
 	switch driver {
-	case "sqlite":
-		template = sqliteTemplate
+	case DRIVER_SQLITE, DRIVER_POSTGRES:
+		// OK
 	default:
 		fmt.Fprintf(os.Stderr, "Unknown driver: %s", driver)
 		usage()
 	}
 
-	err := render(template, defsPath, outputPath)
+	err := render(driver, defsPath, outputPath)
 	if err != nil {
 		fmt.Fprintf(os.Stderr, "Error: %v", err)
 		os.Exit(1)