package datacenter

import (
	"appPlatform/overall/publicmethod"
	"encoding/json"
	"errors"
	"fmt"
	"strings"

	"gorm.io/driver/mysql"
	"gorm.io/driver/postgres"
	"gorm.io/driver/sqlite"
	"gorm.io/driver/sqlserver"
	"gorm.io/gorm"
)

/*
*
@ 作者: 秦东
@ 时间: 2024-12-24 11:28:46
@ 功能: 建立SqlServer
@ 参数

	#

@ 返回值

	#

@ 方法原型

	#
*/
func (c *DatabaseConfig) ConnectToSqlServer() (*gorm.DB, error) {
	// connStr := "provider=SQLOLEDB;data source=你的服务器名;initial catalog=你的数据库名;user id=你的用户名;password=你的密码;"
	// db, err := sql.Open("adodb", connStr)
	// connStr := fmt.Sprintf("server=%v;user id=<%v>;password=<%v>;port=<%v>;database=<%v>;", c.Host,
	// 	c.User, c.Password, c.Port, c.Name)
	// db, err := sql.Open("mssql", connStr)
	dsn := fmt.Sprintf("sqlserver://gorm:LoremIpsum86@%v:%v?database=%v", c.Host, c.Port, c.Name)
	db, err := gorm.Open(sqlserver.Open(dsn), &gorm.Config{})
	if err != nil {
		return nil, err
	}
	return db, nil
}

/*
*
@ 作者: 秦东
@ 时间: 2024-12-24 09:59:12
@ 功能: 建立PostgreSQL
@ 参数

	#

@ 返回值

	#

@ 方法原型

	#
*/
func (c *DatabaseConfig) ConnectToPostgreSQL() (*gorm.DB, error) {
	connStr := fmt.Sprintf("user=%s password=%s dbname=%s host=%s port=%d sslmode=disable",
		c.User, c.Password, c.Name, c.Host, c.Port)
	// db, err := sql.Open("postgres", connStr)
	db, err := gorm.Open(postgres.Open(connStr), &gorm.Config{})
	if err != nil {
		return nil, err
	}
	return db, nil
}

/*
*
@ 作者: 秦东
@ 时间: 2024-12-24 09:59:12
@ 功能: 建立MySQL
@ 参数

	#

@ 返回值

	#

@ 方法原型

	#
*/
func (c *DatabaseConfig) ConnectToMySQL() (*gorm.DB, error) {
	connStr := fmt.Sprintf("%v:%v@tcp(%v:%v)/%v?charset=utf8mb4", c.User, c.Password, c.Host, c.Port, c.Name)
	// db, err := sql.Open("mysql", connStr)
	fmt.Printf("sqlOpen: %v\n", connStr)

	sqlConfig := mysql.Config{
		DSN:                       connStr, // DSN
		DefaultStringSize:         255,     // string 类型字段的默认长度
		DisableDatetimePrecision:  true,    // 禁用 datetime 精度，MySQL 5.6 之前的数据库不支持
		DontSupportRenameIndex:    true,    // 重命名索引时采用删除并新建的方式，MySQL 5.7 之前的数据库和 MariaDB 不支持重命名索引
		DontSupportRenameColumn:   true,    // 用 `change` 重命名列，MySQL 8 之前的数据库和 MariaDB 不支持重命名列
		SkipInitializeWithVersion: false,   // 根据版本自动配置
	}

	db, err := gorm.Open(mysql.New(sqlConfig))
	if err != nil {
		return nil, err
	}

	return db, nil
}

/*
*
@ 作者: 秦东
@ 时间: 2024-12-24 09:59:12
@ 功能: 建立SQLite
@ 参数

	#

@ 返回值

	#

@ 方法原型

	#
*/
func (c *DatabaseConfig) ConnectToSQLite() (*gorm.DB, error) {
	// db, err := sql.Open("sqlite3", c.Name)
	db, err := gorm.Open(sqlite.Open("gorm.db"), &gorm.Config{})
	if err != nil {
		return nil, err
	}
	return db, nil
}

