Add migration, status and pending checks to database

cmd/cmd.go

@@ -16,6 +16,8 @@ func Parse() {
 		run(os.Args[2:])
 	case "migrate":
 		migrate(os.Args[2:])
+	case "status":
+		status(os.Args[2:])
 	case "-h":
 	case "--help":
 		fmt.Println("Available commands are:")

cmd/database.go

@@ -3,19 +3,15 @@ package cmd
 import (
 	"flag"
 	"fmt"
-	"log"
 	"os"
 
 	db "github.com/Fluffy-Bean/TastyBites/database"
-	sb "github.com/huandu/go-sqlbuilder"
 )
 
 func migrate(flags []string) {
-
 	cmd := flag.NewFlagSet("migrate", flag.ExitOnError)
 
-	tables := cmd.Bool("tables", false, "Create tables")
-	verbose := cmd.Bool("v", false, "Verbose")
+	downgrade := cmd.Bool("downgrade", false, "Downgrade Database")
 
 	err := cmd.Parse(flags)
 	if err != nil {
@@ -23,36 +19,17 @@ func migrate(flags []string) {
 		os.Exit(1)
 	}
 
-	if !*tables {
-		builder := db.ItemStruct.InsertInto("Item")
-		for _, item := range db.SampleData {
-			builder.Values(item.UUID, item.Name, item.Price, item.Description)
-		}
-		query, args := builder.Build()
-
-		_, err = db.Conn.Exec(query, args...)
-		if err != nil {
-			log.Fatal(err)
-		} else {
-			fmt.Println("Success")
-		}
-	} else {
-		itemTable := sb.CreateTable("Item").IfNotExists()
-		itemTable.Define("uuid", "TEXT", "NOT NULL", "PRIMARY KEY")
-		itemTable.Define("name", "TEXT", "NOT NULL")
-		itemTable.Define("price", "INTEGER", "NOT NULL")
-		itemTable.Define("description", "TEXT")
-		query, args := itemTable.BuildWithFlavor(sb.SQLite)
-
-		if *verbose {
-			fmt.Println(query, args)
-		}
-
-		_, err := db.Conn.Exec(query, args...)
-		if err != nil {
-			log.Fatal(err)
-		} else {
-			fmt.Println("Success")
-		}
+	err = db.Migrate(!*downgrade)
+	if err != nil {
+		fmt.Println(err)
+		os.Exit(1)
+	}
+}
+
+func status(flags []string) {
+	err := db.Status()
+	if err != nil {
+		fmt.Println(err)
+		os.Exit(1)
 	}
 }

cmd/run.go

@@ -6,12 +6,14 @@ import (
 	"os"
 
 	"github.com/Fluffy-Bean/TastyBites/api"
+	db "github.com/Fluffy-Bean/TastyBites/database"
 )
 
 func run(flags []string) {
 	cmd := flag.NewFlagSet("run", flag.ExitOnError)
 
 	host := cmd.String("host", "127.0.0.1:8080", "Host address, such as 0.0.0.0:8080")
+	skip := cmd.Bool("skipMigrations", false, "Skip any pending migrations")
 	logging := cmd.Bool("log", false, "Enable logging for the application")
 
 	err := cmd.Parse(flags)
@@ -20,6 +22,20 @@ func run(flags []string) {
 		os.Exit(1)
 	}
 
+	if !*skip {
+		pending, err := db.Pending()
+		if err != nil {
+			fmt.Println(err)
+			os.Exit(1)
+		}
+
+		if pending {
+			fmt.Println("Pending migrations!")
+			fmt.Println("Run `TastyBites status` or use the flag -skipMigrations")
+			os.Exit(1)
+		}
+	}
+
 	api.Serve(api.Config{
 		Host:    *host,
 		Logging: *logging,