mirai-js/src/core/fs/uploadGroupFIle.js

QQ robot development framework based on Mirai-api-http.

const { errCodeMap } = require('../../util/errCode');
const axios = require('axios').default;
const { URL } = require('../../polyfill/URL');
const errorHandler = require('../../util/errorHandler');
const path = require('path');
const { isBrowserEnv } = require('../../util/isBrowserEnv');
const locationStr = !isBrowserEnv() ? `core.${path.basename(__filename, path.extname(__filename))}` : 'borwser';
const FormData = require('form-data');


/**
 * @description 上传文件至服务器并发送，返回文件 id
 * @param {string}  baseUrl    mirai-api-http server 的地址
 * @param {string}  sessionKey 会话标识
 * @param {string}  type       "friend" 或 "group"，目前仅支持 group
 * @param {string}  target     群/好友号
 * @param {string}  path       上传目录id
 * @param {Buffer}  file       文件二进制数据
 * @returns {string} 文件 id
 */
module.exports = async ({ baseUrl, sessionKey, type, target, path, file }) => {
    try {
        // 拼接 url
        const targetUrl = new URL('/file/upload', baseUrl).toString();

        // 构造 fromdata
        const form = new FormData();
        form.append('sessionKey', sessionKey);
        form.append('type', type);
        form.append('target', target);
        form.append('path', path);
        form.append('file', file, { filename: 'payload' });

        // 请求
        const responseData = await axios.post(targetUrl, form, {
            // formdata.getHeaders 将会指定 content-type，同时给定随
            // 机生成的 boundary，即分隔符，用以分隔多个表单项而不会造成混乱
            headers: form.getHeaders(),
        });

        try {
            var {
                data: { msg: message, code, id }
            } = responseData;
        } catch (error) {
            throw new Error(('请求返回格式出错，请检查 mirai-console'));
        }

        // 抛出 mirai 的异常，到 catch 中处理后再抛出
        if (code in errCodeMap) {
            throw new Error(message);
        }
        return id;
    } catch (error) {
        console.error(`mirai-js: error ${locationStr}`);
        errorHandler(error);
    }
};