type JWT struct {
    header    string
    Payload   string
    signature string
}

func encodeBase64(data string) string {
	return base64.RawURLEncoding.EncodeToString([]byte(data))
}

func generateSignature(key []byte, data []byte) (string, error) {
	// 創(chuàng)建一個(gè)哈希對(duì)象
	hash := hmac.New(sha256.New, key)
	// 將要簽名的信息寫(xiě)入哈希對(duì)象中 hash.Write(data)
	_, err := hash.Write(data)
	if err != nil {
		return "", err
	}
	// hash.Sum()計(jì)算簽名，在這里會(huì)返回簽名內(nèi)容
	// 將簽名經(jīng)過(guò)base64編碼生成字符串形式返回。
	return encodeBase64(string(hash.Sum(nil))), nil
}

func CreateToken(key []byte, payloadData any) (string, error) {
    // 標(biāo)準(zhǔn)頭部
	header := `{"alg":"HS256","typ":"JWT"}`
    // 將負(fù)載的數(shù)據(jù)轉(zhuǎn)換為json
	payload, jsonErr := json.Marshal(payloadData)
	if jsonErr != nil {
		return "", fmt.Errorf("負(fù)載json解析錯(cuò)誤")
	}
    // 將頭部和負(fù)載通過(guò)base64編碼，并使用.作為分隔進(jìn)行連接
	encodedHeader := encodeBase64(header)
	encodedPayload := encodeBase64(string(payload))
	HeaderAndPayload := encodedHeader + "." + encodedPayload
    // 使用簽名使用的key將傳入的頭部和負(fù)載連接所得的數(shù)據(jù)進(jìn)行簽名
	signature, err := generateSignature(key, []byte(HeaderAndPayload))
	if err != nil {
		return "", err
	}
    // 將token的三個(gè)部分使用.進(jìn)行連接并返回
	return HeaderAndPayload + "." + signature, nil
}

func ParseJwt(token string, key []byte) (*JWT, error) {
    // 分解規(guī)定，我們使用.進(jìn)行分隔，所以我們通過(guò).進(jìn)行分隔成三個(gè)字符串的數(shù)組
	jwtParts := strings.Split(token, ".")
    // 數(shù)據(jù)數(shù)組長(zhǎng)度不是3就說(shuō)明token在格式上就不合法
	if len(jwtParts) != 3 {
		return nil, fmt.Errorf("非法token")
	}
	// 分別拿出
	encodedHeader := jwtParts[0]
	encodedPayload := jwtParts[1]
	signature := jwtParts[2]
	// 使用key將token中的頭部和負(fù)載用.連接后進(jìn)行簽名
    // 這個(gè)簽名應(yīng)該個(gè)token中第三部分的簽名一致
	confirmSignature, err := generateSignature(key, []byte(encodedHeader+"."+encodedPayload))
	if err != nil {
		return nil, fmt.Errorf("生成簽名錯(cuò)誤")
	}
    // 如果不一致
	if signature != confirmSignature {
		return nil, fmt.Errorf("token驗(yàn)證失敗")
	}
	// 將payload解base64編碼
	dstPayload, _ := base64.RawURLEncoding.DecodeString(encodedPayload)
    // 返回我們的JWT對(duì)象以供后續(xù)使用
	return &JWT{encodedHeader, string(dstPayload), signature}, nil
}

type UserInfo struct {
	Name     string `json:"name"`
	Password string `json:"password"`
}

var Key []byte = []byte("12346")

func jwtConfirm(context *gin.Context) {
	// 登錄不需要token
	if context.Request.RequestURI == "/login" {
		return
	}
	// 拿出token
	token := context.GetHeader("Token")
    // 進(jìn)行解析驗(yàn)證
	jwt, err := utils.ParseJwt(token, Key)
	if err != nil {
		context.JSON(200, gin.H{
			"msg": err.Error(),
		})
        // 有問(wèn)題就流產(chǎn)掉（我也不知道怎么翻譯好了，香蕉貓.jpg）
		context.Abort()
	}
    // 驗(yàn)證通過(guò)就將負(fù)載返回回去
	context.JSON(200, gin.H{
		"payload": jwt.Payload,
	})
}

func main() {
    // 使用默認(rèn)路由
	router := gin.Default()
    // 注冊(cè)中間件
	router.Use(jwtConfirm)
    // 簡(jiǎn)單做兩個(gè)服務(wù)
	router.POST("/login", func(context *gin.Context) {
		// 接收用戶參數(shù)
		var userInfo UserInfo = UserInfo{}
        // 使用jsonbind接收
		bindErr := context.ShouldBindJSON(&userInfo)
		if bindErr != nil {
			context.JSON(200, gin.H{
				"msg": bindErr.Error(),
			})
		}
		// 使用密鑰做出token
		jwt, err := utils.CreateJwt(Key, userInfo)
		if err != nil {
			fmt.Println(err)
		}
         // 我們將token直接返回用于測(cè)試
		context.JSON(200, gin.H{
			"token": jwt,
		})
	})
	router.GET("/doing")
	router.Run()
}