From 1ee527ceb84430ed4fbde0218f809535aa212323 Mon Sep 17 00:00:00 2001
From: archer <545436317@qq.com>
Date: Sun, 16 Apr 2023 23:33:12 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20=E9=AA=8C=E8=AF=81=E7=A0=81=E7=A8=8B?=
 =?UTF-8?q?=E5=BA=A6?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .env.template                      |  4 ++++
 src/pages/api/user/sendAuthCode.ts |  7 ++++---
 src/service/utils/sendNote.ts      | 19 +++++++++++++++++++
 3 files changed, 27 insertions(+), 3 deletions(-)

diff --git a/.env.template b/.env.template
index 78caad673..c2fbae06c 100644
--- a/.env.template
+++ b/.env.template
@@ -4,6 +4,10 @@ AXIOS_PROXY_PORT_NORMAL=7890
 MONGODB_URI=mongodb://username:password@0.0.0.0:27017/?authSource=admin&readPreference=primary&appname=MongoDB%20Compass&ssl=false
 MY_MAIL=11111111@qq.com
 MAILE_CODE=sdasadasfasfad
+aliAccessKeyId=阿里云keyid
+aliAccessKeySecret=阿里云keysecret
+aliSignName=短信签名
+aliTemplateCode=SMS_号
 TOKEN_KEY=sssssssss
 OPENAIKEY=sk-afadfadfadfsd
 REDIS_URL=redis://default:password@0.0.0.0:8100
\ No newline at end of file
diff --git a/src/pages/api/user/sendAuthCode.ts b/src/pages/api/user/sendAuthCode.ts
index a79c30394..494521db9 100644
--- a/src/pages/api/user/sendAuthCode.ts
+++ b/src/pages/api/user/sendAuthCode.ts
@@ -5,8 +5,6 @@ import { AuthCode } from '@/service/models/authCode';
 import { connectToDatabase } from '@/service/mongo';
 import { sendPhoneCode, sendEmailCode } from '@/service/utils/sendNote';
 import { UserAuthTypeEnum } from '@/constants/common';
-import { customAlphabet } from 'nanoid';
-const nanoid = customAlphabet('1234567890', 6);
 
 export default async function handler(req: NextApiRequest, res: NextApiResponse) {
   try {
@@ -18,7 +16,10 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
 
     await connectToDatabase();
 
-    let code = nanoid();
+    let code = '';
+    for (let i = 0; i < 6; i++) {
+      code += Math.floor(Math.random() * 10);
+    }
 
     // 判断 1 分钟内是否有重复数据
     const authCode = await AuthCode.findOne({
diff --git a/src/service/utils/sendNote.ts b/src/service/utils/sendNote.ts
index 4ef222c3c..996aa2e1c 100644
--- a/src/service/utils/sendNote.ts
+++ b/src/service/utils/sendNote.ts
@@ -48,6 +48,25 @@ export const sendEmailCode = (email: string, code: string, type: `${UserAuthType
   });
 };
 
+export const sendTrainSucceed = (email: string, modelName: string) => {
+  return new Promise((resolve, reject) => {
+    const options = {
+      from: `"FastGPT" ${myEmail}`,
+      to: email,
+      subject: '模型训练完成通知',
+      html: `你的模型 ${modelName} 已于 ${dayjs().format('YYYY-MM-DD HH:mm')} 训练完成！`
+    };
+    mailTransport.sendMail(options, function (err, msg) {
+      if (err) {
+        console.log('send email  error->', err);
+        reject('邮箱异常');
+      } else {
+        resolve('');
+      }
+    });
+  });
+};
+
 export const sendPhoneCode = async (phone: string, code: string) => {
   const accessKeyId = process.env.aliAccessKeyId;
   const accessKeySecret = process.env.aliAccessKeySecret;