package config
import (
"github.com/go-chi/chi/v5"
"github.com/weibaohui/k8m/internal/dao"
"github.com/weibaohui/k8m/pkg/comm/utils"
"github.com/weibaohui/k8m/pkg/comm/utils/amis"
"github.com/weibaohui/k8m/pkg/models"
"github.com/weibaohui/k8m/pkg/response"
"gorm.io/gorm"
)
type SSOConfigController struct {
}
func RegisterSSOConfigRoutes(admin chi.Router) {
ctrl := &SSOConfigController{}
// SSO 配置 - Gin到Chi迁移
admin.Get("/config/sso/list", response.Adapter(ctrl.List))
admin.Post("/config/sso/save", response.Adapter(ctrl.Save))
admin.Post("/config/sso/delete/{ids}", response.Adapter(ctrl.Delete))
admin.Post("/config/sso/save/id/{id}/status/{enabled}", response.Adapter(ctrl.QuickSave))
}
// @Summary 获取SSO配置列表
// @Security BearerAuth
// @Success 200 {object} string
// @Router /admin/config/sso/list [get]
func (sc *SSOConfigController) List(c *response.Context) {
params := dao.BuildParams(c)
m := &models.SSOConfig{}
items, total, err := m.List(params)
if err != nil {
amis.WriteJsonError(c, err)
return
}
amis.WriteJsonListWithTotal(c, total, items)
}
// @Summary 创建或更新SSO配置
// @Security BearerAuth
// @Success 200 {object} string
// @Router /admin/config/sso/save [post]
func (sc *SSOConfigController) Save(c *response.Context) {
params := dao.BuildParams(c)
m := models.SSOConfig{}
err := c.ShouldBindJSON(&m)
if err != nil {
amis.WriteJsonError(c, err)
return
}
err = m.Save(params, func(db *gorm.DB) *gorm.DB {
return db.Select([]string{"name", "type", "client_id", "client_secret", "issuer", "prefer_user_name_keys", "scopes"})
})
if err != nil {
amis.WriteJsonError(c, err)
return
}
amis.WriteJsonData(c, response.H{
"id": m.ID,
})
}
// @Summary 删除SSO配置
// @Security BearerAuth
// @Param ids path string true "SSO配置ID，多个用逗号分隔"
// @Success 200 {object} string
// @Router /admin/config/sso/delete/{ids} [post]
func (sc *SSOConfigController) Delete(c *response.Context) {
ids := c.Param("ids")
params := dao.BuildParams(c)
m := &models.SSOConfig{}
err := m.Delete(params, ids)
if err != nil {
amis.WriteJsonError(c, err)
return
}
amis.WriteJsonOK(c)
}
// @Summary 快速更新SSO配置状态
// @Security BearerAuth
// @Param id path int true "SSO配置ID"
// @Param enabled path string true "状态，例如：true、false"
// @Success 200 {object} string
// @Router /admin/config/sso/save/id/{id}/status/{enabled} [post]
func (sc *SSOConfigController) QuickSave(c *response.Context) {
id := c.Param("id")
enabled := c.Param("enabled")
var entity models.SSOConfig
entity.ID = utils.ToUInt(id)
if enabled == "true" {
entity.Enabled = true
} else {
entity.Enabled = false
}
err := dao.DB().Model(&entity).Select("enabled").Updates(entity).Error
if err != nil {
amis.WriteJsonError(c, err)
return
}
amis.WriteJsonErrorOrOK(c, err)
}
