option.go 25 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583
  1. package model
  2. import (
  3. "strconv"
  4. "strings"
  5. "time"
  6. "github.com/QuantumNous/new-api/common"
  7. "github.com/QuantumNous/new-api/setting"
  8. "github.com/QuantumNous/new-api/setting/config"
  9. "github.com/QuantumNous/new-api/setting/operation_setting"
  10. "github.com/QuantumNous/new-api/setting/performance_setting"
  11. "github.com/QuantumNous/new-api/setting/ratio_setting"
  12. "github.com/QuantumNous/new-api/setting/system_setting"
  13. )
  14. type Option struct {
  15. Key string `json:"key" gorm:"primaryKey"`
  16. Value string `json:"value"`
  17. }
  18. func AllOption() ([]*Option, error) {
  19. var options []*Option
  20. var err error
  21. err = DB.Find(&options).Error
  22. return options, err
  23. }
  24. func InitOptionMap() {
  25. common.OptionMapRWMutex.Lock()
  26. common.OptionMap = make(map[string]string)
  27. // 添加原有的系统配置
  28. common.OptionMap["FileUploadPermission"] = strconv.Itoa(common.FileUploadPermission)
  29. common.OptionMap["FileDownloadPermission"] = strconv.Itoa(common.FileDownloadPermission)
  30. common.OptionMap["ImageUploadPermission"] = strconv.Itoa(common.ImageUploadPermission)
  31. common.OptionMap["ImageDownloadPermission"] = strconv.Itoa(common.ImageDownloadPermission)
  32. common.OptionMap["PasswordLoginEnabled"] = strconv.FormatBool(common.PasswordLoginEnabled)
  33. common.OptionMap["PasswordRegisterEnabled"] = strconv.FormatBool(common.PasswordRegisterEnabled)
  34. common.OptionMap["EmailVerificationEnabled"] = strconv.FormatBool(common.EmailVerificationEnabled)
  35. common.OptionMap["GitHubOAuthEnabled"] = strconv.FormatBool(common.GitHubOAuthEnabled)
  36. common.OptionMap["LinuxDOOAuthEnabled"] = strconv.FormatBool(common.LinuxDOOAuthEnabled)
  37. common.OptionMap["TelegramOAuthEnabled"] = strconv.FormatBool(common.TelegramOAuthEnabled)
  38. common.OptionMap["WeChatAuthEnabled"] = strconv.FormatBool(common.WeChatAuthEnabled)
  39. common.OptionMap["TurnstileCheckEnabled"] = strconv.FormatBool(common.TurnstileCheckEnabled)
  40. common.OptionMap["RegisterEnabled"] = strconv.FormatBool(common.RegisterEnabled)
  41. common.OptionMap["AutomaticDisableChannelEnabled"] = strconv.FormatBool(common.AutomaticDisableChannelEnabled)
  42. common.OptionMap["AutomaticEnableChannelEnabled"] = strconv.FormatBool(common.AutomaticEnableChannelEnabled)
  43. common.OptionMap["LogConsumeEnabled"] = strconv.FormatBool(common.LogConsumeEnabled)
  44. common.OptionMap["DisplayInCurrencyEnabled"] = strconv.FormatBool(common.DisplayInCurrencyEnabled)
  45. common.OptionMap["DisplayTokenStatEnabled"] = strconv.FormatBool(common.DisplayTokenStatEnabled)
  46. common.OptionMap["DrawingEnabled"] = strconv.FormatBool(common.DrawingEnabled)
  47. common.OptionMap["TaskEnabled"] = strconv.FormatBool(common.TaskEnabled)
  48. common.OptionMap["DataExportEnabled"] = strconv.FormatBool(common.DataExportEnabled)
  49. common.OptionMap["ChannelDisableThreshold"] = strconv.FormatFloat(common.ChannelDisableThreshold, 'f', -1, 64)
  50. common.OptionMap["EmailDomainRestrictionEnabled"] = strconv.FormatBool(common.EmailDomainRestrictionEnabled)
  51. common.OptionMap["EmailAliasRestrictionEnabled"] = strconv.FormatBool(common.EmailAliasRestrictionEnabled)
  52. common.OptionMap["EmailDomainWhitelist"] = strings.Join(common.EmailDomainWhitelist, ",")
  53. common.OptionMap["SMTPServer"] = ""
  54. common.OptionMap["SMTPFrom"] = ""
  55. common.OptionMap["SMTPPort"] = strconv.Itoa(common.SMTPPort)
  56. common.OptionMap["SMTPAccount"] = ""
  57. common.OptionMap["SMTPToken"] = ""
  58. common.OptionMap["SMTPSSLEnabled"] = strconv.FormatBool(common.SMTPSSLEnabled)
  59. common.OptionMap["SMTPForceAuthLogin"] = strconv.FormatBool(common.SMTPForceAuthLogin)
  60. common.OptionMap["Notice"] = ""
  61. common.OptionMap["About"] = ""
  62. common.OptionMap["HomePageContent"] = ""
  63. common.OptionMap["Footer"] = common.Footer
  64. common.OptionMap["SystemName"] = common.SystemName
  65. common.OptionMap["Logo"] = common.Logo
  66. common.OptionMap["ServerAddress"] = ""
  67. common.OptionMap["WorkerUrl"] = system_setting.WorkerUrl
  68. common.OptionMap["WorkerValidKey"] = system_setting.WorkerValidKey
  69. common.OptionMap["WorkerAllowHttpImageRequestEnabled"] = strconv.FormatBool(system_setting.WorkerAllowHttpImageRequestEnabled)
  70. common.OptionMap["PayAddress"] = ""
  71. common.OptionMap["CustomCallbackAddress"] = ""
  72. common.OptionMap["EpayId"] = ""
  73. common.OptionMap["EpayKey"] = ""
  74. common.OptionMap["Price"] = strconv.FormatFloat(operation_setting.Price, 'f', -1, 64)
  75. common.OptionMap["USDExchangeRate"] = strconv.FormatFloat(operation_setting.USDExchangeRate, 'f', -1, 64)
  76. common.OptionMap["MinTopUp"] = strconv.Itoa(operation_setting.MinTopUp)
  77. common.OptionMap["StripeMinTopUp"] = strconv.Itoa(setting.StripeMinTopUp)
  78. common.OptionMap["StripeApiSecret"] = setting.StripeApiSecret
  79. common.OptionMap["StripeWebhookSecret"] = setting.StripeWebhookSecret
  80. common.OptionMap["StripePriceId"] = setting.StripePriceId
  81. common.OptionMap["StripeUnitPrice"] = strconv.FormatFloat(setting.StripeUnitPrice, 'f', -1, 64)
  82. common.OptionMap["StripePromotionCodesEnabled"] = strconv.FormatBool(setting.StripePromotionCodesEnabled)
  83. common.OptionMap["CreemApiKey"] = setting.CreemApiKey
  84. common.OptionMap["CreemProducts"] = setting.CreemProducts
  85. common.OptionMap["CreemTestMode"] = strconv.FormatBool(setting.CreemTestMode)
  86. common.OptionMap["CreemWebhookSecret"] = setting.CreemWebhookSecret
  87. common.OptionMap["WaffoEnabled"] = strconv.FormatBool(setting.WaffoEnabled)
  88. common.OptionMap["WaffoApiKey"] = setting.WaffoApiKey
  89. common.OptionMap["WaffoPrivateKey"] = setting.WaffoPrivateKey
  90. common.OptionMap["WaffoPublicCert"] = setting.WaffoPublicCert
  91. common.OptionMap["WaffoSandboxPublicCert"] = setting.WaffoSandboxPublicCert
  92. common.OptionMap["WaffoSandboxApiKey"] = setting.WaffoSandboxApiKey
  93. common.OptionMap["WaffoSandboxPrivateKey"] = setting.WaffoSandboxPrivateKey
  94. common.OptionMap["WaffoSandbox"] = strconv.FormatBool(setting.WaffoSandbox)
  95. common.OptionMap["WaffoMerchantId"] = setting.WaffoMerchantId
  96. common.OptionMap["WaffoNotifyUrl"] = setting.WaffoNotifyUrl
  97. common.OptionMap["WaffoReturnUrl"] = setting.WaffoReturnUrl
  98. common.OptionMap["WaffoSubscriptionReturnUrl"] = setting.WaffoSubscriptionReturnUrl
  99. common.OptionMap["WaffoCurrency"] = setting.WaffoCurrency
  100. common.OptionMap["WaffoUnitPrice"] = strconv.FormatFloat(setting.WaffoUnitPrice, 'f', -1, 64)
  101. common.OptionMap["WaffoMinTopUp"] = strconv.Itoa(setting.WaffoMinTopUp)
  102. common.OptionMap["WaffoPayMethods"] = setting.WaffoPayMethods2JsonString()
  103. common.OptionMap["WaffoPancakeEnabled"] = strconv.FormatBool(setting.WaffoPancakeEnabled)
  104. common.OptionMap["WaffoPancakeSandbox"] = strconv.FormatBool(setting.WaffoPancakeSandbox)
  105. common.OptionMap["WaffoPancakeMerchantID"] = setting.WaffoPancakeMerchantID
  106. common.OptionMap["WaffoPancakePrivateKey"] = setting.WaffoPancakePrivateKey
  107. common.OptionMap["WaffoPancakeWebhookPublicKey"] = setting.WaffoPancakeWebhookPublicKey
  108. common.OptionMap["WaffoPancakeWebhookTestKey"] = setting.WaffoPancakeWebhookTestKey
  109. common.OptionMap["WaffoPancakeStoreID"] = setting.WaffoPancakeStoreID
  110. common.OptionMap["WaffoPancakeProductID"] = setting.WaffoPancakeProductID
  111. common.OptionMap["WaffoPancakeReturnURL"] = setting.WaffoPancakeReturnURL
  112. common.OptionMap["WaffoPancakeCurrency"] = setting.WaffoPancakeCurrency
  113. common.OptionMap["WaffoPancakeUnitPrice"] = strconv.FormatFloat(setting.WaffoPancakeUnitPrice, 'f', -1, 64)
  114. common.OptionMap["WaffoPancakeMinTopUp"] = strconv.Itoa(setting.WaffoPancakeMinTopUp)
  115. common.OptionMap["TopupGroupRatio"] = common.TopupGroupRatio2JSONString()
  116. common.OptionMap["Chats"] = setting.Chats2JsonString()
  117. common.OptionMap["AutoGroups"] = setting.AutoGroups2JsonString()
  118. common.OptionMap["DefaultUseAutoGroup"] = strconv.FormatBool(setting.DefaultUseAutoGroup)
  119. common.OptionMap["PayMethods"] = operation_setting.PayMethods2JsonString()
  120. common.OptionMap["GitHubClientId"] = ""
  121. common.OptionMap["GitHubClientSecret"] = ""
  122. common.OptionMap["TelegramBotToken"] = ""
  123. common.OptionMap["TelegramBotName"] = ""
  124. common.OptionMap["WeChatServerAddress"] = ""
  125. common.OptionMap["WeChatServerToken"] = ""
  126. common.OptionMap["WeChatAccountQRCodeImageURL"] = ""
  127. common.OptionMap["TurnstileSiteKey"] = ""
  128. common.OptionMap["TurnstileSecretKey"] = ""
  129. common.OptionMap["QuotaForNewUser"] = strconv.Itoa(common.QuotaForNewUser)
  130. common.OptionMap["QuotaForInviter"] = strconv.Itoa(common.QuotaForInviter)
  131. common.OptionMap["QuotaForInvitee"] = strconv.Itoa(common.QuotaForInvitee)
  132. common.OptionMap["QuotaRemindThreshold"] = strconv.Itoa(common.QuotaRemindThreshold)
  133. common.OptionMap["PreConsumedQuota"] = strconv.Itoa(common.PreConsumedQuota)
  134. common.OptionMap["ModelRequestRateLimitCount"] = strconv.Itoa(setting.ModelRequestRateLimitCount)
  135. common.OptionMap["ModelRequestRateLimitDurationMinutes"] = strconv.Itoa(setting.ModelRequestRateLimitDurationMinutes)
  136. common.OptionMap["ModelRequestRateLimitSuccessCount"] = strconv.Itoa(setting.ModelRequestRateLimitSuccessCount)
  137. common.OptionMap["ModelRequestRateLimitGroup"] = setting.ModelRequestRateLimitGroup2JSONString()
  138. common.OptionMap["ModelRatio"] = ratio_setting.ModelRatio2JSONString()
  139. common.OptionMap["ModelPrice"] = ratio_setting.ModelPrice2JSONString()
  140. common.OptionMap["CacheRatio"] = ratio_setting.CacheRatio2JSONString()
  141. common.OptionMap["CreateCacheRatio"] = ratio_setting.CreateCacheRatio2JSONString()
  142. common.OptionMap["GroupRatio"] = ratio_setting.GroupRatio2JSONString()
  143. common.OptionMap["GroupGroupRatio"] = ratio_setting.GroupGroupRatio2JSONString()
  144. common.OptionMap["UserUsableGroups"] = setting.UserUsableGroups2JSONString()
  145. common.OptionMap["CompletionRatio"] = ratio_setting.CompletionRatio2JSONString()
  146. common.OptionMap["ImageRatio"] = ratio_setting.ImageRatio2JSONString()
  147. common.OptionMap["AudioRatio"] = ratio_setting.AudioRatio2JSONString()
  148. common.OptionMap["AudioCompletionRatio"] = ratio_setting.AudioCompletionRatio2JSONString()
  149. common.OptionMap["TopUpLink"] = common.TopUpLink
  150. //common.OptionMap["ChatLink"] = common.ChatLink
  151. //common.OptionMap["ChatLink2"] = common.ChatLink2
  152. common.OptionMap["QuotaPerUnit"] = strconv.FormatFloat(common.QuotaPerUnit, 'f', -1, 64)
  153. common.OptionMap["RetryTimes"] = strconv.Itoa(common.RetryTimes)
  154. common.OptionMap["DataExportInterval"] = strconv.Itoa(common.DataExportInterval)
  155. common.OptionMap["DataExportDefaultTime"] = common.DataExportDefaultTime
  156. common.OptionMap["DefaultCollapseSidebar"] = strconv.FormatBool(common.DefaultCollapseSidebar)
  157. common.OptionMap["MjNotifyEnabled"] = strconv.FormatBool(setting.MjNotifyEnabled)
  158. common.OptionMap["MjAccountFilterEnabled"] = strconv.FormatBool(setting.MjAccountFilterEnabled)
  159. common.OptionMap["MjModeClearEnabled"] = strconv.FormatBool(setting.MjModeClearEnabled)
  160. common.OptionMap["MjForwardUrlEnabled"] = strconv.FormatBool(setting.MjForwardUrlEnabled)
  161. common.OptionMap["MjActionCheckSuccessEnabled"] = strconv.FormatBool(setting.MjActionCheckSuccessEnabled)
  162. common.OptionMap["CheckSensitiveEnabled"] = strconv.FormatBool(setting.CheckSensitiveEnabled)
  163. common.OptionMap["DemoSiteEnabled"] = strconv.FormatBool(operation_setting.DemoSiteEnabled)
  164. common.OptionMap["SelfUseModeEnabled"] = strconv.FormatBool(operation_setting.SelfUseModeEnabled)
  165. common.OptionMap["ModelRequestRateLimitEnabled"] = strconv.FormatBool(setting.ModelRequestRateLimitEnabled)
  166. common.OptionMap["CheckSensitiveOnPromptEnabled"] = strconv.FormatBool(setting.CheckSensitiveOnPromptEnabled)
  167. common.OptionMap["StopOnSensitiveEnabled"] = strconv.FormatBool(setting.StopOnSensitiveEnabled)
  168. common.OptionMap["SensitiveWords"] = setting.SensitiveWordsToString()
  169. common.OptionMap["StreamCacheQueueLength"] = strconv.Itoa(setting.StreamCacheQueueLength)
  170. common.OptionMap["AutomaticDisableKeywords"] = operation_setting.AutomaticDisableKeywordsToString()
  171. common.OptionMap["AutomaticDisableStatusCodes"] = operation_setting.AutomaticDisableStatusCodesToString()
  172. common.OptionMap["AutomaticRetryStatusCodes"] = operation_setting.AutomaticRetryStatusCodesToString()
  173. common.OptionMap["ExposeRatioEnabled"] = strconv.FormatBool(ratio_setting.IsExposeRatioEnabled())
  174. // 自动添加所有注册的模型配置
  175. modelConfigs := config.GlobalConfig.ExportAllConfigs()
  176. for k, v := range modelConfigs {
  177. common.OptionMap[k] = v
  178. }
  179. common.OptionMapRWMutex.Unlock()
  180. loadOptionsFromDatabase()
  181. }
  182. func loadOptionsFromDatabase() {
  183. options, _ := AllOption()
  184. for _, option := range options {
  185. err := updateOptionMap(option.Key, option.Value)
  186. if err != nil {
  187. common.SysLog("failed to update option map: " + err.Error())
  188. }
  189. }
  190. }
  191. func SyncOptions(frequency int) {
  192. for {
  193. time.Sleep(time.Duration(frequency) * time.Second)
  194. common.SysLog("syncing options from database")
  195. loadOptionsFromDatabase()
  196. }
  197. }
  198. func UpdateOption(key string, value string) error {
  199. // Save to database first
  200. option := Option{
  201. Key: key,
  202. }
  203. // https://gorm.io/docs/update.html#Save-All-Fields
  204. DB.FirstOrCreate(&option, Option{Key: key})
  205. option.Value = value
  206. // Save is a combination function.
  207. // If save value does not contain primary key, it will execute Create,
  208. // otherwise it will execute Update (with all fields).
  209. DB.Save(&option)
  210. // Update OptionMap
  211. return updateOptionMap(key, value)
  212. }
  213. func updateOptionMap(key string, value string) (err error) {
  214. common.OptionMapRWMutex.Lock()
  215. defer common.OptionMapRWMutex.Unlock()
  216. common.OptionMap[key] = value
  217. // 检查是否是模型配置 - 使用更规范的方式处理
  218. if handleConfigUpdate(key, value) {
  219. return nil // 已由配置系统处理
  220. }
  221. // 处理传统配置项...
  222. if strings.HasSuffix(key, "Permission") {
  223. intValue, _ := strconv.Atoi(value)
  224. switch key {
  225. case "FileUploadPermission":
  226. common.FileUploadPermission = intValue
  227. case "FileDownloadPermission":
  228. common.FileDownloadPermission = intValue
  229. case "ImageUploadPermission":
  230. common.ImageUploadPermission = intValue
  231. case "ImageDownloadPermission":
  232. common.ImageDownloadPermission = intValue
  233. }
  234. }
  235. if strings.HasSuffix(key, "Enabled") || key == "DefaultCollapseSidebar" || key == "DefaultUseAutoGroup" || key == "SMTPForceAuthLogin" {
  236. boolValue := value == "true"
  237. switch key {
  238. case "PasswordRegisterEnabled":
  239. common.PasswordRegisterEnabled = boolValue
  240. case "PasswordLoginEnabled":
  241. common.PasswordLoginEnabled = boolValue
  242. case "EmailVerificationEnabled":
  243. common.EmailVerificationEnabled = boolValue
  244. case "GitHubOAuthEnabled":
  245. common.GitHubOAuthEnabled = boolValue
  246. case "LinuxDOOAuthEnabled":
  247. common.LinuxDOOAuthEnabled = boolValue
  248. case "WeChatAuthEnabled":
  249. common.WeChatAuthEnabled = boolValue
  250. case "TelegramOAuthEnabled":
  251. common.TelegramOAuthEnabled = boolValue
  252. case "TurnstileCheckEnabled":
  253. common.TurnstileCheckEnabled = boolValue
  254. case "RegisterEnabled":
  255. common.RegisterEnabled = boolValue
  256. case "EmailDomainRestrictionEnabled":
  257. common.EmailDomainRestrictionEnabled = boolValue
  258. case "EmailAliasRestrictionEnabled":
  259. common.EmailAliasRestrictionEnabled = boolValue
  260. case "AutomaticDisableChannelEnabled":
  261. common.AutomaticDisableChannelEnabled = boolValue
  262. case "AutomaticEnableChannelEnabled":
  263. common.AutomaticEnableChannelEnabled = boolValue
  264. case "LogConsumeEnabled":
  265. common.LogConsumeEnabled = boolValue
  266. case "DisplayInCurrencyEnabled":
  267. // 兼容旧字段:同步到新配置 general_setting.quota_display_type(运行时生效)
  268. // true -> USD, false -> TOKENS
  269. newVal := "USD"
  270. if !boolValue {
  271. newVal = "TOKENS"
  272. }
  273. if cfg := config.GlobalConfig.Get("general_setting"); cfg != nil {
  274. _ = config.UpdateConfigFromMap(cfg, map[string]string{"quota_display_type": newVal})
  275. }
  276. case "DisplayTokenStatEnabled":
  277. common.DisplayTokenStatEnabled = boolValue
  278. case "DrawingEnabled":
  279. common.DrawingEnabled = boolValue
  280. case "TaskEnabled":
  281. common.TaskEnabled = boolValue
  282. case "DataExportEnabled":
  283. common.DataExportEnabled = boolValue
  284. case "DefaultCollapseSidebar":
  285. common.DefaultCollapseSidebar = boolValue
  286. case "MjNotifyEnabled":
  287. setting.MjNotifyEnabled = boolValue
  288. case "MjAccountFilterEnabled":
  289. setting.MjAccountFilterEnabled = boolValue
  290. case "MjModeClearEnabled":
  291. setting.MjModeClearEnabled = boolValue
  292. case "MjForwardUrlEnabled":
  293. setting.MjForwardUrlEnabled = boolValue
  294. case "MjActionCheckSuccessEnabled":
  295. setting.MjActionCheckSuccessEnabled = boolValue
  296. case "CheckSensitiveEnabled":
  297. setting.CheckSensitiveEnabled = boolValue
  298. case "DemoSiteEnabled":
  299. operation_setting.DemoSiteEnabled = boolValue
  300. case "SelfUseModeEnabled":
  301. operation_setting.SelfUseModeEnabled = boolValue
  302. case "CheckSensitiveOnPromptEnabled":
  303. setting.CheckSensitiveOnPromptEnabled = boolValue
  304. case "ModelRequestRateLimitEnabled":
  305. setting.ModelRequestRateLimitEnabled = boolValue
  306. case "StopOnSensitiveEnabled":
  307. setting.StopOnSensitiveEnabled = boolValue
  308. case "SMTPSSLEnabled":
  309. common.SMTPSSLEnabled = boolValue
  310. case "SMTPForceAuthLogin":
  311. common.SMTPForceAuthLogin = boolValue
  312. case "WorkerAllowHttpImageRequestEnabled":
  313. system_setting.WorkerAllowHttpImageRequestEnabled = boolValue
  314. case "DefaultUseAutoGroup":
  315. setting.DefaultUseAutoGroup = boolValue
  316. case "ExposeRatioEnabled":
  317. ratio_setting.SetExposeRatioEnabled(boolValue)
  318. }
  319. }
  320. switch key {
  321. case "EmailDomainWhitelist":
  322. common.EmailDomainWhitelist = strings.Split(value, ",")
  323. case "SMTPServer":
  324. common.SMTPServer = value
  325. case "SMTPPort":
  326. intValue, _ := strconv.Atoi(value)
  327. common.SMTPPort = intValue
  328. case "SMTPAccount":
  329. common.SMTPAccount = value
  330. case "SMTPFrom":
  331. common.SMTPFrom = value
  332. case "SMTPToken":
  333. common.SMTPToken = value
  334. case "ServerAddress":
  335. system_setting.ServerAddress = value
  336. case "WorkerUrl":
  337. system_setting.WorkerUrl = value
  338. case "WorkerValidKey":
  339. system_setting.WorkerValidKey = value
  340. case "PayAddress":
  341. operation_setting.PayAddress = value
  342. case "Chats":
  343. err = setting.UpdateChatsByJsonString(value)
  344. case "AutoGroups":
  345. err = setting.UpdateAutoGroupsByJsonString(value)
  346. case "CustomCallbackAddress":
  347. operation_setting.CustomCallbackAddress = value
  348. case "EpayId":
  349. operation_setting.EpayId = value
  350. case "EpayKey":
  351. operation_setting.EpayKey = value
  352. case "Price":
  353. operation_setting.Price, _ = strconv.ParseFloat(value, 64)
  354. case "USDExchangeRate":
  355. operation_setting.USDExchangeRate, _ = strconv.ParseFloat(value, 64)
  356. case "MinTopUp":
  357. operation_setting.MinTopUp, _ = strconv.Atoi(value)
  358. case "StripeApiSecret":
  359. setting.StripeApiSecret = value
  360. case "StripeWebhookSecret":
  361. setting.StripeWebhookSecret = value
  362. case "StripePriceId":
  363. setting.StripePriceId = value
  364. case "StripeUnitPrice":
  365. setting.StripeUnitPrice, _ = strconv.ParseFloat(value, 64)
  366. case "StripeMinTopUp":
  367. setting.StripeMinTopUp, _ = strconv.Atoi(value)
  368. case "StripePromotionCodesEnabled":
  369. setting.StripePromotionCodesEnabled = value == "true"
  370. case "CreemApiKey":
  371. setting.CreemApiKey = value
  372. case "CreemProducts":
  373. setting.CreemProducts = value
  374. case "CreemTestMode":
  375. setting.CreemTestMode = value == "true"
  376. case "CreemWebhookSecret":
  377. setting.CreemWebhookSecret = value
  378. case "WaffoEnabled":
  379. setting.WaffoEnabled = value == "true"
  380. case "WaffoApiKey":
  381. setting.WaffoApiKey = value
  382. case "WaffoPrivateKey":
  383. setting.WaffoPrivateKey = value
  384. case "WaffoPublicCert":
  385. setting.WaffoPublicCert = value
  386. case "WaffoSandboxPublicCert":
  387. setting.WaffoSandboxPublicCert = value
  388. case "WaffoSandboxApiKey":
  389. setting.WaffoSandboxApiKey = value
  390. case "WaffoSandboxPrivateKey":
  391. setting.WaffoSandboxPrivateKey = value
  392. case "WaffoSandbox":
  393. setting.WaffoSandbox = value == "true"
  394. case "WaffoMerchantId":
  395. setting.WaffoMerchantId = value
  396. case "WaffoNotifyUrl":
  397. setting.WaffoNotifyUrl = value
  398. case "WaffoReturnUrl":
  399. setting.WaffoReturnUrl = value
  400. case "WaffoSubscriptionReturnUrl":
  401. setting.WaffoSubscriptionReturnUrl = value
  402. case "WaffoCurrency":
  403. setting.WaffoCurrency = value
  404. case "WaffoUnitPrice":
  405. setting.WaffoUnitPrice, _ = strconv.ParseFloat(value, 64)
  406. case "WaffoMinTopUp":
  407. setting.WaffoMinTopUp, _ = strconv.Atoi(value)
  408. case "WaffoPancakeEnabled":
  409. setting.WaffoPancakeEnabled = value == "true"
  410. case "WaffoPancakeSandbox":
  411. setting.WaffoPancakeSandbox = value == "true"
  412. case "WaffoPancakeMerchantID":
  413. setting.WaffoPancakeMerchantID = value
  414. case "WaffoPancakePrivateKey":
  415. setting.WaffoPancakePrivateKey = value
  416. case "WaffoPancakeWebhookPublicKey":
  417. setting.WaffoPancakeWebhookPublicKey = value
  418. case "WaffoPancakeWebhookTestKey":
  419. setting.WaffoPancakeWebhookTestKey = value
  420. case "WaffoPancakeStoreID":
  421. setting.WaffoPancakeStoreID = value
  422. case "WaffoPancakeProductID":
  423. setting.WaffoPancakeProductID = value
  424. case "WaffoPancakeReturnURL":
  425. setting.WaffoPancakeReturnURL = value
  426. case "WaffoPancakeCurrency":
  427. setting.WaffoPancakeCurrency = value
  428. case "WaffoPancakeUnitPrice":
  429. setting.WaffoPancakeUnitPrice, _ = strconv.ParseFloat(value, 64)
  430. case "WaffoPancakeMinTopUp":
  431. setting.WaffoPancakeMinTopUp, _ = strconv.Atoi(value)
  432. case "TopupGroupRatio":
  433. err = common.UpdateTopupGroupRatioByJSONString(value)
  434. case "GitHubClientId":
  435. common.GitHubClientId = value
  436. case "GitHubClientSecret":
  437. common.GitHubClientSecret = value
  438. case "LinuxDOClientId":
  439. common.LinuxDOClientId = value
  440. case "LinuxDOClientSecret":
  441. common.LinuxDOClientSecret = value
  442. case "LinuxDOMinimumTrustLevel":
  443. common.LinuxDOMinimumTrustLevel, _ = strconv.Atoi(value)
  444. case "Footer":
  445. common.Footer = value
  446. case "SystemName":
  447. common.SystemName = value
  448. case "Logo":
  449. common.Logo = value
  450. case "WeChatServerAddress":
  451. common.WeChatServerAddress = value
  452. case "WeChatServerToken":
  453. common.WeChatServerToken = value
  454. case "WeChatAccountQRCodeImageURL":
  455. common.WeChatAccountQRCodeImageURL = value
  456. case "TelegramBotToken":
  457. common.TelegramBotToken = value
  458. case "TelegramBotName":
  459. common.TelegramBotName = value
  460. case "TurnstileSiteKey":
  461. common.TurnstileSiteKey = value
  462. case "TurnstileSecretKey":
  463. common.TurnstileSecretKey = value
  464. case "QuotaForNewUser":
  465. common.QuotaForNewUser, _ = strconv.Atoi(value)
  466. case "QuotaForInviter":
  467. common.QuotaForInviter, _ = strconv.Atoi(value)
  468. case "QuotaForInvitee":
  469. common.QuotaForInvitee, _ = strconv.Atoi(value)
  470. case "QuotaRemindThreshold":
  471. common.QuotaRemindThreshold, _ = strconv.Atoi(value)
  472. case "PreConsumedQuota":
  473. common.PreConsumedQuota, _ = strconv.Atoi(value)
  474. case "ModelRequestRateLimitCount":
  475. setting.ModelRequestRateLimitCount, _ = strconv.Atoi(value)
  476. case "ModelRequestRateLimitDurationMinutes":
  477. setting.ModelRequestRateLimitDurationMinutes, _ = strconv.Atoi(value)
  478. case "ModelRequestRateLimitSuccessCount":
  479. setting.ModelRequestRateLimitSuccessCount, _ = strconv.Atoi(value)
  480. case "ModelRequestRateLimitGroup":
  481. err = setting.UpdateModelRequestRateLimitGroupByJSONString(value)
  482. case "RetryTimes":
  483. common.RetryTimes, _ = strconv.Atoi(value)
  484. case "DataExportInterval":
  485. common.DataExportInterval, _ = strconv.Atoi(value)
  486. case "DataExportDefaultTime":
  487. common.DataExportDefaultTime = value
  488. case "ModelRatio":
  489. err = ratio_setting.UpdateModelRatioByJSONString(value)
  490. case "GroupRatio":
  491. err = ratio_setting.UpdateGroupRatioByJSONString(value)
  492. case "GroupGroupRatio":
  493. err = ratio_setting.UpdateGroupGroupRatioByJSONString(value)
  494. case "UserUsableGroups":
  495. err = setting.UpdateUserUsableGroupsByJSONString(value)
  496. case "CompletionRatio":
  497. err = ratio_setting.UpdateCompletionRatioByJSONString(value)
  498. case "ModelPrice":
  499. err = ratio_setting.UpdateModelPriceByJSONString(value)
  500. case "CacheRatio":
  501. err = ratio_setting.UpdateCacheRatioByJSONString(value)
  502. case "CreateCacheRatio":
  503. err = ratio_setting.UpdateCreateCacheRatioByJSONString(value)
  504. case "ImageRatio":
  505. err = ratio_setting.UpdateImageRatioByJSONString(value)
  506. case "AudioRatio":
  507. err = ratio_setting.UpdateAudioRatioByJSONString(value)
  508. case "AudioCompletionRatio":
  509. err = ratio_setting.UpdateAudioCompletionRatioByJSONString(value)
  510. case "TopUpLink":
  511. common.TopUpLink = value
  512. //case "ChatLink":
  513. // common.ChatLink = value
  514. //case "ChatLink2":
  515. // common.ChatLink2 = value
  516. case "ChannelDisableThreshold":
  517. common.ChannelDisableThreshold, _ = strconv.ParseFloat(value, 64)
  518. case "QuotaPerUnit":
  519. common.QuotaPerUnit, _ = strconv.ParseFloat(value, 64)
  520. case "SensitiveWords":
  521. setting.SensitiveWordsFromString(value)
  522. case "AutomaticDisableKeywords":
  523. operation_setting.AutomaticDisableKeywordsFromString(value)
  524. case "AutomaticDisableStatusCodes":
  525. err = operation_setting.AutomaticDisableStatusCodesFromString(value)
  526. case "AutomaticRetryStatusCodes":
  527. err = operation_setting.AutomaticRetryStatusCodesFromString(value)
  528. case "StreamCacheQueueLength":
  529. setting.StreamCacheQueueLength, _ = strconv.Atoi(value)
  530. case "PayMethods":
  531. err = operation_setting.UpdatePayMethodsByJsonString(value)
  532. case "WaffoPayMethods":
  533. // WaffoPayMethods is read directly from OptionMap via setting.GetWaffoPayMethods().
  534. // The value is already stored in OptionMap at the top of this function (line: common.OptionMap[key] = value).
  535. // No additional in-memory variable to update.
  536. }
  537. return err
  538. }
  539. // handleConfigUpdate 处理分层配置更新,返回是否已处理
  540. func handleConfigUpdate(key, value string) bool {
  541. parts := strings.SplitN(key, ".", 2)
  542. if len(parts) != 2 {
  543. return false // 不是分层配置
  544. }
  545. configName := parts[0]
  546. configKey := parts[1]
  547. // 获取配置对象
  548. cfg := config.GlobalConfig.Get(configName)
  549. if cfg == nil {
  550. return false // 未注册的配置
  551. }
  552. // 更新配置
  553. configMap := map[string]string{
  554. configKey: value,
  555. }
  556. config.UpdateConfigFromMap(cfg, configMap)
  557. // 特定配置的后处理
  558. if configName == "performance_setting" {
  559. // 同步磁盘缓存配置到 common 包
  560. performance_setting.UpdateAndSync()
  561. }
  562. return true // 已处理
  563. }