package gateway

import (
	"yunyan/lego/utils/crypto/md5"
	"yunyan/pb"
	"fmt"
	"net/http"
	"reflect"
	"strings"

	"github.com/gin-gonic/gin"
	"github.com/golang-jwt/jwt/v4"
)

var (
	httpReqTyoe, httpRespTyoe reflect.Type = reflect.TypeOf(&pb.Rpc_GatewayHttpRouteReq{}), reflect.TypeOf(&pb.Rpc_GatewayHttpRouteResp{})
)

// 判断接口名是否在白名单中
func isInWhiteList(msgName string, whiteList []string) bool {
	for _, name := range whiteList {
		if name == msgName {
			return true
		}
	}
	return false
}

// 设置跨域
func cors() gin.HandlerFunc {
	return func(c *gin.Context) {
		method := c.Request.Method
		origin := c.Request.Header.Get("Origin")
		if origin != "" {
			c.Header("Access-Control-Allow-Origin", origin) // 回显请求的 Origin，配合 Credentials 使用
			c.Header("Access-Control-Allow-Methods", "POST, GET, OPTIONS, PUT, DELETE, UPDATE")
			c.Header("Access-Control-Allow-Headers", "Origin, X-Requested-With, Content-Type, Accept, Authorization")
			c.Header("Access-Control-Expose-Headers", "Content-Length, Access-Control-Allow-Origin, Access-Control-Allow-Headers, Cache-Control, Content-Language, Content-Type")
			c.Header("Access-Control-Allow-Credentials", "true")
		}
		if method == "OPTIONS" {
			c.AbortWithStatus(http.StatusNoContent)
		}
		c.Next()
	}
}

func parseToken(tokenString string, secretKey []byte) (*jwt.RegisteredClaims, error) {
	parsedToken, err := jwt.ParseWithClaims(tokenString, &jwt.RegisteredClaims{}, func(token *jwt.Token) (interface{}, error) {
		// Ensure the token method conforms to "SigningMethodHMAC"
		if _, ok := token.Method.(*jwt.SigningMethodHMAC); !ok {
			return nil, fmt.Errorf("unexpected signing method: %v", token.Header["alg"])
		}
		return secretKey, nil
	})
	if err != nil {
		return nil, err
	}

	// Validate the token and extract the claims
	if claims, ok := parsedToken.Claims.(*jwt.RegisteredClaims); ok && parsedToken.Valid {
		return claims, nil
	} else {
		return nil, fmt.Errorf("invalid token")
	}
}

// 签名接口
func ParamSign(key string, ts string) (sign string) {
	builder := strings.Builder{}
	builder.WriteString("ts=" + ts)
	builder.WriteString("key=" + key)
	orgin := builder.String()
	// fmt.Println(orgin)
	sign = md5.MD5EncToLower(orgin)
	return
}