package starrocks

import (
	"fmt"
	"github.com/google/uuid"
	"gitlab.fjmaimaimai.com/allied-creation/character-library-metadata-bastion/pkg/domain"
	"gorm.io/gorm"
)

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()
			f.Value = id.String()
		}
		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
}