package domain

import (
	"fmt"
	"time"
)

// 风险等级
const (
	RiskLevelHigh   = 1 // 1:高
	RiskLevelMiddle = 2 // 2:中
	RiskLevelLow    = 3 // 3:低
)

const (
	DeviceStatusNormal   = 1 // 1:正常
	DeviceStatusArchive  = 2 // 2:封存
	DeviceStatusScrapped = 3 // 3:报废
)

var (
	mapRiskLevel = map[string]int{
		"高": RiskLevelHigh,
		"中": RiskLevelMiddle,
		"低": RiskLevelLow,
	}
	mapDeviceStatus = map[string]int{
		"正常": DeviceStatusNormal,
		"封存": DeviceStatusArchive,
		"报废": DeviceStatusScrapped,
	}
)

// 设备
type Device struct {
	// 设备Id
	DeviceId int `json:"deviceId,omitempty"`
	// 企业id
	CompanyId int `json:"companyId,omitempty"`
	// 组织ID
	OrgId int `json:"orgId,omitempty"`
	// 设备编号
	DeviceCode string `json:"deviceCode,omitempty"`
	// 设备名称
	DeviceName string `json:"deviceName,omitempty"`
	// 设备型号
	DeviceModel string `json:"deviceModel,omitempty"`
	// 设备类型
	DeviceType string `json:"deviceType,omitempty"`
	// 品牌
	Brand string `json:"brand,omitempty"`
	// 设备状态 1:正常 2:封存 3:报废
	DeviceStatus int `json:"deviceStatus,omitempty"`
	// 风险等级 1:高 2:中 3:低
	RiskLevel int `json:"riskLevel,omitempty"`
	// 创建时间
	CreatedAt time.Time `json:"createdAt,omitempty"`
	// 更新时间
	UpdatedAt time.Time `json:"updatedAt,omitempty"`
	// 删除时间
	DeletedAt time.Time `json:"deletedAt,omitempty"`
	// 所属位置
	WorkStation *WorkStation `json:"workStation,omitempty"`
	// 扩展数据
	Ext *Ext `json:"ext,omitempty"`
}

type DeviceRepository interface {
	Save(device *Device) (*Device, error)
	Remove(device *Device) (*Device, error)
	FindOne(queryOptions map[string]interface{}) (*Device, error)
	Find(queryOptions map[string]interface{}) (int64, []*Device, error)
}

func (device *Device) Identify() interface{} {
	if device.DeviceId == 0 {
		return nil
	}
	return device.DeviceId
}

func (device *Device) Update(data map[string]interface{}) error {
	if deviceCode, ok := data["deviceCode"]; ok {
		device.DeviceCode = deviceCode.(string)
	}
	if deviceName, ok := data["deviceName"]; ok {
		device.DeviceName = deviceName.(string)
	}
	if deviceModel, ok := data["deviceModel"]; ok {
		device.DeviceModel = deviceModel.(string)
	}
	if deviceType, ok := data["deviceType"]; ok {
		device.DeviceType = deviceType.(string)
	}
	if brand, ok := data["brand"]; ok {
		device.Brand = brand.(string)
	}
	if deviceStatus, ok := data["deviceStatus"]; ok {
		device.DeviceStatus = deviceStatus.(int)
	}
	if riskLevel, ok := data["riskLevel"]; ok {
		device.RiskLevel = riskLevel.(int)
	}
	if device.Ext == nil {
		device.Ext = &Ext{}
	}
	if device.Ext != nil && device.Ext.DeviceExt == nil {
		device.Ext.DeviceExt = &DeviceExt{}
	}
	if orgName, ok := data["orgName"]; ok {
		device.Ext.OrgName = orgName.(string)
	}
	if unitProductionSecTime, ok := data["unitProductionSecTime"]; ok {
		device.Ext.DeviceExt.UnitProductionSecTime = unitProductionSecTime.(int)
	}
	device.UpdatedAt = time.Now()
	return nil
}

func (device *Device) Valid() error {
	if !(device.RiskLevel == RiskLevelHigh || device.RiskLevel == RiskLevelMiddle || device.RiskLevel == RiskLevelLow) {
		return fmt.Errorf("风险等级参数有误")
	}
	if !(device.DeviceStatus == DeviceStatusNormal || device.DeviceStatus == DeviceStatusArchive || device.DeviceStatus == DeviceStatusScrapped) {
		return fmt.Errorf("设备状态参数有误")
	}
	return nil
}

// 标记为生产设备
func (device *Device) MarkAsProductDevice() bool {
	if device.Ext.DeviceExt.IsProductDevice == 1 {
		return false
	}
	device.Ext.DeviceExt.IsProductDevice = 1
	device.UpdatedAt = time.Now()
	return true
}

// 导入设备数据体
type ImportDeviceItem struct {
	// 设备编号
	DeviceCode string `json:"deviceCode,omitempty"`
	// 设备名称
	DeviceName string `json:"deviceName,omitempty"`
	// 设备型号
	DeviceModel string `json:"deviceModel,omitempty"`
	// 设备类型
	DeviceType string `json:"deviceType,omitempty"`
	// 品牌
	Brand string `json:"brand,omitempty"`
	// 设备状态 1:正常 2:封存 3:报废
	Status string `json:"status,omitempty"`
	// 风险等级 1:高 2:中 3:低
	Level string `json:"level,omitempty"`
	// 失败理由
	FailReason string `json:"failReason"`

	// 设备状态 1:正常 2:封存 3:报废
	DeviceStatus int `json:"-"`
	// 风险等级 1:高 2:中 3:低
	RiskLevel int `json:"-"`
}

func (item *ImportDeviceItem) Valid() error {
	if len(item.DeviceCode) == 0 {
		return fmt.Errorf("设备编号不能为空")
	}
	if len(item.DeviceName) == 0 {
		return fmt.Errorf("设备名称不能为空")
	}
	if len(item.DeviceModel) == 0 {
		return fmt.Errorf("设备型号不能为空")
	}
	if len(item.DeviceType) == 0 {
		return fmt.Errorf("设备类型不能为空")
	}
	if len(item.Status) == 0 {
		return fmt.Errorf("设备状态不能为空")
	}
	if len(item.Level) == 0 {
		return fmt.Errorf("风险等级不能为空")
	}
	if v, ok := mapDeviceStatus[item.Status]; !ok {
		return fmt.Errorf("设备状态格式有误")
	} else {
		item.DeviceStatus = v
	}
	if v, ok := mapRiskLevel[item.Level]; !ok {
		return fmt.Errorf("风险等级格式有误")
	} else {
		item.RiskLevel = v
	}
	return nil
}

type Devices []*Device

func (devices Devices) ToMap() map[string]*Device {
	var resp = make(map[string]*Device)
	for _, v := range devices {
		resp[v.DeviceCode] = v
	}
	return resp
}