mlogclub
c6c1c00141
- Implement role management endpoints including listing, creating, updating, and deleting roles. - Add session management endpoints for viewing and revoking sessions. - Create skill definition management endpoints for CRUD operations and status updates. - Introduce tag management endpoints for handling tags with sorting and status updates. - Develop ticket management endpoints for viewing, creating, updating, and managing ticket progress. - Implement user management endpoints for CRUD operations, role assignments, and password resets. - Add WeChat callback handlers for verifying and processing messages.
103 lines
2.5 KiB
Go
103 lines
2.5 KiB
Go
package api
|
|
|
|
import (
|
|
"cs-agent/internal/pkg/config"
|
|
"cs-agent/internal/pkg/dto/request"
|
|
"cs-agent/internal/pkg/httpx"
|
|
"cs-agent/internal/pkg/httpx/params"
|
|
"cs-agent/internal/services"
|
|
"net/http"
|
|
"net/url"
|
|
"strings"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
func Login(ctx *gin.Context) {
|
|
cfg := config.Current()
|
|
req := request.LoginRequest{}
|
|
if err := params.ReadJSON(ctx, &req); err != nil {
|
|
httpx.WriteJSON(ctx, err)
|
|
return
|
|
}
|
|
|
|
ret, err := services.AuthService.Login(req, cfg.Auth, ctx.ClientIP(), ctx.GetHeader("User-Agent"))
|
|
if err != nil {
|
|
httpx.WriteJSON(ctx, err)
|
|
return
|
|
}
|
|
httpx.WriteJSON(ctx, ret)
|
|
}
|
|
|
|
func WxWorkLogin(ctx *gin.Context) {
|
|
loginURL, err := services.WxWorkLoginService.BuildWxWorkLoginURL(ctx.Query("next"))
|
|
if err != nil {
|
|
ctx.Redirect(http.StatusFound, "/login?wxworkError="+url.QueryEscape(wxWorkErrorMessage(err.Error())))
|
|
return
|
|
}
|
|
ctx.Redirect(http.StatusFound, loginURL)
|
|
}
|
|
|
|
func WxWorkQRLogin(ctx *gin.Context) {
|
|
loginURL, err := services.WxWorkLoginService.BuildWxWorkQRCodeLoginURL(ctx.Query("next"))
|
|
if err != nil {
|
|
ctx.Redirect(http.StatusFound, "/login?wxworkError="+url.QueryEscape(wxWorkErrorMessage(err.Error())))
|
|
return
|
|
}
|
|
ctx.Redirect(http.StatusFound, loginURL)
|
|
}
|
|
|
|
func WxWorkCallback(ctx *gin.Context) {
|
|
cfg := config.Current()
|
|
ticket, next, err := services.WxWorkLoginService.LoginByWxWork(
|
|
ctx.Query("code"),
|
|
ctx.Query("state"),
|
|
cfg.Auth,
|
|
ctx.ClientIP(),
|
|
ctx.GetHeader("User-Agent"),
|
|
)
|
|
if err != nil {
|
|
ctx.Redirect(http.StatusFound, "/login?wxworkError="+url.QueryEscape(wxWorkErrorMessage(err.Error())))
|
|
return
|
|
}
|
|
ctx.Redirect(http.StatusFound, "/dashboard/login/wxwork/callback?ticket="+url.QueryEscape(ticket)+"&next="+url.QueryEscape(next))
|
|
}
|
|
|
|
func WxWorkExchange(ctx *gin.Context) {
|
|
req := request.WxWorkExchangeRequest{}
|
|
if err := params.ReadJSON(ctx, &req); err != nil {
|
|
httpx.WriteJSON(ctx, err)
|
|
return
|
|
}
|
|
ret, err := services.WxWorkLoginService.ExchangeWxWorkLoginTicket(req.Ticket)
|
|
if err != nil {
|
|
httpx.WriteJSON(ctx, err)
|
|
return
|
|
}
|
|
httpx.WriteJSON(ctx, ret)
|
|
}
|
|
|
|
func Logout(ctx *gin.Context) {
|
|
if err := services.AuthService.Logout(ctx.GetHeader("Authorization")); err != nil {
|
|
httpx.WriteJSON(ctx, err)
|
|
return
|
|
}
|
|
httpx.WriteJSON(ctx, nil)
|
|
}
|
|
|
|
func Profile(ctx *gin.Context) {
|
|
ret, err := services.AuthService.CurrentProfile(ctx)
|
|
if err != nil {
|
|
httpx.WriteJSON(ctx, err)
|
|
return
|
|
}
|
|
httpx.WriteJSON(ctx, ret)
|
|
}
|
|
|
|
func wxWorkErrorMessage(message string) string {
|
|
if idx := strings.Index(message, ": "); idx >= 0 {
|
|
message = message[idx+2:]
|
|
}
|
|
return message
|
|
}
|