package stock

// CapacityUnitWeight 重量单位枚举
type CapacityUnitWeight string

const (
	CapacityUnitWeightKG  CapacityUnitWeight = "KG"  // 千克
	CapacityUnitWeightTon CapacityUnitWeight = "TON" // 吨
	CapacityUnitWeightG   CapacityUnitWeight = "G"   // 克
	CapacityUnitWeightMG  CapacityUnitWeight = "MG"  // 毫克
	CapacityUnitWeightLB  CapacityUnitWeight = "LB"  // 磅
	CapacityUnitWeightOZ  CapacityUnitWeight = "OZ"  // 盎司
)

// GetAllCapacityUnitWeights 获取所有重量单位
func GetAllCapacityUnitWeights() []CapacityUnitWeight {
	return []CapacityUnitWeight{
		CapacityUnitWeightKG,
		CapacityUnitWeightTon,
		CapacityUnitWeightG,
		CapacityUnitWeightMG,
		CapacityUnitWeightLB,
		CapacityUnitWeightOZ,
	}
}

type CapacityUnitWeightKeyValue struct {
	Key   CapacityUnitWeight
	Value string
}

var (
	CapacityUnitWeightKGKeyValue  = CapacityUnitWeightKeyValue{Key: CapacityUnitWeightKG, Value: "千克"}
	CapacityUnitWeightTonKeyValue = CapacityUnitWeightKeyValue{Key: CapacityUnitWeightTon, Value: "吨"}
	CapacityUnitWeightGKeyValue   = CapacityUnitWeightKeyValue{Key: CapacityUnitWeightG, Value: "克"}
	CapacityUnitWeightMGKeyValue  = CapacityUnitWeightKeyValue{Key: CapacityUnitWeightMG, Value: "毫克"}
	CapacityUnitWeightLBKeyValue  = CapacityUnitWeightKeyValue{Key: CapacityUnitWeightLB, Value: "磅"}
	CapacityUnitWeightOZKeyValue  = CapacityUnitWeightKeyValue{Key: CapacityUnitWeightOZ, Value: "盎司"}
)

func GetAllCapacityUnitWeightKeyValue() []CapacityUnitWeightKeyValue {
	return []CapacityUnitWeightKeyValue{
		CapacityUnitWeightKGKeyValue,
		CapacityUnitWeightTonKeyValue,
		CapacityUnitWeightGKeyValue,
		CapacityUnitWeightMGKeyValue,
		CapacityUnitWeightLBKeyValue,
		CapacityUnitWeightOZKeyValue,
	}
}

var capacityUnitWeightValueMap = map[CapacityUnitWeight]string{
	CapacityUnitWeightKG:  CapacityUnitWeightKGKeyValue.Value,
	CapacityUnitWeightTon: CapacityUnitWeightTonKeyValue.Value,
	CapacityUnitWeightG:   CapacityUnitWeightGKeyValue.Value,
	CapacityUnitWeightMG:  CapacityUnitWeightMGKeyValue.Value,
	CapacityUnitWeightLB:  CapacityUnitWeightLBKeyValue.Value,
	CapacityUnitWeightOZ:  CapacityUnitWeightOZKeyValue.Value,
}

func GetCapacityUnitWeightValueByKey(key CapacityUnitWeight) (value string) {
	value, exists := capacityUnitWeightValueMap[key]
	if !exists {
		value = "未知单位"
	}
	return
}