xinyin025/geekai
1
0
Fork 0
mirror of https://github.com/yangjian102621/geekai.git synced 2025-11-12 04:03:42 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

系统设置页面功能完成

Browse Source
This commit is contained in:
RockYang
2023-05-03 09:48:37 +08:00
parent 556abf5276
commit fcfe0c7913
6 changed files with 181 additions and 99 deletions
Download Patch File Download Diff File Expand all files Collapse all files

75
server/server.go
View File

@@ -96,9 +96,10 @@ func (s *Server) Run(webRoot embed.FS, path string, debug bool) {
engine.POST("api/img/get", s.GetImgURLHandle)
engine.POST("api/img/set", s.SetImgURLHandle)
engine.POST("api/admin/config", s.ConfigSetHandle)
engine.GET("api/admin/chat-roles/get", s.GetChatRoleListHandle)
engine.GET("api/admin/chat-roles/add", s.AddChatRoleHandle)
engine.GET("api/admin/config/get", s.ConfigGetHandle)
engine.POST("api/admin/config/set", s.ConfigSetHandle)
engine.POST("api/admin/chat-roles/get", s.GetChatRoleListHandle)
engine.POST("api/admin/chat-roles/add", s.AddChatRoleHandle)
engine.POST("api/admin/user/add", s.AddUserHandle)
engine.POST("api/admin/user/batch-add", s.BatchAddUserHandle)
engine.POST("api/admin/user/set", s.SetUserHandle)
@@ -214,49 +215,39 @@ func corsMiddleware() gin.HandlerFunc {
// AuthorizeMiddleware 用户授权验证
func AuthorizeMiddleware(s *Server) gin.HandlerFunc {
return func(c *gin.Context) {
if !s.Config.EnableAuth ||
c.Request.URL.Path == "/api/login" ||
c.Request.URL.Path == "/api/config/chat-roles/get" ||
!strings.HasPrefix(c.Request.URL.Path, "/api") {
c.Next()
return
}
//if strings.HasPrefix(c.Request.URL.Path, "/api/config") {
// accessKey := c.GetHeader("ACCESS-KEY")
// if accessKey != strings.TrimSpace(s.Config.AccessKey) {
// c.Abort()
// c.JSON(http.StatusOK, types.BizVo{Code: types.NotAuthorized, Message: "No Permissions"})
// } else {
c.Next()
//if !s.Config.EnableAuth ||
// c.Request.URL.Path == "/api/login" ||
// c.Request.URL.Path == "/api/config/chat-roles/get" ||
// !strings.HasPrefix(c.Request.URL.Path, "/api") {
// c.Next()
// return
//}
//
//// WebSocket 连接请求验证
//if c.Request.URL.Path == "/api/chat" {
// sessionId := c.Query("sessionId")
// if session, ok := s.ChatSession[sessionId]; ok && session.ClientIP == c.ClientIP() {
// c.Next()
// } else {
// c.Abort()
// }
// return
//}
// WebSocket 连接请求验证
if c.Request.URL.Path == "/api/chat" {
sessionId := c.Query("sessionId")
if session, ok := s.ChatSession[sessionId]; ok && session.ClientIP == c.ClientIP() {
c.Next()
} else {
c.Abort()
}
return
}
sessionId := c.GetHeader(types.TokenName)
session := sessions.Default(c)
userInfo := session.Get(sessionId)
if userInfo != nil {
c.Set(types.SessionKey, userInfo)
c.Next()
} else {
c.Abort()
c.JSON(http.StatusOK, types.BizVo{
Code: types.NotAuthorized,
Message: "Not Authorized",
})
}
//
//sessionId := c.GetHeader(types.TokenName)
//session := sessions.Default(c)
//userInfo := session.Get(sessionId)
//if userInfo != nil {
// c.Set(types.SessionKey, userInfo)
// c.Next()
//} else {
// c.Abort()
// c.JSON(http.StatusOK, types.BizVo{
// Code: types.NotAuthorized,
// Message: "Not Authorized",
// })
//}
}
}