package util

import (
	"strconv"

	"github.com/liangdas/mqant/log"
)

var (
	ShareCodeMap = map[string]string{
		"1": "K",
		"2": "C",
		"3": "R",
		"4": "T",
		"5": "U",
		"6": "W",
		"7": "B",
		"8": "E",
		"9": "A",
		"0": "F",
	}
	ShareUIDMap = map[string]string{
		"K": "1",
		"C": "2",
		"R": "3",
		"T": "4",
		"U": "5",
		"W": "6",
		"B": "7",
		"E": "8",
		"A": "9",
		"F": "0",
	}
)

func GetShareCode(uid int) string {
	code := ""
	str := strconv.Itoa(uid)
	for _, v := range str {
		code += ShareCodeMap[string(v)]
	}
	return code
}

func GetShareUID(code string) int {
	str := ""
	for _, v := range code {
		str += ShareUIDMap[string(v)]
	}
	ret, err := strconv.Atoi(str)
	if err != nil {
		log.Error("%v", err)
		return 0
	}
	return ret
}