🖼️ chore: format code file

2025-08-10 12:11:31 +08:00
parent ca1f3c6e4c
commit 1d578b73ce
14 changed files with 776 additions and 776 deletions
--- a/controller/missing_models.go
+++ b/controller/missing_models.go
@@ -1,27 +1,27 @@
 package controller

 import (
-    "net/http"
-    "one-api/model"
+	"net/http"
+	"one-api/model"

-    "github.com/gin-gonic/gin"
+	"github.com/gin-gonic/gin"
 )

 // GetMissingModels returns the list of model names that are referenced by channels
 // but do not have corresponding records in the models meta table.
 // This helps administrators quickly discover models that need configuration.
 func GetMissingModels(c *gin.Context) {
-    missing, err := model.GetMissingModels()
-    if err != nil {
-        c.JSON(http.StatusOK, gin.H{
-            "success": false,
-            "message": err.Error(),
-        })
-        return
-    }
+	missing, err := model.GetMissingModels()
+	if err != nil {
+		c.JSON(http.StatusOK, gin.H{
+			"success": false,
+			"message": err.Error(),
+		})
+		return
+	}

-    c.JSON(http.StatusOK, gin.H{
-        "success": true,
-        "data":    missing,
-    })
+	c.JSON(http.StatusOK, gin.H{
+		"success": true,
+		"data":    missing,
+	})
 }
