feat: 手机验证码作为用户凭证

src/pages/api/user/loginByPassword.ts

@@ -7,24 +7,24 @@ import { generateToken } from '@/service/utils/tools';
 
 export default async function handler(req: NextApiRequest, res: NextApiResponse) {
   try {
-    const { email, password } = req.body;
+    const { username, password } = req.body;
 
-    if (!email || !password) {
+    if (!username || !password) {
       throw new Error('缺少参数');
     }
 
     await connectToDatabase();
 
-    // 检测邮箱是否存在
-    const authEmail = await User.findOne({
-      email
+    // 检测用户是否存在
+    const authUser = await User.findOne({
+      username
     });
-    if (!authEmail) {
-      throw new Error('邮箱未注册');
+    if (!authUser) {
+      throw new Error('用户未注册');
     }
 
     const user = await User.findOne({
-      email,
+      username,
       password
     });
 

src/pages/api/user/register.ts

@@ -5,23 +5,29 @@ import { User } from '@/service/models/user';
 import { AuthCode } from '@/service/models/authCode';
 import { connectToDatabase } from '@/service/mongo';
 import { generateToken } from '@/service/utils/tools';
-import { EmailTypeEnum } from '@/constants/common';
+import { UserAuthTypeEnum } from '@/constants/common';
 
 export default async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
   try {
-    const { email, code, password } = req.body;
+    const { phone, code, password } = req.body;
 
-    if (!email || !code || !password) {
+    if (!phone || !code || !password) {
       throw new Error('缺少参数');
     }
 
+    const reg = /^1[3456789]\d{9}$/;
+
+    if (!reg.test(phone)) {
+      throw new Error('手机号格式错误');
+    }
+
     await connectToDatabase();
 
-    // 验证码校验
+    // 验证码校验. 注册只接收手机号
     const authCode = await AuthCode.findOne({
-      email,
+      username: phone,
       code,
-      type: EmailTypeEnum.register,
+      type: UserAuthTypeEnum.register,
       expiredTime: { $gte: Date.now() }
     });
 
@@ -31,15 +37,15 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<
 
     // 重名校验
     const authRepeat = await User.findOne({
-      email
+      username: phone
     });
 
     if (authRepeat) {
-      throw new Error('邮箱已被注册');
+      throw new Error('手机号已被注册');
     }
 
     const response = await User.create({
-      email,
+      username: phone,
       password
     });
 
@@ -50,6 +56,11 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<
       throw new Error('获取用户信息异常');
     }
 
+    // 删除验证码记录
+    await AuthCode.deleteMany({
+      username: phone
+    });
+
     jsonRes(res, {
       data: {
         token: generateToken(user._id),

src/pages/api/user/sendAuthCode.ts

@@ -2,28 +2,27 @@
 import type { NextApiRequest, NextApiResponse } from 'next';
 import { jsonRes } from '@/service/response';
 import { AuthCode } from '@/service/models/authCode';
-import { connectToDatabase, User } from '@/service/mongo';
-import { sendCode } from '@/service/utils/sendEmail';
-import { EmailTypeEnum } from '@/constants/common';
+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 {
-    const { email, type } = req.query as { email: string; type: `${EmailTypeEnum}` };
+    const { username, type } = req.query as { username: string; type: `${UserAuthTypeEnum}` };
 
-    if (!email || !type) {
+    if (!username || !type) {
       throw new Error('缺少参数');
     }
 
     await connectToDatabase();
 
-    let code = '';
-    for (let i = 0; i < 6; i++) {
-      code += Math.floor(Math.random() * 10);
-    }
+    let code = nanoid();
 
     // 判断 1 分钟内是否有重复数据
     const authCode = await AuthCode.findOne({
-      email,
+      username,
       type,
       expiredTime: { $gte: Date.now() + 4 * 60 * 1000 } // 如果有一个记录的过期时间，大于当前+4分钟，说明距离上次发送还没到1分钟。（因为默认创建时，过期时间是未来5分钟）
     });
@@ -34,13 +33,17 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
 
     // 创建 auth 记录
     await AuthCode.create({
-      email,
+      username,
       type,
       code
     });
 
-    // 发送验证码
-    await sendCode(email as string, code, type as `${EmailTypeEnum}`);
+    if (username.includes('@')) {
+      await sendEmailCode(username, code, type);
+    } else {
+      // 发送验证码
+      await sendPhoneCode(username, code);
+    }
 
     jsonRes(res, {
       message: '发送验证码成功'

src/pages/api/user/updatePasswordByCode.ts

@@ -5,13 +5,13 @@ import { User } from '@/service/models/user';
 import { AuthCode } from '@/service/models/authCode';
 import { connectToDatabase } from '@/service/mongo';
 import { generateToken } from '@/service/utils/tools';
-import { EmailTypeEnum } from '@/constants/common';
+import { UserAuthTypeEnum } from '@/constants/common';
 
 export default async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
   try {
-    const { email, code, password } = req.body;
+    const { username, code, password } = req.body;
 
-    if (!email || !code || !password) {
+    if (!username || !code || !password) {
       throw new Error('缺少参数');
     }
 
@@ -19,9 +19,9 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<
 
     // 验证码校验
     const authCode = await AuthCode.findOne({
-      email,
+      username,
       code,
-      type: EmailTypeEnum.findPassword,
+      type: UserAuthTypeEnum.findPassword,
       expiredTime: { $gte: Date.now() }
     });
 
@@ -32,16 +32,16 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<
     // 更新对应的记录
     await User.updateOne(
       {
-        email
+        username
       },
       {
         password
       }
     );
 
-    // 根据 email 获取用户信息
+    // 根据 username 获取用户信息
     const user = await User.findOne({
-      email
+      username
     });
 
     if (!user) {