xinyin025/one-api
1
0
Fork 0
mirror of https://github.com/songquanpeng/one-api.git synced 2025-11-17 21:53:42 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

🔖 chore: migration constants

Browse Source
This commit is contained in:
MartialBE
2024-05-29 01:56:14 +08:00
parent ce12558ad6
commit 3d8a51e139
91 changed files with 670 additions and 614 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
common/stmp/email.go
View File

@@ -3,6 +3,7 @@ package stmp
import (
"fmt"
"one-api/common"
"one-api/common/config"
"one-api/common/utils"
"strings"
@@ -38,7 +39,7 @@ func (s *StmpConfig) Send(to, subject, body string) error {
message.Subject(subject)
message.SetGenHeader("References", s.getReferences())
message.SetBodyString(mail.TypeTextHTML, body)
message.SetUserAgent(fmt.Sprintf("One API %s // https://github.com/MartialBE/one-api", common.Version))
message.SetUserAgent(fmt.Sprintf("One API %s // https://github.com/MartialBE/one-api", config.Version))
client, err := mail.NewClient(
s.Host,
@@ -78,11 +79,11 @@ func (s *StmpConfig) Render(to, subject, content string) error {
}
func GetSystemStmp() (*StmpConfig, error) {
if common.SMTPServer == "" || common.SMTPPort == 0 || common.SMTPAccount == "" || common.SMTPToken == "" {
if config.SMTPServer == "" || config.SMTPPort == 0 || config.SMTPAccount == "" || config.SMTPToken == "" {
return nil, fmt.Errorf("SMTP 信息未配置")
}
return NewStmp(common.SMTPServer, common.SMTPPort, common.SMTPAccount, common.SMTPToken, common.SMTPFrom), nil
return NewStmp(config.SMTPServer, config.SMTPPort, config.SMTPAccount, config.SMTPToken, config.SMTPFrom), nil
}
func SendPasswordResetEmail(userName, email, link string) error {
@@ -106,7 +107,7 @@ func SendPasswordResetEmail(userName, email, link string) error {
</p>
<p style="color: #858585;">重置链接 %d 分钟内有效，如果不是本人操作，请忽略。</p>`
subject := fmt.Sprintf("%s密码重置", common.SystemName)
subject := fmt.Sprintf("%s密码重置", config.SystemName)
content := fmt.Sprintf(contentTemp, userName, link, link, common.VerificationValidMinutes)
return stmp.Render(email, subject, content)
@@ -132,7 +133,7 @@ func SendVerificationCodeEmail(email, code string) error {
验证码 %d 分钟内有效，如果不是本人操作，请忽略。
</p>`
subject := fmt.Sprintf("%s邮箱验证邮件", common.SystemName)
subject := fmt.Sprintf("%s邮箱验证邮件", config.SystemName)
content := fmt.Sprintf(contentTemp, code, common.VerificationValidMinutes)
return stmp.Render(email, subject, content)
@@ -162,7 +163,7 @@ func SendQuotaWarningCodeEmail(userName, email string, quota int, noMoreQuota bo
if noMoreQuota {
subject = "您的额度已用尽"
}
topUpLink := fmt.Sprintf("%s/topup", common.ServerAddress)
topUpLink := fmt.Sprintf("%s/topup", config.ServerAddress)
content := fmt.Sprintf(contentTemp, userName, subject, quota, topUpLink, topUpLink)

3
common/stmp/email_test.go
View File

@@ -2,6 +2,7 @@ package stmp_test
import (
"fmt"
"one-api/common/config"
"testing"
"one-api/common"
@@ -56,7 +57,7 @@ func TestSend(t *testing.T) {
验证码 %d 分钟内有效，如果不是本人操作，请忽略。
</p>`
subject := fmt.Sprintf("%s邮箱验证邮件", common.SystemName)
subject := fmt.Sprintf("%s邮箱验证邮件", config.SystemName)
content := fmt.Sprintf(contentTemp, code, common.VerificationValidMinutes)
err := stmpClient.Render(email, subject, content)

10
common/stmp/template.go
View File

@@ -1,17 +1,17 @@
package stmp
import (
"one-api/common"
"one-api/common/config"
)
func getLogo() string {
if common.Logo == "" {
if config.Logo == "" {
return ""
}
return `<table class="logo" width="100%">
<tr>
<td>
<img src="` + common.Logo + `" width="130" style="max-width: 100%"
<img src="` + config.Logo + `" width="130" style="max-width: 100%"
/>
</td>
</tr>
@@ -19,11 +19,11 @@ func getLogo() string {
}
func getSystemName() string {
if common.SystemName == "" {
if config.SystemName == "" {
return "One API"
}
return common.SystemName
return config.SystemName
}
func getDefaultTemplate(content string) string {