mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-09-29 22:06:38 +08:00
ryan
63cd906cfc
- Merge all files inside openflare-server to the repository root directory. - Relocate agent, relay, and flared subprojects from internal/ to internal/apps/. - Combine docker-compose files and update build context paths to root. - Update GitHub workflows and Dockerfiles to refer to new directories and package names. - Rewrite Go package imports across all files. - Resolve database renew test race condition and clean up docs.
246 lines
8.4 KiB
Go
246 lines
8.4 KiB
Go
// Copyright 2026 Arctel.net
|
||
// SPDX-License-Identifier: Apache-2.0
|
||
|
||
// Package auth_source 提供认证源管理功能
|
||
package auth_source
|
||
|
||
import (
|
||
"errors"
|
||
"fmt"
|
||
"net/http"
|
||
"strings"
|
||
|
||
"github.com/Rain-kl/Wavelet/internal/apps/admin"
|
||
"github.com/Rain-kl/Wavelet/internal/apps/oauth"
|
||
"github.com/Rain-kl/Wavelet/internal/model"
|
||
"github.com/gin-gonic/gin"
|
||
|
||
"github.com/Rain-kl/Wavelet/internal/common/response"
|
||
)
|
||
|
||
// AuthSourceRequest 创建或更新认证源的请求参数
|
||
type AuthSourceRequest struct {
|
||
Name string `json:"name"`
|
||
Type string `json:"type"`
|
||
DisplayName string `json:"display_name"`
|
||
IsActive bool `json:"is_active"`
|
||
ClientID string `json:"client_id"`
|
||
ClientSecret string `json:"client_secret"`
|
||
OpenIDDiscoveryURL string `json:"openid_discovery_url"`
|
||
Scopes string `json:"scopes"`
|
||
IconURL string `json:"icon_url"`
|
||
}
|
||
|
||
// ToggleAuthSourceRequest 切换认证源启用状态的请求参数
|
||
type ToggleAuthSourceRequest struct {
|
||
IsActive bool `json:"is_active"`
|
||
}
|
||
|
||
// ListAuthSources 获取认证源列表
|
||
// @Summary 获取认证源列表
|
||
// @Description 返回所有已配置的 OAuth/OIDC 认证源列表，包括已启用和未启用的，需要管理员权限
|
||
// @Tags admin
|
||
// @Produce json
|
||
// @Security SessionCookie
|
||
// @Success 200 {object} response.Any{data=[]model.AuthSource} "认证源列表"
|
||
// @Failure 401 {object} response.Any "未登录"
|
||
// @Failure 403 {object} response.Any "无管理员权限"
|
||
// @Failure 500 {object} response.Any "内部错误"
|
||
// @Router /api/v1/admin/auth-sources [get]
|
||
func ListAuthSources(c *gin.Context) {
|
||
sources, err := model.GetAuthSources(c.Request.Context())
|
||
if err != nil {
|
||
response.AbortInternal(c, err.Error())
|
||
return
|
||
}
|
||
c.JSON(http.StatusOK, response.OK(sources))
|
||
}
|
||
|
||
// CreateAuthSource 创建认证源
|
||
// @Summary 创建认证源
|
||
// @Description 创建一个新的 OAuth/OIDC 认证源配置，认证源名称必须唯一且符合命名规范，需要管理员权限
|
||
// @Tags admin
|
||
// @Accept json
|
||
// @Produce json
|
||
// @Security SessionCookie
|
||
// @Param request body auth_source.AuthSourceRequest true "创建认证源参数"
|
||
// @Success 200 {object} response.Any{data=model.AuthSource} "创建成功，返回认证源信息"
|
||
// @Failure 400 {object} response.Any "参数错误或验证失败"
|
||
// @Failure 401 {object} response.Any "未登录"
|
||
// @Failure 403 {object} response.Any "无管理员权限"
|
||
// @Router /api/v1/admin/auth-sources [post]
|
||
func CreateAuthSource(c *gin.Context) {
|
||
var req AuthSourceRequest
|
||
if err := c.ShouldBindJSON(&req); err != nil {
|
||
response.AbortBadRequest(c, err.Error())
|
||
return
|
||
}
|
||
|
||
source := model.AuthSource{
|
||
Name: req.Name,
|
||
Type: req.Type,
|
||
DisplayName: req.DisplayName,
|
||
IsActive: req.IsActive,
|
||
ClientID: req.ClientID,
|
||
ClientSecret: req.ClientSecret,
|
||
OpenIDDiscoveryURL: req.OpenIDDiscoveryURL,
|
||
Scopes: req.Scopes,
|
||
IconURL: req.IconURL,
|
||
}
|
||
if err := model.CreateAuthSource(c.Request.Context(), &source); err != nil {
|
||
response.AbortBadRequest(c, err.Error())
|
||
return
|
||
}
|
||
source.Sanitize()
|
||
c.JSON(http.StatusOK, response.OK(source))
|
||
}
|
||
|
||
// UpdateAuthSource 更新认证源
|
||
// @Summary 更新认证源
|
||
// @Description 更新指定 ID 的认证源配置。若 client_secret 字段为空，则保留原有密钥不变，需要管理员权限
|
||
// @Tags admin
|
||
// @Accept json
|
||
// @Produce json
|
||
// @Security SessionCookie
|
||
// @Param id path uint64 true "认证源 ID 或名称"
|
||
// @Param request body auth_source.AuthSourceRequest true "更新认证源参数"
|
||
// @Success 200 {object} response.Any{data=model.AuthSource} "更新成功，返回更新后的认证源信息"
|
||
// @Failure 400 {object} response.Any "参数错误或验证失败"
|
||
// @Failure 401 {object} response.Any "未登录"
|
||
// @Failure 403 {object} response.Any "无管理员权限"
|
||
// @Failure 500 {object} response.Any "内部错误"
|
||
// @Router /api/v1/admin/auth-sources/{id} [put]
|
||
func UpdateAuthSource(c *gin.Context) {
|
||
id, err := parseSourceID(c)
|
||
if err != nil {
|
||
response.AbortBadRequest(c, err.Error())
|
||
return
|
||
}
|
||
|
||
var req AuthSourceRequest
|
||
if err := c.ShouldBindJSON(&req); err != nil {
|
||
response.AbortBadRequest(c, err.Error())
|
||
return
|
||
}
|
||
|
||
// 记录更新前的 Discovery URL，以便更新成功后清除旧缓存条目。
|
||
existing, _ := model.GetAuthSourceByID(c.Request.Context(), id)
|
||
|
||
source := model.AuthSource{
|
||
ID: id,
|
||
Name: req.Name,
|
||
Type: req.Type,
|
||
DisplayName: req.DisplayName,
|
||
IsActive: req.IsActive,
|
||
ClientID: req.ClientID,
|
||
ClientSecret: req.ClientSecret,
|
||
OpenIDDiscoveryURL: req.OpenIDDiscoveryURL,
|
||
Scopes: req.Scopes,
|
||
IconURL: req.IconURL,
|
||
}
|
||
keepSecret := source.ClientSecret == ""
|
||
if err := model.UpdateAuthSource(c.Request.Context(), &source, keepSecret); err != nil {
|
||
response.AbortBadRequest(c, err.Error())
|
||
return
|
||
}
|
||
|
||
// Discovery URL 可能已变更，清除旧、新 issuer 的 provider 缓存，
|
||
// 确保下次登录时重新拉取最新 OIDC 元数据。
|
||
if existing != nil {
|
||
oauth.InvalidateOIDCProviderCache(normalizeIssuer(existing.OpenIDDiscoveryURL))
|
||
}
|
||
oauth.InvalidateOIDCProviderCache(normalizeIssuer(req.OpenIDDiscoveryURL))
|
||
|
||
updated, err := model.GetAuthSourceByID(c.Request.Context(), id)
|
||
if err != nil {
|
||
response.AbortInternal(c, err.Error())
|
||
return
|
||
}
|
||
updated.Sanitize()
|
||
c.JSON(http.StatusOK, response.OK(updated))
|
||
}
|
||
|
||
// ToggleAuthSource 切换认证源启用状态
|
||
// @Summary 切换认证源启用状态
|
||
// @Description 启用或禁用指定认证源。尝试启用时将验证 Client ID 和 Client Secret 是否已配置，需要管理员权限
|
||
// @Tags admin
|
||
// @Accept json
|
||
// @Produce json
|
||
// @Security SessionCookie
|
||
// @Param id path uint64 true "认证源 ID 或名称"
|
||
// @Param request body auth_source.ToggleAuthSourceRequest true "启用状态"
|
||
// @Success 200 {object} response.Any{data=string} "切换成功"
|
||
// @Failure 400 {object} response.Any "验证失败或认证源不存在"
|
||
// @Failure 401 {object} response.Any "未登录"
|
||
// @Failure 403 {object} response.Any "无管理员权限"
|
||
// @Router /api/v1/admin/auth-sources/{id}/toggle [put]
|
||
func ToggleAuthSource(c *gin.Context) {
|
||
id, err := parseSourceID(c)
|
||
if err != nil {
|
||
response.AbortBadRequest(c, err.Error())
|
||
return
|
||
}
|
||
|
||
var req ToggleAuthSourceRequest
|
||
if err := c.ShouldBindJSON(&req); err != nil {
|
||
response.AbortBadRequest(c, err.Error())
|
||
return
|
||
}
|
||
|
||
if err := model.ToggleAuthSource(c.Request.Context(), id, req.IsActive); err != nil {
|
||
response.AbortBadRequest(c, err.Error())
|
||
return
|
||
}
|
||
c.JSON(http.StatusOK, response.OKNil())
|
||
}
|
||
|
||
// DeleteAuthSource 删除认证源
|
||
// @Summary 删除认证源
|
||
// @Description 删除指定认证源及其关联的所有外部帐号绑定记录，警告：删除后相关用户将无法通过该源登录，需要管理员权限
|
||
// @Tags admin
|
||
// @Produce json
|
||
// @Security SessionCookie
|
||
// @Param id path uint64 true "认证源 ID 或名称"
|
||
// @Success 200 {object} response.Any{data=string} "删除成功"
|
||
// @Failure 400 {object} response.Any "ID 无效或删除失败"
|
||
// @Failure 401 {object} response.Any "未登录"
|
||
// @Failure 403 {object} response.Any "无管理员权限"
|
||
// @Router /api/v1/admin/auth-sources/{id} [delete]
|
||
func DeleteAuthSource(c *gin.Context) {
|
||
id, err := parseSourceID(c)
|
||
if err != nil {
|
||
response.AbortBadRequest(c, err.Error())
|
||
return
|
||
}
|
||
if err := model.DeleteAuthSource(c.Request.Context(), id); err != nil {
|
||
response.AbortBadRequest(c, err.Error())
|
||
return
|
||
}
|
||
c.JSON(http.StatusOK, response.OKNil())
|
||
}
|
||
|
||
func parseSourceID(c *gin.Context) (uint64, error) {
|
||
raw := c.Param("id")
|
||
if raw == "" {
|
||
return 0, errors.New(admin.InvalidAuthSourceID)
|
||
}
|
||
source, err := model.GetAuthSourceByName(c.Request.Context(), raw)
|
||
if err == nil {
|
||
return source.ID, nil
|
||
}
|
||
var id uint64
|
||
if _, scanErr := fmt.Sscanf(raw, "%d", &id); scanErr != nil || id == 0 {
|
||
return 0, errors.New(admin.InvalidAuthSourceID)
|
||
}
|
||
return id, nil
|
||
}
|
||
|
||
// normalizeIssuer 将 Discovery URL 规范化为 issuer 基础 URL，
|
||
// 与 oauth.buildOAuthConfig 中的规范化逻辑保持一致。
|
||
func normalizeIssuer(discoveryURL string) string {
|
||
issuer := strings.TrimSuffix(strings.TrimSpace(discoveryURL), "/")
|
||
issuer = strings.TrimSuffix(issuer, "/.well-known/openid-configuration")
|
||
issuer = strings.TrimSuffix(issuer, "/.well-known/oauth-authorization-server")
|
||
return issuer
|
||
}
|