/*
*
@ 作者: 秦东
@ 时间: 2024-12-24 16:25:17
@ 功能: Tidb
@ 参数

	#

@ 返回值

	#

@ 方法原型

	#
*/
func (c *DatabaseConfig) ConnectToTiDb() (*gorm.DB, error) {
	sql := fmt.Sprintf("%v:%v@tcp(%v:%v)/%v", c.User, c.Password, c.Host, c.Port, c.Name)
	db, err := gorm.Open(mysql.Open(sql), &gorm.Config{})
	if err != nil {
		return nil, err
	}
	return db, nil
}

/*
*
@ 作者: 秦东
@ 时间: 2024-12-24 10:04:00
@ 功能: 启动数据库
@ 参数

	#

@ 返回值

	#

@ 方法原型

	#
*/
func (d *DataBastType) StartDataBast() (openDb *gorm.DB, err error) {
	var configInfo DatabaseConfig

	configInfo.Name = d.DataBaseName
	configInfo.Host = d.Ip
	configInfo.Port = d.Port
	configInfo.User = d.UserName
	configInfo.Password = d.Pwd
	switch strings.ToLower(d.Type) {
	case "PostgreSQL", "postgresql", "postgre sql":
		openDb, err = configInfo.ConnectToPostgreSQL()
	case "MySQL", "mysql":
		openDb, err = configInfo.ConnectToMySQL()
	case "SQLite", "sqlite":
		openDb, err = configInfo.ConnectToSQLite()
	case "SqlServer", "sql server", "sqlserver":
		openDb, err = configInfo.ConnectToSqlServer()
	case "Tidb", "tidb", "TiDb", "TIDB":
		openDb, err = configInfo.ConnectToMySQL()
	default:
		msg := fmt.Sprintf("对不起！当前不支持'%v'数据库。", d.DataBaseName)
		err = errors.New(msg)
	}
	return
}

/*
*
@ 作者: 秦东
@ 时间: 2024-12-24 13:07:56
@ 功能: 获取数据库中说有表
@ 参数

	#

@ 返回值

	#

@ 方法原型

	#获取数据库表comment
*/
func (d *DataBastType) ObtainDataBaseAllTableMysql() (tableKeyValue []DataBaseTableInfo, err error) {
	// tableKeyValue := publicmethod.MapOut[string]()
	sqlDb, err := d.StartDataBast()
	if err != nil {
		return
	}
	var valData []string
	// // sql := fmt.Sprintf("SHOW  %v", d.DataBaseName)
	sqlDb.Raw("SHOW TABLES").Scan(&valData)
	// if err != nil {
	// 	return tableKeyValue, err
	// }
	// fmt.Println(rows)
	fmt.Printf("%T----->%v\n", valData, valData)
	// var
	for _, v := range valData {
		fmt.Printf("%T----->%v\n", v, v)
		// sqlStr := fmt.Sprintf("SHOW  `%v`", v)

		// sql := "SELECT TABLE_NAME AS `Table`, TABLE_COMMENT AS `Table Comment` FROM INFORMATION_SCHEMA.TABLES WHERE TABLE_SCHEMA = ?"

		// var list []Result
		// err = sqlDb.Raw(sqlStr).Scan(&list).Error
		// tableKeyValue[v] = list
		var tableName string
		sql := fmt.Sprintf("SELECT TABLE_COMMENT  FROM INFORMATION_SCHEMA.TABLES WHERE TABLE_SCHEMA = \"%v\" AND TABLE_NAME  = \"%v\";", d.DataBaseName, v)
		err = sqlDb.Raw(sql).Scan(&tableName).Error
		// tableKeyValue[v] = tableInfo
		var tableInfo DataBaseTableInfo
		tableInfo.TableName = tableName
		tableInfo.TableKey = v
		sqlStr := fmt.Sprintf("SHOW FULL COLUMNS FROM `%v`", v)
		err = sqlDb.Raw(sqlStr).Scan(&tableInfo.Fields).Error
		tableKeyValue = append(tableKeyValue, tableInfo)
	}
	// defer rows.Close()
	// var tableName []string
	// for rows.Next() {
	// 	var table string
	// 	err = rows.Scan(&table)
	// 	if err == nil {
	// 		tableName = append(tableName, table)
	// 	}
	// }
	// if len(tableName) > 0 {

	// rows, err = sqlDb.Query("SHOW TABLE_COMMENT FROM INFORMATION_SCHEMA.TABLES WHERE TABLE_SCHEMA  IS NOT NULL")
	// for _,v := range tableName{
	// 	sql := fmt.Sprintf("SHOW TABLE_COMMENT FROM INFORMATION_SCHEMA.TABLES WHERE TABLE_SCHEMA = %v AND TABLE_NAME  = %v",d.DataBaseName,v)
	// 	rows, err = sqlDb.Query("SHOW TABLE_COMMENT FROM INFORMATION_SCHEMA.TABLES WHERE TABLE_SCHEMA  IS NOT NULL")
	// }
	// if err != nil {
	// 	return tableKeyValue, err
	// }
	// defer rows.Close()

	// var tableComments = make(map[string]string)
	// for rows.Next() {
	// 	var table, comment string
	// 	if err := rows.Scan(&table, &comment); err != nil {
	// 		// log.Fatal(err)
	// 		fmt.Printf("rows----err---->:%v\n", err)
	// 	}
	// 	tableComments[table] = comment
	// }
	// fmt.Printf("rows-------->:%v\n", tableComments)
	// }

	return tableKeyValue, err
}

