ic-ctbu-backend/apis/Lepao/Face/BeginFaceReco.js

const API = require("../../../lib/API.js")
const db = require("../../../plugin/DataBase/db.js")
const axios = require('axios')
const config = require('../../../config.json')
const { BaseStdResponse } = require("../../../BaseStdResponse.js")

class BeginFaceReco extends API {
    constructor() {
        super()

        this.runpy = config.runpy

        this.setPath('/Face/BeginFaceReco')
        this.setMethod('POST')
    }

    /**
     * 获取图片并转换为Base64
     * @param {string} url - 图片链接
     * @returns {Promise<string>} - Base64字符串
     */
    async getImageBase64(url) {
        try {
            const response = await axios.get(url, {
                proxy: false,
                responseType: "arraybuffer"
            })
            const base64 = Buffer.from(response.data, "binary").toString("base64")

            // 获取图片MIME类型
            const contentType = response.headers["content-type"]
            return `data:${contentType};base64,${base64}`
        } catch (error) {
            throw new Error(`获取图片失败: ${error.message}`)
        }
    }

    async onRequest(req, res) {
        let { student_num, name } = req.body

        if ([student_num, name].some(value => value === '' || value === null || value === undefined))
            return res.json({
                ...BaseStdResponse.MISSING_PARAMETER
            })

        try {
            let sql = 'SELECT student_num, name, user_avatar, academy_name FROM lepao_account WHERE student_num = ? AND name = ?'
            let rows = await db.query(sql, [student_num, name])
            if (!rows)
                return res.json({
                    ...BaseStdResponse.DATABASE_ERR,
                    msg: '获取用户人脸信息失败，请重试'
                })
            if (rows.length !== 1)
                return res.json({
                    ...BaseStdResponse.ERR,
                    msg: '该用户尚未在RunForge系统中添加，请先添加账号'
                })

            const face_img = await this.getImageBase64('https://lepao-cloud.xxoo365.top/view.php/d0b85269c3683ed48da1fc5e468108c7.jpg')
            // 此时应该从乐跑获取人脸照片
            let resData = {
                ...rows[0],
                face_img
            }

            res.json({
                ...BaseStdResponse.OK,
                data: resData
            })
        } catch (error) {
            this.logger.error(`获取用户人脸信息失败。${error.stack}`)
            return res.json({
                ...BaseStdResponse.ERR,
                msg: '获取用户人脸信息失败，请重试'
            })
        }
    }
}

module.exports.BeginFaceReco = BeginFaceReco