HI-VPN/hi-server
4
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update: Device Middleware

Browse Source
This commit is contained in:
EUForest 2025-10-14 15:08:59 +08:00
parent f67c2e34dc
commit 571628710b
1 changed files with 6 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
internal/middleware/deviceMiddleware.go
View File

@ -40,20 +40,18 @@ func DeviceMiddleware(srvCtx *svc.ServiceContext) func(c *gin.Context) {
return
}
var loginType string
if c.Value(constant.CtxKeyUser) == nil {
loginType = c.GetHeader("Login-Type")
} else {
loginType = c.GetString(string(constant.LoginType))
ctx := c.Request.Context()
if ctx.Value(constant.CtxKeyUser) == nil && c.GetHeader("Login-Type") != "" {
ctx = context.WithValue(ctx, constant.LoginType, c.GetHeader("Login-Type"))
c.Request = c.Request.WithContext(ctx)
}
if loginType != "device" {
loginType, ok := ctx.Value(constant.LoginType).(string)
if !ok || loginType != "device" {
c.Next()
return
}
c.Request = c.Request.WithContext(context.WithValue(c.Request.Context(), constant.LoginType, loginType))
rw := NewResponseWriter(c, srvCtx)
if !rw.Decrypt() {
result.HttpResult(c, nil, errors.Wrapf(xerr.NewErrCode(xerr.InvalidCiphertext), "Invalid ciphertext"))