package starrocks

import (
	"bytes"
	"fmt"
	"gitlab.fjmaimaimai.com/allied-creation/character-library-metadata-bastion/pkg/domain"
	"gitlab.fjmaimaimai.com/allied-creation/character-library-metadata-bastion/pkg/infrastructure/utils"
	"gorm.io/gorm"
	"html"
	"html/template"
)

func Insert(db *gorm.DB, tableName string, fields []*domain.FieldValue) error {
	value := map[string]interface{}{}
	for _, f := range fields {
		if f.Field.Flag == domain.PKField && f.Value == "" {
			//continue
			//id, _ := uuid.NewUUID()
			// idString := id.String()
			// 需要调用分布式id生成
			id, _ := utils.NewSnowflakeId()
			idString := fmt.Sprintf("%d", id)
			f.Value = idString
		}
		value[f.Field.SQLName] = f.TypeValue()
	}
	//tx := db.Table(tableName).Create(value)
	//return tx.Error
	sql := db.ToSQL(func(tx *gorm.DB) *gorm.DB {
		return tx.Table(tableName).Create(value)
	})
	tx := db.Exec(sql)
	return tx.Error
}

func Update(db *gorm.DB, tableName string, fields []*domain.FieldValue) error {
	value := map[string]interface{}{}
	var pk *domain.FieldValue
	for _, f := range fields {
		if f.Field.Flag == domain.PKField && f.Value != "" {
			pk = f
		}
		value[f.Field.SQLName] = f.TypeValue()
	}
	if pk == nil && pk.Value != "" && pk.TypeValue() != nil {
		return fmt.Errorf("主键不能为空")
	}
	//tx := db.Table(tableName).Where("? = ?", pk.Field.SQLName, pk.Value).Updates(value)
	//return tx.Error
	sql := db.ToSQL(func(tx *gorm.DB) *gorm.DB {
		return tx.Table(tableName).Create(value)
	})
	tx := db.Exec(sql)
	return tx.Error
}

func Delete(db *gorm.DB, tableName string, fields []*domain.FieldValue) error {
	var pk *domain.FieldValue
	for _, f := range fields {
		if f.Field.Flag == domain.PKField && f.Value != "" {
			pk = f
		}
	}
	if pk == nil {
		return fmt.Errorf("主键不能为空")
	}
	//tx := db.Table(tableName).Delete("? = ?", pk.Field.SQLName, pk.Value)
	//return tx.Error
	sql := db.ToSQL(func(tx *gorm.DB) *gorm.DB {
		return tx.Table(tableName).Delete("? = ?", pk.Field.SQLName, pk.Value)
	})
	tx := db.Exec(sql)
	return tx.Error
}

//DROP VIEW IF EXISTS

func DropView(db *gorm.DB, tableName string) error {
	tx := db.Exec("DROP VIEW IF EXISTS " + tableName)
	return tx.Error
}

func Exec(db *gorm.DB, sql string) error {
	tx := db.Exec(sql)
	return tx.Error
}

func CreateViewSql(viewName string, fields []*domain.Field, selectSql string) string {
	sql := `
DROP VIEW IF EXISTS {{.ViewName}};
create view {{.ViewName}}(
{{.Field}} 
) as(
{{.Sql}}
)
`
	tmp := template.New("ViewCreator")
	tmp.Parse(sql)

	buf := bytes.NewBuffer(nil)
	bufField := bytes.NewBuffer(nil)
	for i, f := range fields {
		bufField.WriteString(fmt.Sprintf(`%s  COMMENT "%v"`, f.SQLName, f.Name))
		if i != len(fields)-1 {
			bufField.WriteString(",\n")
		}
	}
	tmp.Execute(buf, map[string]interface{}{
		"ViewName": viewName,
		"Field":    bufField.String(),
		"Sql":      selectSql,
	})
	return html.UnescapeString(buf.String())
}

func DropTable(db *gorm.DB, tableName string) error {
	tx := db.Exec("DROP TABLE IF EXISTS " + tableName)
	return tx.Error
}

func CreateTableSql(viewName string, fields []*domain.Field, data []string) string {
	sql := `
## DROP VIEW IF EXISTS {{.ViewName}};
## DROP TABLE IF EXISTS {{.ViewName}};
Create TABLE {{.ViewName}} (
id BIGINT,
{{.Field}} 
) 
UNIQUE KEY(id)
DISTRIBUTED BY HASH(id) BUCKETS 3
PROPERTIES("replication_num"="1");

TRUNCATE TABLE {{.ViewName}};

insert into {{.ViewName}} Values {{.Values}}
`
	tmp := template.New("ViewCreator")
	tmp.Parse(sql)

	buf := bytes.NewBuffer(nil)
	bufField := bytes.NewBuffer(nil)
	for i, f := range fields {
		bufField.WriteString(fmt.Sprintf(`%s VARCHAR(65533)`, f.SQLName)) //, f.SqlTypeToDB()
		if i != len(fields)-1 {
			bufField.WriteString(",\n")
		}
	}

	bufInsert := bytes.NewBuffer(nil)
	for _, d := range data {
		id, _ := utils.NewSnowflakeId()
		bufInsert.WriteString(fmt.Sprintf("(%d,'%s')", id, d))
	}

	tmp.Execute(buf, map[string]interface{}{
		"ViewName": viewName,
		"Field":    bufField.String(),
		"Values":   bufInsert.String(),
	})
	return html.UnescapeString(buf.String())
}