--- a/controller/model_meta.go
+++ b/controller/model_meta.go
@@ -1,178 +1,178 @@
 package controller

 import (
-    "encoding/json"
-    "strconv"
+	"encoding/json"
+	"strconv"

-    "one-api/common"
-    "one-api/model"
+	"one-api/common"
+	"one-api/model"

-    "github.com/gin-gonic/gin"
+	"github.com/gin-gonic/gin"
 )

 // GetAllModelsMeta 获取模型列表（分页）
 func GetAllModelsMeta(c *gin.Context) {

-    pageInfo := common.GetPageQuery(c)
-    modelsMeta, err := model.GetAllModels(pageInfo.GetStartIdx(), pageInfo.GetPageSize())
-    if err != nil {
-        common.ApiError(c, err)
-        return
-    }
-    // 填充附加字段
-    for _, m := range modelsMeta {
-        fillModelExtra(m)
-    }
-    var total int64
-    model.DB.Model(&model.Model{}).Count(&total)
+	pageInfo := common.GetPageQuery(c)
+	modelsMeta, err := model.GetAllModels(pageInfo.GetStartIdx(), pageInfo.GetPageSize())
+	if err != nil {
+		common.ApiError(c, err)
+		return
+	}
+	// 填充附加字段
+	for _, m := range modelsMeta {
+		fillModelExtra(m)
+	}
+	var total int64
+	model.DB.Model(&model.Model{}).Count(&total)

-    // 统计供应商计数（全部数据，不受分页影响）
-    vendorCounts, _ := model.GetVendorModelCounts()
+	// 统计供应商计数（全部数据，不受分页影响）
+	vendorCounts, _ := model.GetVendorModelCounts()

-    pageInfo.SetTotal(int(total))
-    pageInfo.SetItems(modelsMeta)
-    common.ApiSuccess(c, gin.H{
-        "items":         modelsMeta,
-        "total":         total,
-        "page":          pageInfo.GetPage(),
-        "page_size":     pageInfo.GetPageSize(),
-        "vendor_counts": vendorCounts,
-    })
+	pageInfo.SetTotal(int(total))
+	pageInfo.SetItems(modelsMeta)
+	common.ApiSuccess(c, gin.H{
+		"items":         modelsMeta,
+		"total":         total,
+		"page":          pageInfo.GetPage(),
+		"page_size":     pageInfo.GetPageSize(),
+		"vendor_counts": vendorCounts,
+	})
 }

 // SearchModelsMeta 搜索模型列表
 func SearchModelsMeta(c *gin.Context) {

-    keyword := c.Query("keyword")
-    vendor := c.Query("vendor")
-    pageInfo := common.GetPageQuery(c)
+	keyword := c.Query("keyword")
+	vendor := c.Query("vendor")
+	pageInfo := common.GetPageQuery(c)

-    modelsMeta, total, err := model.SearchModels(keyword, vendor, pageInfo.GetStartIdx(), pageInfo.GetPageSize())
-    if err != nil {
-        common.ApiError(c, err)
-        return
-    }
-    for _, m := range modelsMeta {
-        fillModelExtra(m)
-    }
-    pageInfo.SetTotal(int(total))
-    pageInfo.SetItems(modelsMeta)
-    common.ApiSuccess(c, pageInfo)
+	modelsMeta, total, err := model.SearchModels(keyword, vendor, pageInfo.GetStartIdx(), pageInfo.GetPageSize())
+	if err != nil {
+		common.ApiError(c, err)
+		return
+	}
+	for _, m := range modelsMeta {
+		fillModelExtra(m)
+	}
+	pageInfo.SetTotal(int(total))
+	pageInfo.SetItems(modelsMeta)
+	common.ApiSuccess(c, pageInfo)
 }

 // GetModelMeta 根据 ID 获取单条模型信息
 func GetModelMeta(c *gin.Context) {
-    idStr := c.Param("id")
-    id, err := strconv.Atoi(idStr)
-    if err != nil {
-        common.ApiError(c, err)
-        return
-    }
-    var m model.Model
-    if err := model.DB.First(&m, id).Error; err != nil {
-        common.ApiError(c, err)
-        return
-    }
-    fillModelExtra(&m)
-    common.ApiSuccess(c, &m)
+	idStr := c.Param("id")
+	id, err := strconv.Atoi(idStr)
+	if err != nil {
+		common.ApiError(c, err)
+		return
+	}
+	var m model.Model
+	if err := model.DB.First(&m, id).Error; err != nil {
+		common.ApiError(c, err)
+		return
+	}
+	fillModelExtra(&m)
+	common.ApiSuccess(c, &m)
 }

 // CreateModelMeta 新建模型
 func CreateModelMeta(c *gin.Context) {
-    var m model.Model
-    if err := c.ShouldBindJSON(&m); err != nil {
-        common.ApiError(c, err)
-        return
-    }
-    if m.ModelName == "" {
-        common.ApiErrorMsg(c, "模型名称不能为空")
-        return
-    }
-    // 名称冲突检查
-    if dup, err := model.IsModelNameDuplicated(0, m.ModelName); err != nil {
-        common.ApiError(c, err)
-        return
-    } else if dup {
-        common.ApiErrorMsg(c, "模型名称已存在")
-        return
-    }
+	var m model.Model
+	if err := c.ShouldBindJSON(&m); err != nil {
+		common.ApiError(c, err)
+		return
+	}
+	if m.ModelName == "" {
+		common.ApiErrorMsg(c, "模型名称不能为空")
+		return
+	}
+	// 名称冲突检查
+	if dup, err := model.IsModelNameDuplicated(0, m.ModelName); err != nil {
+		common.ApiError(c, err)
+		return
+	} else if dup {
+		common.ApiErrorMsg(c, "模型名称已存在")
+		return
+	}

-    if err := m.Insert(); err != nil {
-        common.ApiError(c, err)
-        return
-    }
-    model.RefreshPricing()
-    common.ApiSuccess(c, &m)
+	if err := m.Insert(); err != nil {
+		common.ApiError(c, err)
+		return
+	}
+	model.RefreshPricing()
+	common.ApiSuccess(c, &m)
 }

 // UpdateModelMeta 更新模型
 func UpdateModelMeta(c *gin.Context) {
-    statusOnly := c.Query("status_only") == "true"
+	statusOnly := c.Query("status_only") == "true"

-    var m model.Model
-    if err := c.ShouldBindJSON(&m); err != nil {
-        common.ApiError(c, err)
-        return
-    }
-    if m.Id == 0 {
-        common.ApiErrorMsg(c, "缺少模型 ID")
-        return
-    }
+	var m model.Model
+	if err := c.ShouldBindJSON(&m); err != nil {
+		common.ApiError(c, err)
+		return
+	}
+	if m.Id == 0 {
+		common.ApiErrorMsg(c, "缺少模型 ID")
+		return
+	}

-    if statusOnly {
-        // 只更新状态，防止误清空其他字段
-        if err := model.DB.Model(&model.Model{}).Where("id = ?", m.Id).Update("status", m.Status).Error; err != nil {
-            common.ApiError(c, err)
-            return
-        }
-    } else {
-        // 名称冲突检查
-        if dup, err := model.IsModelNameDuplicated(m.Id, m.ModelName); err != nil {
-            common.ApiError(c, err)
-            return
-        } else if dup {
-            common.ApiErrorMsg(c, "模型名称已存在")
-            return
-        }
+	if statusOnly {
+		// 只更新状态，防止误清空其他字段
+		if err := model.DB.Model(&model.Model{}).Where("id = ?", m.Id).Update("status", m.Status).Error; err != nil {
+			common.ApiError(c, err)
+			return
+		}
+	} else {
+		// 名称冲突检查
+		if dup, err := model.IsModelNameDuplicated(m.Id, m.ModelName); err != nil {
+			common.ApiError(c, err)
+			return
+		} else if dup {
+			common.ApiErrorMsg(c, "模型名称已存在")
+			return
+		}

-        if err := m.Update(); err != nil {
-            common.ApiError(c, err)
-            return
-        }
-    }
-    model.RefreshPricing()
-    common.ApiSuccess(c, &m)
+		if err := m.Update(); err != nil {
+			common.ApiError(c, err)
+			return
+		}
+	}
+	model.RefreshPricing()
+	common.ApiSuccess(c, &m)
 }

 // DeleteModelMeta 删除模型
 func DeleteModelMeta(c *gin.Context) {
-    idStr := c.Param("id")
-    id, err := strconv.Atoi(idStr)
-    if err != nil {
-        common.ApiError(c, err)
-        return
-    }
-    if err := model.DB.Delete(&model.Model{}, id).Error; err != nil {
-        common.ApiError(c, err)
-        return
-    }
-    model.RefreshPricing()
-    common.ApiSuccess(c, nil)
+	idStr := c.Param("id")
+	id, err := strconv.Atoi(idStr)
+	if err != nil {
+		common.ApiError(c, err)
+		return
+	}
+	if err := model.DB.Delete(&model.Model{}, id).Error; err != nil {
+		common.ApiError(c, err)
+		return
+	}
+	model.RefreshPricing()
+	common.ApiSuccess(c, nil)
 }

 // 辅助函数：填充 Endpoints 和 BoundChannels 和 EnableGroups
 func fillModelExtra(m *model.Model) {
-    if m.Endpoints == "" {
-        eps := model.GetModelSupportEndpointTypes(m.ModelName)
-        if b, err := json.Marshal(eps); err == nil {
-            m.Endpoints = string(b)
-        }
-    }
-    if channels, err := model.GetBoundChannels(m.ModelName); err == nil {
-        m.BoundChannels = channels
-    }
-    // 填充启用分组
-    m.EnableGroups = model.GetModelEnableGroups(m.ModelName)
-    // 填充计费类型
-    m.QuotaType = model.GetModelQuotaType(m.ModelName)
+	if m.Endpoints == "" {
+		eps := model.GetModelSupportEndpointTypes(m.ModelName)
+		if b, err := json.Marshal(eps); err == nil {
+			m.Endpoints = string(b)
+		}
+	}
+	if channels, err := model.GetBoundChannels(m.ModelName); err == nil {
+		m.BoundChannels = channels
+	}
+	// 填充启用分组
+	m.EnableGroups = model.GetModelEnableGroups(m.ModelName)
+	// 填充计费类型
+	m.QuotaType = model.GetModelQuotaType(m.ModelName)
 }
