ark-shop-tools/router/Router.go

package router

import (
	"ARKItems/configs"
	"ARKItems/entity/ark"
	"ARKItems/util"
	"github.com/gin-gonic/gin"
	"github.com/spf13/cast"
)

var ARKConfig ark.Config

func LoadRouter(router *gin.Engine) {
	router.POST("/login", Login)
	router.Use(Calibration())
	ARKConfigRouter(router.Group("/ark"))
}

// Calibration 登录校验
func Calibration() gin.HandlerFunc {
	return func(c *gin.Context) {
		header := c.GetHeader("auth-sign")
		if header == "" {
			c.JSON(200, CreateResultError(500, "用户未登录"))
			c.Abort()
		}
		token, err := util.ParseToken(header)
		if err != nil {
			c.JSON(200, CreateResultError(500, "token解析失败"))
			c.Abort()
			return
		}

		if !util.JudgmentUserInfo(token) {
			c.JSON(200, CreateResultError(500, "token过期，或无效"))
			c.Abort()
			return
		}
	}
}

func Login(c *gin.Context) {
	param := util.GetJsonAnyParam(c)
	username, _ := param("username")
	password, _ := param("password")
	if username == configs.Manager.Username && password == configs.Manager.Password {
		token, err := util.GenerateToken(cast.ToString(username))
		if err != nil {
			c.JSON(200, CreateResultError(500, "token生成失败"))
			return
		}
		c.JSON(200, CreateResultData(gin.H{
			"token": token,
		}))
		return
	}
	c.JSON(200, CreateResultError(500, "用户名或密码错误"))
}

//===============================================================

func CreateResult() gin.H {
	return gin.H{
		"code": 200,
		"msg":  "success",
	}
}
func CreateResultData(Data any) gin.H {
	return gin.H{
		"code": 200,
		"msg":  "success",
		"data": Data,
	}
}

func CreateResultError(errCode int, errMsg string) gin.H {
	return gin.H{
		"code": errCode,
		"msg":  errMsg,
	}
}