/*
*
@ 作者: 秦东
@ 时间: 2024-12-30 16:49:14
@ 功能: 获取数据源
@ 参数

	#

@ 返回值

	#

@ 方法原型

	#
*/
func GainDataStorce(dataId string) (openDb *gorm.DB, err error) {
	sendUrl := fmt.Sprintf("http://120.224.6.6:29911/prod-api/database/app/datasource/page?pageNum=1&pageSize=%v&databaseName=&dataType=", 100000)
	// fmt.Printf("1获取数据源===>%v\n", sendUrl)
	getData := publicmethod.CurlGet(sendUrl)
	// fmt.Printf("2获取数据源===>%v\n", getData)
	var sendMap map[string]interface{}
	json.Unmarshal(getData, &sendMap)
	var sqlDb DataBastType
	// fmt.Printf("3获取数据源===>%v\n", sendMap)
	if code, ok := sendMap["code"]; ok {
		// fmt.Printf("4获取数据源===>%v\n", code)

		codeInt, _ := publicmethod.StringToInt64(code)
		if codeInt == 200 {
			// jsonval, _ := json.Marshal(sendMap)
			// fmt.Printf("\n\n\n6获取数据源===>%v\n\n\n", string(jsonval))
			if dataval, isOk := sendMap["data"]; isOk {
				if dataMap, ok := dataval.(map[string]interface{}); ok {
					if records, isOk := dataMap["records"]; isOk {
						if recordAry, isOk := records.([]interface{}); isOk {
							for _, v := range recordAry {
								if vMAp, ok := v.(map[string]interface{}); ok {
									// fmt.Printf("5获取数据源===>%T\n===>%v\n\n\n", vMAp, vMAp["id"])
									idStr := publicmethod.TypeToInterface(vMAp["id"])
									if idStr == dataId {
										typeVal, _ := publicmethod.StringToInt64(vMAp["datasourceType"])
										sqlDb.Type = publicmethod.DatabaseIntToString(typeVal)
										sqlDb.Ip = publicmethod.TypeToInterface(vMAp["ipAddress"])
										sqlDb.DataBaseName = publicmethod.TypeToInterface(vMAp["databaseName"])
										sqlDb.Port, _ = publicmethod.StringToInt64(vMAp["port"])
										sqlDb.UserName = publicmethod.TypeToInterface(vMAp["account"])
										sqlDb.Pwd = publicmethod.TypeToInterface(vMAp["password"])
									}

								}
								// fmt.Printf("5获取数据源===>%T\n===>%v\n\n\n", v, v)
							}
						}
					}
				}

			}
		}
	}
	// fmt.Printf("7获取数据源===>%T\n===>%v\n\n\n", sqlDb, sqlDb)
	openDb, err = sqlDb.StartDataBast()
	return
}