package enums

import (
	"bytes"
	"fmt"
	"gitee.com/xlizy/common-go/base/models"
	"gitee.com/xlizy/common-go/utils/json"
	"reflect"
	"strconv"
)

type BaseValEnum struct {
	Enum string `json:"enum"`
	Val  int    `json:"val"`
	Des  string `json:"des"`
}

func SelectOptions[T any](enums []T) []models.SelectOptions {
	res := make([]models.SelectOptions, 0)
	for _, enumVal := range enums {
		b := reflect.ValueOf(enumVal).MethodByName("MarshalJSON").Call(nil)[0]
		be := BaseValEnum{}
		json.ToObj(fmt.Sprintf("%s", b), &be)
		res = append(res, models.SelectOptions{Value: be.Val, Desc: be.Des})
	}
	return res
}

func JsonObj(val int, enum string, des string) ([]byte, error) {
	var buf bytes.Buffer
	str := "{\"val\":" + strconv.Itoa(val) + ",\"enum\":\"" + enum + "\",\"des\":\"" + des + "\"}"
	//_, err := buf.WriteString(json.ToJsonStr(BaseValEnum{Val: val, Enum: enum, Des: des}))
	_, err := buf.WriteString(str)
	return buf.Bytes(), err
}

func BE(val any) BaseValEnum {
	b := reflect.ValueOf(val).MethodByName("MarshalJSON").Call(nil)[0]
	be := BaseValEnum{}
	json.ToObj(fmt.Sprintf("%s", b), &be)
	return be
}

func UnmarshalEnum(value []byte) int {
	valueStr := string(value)
	val, err := strconv.Atoi(valueStr)
	if err != nil {
		bve := BaseValEnum{}
		json.ToObj(string(value), &bve)
		return bve.Val
	} else {
		return val
	}
}