Translate Chinese comments, log/error messages, validator labels and Swagger annotations to English throughout the source code, generated Swagger docs, config files and CI workflows. Make the English README primary: README.md now holds the English docs and README_EN.md holds the Chinese version, with cross-language links updated accordingly. Note: the generated docs/ swagger files were translated in place; run `go generate ./...` (swag init) to regenerate them from the now-English annotations when a Go toolchain is available. Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
70 lines
1.3 KiB
Go
70 lines
1.3 KiB
Go
package middleware
|
|
|
|
import (
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/lejianwen/rustdesk-api/v2/global"
|
|
"github.com/lejianwen/rustdesk-api/v2/service"
|
|
)
|
|
|
|
func RustAuth() gin.HandlerFunc {
|
|
return func(c *gin.Context) {
|
|
//fmt.Println(c.Request.URL, c.Request.Header)
|
|
//getHTTP_AUTHORIZATION
|
|
token := c.GetHeader("Authorization")
|
|
if token == "" {
|
|
c.JSON(401, gin.H{
|
|
"error": "Unauthorized",
|
|
})
|
|
c.Abort()
|
|
return
|
|
}
|
|
if len(token) <= 7 {
|
|
c.JSON(401, gin.H{
|
|
"error": "Unauthorized",
|
|
})
|
|
c.Abort()
|
|
return
|
|
}
|
|
//extracttoken, format is Bearer {token}
|
|
//this is just a simple extraction
|
|
token = token[7:]
|
|
|
|
// verify the token
|
|
|
|
//check whether it is set jwt key
|
|
if len(global.Jwt.Key) > 0 {
|
|
uid, _ := service.AllService.UserService.VerifyJWT(token)
|
|
if uid == 0 {
|
|
c.JSON(401, gin.H{
|
|
"error": "Unauthorized",
|
|
})
|
|
c.Abort()
|
|
return
|
|
}
|
|
}
|
|
|
|
user, ut := service.AllService.UserService.InfoByAccessToken(token)
|
|
if user.Id == 0 {
|
|
c.JSON(401, gin.H{
|
|
"error": "Unauthorized",
|
|
})
|
|
c.Abort()
|
|
return
|
|
}
|
|
if !service.AllService.UserService.CheckUserEnable(user) {
|
|
c.JSON(401, gin.H{
|
|
"error": "Unauthorized",
|
|
})
|
|
c.Abort()
|
|
return
|
|
}
|
|
|
|
c.Set("curUser", user)
|
|
c.Set("token", token)
|
|
|
|
service.AllService.UserService.AutoRefreshAccessToken(ut)
|
|
|
|
c.Next()
|
|
}
|
|
}
|