package dbs

import (
	"fmt"
	"github.com/spf13/viper"
	"gorm.io/driver/mysql"
	"gorm.io/gen"
	"gorm.io/gorm"
)

var isInit = false
var mysqlUser string
var mysqlPassword string
var mysqlUrl string
var mysqlDatabase string

func GetGormDB() (db *gorm.DB) {
	var MySQLDSN string

	if !isInit {
		viper.SetConfigName("database")
		viper.SetConfigType("yaml")
		viper.AddConfigPath("./")

		err := viper.ReadInConfig()
		if err != nil {
			panic(fmt.Errorf("fatal error config file: %w", err))
		}
		mysqlUser = viper.GetString("mysql.username")
		mysqlPassword = viper.GetString("mysql.password")
		mysqlUrl = viper.GetString("mysql.url")
		mysqlDatabase = viper.GetString("mysql.databasename")
		isInit = true
	}

	if mysqlPassword == "" || len(mysqlPassword) == 0 {
		MySQLDSN = fmt.Sprintf("%s@tcp(%s)/%s?charset=utf8mb4&parseTime=True&loc=Local",
			mysqlUser, mysqlUrl, mysqlDatabase)
	} else {
		MySQLDSN = fmt.Sprintf("%s:%s@tcp(%s)/%s?charset=utf8mb4&parseTime=True&loc=Local",
			mysqlUser, mysqlPassword, mysqlUrl, mysqlDatabase)
	}

	db, err2 := gorm.Open(mysql.Open(MySQLDSN))
	if err2 != nil {
		panic(fmt.Errorf("cannot establish db connection: %w", err2))
	}

	return
}

func GenerateModel() {
	g := gen.NewGenerator(gen.Config{
		OutPath: "../../internal/query",
	})

	g.UseDB(GetGormDB())
	g.ApplyBasic(
		g.GenerateModel("cameras"),
		g.GenerateModelAs("person_detection_record", "PersonDetectionRecord"),
		g.GenerateModelAs("trajectory_detection_record", "TrajectoryDetectionRecord"),
		g.GenerateModelAs("person", "Person"))
	g.Execute()
}

func insertlocal() {
	//Create_local_component("hadoop", "3.0.0", "../../shell/init/hadoop.sh", "node1")
	//Create_local_component("spark", "3.3.0", "../../shell/init/spark.sh", "node1")
	//Create_local_component("hbase", "2.0.0", "../../shell/init/hbase.sh", "node1")
	//Create_local_component("zookeeper", "3.7.1", "../../shell/init/zookeeper.sh", "node1")
	//Create_local_component("hive", "3.1.2", "../../shell/init/hive.sh", "node1")
	//Create_local_component("flink", "1.16.2", "../../shell/init/flink.sh", "node1")
	//Create_local_component("streampark", "2.0.0", "../../shell/init/streampark.sh", "node1")
	//Create_local_component("zeppelin", "0.10.1", "../../shell/init/zeppelin.sh", "node1")
}

func insertK8s() {
	//Create_k8s_component("hadoop", "hadoop:3.0.0")
	//Create_k8s_component("spark", "spark:3.3.0")
	//Create_k8s_component("hbase", "hadpp:2.0.0")
	//Create_k8s_component("zookeeper", "zookeeper:3.7.1")
	//Create_k8s_component("hive", "hadpp:3.1.2")
	//Create_k8s_component("flink", "flink:1.16.2")
	//Create_k8s_component("streampark", "streampark:2.0.0")
	//Create_k8s_component("zeppelin", "zeppelin:0.10.1")
}

func insertlocalStatus() {
	//created := "created"
	//node := "node1"
	//Create_local_status("hadoop", created, node)
	//Create_local_status("spark", created, node)
	//Create_local_status("hbase", created, node)
	//Create_local_status("zookeeper", created, node)
	//Create_local_status("hive", created, node)
	//Create_local_status("flink", created, node)
	//Create_local_status("streampark", created, node)
	//Create_local_status("zeppelin", created, node)

}