api_user/handler/users.js

import { DBModel } from "../models/index.js";
import ResponseUtil from "../utils/api_response.js";
import config from "../conf.json" with { type: "json" };

class HandlerUser {
  constructor() {
  }

  // 获取微信的手机号码
  async wxGetPhoneNumber(ctx) {
    try {
      const { code, appId } = ctx.request.body;
      if (!code || !appId) {
        return ResponseUtil.badRequest(ctx, "缺少手机号凭证 code 或 appId");
      }

      let app = config.app[appId];
      if (!app) {
        return ResponseUtil.badRequest(ctx, `未配置 appId: ${appId}`);
      }

      // 获取access_token
      const client_credential_url = `https://api.weixin.qq.com/cgi-bin/token?appid=${app.appid}&secret=${app.secret}&grant_type=client_credential`;
      const fetch = (await import("node-fetch")).default;
      let sessionRes = await fetch(client_credential_url);
      const resp = await sessionRes.json();
      if (!resp.access_token) {
        return ResponseUtil.internalError(ctx, "获取微信 access_token 失败");
      }

      // 获取phoneNumber
      const phoneUrl = `https://api.weixin.qq.com/wxa/business/getuserphonenumber?access_token=${resp.access_token}`;
      const phoneRes = await fetch(phoneUrl, {
        method: 'POST',
        headers: { 'Content-Type': 'application/json' },
        body: JSON.stringify({ code: code })
      });
      const phoneData = await phoneRes.json();
      if (phoneData.errcode) {
        return ResponseUtil.error(ctx, `获取手机号失败: ${phoneData.errmsg}`, null, 400);
      }

      // 从上图获取手机号
      const phoneNumber = phoneData.phone_info?.phoneNumber;
      return ResponseUtil.success(ctx, {phoneNumber}, "获取手机号成功");
    } catch (err) {
      return ResponseUtil.internalError(ctx, err.message);
    }
  }

  // 微信登录
  async wxSignin(ctx) {
    try {
      const { code, phoneNumber, appId } = ctx.request.body;
      if (!code || !appId) {
        return ResponseUtil.badRequest(ctx, "缺少微信登录凭证 code 或 appId");
      }

      let app = config.app[appId];
      if (!app) {
        return ResponseUtil.badRequest(ctx, `未配置 appId: ${appId}`);
      }

      // 通过 code 换取 openid/session_key
      const sessionUrl = `https://api.weixin.qq.com/sns/jscode2session?appid=${app.appid}&secret=${app.secret}&js_code=${code}&grant_type=authorization_code`;
      const wxSessionRes = await fetch(sessionUrl);
      const sessionData = await wxSessionRes.json();
      if (sessionData.errcode) {
        return ResponseUtil.error(ctx, `微信接口错误: ${sessionData.errmsg}`, null, 400);
      }

      const { openid } = sessionData;
      if (!openid) {
        return ResponseUtil.error(ctx, "微信登录失败，未获取到 openid", null, 400);
      }

      let user = await DBModel.User.findOne({ "social.wechat.openid": openid });
      if (!user) {
        if (!phoneNumber) {
          return ResponseUtil.badRequest(ctx, "缺少手机号");
        }

        const newUser = {
          profile: { name: phoneNumber, mobile: phoneNumber, },
          social: {
            wechat: { openid: openid },
          },
          status: { account: "normal", },
          app: {
            attendant: { role: ["patient"], },
          },
        };
        user = await DBModel.User.setUser(newUser);
      } else if (phoneNumber && phoneNumber.length > 0 && user.profile.mobile !== phoneNumber) {
        user.profile.mobile = phoneNumber;
      }

      const token = await this.genToken(user._id.toString());
      user.security.token = token;
      user.security.tokenExpiry = new Date(Date.now() + 7 * 24 * 60 * 60 * 1000);
      await user.save();

      // 安全起见删除密码相关字段
      delete user.security.passwd;
      delete user.security.passwdSalt;

      return ResponseUtil.success(ctx, { user }, "登录成功");
    } catch (err) {
      return ResponseUtil.internalError(ctx, err.message);
    }
  }

  // update user
  async updateUser(ctx) {
    try {
      const userInfo = ctx.request.body;
      if (!userInfo) {
        return ResponseUtil.badRequest(ctx, "缺少用户信息");
      }

      // 从 token 获取当前用户
      const token = ctx.request.body?.token
        || ctx.request.query?.token
        || ctx.header?.authorization
        || ctx.header?.token;

      if (!token) {
        return ResponseUtil.badRequest(ctx, "缺少 token");
      }

      const user = await DBModel.User.findOne({ "security.token": token });
      if (!user) {
        return ResponseUtil.unauthorized(ctx, "用户未登录或 token 无效");
      }

      // 检查 token 是否过期
      if (user.security.tokenExpiry && new Date() > user.security.tokenExpiry) {
        return ResponseUtil.unauthorized(ctx, "登录已过期，请重新登录");
      }

      const updatedUser = await DBModel.User.updateFromUserInfo(
        user._id,
        userInfo
      );

      if (!updatedUser) {
        return ResponseUtil.internalError(ctx, "更新用户失败");
      }

      // 安全起见删除密码相关字段
      const safeUser = updatedUser.toObject();
      delete safeUser.security?.passwd;
      delete safeUser.security?.passwdSalt;

      return ResponseUtil.success(ctx, { user: safeUser }, "更新成功");
    } catch (err) {
      return ResponseUtil.internalError(ctx, err.message);
    }
  }

  // 退出登录
  async signout(ctx) {
    const token = ctx.request.body?.token
      || ctx.request.query?.token
      || ctx.header?.authorization
      || ctx.header?.token;

    if (!token) {
      return ResponseUtil.badRequest(ctx, "缺少 token");
    }

    const user = await DBModel.User.findOne({ "security.token": token });
    if (user) {
      user.security.token = null;
      user.security.tokenExpiry = null;
      await user.save();
    }

    return ResponseUtil.success(ctx, null, "退出登录成功");
  }

  // 生成 token
  async genToken(uid) {
    const crypto = await import("crypto");
    const hash = crypto.createHash("md5");
    hash.update(uid + Date.now() + Math.random());
    return hash.digest("hex");
  }
}

export { HandlerUser };