package agent

import (
	"fmt"
	"git.inspur.com/sbg-jszt/cfn/cfn-schedule/internal/model"
	"git.inspur.com/sbg-jszt/cfn/cfn-schedule/internal/pkg/utils"
	"gorm.io/gorm"
	"time"
)

const (
	OS_WINDOWS       = "Windows"
	OS_LINUX         = "Linux"
	OS_LINUX_LIST    = "Ubuntu,CentOS,KylinOS"
	OS_LINUX_UBUNTU  = "Ubuntu"
	OS_LINUX_CENTOS  = "CentOS"
	OS_LINUX_KylinOS = "KylinOS"
)

// 小助手信息
// 表名:perc_agent_info
type AgentInfo struct {
	Id              string    `gorm:"column:id;primaryKey" json:"id"`                       //type:string       comment:主键                                                                   version:2023-9-11 14:25
	Version         string    `gorm:"column:version" json:"version" validate:"required"`    //type:string       comment:版本                                                                   version:2023-9-11 14:25
	VersionNum      int64     `gorm:"column:version_num" json:"versionNum"`                 //type:int       comment:版本排序号
	OsPlatform      string    `gorm:"column:os_platform" json:"osPlatform"`                 //type:string       comment:操作系统平台;字典：Windows\Linux
	OsType          string    `gorm:"column:os_type" json:"osType" validate:"required"`     //type:string       comment:操作系统类型;字典：Windows\Ubuntu\CentOS\KylinOS   version:2023-9-11 14:25
	ArchType        string    `gorm:"column:arch_type" json:"archType" validate:"required"` //type:string       comment:芯片架构类型;字典: X86\ARM                                             version:2023-9-11 14:25
	PackageUrl      string    `gorm:"column:package_url" json:"packageUrl"`                 //type:string       comment:程序包地址;oss地址                                                     version:2023-9-11 14:25
	BucketName      string    `gorm:"column:bucket_name" json:"bucketName"`                 //type:string       comment:桶名称
	Filename        string    `gorm:"column:filename" json:"filename"`                      //type:string       comment:文件名称
	CompressionType string    `gorm:"column:compression_type" json:"compressionType"`       //type:string       comment:压缩格式
	Command         string    `gorm:"column:command" json:"command"`                        //type:string       comment:启动命令
	Describe        string    `gorm:"column:describe" json:"describe"`                      //type:string       comment:描述                                                                   version:2023-9-11 14:25
	CreateBy        string    `gorm:"column:create_by" json:"createBy"`                     //type:string       comment:创建人
	CreateTime      time.Time `gorm:"column:create_time" json:"createTime"`                 //type:*time.Time   comment:创建时间                                                               version:2023-9-11 14:25
	UpdateTime      time.Time `gorm:"column:update_time" json:"updateTime"`                 //type:*time.Time   comment:更新时间
}

func (AgentInfo) TableName() string {
	return "perc_agent_info"
}

func (a *AgentInfo) Create() (int64, error) {
	a.CreateTime = time.Now()
	a.UpdateTime = time.Now()
	a.Id = utils.NewAgentID()

	tx := model.DB().Create(a)

	if tx.Error != nil {
		return 0, tx.Error
	}
	return tx.RowsAffected, nil
}

func (a *AgentInfo) CreateBatch(agents []AgentInfo) (int64, error) {
	for i, _ := range agents {
		agents[i].CreateTime = time.Now()
		agents[i].UpdateTime = time.Now()
		agents[i].Id = utils.NewAgentID()
	}
	tx := model.DB().CreateInBatches(agents, len(agents))
	if tx.Error != nil {
		return 0, tx.Error
	}
	return tx.RowsAffected, nil
}

func (a *AgentInfo) Delete() (int64, error) {
	tx := model.DB().Delete(a)
	if tx.Error != nil {
		return 0, tx.Error
	}
	return tx.RowsAffected, nil
}

func (a *AgentInfo) Update(fields []string) (int64, error) {
	a.UpdateTime = time.Now()
	tx := model.DB().Model(&a).Select(fields).Updates(a)
	if tx.Error != nil {
		return 0, tx.Error
	}
	return tx.RowsAffected, nil
}

func (a *AgentInfo) Get(fields map[string]interface{}) (AgentInfo, error) {
	agent := AgentInfo{}
	tx := model.DB().Where(fields).First(&agent)
	if tx.Error != nil && tx.Error != gorm.ErrRecordNotFound {
		return agent, tx.Error
	}
	return agent, nil
}

func (a *AgentInfo) HasNewVersion(fields map[string]interface{}) (bool, error) {
	cout := int64(0)
	tx := model.DB().Model(a).Where("os_type=? and arch_type=? and version_num >=?", fields["os_type"], fields["arch_type"], fields["version_num"]).Count(&cout)
	if tx.Error != nil || cout > int64(0) {
		return true, tx.Error
	}
	return false, nil
}

// 获取最近的版本
func (a *AgentInfo) ListLastVersion() ([]AgentInfo, error) {
	agents := []AgentInfo{}
	tx := model.DB().Raw("SELECT * FROM perc_agent_info A INNER JOIN (SELECT os_type,arch_type,MAX (version_num) AS version_num FROM perc_agent_info GROUP BY os_type,arch_type) b ON A.arch_type=b.arch_type AND A.os_type=b.os_type AND A.version_num=b.version_num").Find(&agents)
	if tx.Error != nil {
		return nil, tx.Error
	}
	return agents, nil
}

func (a *AgentInfo) Page(page *model.Page[AgentInfo], fields map[string]interface{}) error {
	keyword := fields["keyword"]
	query := model.DB()
	if keyword != "" {
		keyword = fmt.Sprintf("%%%s%%", keyword)
		query = query.Where("concat(version,os_platform,os_type,arch_type,describe) like ?", keyword)
	}
	err := page.SelectPages(query)
	if err != nil {
		return err
	}
	return nil
}