package db

import (
	"fmt"
	//"gorm.io/driver/postgres"
	"os"

	"gorm.io/driver/postgres"
	"gorm.io/driver/sqlite"
	"gorm.io/gorm"
	"zsxyww.com/scheduler/config"
	"zsxyww.com/scheduler/model"
)

var err error

func Connect() {
	switch config.Default.DB.Type {
	case "SQLite":
		connectSQLite()
	case "PostgreSQL":
		connectPGSQL()
		PGSQL()
	default:
		panic("DBType error")
	}
	//Main.AutoMigrate(&model.Member{}, &model.Tweak{})
}

func connectSQLite() {
	Main, err = gorm.Open(sqlite.Open(config.Default.DB.Path), &gorm.Config{})
	if err != nil {
		fmt.Printf("error in connecting to SQLite:")
		fmt.Println(err)
		os.Exit(1)
	}
	if config.InitDB == true {
		Main.AutoMigrate(&model.Member{}, &model.Tweak{})
	}
}

func connectPGSQL() {
	Main, err = gorm.Open(postgres.Open(config.Default.DB.Path), &gorm.Config{})
	if err != nil {
		panic(err)
	}
	if config.InitDB == true {
		Main.AutoMigrate(&model.Member{}, &model.Tweak{})
	}
}