test: 优化邮箱新增测试

chore: 优化首次打开有些慢的问题

Author: Kerwin

service/src/index.ts

@@ -11,8 +11,9 @@ import { Status } from './storage/model'
 import { clearChat, createChatRoom, createUser, deleteChat, deleteChatRoom, existsChatRoom, getChat, getChatRooms, getChats, getUser, getUserById, insertChat, renameChatRoom, updateChat, updateConfig, updateUserInfo, verifyUser } from './storage/mongo'
 import { limiter } from './middleware/limiter'
 import { isNotEmptyString } from './utils/is'
-import { sendMail } from './utils/mail'
+import { sendTestMail, sendVerifyMail } from './utils/mail'
 import { checkUserVerify, getUserVerifyUrl, md5 } from './utils/security'
+import { rootAuth } from './middleware/rootAuth'
 
 const app = express()
 const router = express.Router()
@@ -182,41 +183,46 @@ router.post('/chat-process', [auth, limiter], async (req, res) => {
 })
 
 router.post('/user-register', async (req, res) => {
-  const { username, password } = req.body as { username: string; password: string }
-  const config = await getCacheConfig()
-  if (!config.siteConfig.registerEnabled) {
-    res.send({ status: 'Fail', message: '注册账号功能未启用 | Register account is disabled!', data: null })
-    return
-  }
-  if (isNotEmptyString(config.siteConfig.registerMails)) {
-    let allowSuffix = false
-    const emailSuffixs = config.siteConfig.registerMails.split(',')
-    for (let index = 0; index < emailSuffixs.length; index++) {
-      const element = emailSuffixs[index]
-      allowSuffix = username.toLowerCase().endsWith(element)
-      if (allowSuffix)
-        break
-    }
-    if (!allowSuffix) {
-      res.send({ status: 'Fail', message: '该邮箱后缀不支持 | The email service provider is not allowed', data: null })
+  try {
+    const { username, password } = req.body as { username: string; password: string }
+    const config = await getCacheConfig()
+    if (!config.siteConfig.registerEnabled) {
+      res.send({ status: 'Fail', message: '注册账号功能未启用 | Register account is disabled!', data: null })
       return
     }
-  }
+    if (isNotEmptyString(config.siteConfig.registerMails)) {
+      let allowSuffix = false
+      const emailSuffixs = config.siteConfig.registerMails.split(',')
+      for (let index = 0; index < emailSuffixs.length; index++) {
+        const element = emailSuffixs[index]
+        allowSuffix = username.toLowerCase().endsWith(element)
+        if (allowSuffix)
+          break
+      }
+      if (!allowSuffix) {
+        res.send({ status: 'Fail', message: '该邮箱后缀不支持 | The email service provider is not allowed', data: null })
+        return
+      }
+    }
 
-  const user = await getUser(username)
-  if (user != null) {
-    res.send({ status: 'Fail', message: '邮箱已存在 | The email exists', data: null })
-    return
-  }
-  const newPassword = md5(password)
-  await createUser(username, newPassword)
+    const user = await getUser(username)
+    if (user != null) {
+      res.send({ status: 'Fail', message: '邮箱已存在 | The email exists', data: null })
+      return
+    }
+    const newPassword = md5(password)
+    await createUser(username, newPassword)
 
-  if (username.toLowerCase() === process.env.ROOT_USER) {
-    res.send({ status: 'Success', message: '注册成功 | Register success', data: null })
+    if (username.toLowerCase() === process.env.ROOT_USER) {
+      res.send({ status: 'Success', message: '注册成功 | Register success', data: null })
+    }
+    else {
+      await sendVerifyMail(username, await getUserVerifyUrl(username))
+      res.send({ status: 'Success', message: '注册成功, 去邮箱中验证吧 | Registration is successful, you need to go to email verification', data: null })
+    }
   }
-  else {
-    await sendMail(username, await getUserVerifyUrl(username))
-    res.send({ status: 'Success', message: '注册成功, 去邮箱中验证吧 | Registration is successful, you need to go to email verification', data: null })
+  catch (error) {
+    res.send({ status: 'Fail', message: error.message, data: null })
   }
 })
 
@@ -307,18 +313,13 @@ router.post('/verify', async (req, res) => {
   }
 })
 
-router.post('/setting-base', auth, async (req, res) => {
+router.post('/setting-base', rootAuth, async (req, res) => {
   try {
     const { apiKey, apiModel, apiBaseUrl, accessToken, timeoutMs, socksProxy, httpsProxy } = req.body as Config
-    const userId = new ObjectId(req.headers.userId.toString())
 
     if (apiKey == null && accessToken == null)
       throw new Error('Missing OPENAI_API_KEY or OPENAI_ACCESS_TOKEN environment variable.')
 
-    const user = await getUserById(userId)
-    if (user == null || user.status !== Status.Normal || user.email.toLowerCase() !== process.env.ROOT_USER)
-      throw new Error('无权限 | No permission.')
-
     const thisConfig = await getOriginConfig()
     thisConfig.apiKey = apiKey
     thisConfig.apiModel = apiModel
@@ -338,14 +339,9 @@ router.post('/setting-base', auth, async (req, res) => {
   }
 })
 
-router.post('/setting-site', auth, async (req, res) => {
+router.post('/setting-site', rootAuth, async (req, res) => {
   try {
     const config = req.body as SiteConfig
-    const userId = new ObjectId(req.headers.userId.toString())
-
-    const user = await getUserById(userId)
-    if (user == null || user.status !== Status.Normal || user.email.toLowerCase() !== process.env.ROOT_USER)
-      throw new Error('无权限 | No permission.')
 
     const thisConfig = await getOriginConfig()
     thisConfig.siteConfig = config
@@ -358,14 +354,9 @@ router.post('/setting-site', auth, async (req, res) => {
   }
 })
 
-router.post('/setting-mail', auth, async (req, res) => {
+router.post('/setting-mail', rootAuth, async (req, res) => {
   try {
     const config = req.body as MailConfig
-    const userId = new ObjectId(req.headers.userId.toString())
-
-    const user = await getUserById(userId)
-    if (user == null || user.status !== Status.Normal || user.email.toLowerCase() !== process.env.ROOT_USER)
-      throw new Error('无权限 | No permission.')
 
     const thisConfig = await getOriginConfig()
     thisConfig.mailConfig = config
@@ -378,6 +369,19 @@ router.post('/setting-mail', auth, async (req, res) => {
   }
 })
 
+router.post('/mail-test', rootAuth, async (req, res) => {
+  try {
+    const config = req.body as MailConfig
+    const userId = new ObjectId(req.headers.userId as string)
+    const user = await getUserById(userId)
+    await sendTestMail(user.email, config)
+    res.send({ status: 'Success', message: '发送成功 | Successfully', data: null })
+  }
+  catch (error) {
+    res.send({ status: 'Fail', message: error.message, data: null })
+  }
+})
+
 app.use('', router)
 app.use('/api', router)
 app.set('trust proxy', 1)

service/src/middleware/rootAuth.ts

@@ -0,0 +1,29 @@
+import jwt from 'jsonwebtoken'
+import { ObjectId } from 'mongodb'
+import { Status } from '../storage/model'
+import { getUserById } from '../storage/mongo'
+import { getCacheConfig } from '../storage/config'
+
+const rootAuth = async (req, res, next) => {
+  const config = await getCacheConfig()
+  if (config.siteConfig.loginEnabled) {
+    try {
+      const token = req.header('Authorization').replace('Bearer ', '')
+      const info = jwt.verify(token, config.siteConfig.loginSalt.trim())
+      req.headers.userId = info.userId
+      const user = await getUserById(new ObjectId(info.userId))
+      if (user == null || user.status !== Status.Normal || user.email.toLowerCase() !== process.env.ROOT_USER)
+        res.send({ status: 'Fail', message: '无权限 | No permission.', data: null })
+      else
+        next()
+    }
+    catch (error) {
+      res.send({ status: 'Unauthorized', message: error.message ?? 'Please authenticate.', data: null })
+    }
+  }
+  else {
+    res.send({ status: 'Fail', message: '无权限 | No permission.', data: null })
+  }
+}
+
+export { rootAuth }

service/src/utils/mail.ts

@@ -1,160 +1,41 @@
+import * as fs from 'fs'
+import * as path from 'path'
 import nodemailer from 'nodemailer'
+import type { MailConfig } from '../storage/model'
 import { getCacheConfig } from '../storage/config'
 
-export async function sendMail(toMail: string, verifyUrl: string) {
+export async function sendVerifyMail(toMail: string, verifyUrl: string) {
   const config = (await getCacheConfig())
+
+  const templatesPath = path.join(__dirname, 'templates')
+  const mailTemplatePath = path.join(templatesPath, 'mail.template.html')
+  let mailHtml = fs.readFileSync(mailTemplatePath, 'utf8')
+  mailHtml = mailHtml.replace(/\${VERIFY_URL}/g, verifyUrl)
+  mailHtml = mailHtml.replace(/\${SITE_TITLE}/g, config.siteConfig.siteTitle)
+  sendMail(toMail, `${config.siteConfig.siteTitle} 账号验证`, mailHtml, config.mailConfig)
+}
+
+export async function sendTestMail(toMail: string, config: MailConfig) {
+  return sendMail(toMail, '测试邮件|Test mail', '这是一封测试邮件|This is test mail', config)
+}
+
+async function sendMail(toMail: string, subject: string, html: string, config: MailConfig) {
   const mailOptions = {
-    from: config.mailConfig.smtpUserName, // sender address
-    to: toMail, // list of receivers
-    subject: `${config.siteConfig.siteTitle} 账号验证`, // Subject line
-    html: `
-    <div class="page flex-col">
-    <div class="box_3 flex-col" style="
-    display: flex;
-    position: relative;
-    width: 100%;
-    height: 206px;
-    background: #ef859d2e;
-    top: 0;
-    left: 0;
-    justify-content: center;
-  "><div class="section_1 flex-col" style="
-    background-image: url(&quot;https://ghproxy.com/https://raw.githubusercontent.com/Chanzhaoyu/chatgpt-web/main/src/assets/avatar.jpg&quot;);
-    position: absolute;
-    width: 152px;
-    height: 152px;
-    display: flex;
-    top: 130px;
-    background-size: cover;
-    border-radius: 50%;
-    margin: 10px;
-  "></div></div>
-    <div class="box_4 flex-col" style="
-    margin-top: 92px;
-    display: flex;
-    flex-direction: column;
-    align-items: center;
-  ">
-      <div class="text-group_5 flex-col justify-between" style="
-    display: flex;
-    flex-direction: column;
-    align-items: center;
-    margin: 0 20px;
-  ">
-        <span class="text_1" style="
-    font-size: 26px;
-    font-family: PingFang-SC-Bold, PingFang-SC;
-    font-weight: bold;
-    color: #000000;
-    line-height: 37px;
-    text-align: center;
-    "><target="_blank" style="text-decoration: none; color: #0088cc;">${config.siteConfig.siteTitle}</a> 账号验证</span>
-    
-      <div class="box_2 flex-row" style="
-    margin: 0 20px;
-    min-height: 128px;
-    background: #F7F7F7;
-    border-radius: 12px;
-    margin-top: 34px;
-    display: flex;
-    flex-direction: column;
-    align-items: flex-start;
-    padding: 32px 16px;
-    width: calc(100% - 40px);
-  ">
-        
-        <div class="text-wrapper_4 flex-col justify-between" style="
-    display: flex;
-    flex-direction: column;
-    margin-left: 30px;
-    margin-bottom: 16px;
-  ">
-  <hr>
-  <span class="text_3" style="
-  <div style="font-family: Arial, sans-serif; font-size: 16px; color: #333;">
-  <h1 style="color: #0088cc;">
-    感谢您使用 
-    <target="_blank" style="text-decoration: none; color: #0088cc;">${config.siteConfig.siteTitle}</a>，
-    您的邮箱验证链接为（12小时内有效）：
-  </span>
-        </div><hr style="
-      display: flex;
-      position: relative;
-      border: 1px dashed #ef859d2e;
-      box-sizing: content-box;
-      height: 0px;
-      overflow: visible;
-      width: 100%;
-  "><div class="text-wrapper_4 flex-col justify-between" style="
-    display: flex;
-    flex-direction: column;
-    margin-left: 30px;
-  ">
-  <hr>
-  </h1>
-  <p style="margin-top: 20px;">
-    请点击以下按钮进行验证：
-  <span class="text_4" style="
-  margin-top: 6px;
-  margin-right: 22px;
-  font-size: 16px;
-  font-family: PingFangSC-Regular, PingFang SC;
-  font-weight: 400;
-  color: #000000;
-  line-height: 22px;
-  "></span>
-        </div>
-        
-        <a class="text-wrapper_2 flex-col" style="
-    min-width: 106px;
-    height: 38px;
-    background: #ef859d38;
-    border-radius: 32px;
-    display: flex;
-    align-items: center;
-    justify-content: center;
-    text-decoration: none;
-    margin: auto;
-    margin-top: 32px;
-  " href="${verifyUrl}">
-          <span class="text_5" style="
-    color: #DB214B;
-  ">点我验证</span>
-        </a>
-      </div>
-      <div class="text-group_6 flex-col justify-between" style="
-    display: flex;
-    flex-direction: column;
-    align-items: center;
-    margin-top: 34px;
-  ">
-        <span class="text_6" style="
-    height: 17px;
-    font-size: 12px;
-    font-family: PingFangSC-Regular, PingFang SC;
-    font-weight: 400;
-    color: #00000045;
-    line-height: 17px;
-  ">此邮件由服务器自动发出，直接回复无效。</span>
-      </div>
-    </div>
-  </div>
-    `, // html body
+    from: config.smtpUserName,
+    to: toMail,
+    subject,
+    html,
   }
 
   const transporter = nodemailer.createTransport({
-    host: config.mailConfig.smtpHost,
-    port: config.mailConfig.smtpPort,
-    secure: config.mailConfig.smtpTsl,
+    host: config.smtpHost,
+    port: config.smtpPort,
+    secure: config.smtpTsl,
     auth: {
-      user: config.mailConfig.smtpUserName,
-      pass: config.mailConfig.smtpPassword,
+      user: config.smtpUserName,
+      pass: config.smtpPassword,
     },
   })
-  transporter.sendMail(mailOptions, (error, info) => {
-    if (error)
-      throw error
-    else
-      return info.messageId
-  })
+  const info = await transporter.sendMail(mailOptions)
+  return info.messageId
 }