package utils

import (
	"encoding/json"
	"fmt"
	"regexp"
	"strconv"
	"strings"
	"time"
	"unsafe"
)

var (
	idCertMatrix       = []int{7, 9, 10, 5, 8, 4, 2, 1, 6, 3, 7, 9, 10, 5, 8, 4, 2}
	idCertVerifyCode   = []byte("10X98765432")
	socialCreditMatrix = []int{1, 3, 9, 27, 19, 26, 16, 17, 20, 29, 25, 13, 8, 24, 10, 30, 28}
	socialCreditMap    = map[int]int{'0': 0, '1': 1, '2': 2, '3': 3, '4': 4, '5': 5, '6': 6, '7': 7, '8': 8, '9': 9,
		'A': 10, 'B': 11, 'C': 12, 'D': 13, 'E': 14, 'F': 15, 'G': 16, 'H': 17, 'J': 18,
		'K': 19, 'L': 20, 'M': 21, 'N': 22, 'P': 23, 'Q': 24, 'R': 25, 'T': 26, 'U': 27,
		'W': 28, 'X': 29, 'Y': 30}

	socialCreditMapKeys   = []int{'0', '1', '2', '3', '4', '5', '6', '7', '8', '9', 'A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'J', 'K', 'L', 'M', 'N', 'P', 'Q', 'R', 'T', 'U', 'W', 'X', 'Y'}
	socialCreditMapValues = []int{0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30}
)

//单位证件
func CheckSocialCode(str string) (res bool) {
	defer func() {
		if r := recover(); r != nil {
			res = false
		}
	}()

	socialCode := strings.ToUpper(str)

	if len(socialCode) != 18 {
		return false
	}

	check := 0
	lastLetter := 0
	for index, value := range []byte(socialCode) {
		if index == 17 {
			lastLetter = int(value)
			break
		}

		if value >= 48 && value <= 57 {
			//fmt.Println((int(value)-48),socialCreditMatrix[index])
			check = check + (int(value)-48)*socialCreditMatrix[index]
		} else {
			check = check + socialCreditMap[int(value)]*socialCreditMatrix[index]
		}
	}

	diff := 31 - check%31

	keys := make([]int, 0)
	values := make([]int, 0)

	for k, v := range socialCreditMap {
		keys = append(keys, k)
		values = append(values, v)
	}

	verifyCode := socialCreditMapKeys[socialCreditMapValues[diff]]

	if verifyCode == lastLetter {
		return true
	} else {
		return false
	}
}

//身份证
func CheckIDCert(str string) (res bool) {
	defer func() {
		if r := recover(); r != nil {
			res = false
		}
	}()

	idCert := strings.ToUpper(str)
	if len(idCert) != 18 {
		return false
	}

	year, err := strconv.Atoi(idCert[6:10])

	if err != nil {
		return false
	}

	if !(1900 < year && year < 2100) {
		return false
	}

	check := 0
	lastLetter := 0
	for index, value := range []byte(idCert) {
		if index == 17 {
			lastLetter = int(value)
			break
		}

		if !(value >= 48 && value <= 57) {
			return false
		}

		v := value - 48
		check = check + idCertMatrix[index]*int(v)
	}

	if !((lastLetter >= 48 && lastLetter <= 57) || lastLetter == 'X') {
		return false
	}

	timeLayout := "20060102"                                     //转化所需模板
	loc, _ := time.LoadLocation("Local")                         //重要：获取时区
	_, err = time.ParseInLocation(timeLayout, idCert[6:14], loc) //使用模板在对应时区转化为time.time类型

	if err != nil {
		return false
	}

	verifyCode := int(idCertVerifyCode[check%11])
	if lastLetter == verifyCode {
		return true
	}
	return false
}

// StrToBytes 快读的string转byte
func StrToBytes(s string) []byte {
	x := (*[2]uintptr)(unsafe.Pointer(&s))
	h := [3]uintptr{x[0], x[1], x[1]}
	return *(*[]byte)(unsafe.Pointer(&h))
}

func PrintStruct(prefix string, data interface{}) {
	bytes, _ := json.Marshal(data)
	fmt.Printf("%s, %s\n", prefix, bytes)
}

func VerifyEmailFormat(email string) bool {
	pattern := `\w+([-+.]\w+)*@\w+([-.]\w+)*\.\w+([-.]\w+)*` //匹配电子邮箱
	reg := regexp.MustCompile(pattern)
	return reg.MatchString(email)
}