--- a/controller/prefill_group.go
+++ b/controller/prefill_group.go
@@ -1,90 +1,90 @@
 package controller

 import (
-    "strconv"
+	"strconv"

-    "one-api/common"
-    "one-api/model"
+	"one-api/common"
+	"one-api/model"

-    "github.com/gin-gonic/gin"
+	"github.com/gin-gonic/gin"
 )

 // GetPrefillGroups 获取预填组列表，可通过 ?type=xxx 过滤
 func GetPrefillGroups(c *gin.Context) {
-    groupType := c.Query("type")
-    groups, err := model.GetAllPrefillGroups(groupType)
-    if err != nil {
-        common.ApiError(c, err)
-        return
-    }
-    common.ApiSuccess(c, groups)
+	groupType := c.Query("type")
+	groups, err := model.GetAllPrefillGroups(groupType)
+	if err != nil {
+		common.ApiError(c, err)
+		return
+	}
+	common.ApiSuccess(c, groups)
 }

 // CreatePrefillGroup 创建新的预填组
 func CreatePrefillGroup(c *gin.Context) {
-    var g model.PrefillGroup
-    if err := c.ShouldBindJSON(&g); err != nil {
-        common.ApiError(c, err)
-        return
-    }
-    if g.Name == "" || g.Type == "" {
-        common.ApiErrorMsg(c, "组名称和类型不能为空")
-        return
-    }
-    // 创建前检查名称
-    if dup, err := model.IsPrefillGroupNameDuplicated(0, g.Name); err != nil {
-        common.ApiError(c, err)
-        return
-    } else if dup {
-        common.ApiErrorMsg(c, "组名称已存在")
-        return
-    }
+	var g model.PrefillGroup
+	if err := c.ShouldBindJSON(&g); err != nil {
+		common.ApiError(c, err)
+		return
+	}
+	if g.Name == "" || g.Type == "" {
+		common.ApiErrorMsg(c, "组名称和类型不能为空")
+		return
+	}
+	// 创建前检查名称
+	if dup, err := model.IsPrefillGroupNameDuplicated(0, g.Name); err != nil {
+		common.ApiError(c, err)
+		return
+	} else if dup {
+		common.ApiErrorMsg(c, "组名称已存在")
+		return
+	}

-    if err := g.Insert(); err != nil {
-        common.ApiError(c, err)
-        return
-    }
-    common.ApiSuccess(c, &g)
+	if err := g.Insert(); err != nil {
+		common.ApiError(c, err)
+		return
+	}
+	common.ApiSuccess(c, &g)
 }

 // UpdatePrefillGroup 更新预填组
 func UpdatePrefillGroup(c *gin.Context) {
-    var g model.PrefillGroup
-    if err := c.ShouldBindJSON(&g); err != nil {
-        common.ApiError(c, err)
-        return
-    }
-    if g.Id == 0 {
-        common.ApiErrorMsg(c, "缺少组 ID")
-        return
-    }
-    // 名称冲突检查
-    if dup, err := model.IsPrefillGroupNameDuplicated(g.Id, g.Name); err != nil {
-        common.ApiError(c, err)
-        return
-    } else if dup {
-        common.ApiErrorMsg(c, "组名称已存在")
-        return
-    }
+	var g model.PrefillGroup
+	if err := c.ShouldBindJSON(&g); err != nil {
+		common.ApiError(c, err)
+		return
+	}
+	if g.Id == 0 {
+		common.ApiErrorMsg(c, "缺少组 ID")
+		return
+	}
+	// 名称冲突检查
+	if dup, err := model.IsPrefillGroupNameDuplicated(g.Id, g.Name); err != nil {
+		common.ApiError(c, err)
+		return
+	} else if dup {
+		common.ApiErrorMsg(c, "组名称已存在")
+		return
+	}

-    if err := g.Update(); err != nil {
-        common.ApiError(c, err)
-        return
-    }
-    common.ApiSuccess(c, &g)
+	if err := g.Update(); err != nil {
+		common.ApiError(c, err)
+		return
+	}
+	common.ApiSuccess(c, &g)
 }

 // DeletePrefillGroup 删除预填组
 func DeletePrefillGroup(c *gin.Context) {
-    idStr := c.Param("id")
-    id, err := strconv.Atoi(idStr)
-    if err != nil {
-        common.ApiError(c, err)
-        return
-    }
-    if err := model.DeletePrefillGroupByID(id); err != nil {
-        common.ApiError(c, err)
-        return
-    }
-    common.ApiSuccess(c, nil)
+	idStr := c.Param("id")
+	id, err := strconv.Atoi(idStr)
+	if err != nil {
+		common.ApiError(c, err)
+		return
+	}
+	if err := model.DeletePrefillGroupByID(id); err != nil {
+		common.ApiError(c, err)
+		return
+	}
+	common.ApiSuccess(c, nil)
 }
