mirror of
https://github.com/komari-monitor/komari.git
synced 2025-11-06 15:03:23 +00:00
111 lines
3.3 KiB
Go
111 lines
3.3 KiB
Go
package api
|
||
|
||
import (
|
||
"fmt"
|
||
|
||
"github.com/gin-gonic/gin"
|
||
"github.com/komari-monitor/komari/database/accounts"
|
||
"github.com/komari-monitor/komari/database/auditlog"
|
||
"github.com/komari-monitor/komari/database/config"
|
||
"github.com/komari-monitor/komari/utils"
|
||
"github.com/komari-monitor/komari/utils/oauth"
|
||
)
|
||
|
||
// /api/oauth
|
||
func OAuth(c *gin.Context) {
|
||
cfg, _ := config.Get()
|
||
if !cfg.OAuthEnabled {
|
||
c.JSON(403, gin.H{"status": "error", "error": "OAuth is not enabled"})
|
||
return
|
||
}
|
||
|
||
authURL, state := oauth.CurrentProvider().GetAuthorizationURL(utils.GetCallbackURL(c))
|
||
|
||
c.SetCookie("oauth_state", state, 3600, "/", "", false, true)
|
||
|
||
c.Redirect(302, authURL)
|
||
}
|
||
|
||
// /api/oauth_callback
|
||
func OAuthCallback(c *gin.Context) {
|
||
// 验证state防止CSRF攻击
|
||
state, _ := c.Cookie("oauth_state")
|
||
c.SetCookie("oauth_state", "", -1, "/", "", false, true)
|
||
|
||
// 获取当前OAuth提供商名称
|
||
providerName := oauth.CurrentProvider().GetName()
|
||
|
||
// 对于QQ登录，由于是通过QQ聚合登录平台中转，state可能会不匹配
|
||
// 但我们仍然需要验证state的存在性（不能是空的）
|
||
if providerName == "qq" {
|
||
if state == "" {
|
||
c.JSON(400, gin.H{"status": "error", "error": "Invalid state"})
|
||
return
|
||
}
|
||
} else {
|
||
// 对于其他提供商，严格验证state匹配
|
||
if state == "" || state != c.Query("state") {
|
||
c.JSON(400, gin.H{"status": "error", "error": "Invalid state"})
|
||
return
|
||
}
|
||
}
|
||
|
||
queries := make(map[string]string)
|
||
for key, values := range c.Request.URL.Query() {
|
||
if len(values) > 0 {
|
||
queries[key] = values[0]
|
||
}
|
||
}
|
||
oidcUser, err := oauth.CurrentProvider().OnCallback(c.Request.Context(), state, queries, utils.GetCallbackURL(c))
|
||
if err != nil {
|
||
c.JSON(500, gin.H{"status": "error", "error": "Failed to get user info: " + err.Error()})
|
||
return
|
||
}
|
||
|
||
// ID作为SSO ID
|
||
sso_id := fmt.Sprintf("%s_%s", oauth.CurrentProvider().GetName(), oidcUser.UserId)
|
||
|
||
// 如果cookie中有binding_external_account，说明是绑定外部账号
|
||
// 否则是登录
|
||
uuid, _ := c.Cookie("binding_external_account")
|
||
c.SetCookie("binding_external_account", "", -1, "/", "", false, true)
|
||
if uuid != "" {
|
||
// 绑定外部账号
|
||
session, _ := c.Cookie("session_token")
|
||
user, err := accounts.GetUserBySession(session)
|
||
if err != nil || user.UUID != uuid {
|
||
c.JSON(500, gin.H{"status": "error", "message": "Binding failed"})
|
||
return
|
||
}
|
||
err = accounts.BindingExternalAccount(user.UUID, sso_id)
|
||
if err != nil {
|
||
c.JSON(500, gin.H{"status": "error", "message": "Binding failed"})
|
||
return
|
||
}
|
||
auditlog.Log(c.ClientIP(), user.UUID, "bound external account (OAuth)"+fmt.Sprintf(",sso_id: %s", sso_id), "login")
|
||
c.Redirect(302, "/manage")
|
||
return
|
||
}
|
||
|
||
// 尝试获取用户
|
||
user, err := accounts.GetUserBySSO(sso_id)
|
||
if err != nil {
|
||
c.JSON(401, gin.H{
|
||
"status": "error",
|
||
"message": "please log in and bind your external account first.",
|
||
})
|
||
return
|
||
}
|
||
|
||
// 创建会话
|
||
session, err := accounts.CreateSession(user.UUID, 2592000, c.Request.UserAgent(), c.ClientIP(), "oauth")
|
||
if err != nil {
|
||
c.JSON(500, gin.H{"status": "error", "message": err.Error()})
|
||
return
|
||
}
|
||
|
||
// 设置cookie并返回
|
||
c.SetCookie("session_token", session, 2592000, "/", "", false, true)
|
||
auditlog.Log(c.ClientIP(), user.UUID, "logged in (OAuth)", "login")
|
||
c.Redirect(302, "/admin")
|
||
} |