constants.go 4.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204
  1. package common
  2. import (
  3. //"os"
  4. //"strconv"
  5. "sync"
  6. "time"
  7. "github.com/google/uuid"
  8. )
  9. var StartTime = time.Now().Unix() // unit: second
  10. var Version = "v0.0.0" // this hard coding will be replaced automatically when building, no need to manually change
  11. var SystemName = "New API"
  12. var Footer = ""
  13. var Logo = ""
  14. var TopUpLink = ""
  15. // var ChatLink = ""
  16. // var ChatLink2 = ""
  17. var QuotaPerUnit = 500 * 1000.0 // $0.002 / 1K tokens
  18. // 保留旧变量以兼容历史逻辑,实际展示由 general_setting.quota_display_type 控制
  19. var DisplayInCurrencyEnabled = true
  20. var DisplayTokenStatEnabled = true
  21. var DrawingEnabled = true
  22. var TaskEnabled = true
  23. var DataExportEnabled = true
  24. var DataExportInterval = 5 // unit: minute
  25. var DataExportDefaultTime = "hour" // unit: minute
  26. var DefaultCollapseSidebar = false // default value of collapse sidebar
  27. // Any options with "Secret", "Token" in its key won't be return by GetOptions
  28. var SessionSecret = uuid.New().String()
  29. var CryptoSecret = uuid.New().String()
  30. var OptionMap map[string]string
  31. var OptionMapRWMutex sync.RWMutex
  32. var ItemsPerPage = 10
  33. var MaxRecentItems = 100
  34. var PasswordLoginEnabled = true
  35. var PasswordRegisterEnabled = true
  36. var EmailVerificationEnabled = false
  37. var GitHubOAuthEnabled = false
  38. var LinuxDOOAuthEnabled = false
  39. var WeChatAuthEnabled = false
  40. var TelegramOAuthEnabled = false
  41. var TurnstileCheckEnabled = false
  42. var RegisterEnabled = true
  43. var EmailDomainRestrictionEnabled = false // 是否启用邮箱域名限制
  44. var EmailAliasRestrictionEnabled = false // 是否启用邮箱别名限制
  45. var EmailDomainWhitelist = []string{
  46. "gmail.com",
  47. "163.com",
  48. "126.com",
  49. "qq.com",
  50. "outlook.com",
  51. "hotmail.com",
  52. "icloud.com",
  53. "yahoo.com",
  54. "foxmail.com",
  55. }
  56. var EmailLoginAuthServerList = []string{
  57. "smtp.sendcloud.net",
  58. "smtp.azurecomm.net",
  59. }
  60. var DebugEnabled bool
  61. var MemoryCacheEnabled bool
  62. var LogConsumeEnabled = true
  63. var SMTPServer = ""
  64. var SMTPPort = 587
  65. var SMTPSSLEnabled = false
  66. var SMTPAccount = ""
  67. var SMTPFrom = ""
  68. var SMTPToken = ""
  69. var GitHubClientId = ""
  70. var GitHubClientSecret = ""
  71. var LinuxDOClientId = ""
  72. var LinuxDOClientSecret = ""
  73. var LinuxDOMinimumTrustLevel = 0
  74. var WeChatServerAddress = ""
  75. var WeChatServerToken = ""
  76. var WeChatAccountQRCodeImageURL = ""
  77. var TurnstileSiteKey = ""
  78. var TurnstileSecretKey = ""
  79. var TelegramBotToken = ""
  80. var TelegramBotName = ""
  81. var QuotaForNewUser = 0
  82. var QuotaForInviter = 0
  83. var QuotaForInvitee = 0
  84. var ChannelDisableThreshold = 5.0
  85. var AutomaticDisableChannelEnabled = false
  86. var AutomaticEnableChannelEnabled = false
  87. var QuotaRemindThreshold = 1000
  88. var PreConsumedQuota = 500
  89. var RetryTimes = 0
  90. //var RootUserEmail = ""
  91. var IsMasterNode bool
  92. var requestInterval int
  93. var RequestInterval time.Duration
  94. var SyncFrequency int // unit is second
  95. var BatchUpdateEnabled = false
  96. var BatchUpdateInterval int
  97. var RelayTimeout int // unit is second
  98. var GeminiSafetySetting string
  99. // https://docs.cohere.com/docs/safety-modes Type; NONE/CONTEXTUAL/STRICT
  100. var CohereSafetySetting string
  101. const (
  102. RequestIdKey = "X-Oneapi-Request-Id"
  103. )
  104. const (
  105. RoleGuestUser = 0
  106. RoleCommonUser = 1
  107. RoleAdminUser = 10
  108. RoleRootUser = 100
  109. )
  110. func IsValidateRole(role int) bool {
  111. return role == RoleGuestUser || role == RoleCommonUser || role == RoleAdminUser || role == RoleRootUser
  112. }
  113. var (
  114. FileUploadPermission = RoleGuestUser
  115. FileDownloadPermission = RoleGuestUser
  116. ImageUploadPermission = RoleGuestUser
  117. ImageDownloadPermission = RoleGuestUser
  118. )
  119. // All duration's unit is seconds
  120. // Shouldn't larger then RateLimitKeyExpirationDuration
  121. var (
  122. GlobalApiRateLimitEnable bool
  123. GlobalApiRateLimitNum int
  124. GlobalApiRateLimitDuration int64
  125. GlobalWebRateLimitEnable bool
  126. GlobalWebRateLimitNum int
  127. GlobalWebRateLimitDuration int64
  128. CriticalRateLimitEnable bool
  129. CriticalRateLimitNum = 20
  130. CriticalRateLimitDuration int64 = 20 * 60
  131. UploadRateLimitNum = 10
  132. UploadRateLimitDuration int64 = 60
  133. DownloadRateLimitNum = 10
  134. DownloadRateLimitDuration int64 = 60
  135. )
  136. var RateLimitKeyExpirationDuration = 20 * time.Minute
  137. const (
  138. UserStatusEnabled = 1 // don't use 0, 0 is the default value!
  139. UserStatusDisabled = 2 // also don't use 0
  140. )
  141. const (
  142. TokenStatusEnabled = 1 // don't use 0, 0 is the default value!
  143. TokenStatusDisabled = 2 // also don't use 0
  144. TokenStatusExpired = 3
  145. TokenStatusExhausted = 4
  146. )
  147. const (
  148. RedemptionCodeStatusEnabled = 1 // don't use 0, 0 is the default value!
  149. RedemptionCodeStatusDisabled = 2 // also don't use 0
  150. RedemptionCodeStatusUsed = 3 // also don't use 0
  151. )
  152. const (
  153. ChannelStatusUnknown = 0
  154. ChannelStatusEnabled = 1 // don't use 0, 0 is the default value!
  155. ChannelStatusManuallyDisabled = 2 // also don't use 0
  156. ChannelStatusAutoDisabled = 3
  157. )
  158. const (
  159. TopUpStatusPending = "pending"
  160. TopUpStatusSuccess = "success"
  161. TopUpStatusExpired = "expired"
  162. )