OpenFlare/backend/OpenFlare/plugins/server/admin/auth_source/routers.go

// Copyright 2026 Arctel.net
// SPDX-License-Identifier: Apache-2.0

// Package auth_source 提供认证源管理功能
package auth_source

import (
	"errors"
	"fmt"
	"net/http"
	"strings"

	"Wavelet/OpenFlare/plugins/server/admin"
	"Wavelet/OpenFlare/plugins/server/model"
	"Wavelet/OpenFlare/plugins/server/oauth"
	"Wavelet/OpenFlare/plugins/server/repository"

	"github.com/gin-gonic/gin"

	"Wavelet/pkg/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 := repository.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 := repository.CreateAuthSource(c.Request.Context(), &source); err != nil {
		response.AbortBadRequest(c, err.Error())
		return
	}
	_ = repository.InvalidateAuthSourceCache(c.Request.Context())
	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, _ := repository.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 := repository.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))
	_ = repository.InvalidateAuthSourceCache(c.Request.Context())

	updated, err := repository.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 := repository.ToggleAuthSource(c.Request.Context(), id, req.IsActive); err != nil {
		response.AbortBadRequest(c, err.Error())
		return
	}
	_ = repository.InvalidateAuthSourceCache(c.Request.Context())
	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 := repository.DeleteAuthSource(c.Request.Context(), id); err != nil {
		response.AbortBadRequest(c, err.Error())
		return
	}
	_ = repository.InvalidateAuthSourceCache(c.Request.Context())
	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 := repository.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
}