--- a/controller/pricing.go
+++ b/controller/pricing.go
@@ -39,14 +39,14 @@ func GetPricing(c *gin.Context) {
 	}

 	c.JSON(200, gin.H{
-		"success":      true,
-		"data":         pricing,
-		"vendors":      model.GetVendors(),
-		"group_ratio":  groupRatio,
-        "usable_group": usableGroup,
-        "supported_endpoint": model.GetSupportedEndpointMap(),
-        "auto_groups": setting.AutoGroups,
-    })
+		"success":            true,
+		"data":               pricing,
+		"vendors":            model.GetVendors(),
+		"group_ratio":        groupRatio,
+		"usable_group":       usableGroup,
+		"supported_endpoint": model.GetSupportedEndpointMap(),
+		"auto_groups":        setting.AutoGroups,
+	})
 }

 func ResetModelRatio(c *gin.Context) {
--- a/controller/user.go
+++ b/controller/user.go
@@ -62,7 +62,7 @@ func Login(c *gin.Context) {
 		})
 		return
 	}
-	
+
 	// 检查是否启用2FA
 	if model.IsTwoFAEnabled(user.Id) {
 		// 设置pending session，等待2FA验证
@@ -77,7 +77,7 @@ func Login(c *gin.Context) {
 			})
 			return
 		}
