package util

import (
	"bytes"
	"encoding/hex"
	"github.com/mozillazg/go-pinyin"
	"golang.org/x/text/encoding/simplifiedchinese"
	"golang.org/x/text/transform"
	"io/ioutil"
	"strconv"
	"strings"
	"unsafe"
)

// 获取中文字符串第一个首字母
func GetChineseFirstLetter(chinese string) string {

	// 获取中文字符串第一个字符
	firstChar := string([]rune(chinese)[:1])

	// Utf8 转 GBK2312
	firstCharGbk, err := Utf8ToGbk([]byte(firstChar))
	if err != nil {
		return ""
	}

	// 获取第一个字符的16进制
	firstCharHex := hex.EncodeToString(firstCharGbk)

	// 16进制转十进制
	firstCharDec, err := strconv.ParseInt(firstCharHex, 16, 0)
	if err != nil {
		return ""
	}

	// 十进制落在GB 2312的某个拼音区间即为某个字母
	firstCharDecimalRelative := firstCharDec - 65536
	if firstCharDecimalRelative >= -20319 && firstCharDecimalRelative <= -20284 {
		return "A"
	}
	if firstCharDecimalRelative >= -20283 && firstCharDecimalRelative <= -19776 {
		return "B"
	}
	if firstCharDecimalRelative >= -19775 && firstCharDecimalRelative <= -19219 {
		return "C"
	}
	if firstCharDecimalRelative >= -19218 && firstCharDecimalRelative <= -18711 {
		return "D"
	}
	if firstCharDecimalRelative >= -18710 && firstCharDecimalRelative <= -18527 {
		return "E"
	}
	if firstCharDecimalRelative >= -18526 && firstCharDecimalRelative <= -18240 {
		return "F"
	}
	if firstCharDecimalRelative >= -18239 && firstCharDecimalRelative <= -17923 {
		return "G"
	}
	if firstCharDecimalRelative >= -17922 && firstCharDecimalRelative <= -17418 {
		return "H"
	}
	if firstCharDecimalRelative >= -17417 && firstCharDecimalRelative <= -16475 {
		return "J"
	}
	if firstCharDecimalRelative >= -16474 && firstCharDecimalRelative <= -16213 {
		return "K"
	}
	if firstCharDecimalRelative >= -16212 && firstCharDecimalRelative <= -15641 {
		return "L"
	}
	if firstCharDecimalRelative >= -15640 && firstCharDecimalRelative <= -15166 {
		return "M"
	}
	if firstCharDecimalRelative >= -15165 && firstCharDecimalRelative <= -14923 {
		return "N"
	}
	if firstCharDecimalRelative >= -14922 && firstCharDecimalRelative <= -14915 {
		return "O"
	}
	if firstCharDecimalRelative >= -14914 && firstCharDecimalRelative <= -14631 {
		return "P"
	}
	if firstCharDecimalRelative >= -14630 && firstCharDecimalRelative <= -14150 {
		return "Q"
	}
	if firstCharDecimalRelative >= -14149 && firstCharDecimalRelative <= -14091 {
		return "R"
	}
	if firstCharDecimalRelative >= -14090 && firstCharDecimalRelative <= -13319 {
		return "S"
	}
	if firstCharDecimalRelative >= -13318 && firstCharDecimalRelative <= -12839 {
		return "T"
	}
	if firstCharDecimalRelative >= -12838 && firstCharDecimalRelative <= -12557 {
		return "W"
	}
	if firstCharDecimalRelative >= -12556 && firstCharDecimalRelative <= -11848 {
		return "X"
	}
	if firstCharDecimalRelative >= -11847 && firstCharDecimalRelative <= -11056 {
		return "Y"
	}
	if firstCharDecimalRelative >= -11055 && firstCharDecimalRelative <= -10247 {
		return "Z"
	}
	return ""
}

// Utf8ToGbk
func Utf8ToGbk(s []byte) ([]byte, error) {
	reader := transform.NewReader(bytes.NewReader(s), simplifiedchinese.GBK.NewEncoder())
	d, e := ioutil.ReadAll(reader)
	if e != nil {
		return nil, e
	}
	return d, nil
}

func CharacterUpper(character string) (ch string) {
	anyCharacter := "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ"
	if len(character) > 0 {
		if strings.ContainsAny(character, anyCharacter) {
			ch = character[0:1]
			ch = strings.ToUpper(ch)
			return
		}
	}
	return
}

func HanZi2UpperCapital(name string) string {
	a := pinyin.NewArgs()
	list := pinyin.Pinyin(name, a)
	upStrs := ""
	for i := 0; i < len(list); i++ {
		character := str2bytes(list[i][0])[0]
		str := string(character)
		upStrs += str
	}
	return strings.ToUpper(upStrs)
}

func str2bytes(s string) []byte {
	x := (*[2]uintptr)(unsafe.Pointer(&s))
	h := [3]uintptr{x[0], x[1], x[1]}
	return *(*[]byte)(unsafe.Pointer(&h))
}

func bytes2str(b []byte) string {
	return *(*string)(unsafe.Pointer(&b))
}