package main import ( "bufio" "database/sql" "encoding/json" "fmt" "io/ioutil" "log" "os" "runtime/debug" "strconv" c "github.com/Azareal/Gosora/common" "github.com/Azareal/Gosora/query_gen" _ "github.com/go-sql-driver/mysql" ) var patches = make(map[int]func(*bufio.Scanner) error) func addPatch(index int, handle func(*bufio.Scanner) error) { patches[index] = handle } func main() { scanner := bufio.NewScanner(os.Stdin) // Capture panics instead of closing the window at a superhuman speed before the user can read the message on Windows defer func() { r := recover() if r != nil { fmt.Println(r) debug.PrintStack() pressAnyKey(scanner) log.Fatal("") return } }() log.Print("Loading the configuration data") err := c.LoadConfig() if err != nil { log.Fatal(err) } log.Print("Processing configuration data") err = c.ProcessConfig() if err != nil { log.Fatal(err) } if c.DbConfig.Adapter != "mysql" && c.DbConfig.Adapter != "" { log.Fatal("Only MySQL is supported for upgrades right now, please wait for a newer build of the patcher") } err = prepMySQL() if err != nil { log.Fatal(err) } err = patcher(scanner) if err != nil { log.Fatal(err) } } func pressAnyKey(scanner *bufio.Scanner) { fmt.Println("Please press enter to exit...") for scanner.Scan() { _ = scanner.Text() return } } func prepMySQL() error { return qgen.Builder.Init("mysql", map[string]string{ "host": c.DbConfig.Host, "port": c.DbConfig.Port, "name": c.DbConfig.Dbname, "username": c.DbConfig.Username, "password": c.DbConfig.Password, "collation": "utf8mb4_general_ci", }) } type SchemaFile struct { DBVersion string // Current version of the database schema DynamicFileVersion string MinGoVersion string // TODO: Minimum version of Go needed to install this version MinVersion string // TODO: Minimum version of Gosora to jump to this version, might be tricky as we don't store this in the schema file, maybe store it in the database } func loadSchema() (schemaFile SchemaFile, err error) { fmt.Println("Loading the schema file") data, err := ioutil.ReadFile("./schema/lastSchema.json") if err != nil { return schemaFile, err } err = json.Unmarshal(data, &schemaFile) return schemaFile, err } func patcher(scanner *bufio.Scanner) error { var dbVersion int err := qgen.NewAcc().Select("updates").Columns("dbVersion").QueryRow().Scan(&dbVersion) if err == sql.ErrNoRows { schemaFile, err := loadSchema() if err != nil { return err } dbVersion, err = strconv.Atoi(schemaFile.DBVersion) if err != nil { return err } } else if err != nil { return err } fmt.Println("Applying the patches") var pslice = make([]func(*bufio.Scanner) error, len(patches)) for i := 0; i < len(patches); i++ { pslice[i] = patches[i] } // Run the queued up patches var patched int for index, patch := range pslice { if dbVersion > index { continue } err := patch(scanner) if err != nil { return err } patched++ } if patched > 0 { _, err := qgen.NewAcc().Update("updates").Set("dbVersion = ?").Exec(len(pslice)) if err != nil { return err } } return nil }