-		
+
 		c.JSON(http.StatusOK, gin.H{
 			"message": "请输入两步验证码",
 			"success": true,
@@ -87,7 +87,7 @@ func Login(c *gin.Context) {
 		})
 		return
 	}
-	
+
 	setupLogin(&user, c)
 }

--- a/controller/vendor_meta.go
+++ b/controller/vendor_meta.go
@@ -1,124 +1,124 @@
 package controller

 import (
-    "strconv"
+	"strconv"

-    "one-api/common"
-    "one-api/model"
+	"one-api/common"
+	"one-api/model"

-    "github.com/gin-gonic/gin"
+	"github.com/gin-gonic/gin"
 )

 // GetAllVendors 获取供应商列表（分页）
 func GetAllVendors(c *gin.Context) {
-    pageInfo := common.GetPageQuery(c)
-    vendors, err := model.GetAllVendors(pageInfo.GetStartIdx(), pageInfo.GetPageSize())
-    if err != nil {
-        common.ApiError(c, err)
-        return
-    }
-    var total int64
-    model.DB.Model(&model.Vendor{}).Count(&total)
-    pageInfo.SetTotal(int(total))
-    pageInfo.SetItems(vendors)
-    common.ApiSuccess(c, pageInfo)
+	pageInfo := common.GetPageQuery(c)
+	vendors, err := model.GetAllVendors(pageInfo.GetStartIdx(), pageInfo.GetPageSize())
+	if err != nil {
+		common.ApiError(c, err)
+		return
+	}
+	var total int64
+	model.DB.Model(&model.Vendor{}).Count(&total)
+	pageInfo.SetTotal(int(total))
+	pageInfo.SetItems(vendors)
+	common.ApiSuccess(c, pageInfo)
 }

 // SearchVendors 搜索供应商
 func SearchVendors(c *gin.Context) {
-    keyword := c.Query("keyword")
-    pageInfo := common.GetPageQuery(c)
-    vendors, total, err := model.SearchVendors(keyword, pageInfo.GetStartIdx(), pageInfo.GetPageSize())
-    if err != nil {
-        common.ApiError(c, err)
-        return
-    }
-    pageInfo.SetTotal(int(total))
-    pageInfo.SetItems(vendors)
-    common.ApiSuccess(c, pageInfo)
+	keyword := c.Query("keyword")
+	pageInfo := common.GetPageQuery(c)
+	vendors, total, err := model.SearchVendors(keyword, pageInfo.GetStartIdx(), pageInfo.GetPageSize())
+	if err != nil {
+		common.ApiError(c, err)
+		return
+	}
+	pageInfo.SetTotal(int(total))
+	pageInfo.SetItems(vendors)
+	common.ApiSuccess(c, pageInfo)
 }

 // GetVendorMeta 根据 ID 获取供应商
 func GetVendorMeta(c *gin.Context) {
-    idStr := c.Param("id")
-    id, err := strconv.Atoi(idStr)
-    if err != nil {
-        common.ApiError(c, err)
-        return
-    }
-    v, err := model.GetVendorByID(id)
-    if err != nil {
-        common.ApiError(c, err)
-        return
-    }
-    common.ApiSuccess(c, v)
+	idStr := c.Param("id")
+	id, err := strconv.Atoi(idStr)
+	if err != nil {
+		common.ApiError(c, err)
+		return
+	}
+	v, err := model.GetVendorByID(id)
+	if err != nil {
+		common.ApiError(c, err)
+		return
+	}
+	common.ApiSuccess(c, v)
 }

 // CreateVendorMeta 新建供应商
 func CreateVendorMeta(c *gin.Context) {
-    var v model.Vendor
-    if err := c.ShouldBindJSON(&v); err != nil {
-        common.ApiError(c, err)
-        return
-    }
-    if v.Name == "" {
-        common.ApiErrorMsg(c, "供应商名称不能为空")
-        return
-    }
-    // 创建前先检查名称
-    if dup, err := model.IsVendorNameDuplicated(0, v.Name); err != nil {
-        common.ApiError(c, err)
-        return
-    } else if dup {
-        common.ApiErrorMsg(c, "供应商名称已存在")
-        return
-    }
+	var v model.Vendor
+	if err := c.ShouldBindJSON(&v); err != nil {
+		common.ApiError(c, err)
+		return
+	}
+	if v.Name == "" {
+		common.ApiErrorMsg(c, "供应商名称不能为空")
+		return
+	}
+	// 创建前先检查名称
+	if dup, err := model.IsVendorNameDuplicated(0, v.Name); err != nil {
+		common.ApiError(c, err)
+		return
+	} else if dup {
+		common.ApiErrorMsg(c, "供应商名称已存在")
+		return
+	}

-    if err := v.Insert(); err != nil {
-        common.ApiError(c, err)
-        return
-    }
-    common.ApiSuccess(c, &v)
+	if err := v.Insert(); err != nil {
+		common.ApiError(c, err)
+		return
+	}
+	common.ApiSuccess(c, &v)
 }

 // UpdateVendorMeta 更新供应商
 func UpdateVendorMeta(c *gin.Context) {
-    var v model.Vendor
-    if err := c.ShouldBindJSON(&v); err != nil {
-        common.ApiError(c, err)
-        return
-    }
-    if v.Id == 0 {
-        common.ApiErrorMsg(c, "缺少供应商 ID")
-        return
-    }
-    // 名称冲突检查
-    if dup, err := model.IsVendorNameDuplicated(v.Id, v.Name); err != nil {
-        common.ApiError(c, err)
-        return
-    } else if dup {
-        common.ApiErrorMsg(c, "供应商名称已存在")
-        return
-    }
+	var v model.Vendor
+	if err := c.ShouldBindJSON(&v); err != nil {
+		common.ApiError(c, err)
+		return
+	}
+	if v.Id == 0 {
+		common.ApiErrorMsg(c, "缺少供应商 ID")
+		return
+	}
+	// 名称冲突检查
+	if dup, err := model.IsVendorNameDuplicated(v.Id, v.Name); err != nil {
+		common.ApiError(c, err)
+		return
+	} else if dup {
+		common.ApiErrorMsg(c, "供应商名称已存在")
+		return
+	}

-    if err := v.Update(); err != nil {
-        common.ApiError(c, err)
-        return
-    }
-    common.ApiSuccess(c, &v)
+	if err := v.Update(); err != nil {
+		common.ApiError(c, err)
+		return
+	}
+	common.ApiSuccess(c, &v)
 }

 // DeleteVendorMeta 删除供应商
 func DeleteVendorMeta(c *gin.Context) {
-    idStr := c.Param("id")
-    id, err := strconv.Atoi(idStr)
-    if err != nil {
-        common.ApiError(c, err)
-        return
-    }
-    if err := model.DB.Delete(&model.Vendor{}, id).Error; err != nil {
-        common.ApiError(c, err)
-        return
-    }
-    common.ApiSuccess(c, nil)
-}
+	idStr := c.Param("id")
+	id, err := strconv.Atoi(idStr)
+	if err != nil {
+		common.ApiError(c, err)
+		return
+	}
+	if err := model.DB.Delete(&model.Vendor{}, id).Error; err != nil {
+		common.ApiError(c, err)
+		return
+	}
+	common.ApiSuccess(c, nil)
+}