TastyBites/cmd/database.go

package cmd

import (
	"flag"
	"fmt"
	"os"

	db "github.com/Fluffy-Bean/TastyBites/database"
)

func migrate(flags []string) {
	cmd := flag.NewFlagSet("migrate", flag.ExitOnError)

	confirm := cmd.Bool("y", false, "Skip questioning")

	err := cmd.Parse(flags)
	if err != nil {
		fmt.Println(err)
		os.Exit(1)
	}

	err = db.Migrate(*confirm)
	if err != nil {
		fmt.Println(err)
		os.Exit(1)
	}
}

func downgrade(flags []string) {
	cmd := flag.NewFlagSet("migrate", flag.ExitOnError)

	confirm := cmd.Bool("y", false, "Skip questioning")

	err := cmd.Parse(flags)
	if err != nil {
		fmt.Println(err)
		os.Exit(1)
	}

	err = db.Downgrade(*confirm)
	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)
	}
}