oicq/lib/online-push.js

QQ protocol!

/**
 * 群消息事件入口
 * 好友事件和群事件(禁言、踢人等)
 */
"use strict";
const pb = require("./algo/pb");
const jce = require("./algo/jce");
const { parseGroupMsg, parseDiscussMsg } = require("./message/parser");
const { genC2CMessageId, genGroupMessageId, log } = require("./common");

/**
 * OnlinePush回执
 * @this {import("./ref").Client}
 * @param {number} svrip
 * @param {number} seq
 * @param {Buffer[]} rubbish 
 */
function handleOnlinePush(svrip, seq, rubbish = []) {
    const resp = jce.encodeStruct([
        this.uin, rubbish, svrip & 0xffffffff, null, 0
    ]);
    const extra = {
        req_id: seq,
        service: "OnlinePush",
        method: "SvcRespPushMsg",
    };
    const body = jce.encodeWrapper({ resp }, extra);
    this.writeUni("OnlinePush.RespPush", body);
}

const status_map = {
    1: 11,
    3: 31,
    4: 41,
    5: 50,
    6: 60,
    7: 70,
};

/**
 * @type {({[k: number]: (this: import("./ref").Client, data: import("./ref").Proto, time: number) => void})}
 */
const sub0x27 = {
    80: function (data, time) {
        const o = data[12];
        const group_id = o[3];
        if (!o[4])
            return;
        const group_name = String(o[2][2]);
        try {
            this.gl.get(group_id).group_name = group_name;
        } catch (e) { }
        this.em("notice.group.setting", {
            group_id, time,
            user_id: o[4],
            group_name,
        });
    },
    5: function (data, time) {
        const user_id = data[14][1];
        let nickname;
        try {
            nickname = this.fl.get(user_id).nickname;
            this.fl.delete(user_id);
        } catch (e) { }
        this.logger.info(`更新了好友列表，删除了好友 ${user_id}(${nickname})`);
        this.em("notice.friend.decrease", {
            user_id, nickname, time
        });
    },
    20: function (data, time) {
        // 20002昵称 20009性别 20031生日 23109农历生日 20019说明 20032地区 24002故乡 27372在线状态
        const user_id = data[8][1];
        let o = data[8][2];
        if (Array.isArray(o)) {
            o = o[0];
        }
        let key, value;
        if (o[1] === 20002) {
            key = "nickname";
            value = String(o[2]);
        } else if (o[1] === 20009) {
            key = "sex";
            value = ["unknown", "male", "female"][o[2].toBuffer()[0]];
        } else if (o[1] === 20031) {
            key = "age";
            value = new Date().getFullYear() - o[2].toBuffer().readUInt16BE();
        } else if (o[1] === 20019) {
            key = "description";
            value = String(o[2]);
        } else if (o[1] === 27372 && user_id === this.uin) {
            const status = o[2].toBuffer()[o[2].toBuffer().length - 1];
            this.online_status = status_map[status] || 11;
            return;
        } else {
            return;
        }
        try {
            this.fl.get(user_id)[key] = value;
        } catch (e) { }
        if (user_id === this.uin)
            this[key] = value;
        else {
            const e = { user_id, time };
            e[key] = value;
            this.em("notice.friend.profile", e);
        }
    },
    60: function (data, time) {
        const user_id = data[10][1];
        const sign = String(data[10][2]);
        try {
            this.fl.get(user_id).signature = sign;
        } catch (e) { }
        if (user_id === this.uin)
            this.signature = sign;
        else
            this.em("notice.friend.profile", {
                user_id, signature: sign, time
            });
    },
    40: function (data, time) {
        try {
            const o = data[9][1], uin = o[2];
            if (o[1] > 0) return; //0好友备注 1群备注
            this.fl.get(uin).remark = String(o[3]);
        } catch (e) { }
    },
    21: function (data, time) {
        // 群头像增加 <Buffer 0a 1a 08 00 10 15 5a 14 08 01 10 9f dd 95 a1 04 18 9f dd 95 a1 04 20 f5 ef e8 b1 01>
    }
};

/**
 * @type {({[k: number]: (this: import("./ref").Client, buf: Buffer, time: number) => void})}
 */
const push528 = {
    0x8A: function (buf, time) {
        let data = pb.decode(buf)[1];
        if (Array.isArray(data))
            data = data[0];
        let user_id = data[1], operator_id = data[1], flag = 0;
        if (user_id === this.uin) {
            user_id = data[2];
            flag = 1;
        }
        this.em("notice.friend.recall", {
            user_id, operator_id, message_id: genC2CMessageId(user_id, data[3], data[6], data[5], flag), time
        });
    },
    0x8B: function (buf, time) {
        return push528[0x8A].call(this, buf, time);
    },
    0xB3: function (buf, time) {
        const data = pb.decode(buf)[2];
        const user_id = data[1], nickname = String(data[5]);
        this.fl.set(user_id, {
            user_id, nickname,
            sex: "unknown",
            age: 0,
            area: "unknown",
            remark: nickname,
        });
        this.sl.delete(user_id);
        this.getStrangerInfo(user_id).then(() => {
            this.logger.info(`更新了好友列表，新增了好友 ${user_id}(${nickname})`);
            this.em("notice.friend.increase", {
                user_id, nickname, time
            });
        });
    },
    0xD4: function (buf, time) {
        const group_id = pb.decode(buf)[1];
        this.getGroupInfo(group_id, true);
    },
    0x3B: function (buf, time) {
        const data = pb.decode(buf);
        const group_id = data[2];
        this.em("notice.group.setting", {
            group_id, time,
            enable_show_title: data[3] > 0,
        });
    },
    0x27: function (buf, time) {
        let data = pb.decode(buf)[1];
        if (Array.isArray(data))
            data = data[0];
        if (typeof sub0x27[data[2]] === "function")
            sub0x27[data[2]].call(this, data, time);
    },
    0x122: function (buf, time, uin) {
        const data = pb.decode(buf);
        const eve = { time };
        Object.assign(eve, parsePoke.call(this, data));
        eve.target_id = eve.user_id;
        eve.user_id = uin;
        this.em("notice.friend.poke", eve);
    },
    0x115: function (buf, time) {
        // 正在输入
    },
};

function parsePoke(data) {
    let user_id, operator_id, action, suffix;
    for (let o of data[7]) {
        const name = String(o[1]);
        if (name === "action_str")
            action = String(o[2]);
        if (name === "uin_str1")
            operator_id = parseInt(String(o[2]));
        if (name === "uin_str2")
            user_id = parseInt(String(o[2]));
        if (name === "suffix_str")
            suffix = String(o[2]);
    }
    if (!operator_id)
        operator_id = this.uin;
    if (!user_id)
        user_id = this.uin;
    return { user_id, operator_id, action, suffix };
}

/**
 * @this {import("./ref").Client}
 * @param {number} group_id 
 * @param {string} field 
 * @param {boolean} enable 
 * @param {number} time 
 */
function onGroupSetting(group_id, field, enable, time) {
    if (!field) return;
    const e = {
        group_id, time
    };
    e[field] = !!enable;
    this.em("notice.group.setting", e);
}

/**
 * @type {({[k: number]: (this: import("./ref").Client, group_id: number, buf: Buffer, time: number) => void})}
 */
const push732 = {
    0x0C: function (group_id, buf, time) {
        const operator_id = buf.readUInt32BE(6);
        const user_id = buf.readUInt32BE(16);
        let duration = buf.readUInt32BE(20);
        try {
            if (user_id === 0) {
                duration = duration ? 0xffffffff : 0;
                this.gl.get(group_id).shutup_time_whole = duration;
            }
            else if (user_id === this.uin)
                this.gl.get(group_id).shutup_time_me = duration ? (time + duration) : 0;
            this.gml.get(group_id).get(user_id).shutup_time = duration ? (time + duration) : 0;
        } catch (e) { }
        this.em("notice.group.ban", {
            group_id, operator_id, user_id, duration, time
        });
    },
    0x11: function (group_id, buf, time) {
        const data = pb.decode(buf.slice(7))[11];
        const operator_id = data[1];
        const msg = Array.isArray(data[3]) ? data[3][0] : data[3];
        const user_id = msg[6];
        const message_id = genGroupMessageId(group_id, user_id, msg[1], msg[3], msg[2], Array.isArray(data[3]) ? data[3].length : 1);
        this.em("notice.group.recall", {
            group_id, user_id, operator_id, message_id, time
        });
    },
    0x14: function (group_id, buf, time) {
        const data = pb.decode(buf.slice(7))[26];
        if (data) {
            const eve = { group_id, time };
            Object.assign(eve, parsePoke.call(this, data));
            if (eve.action)
                this.em("notice.group.poke", eve);
        }
    },
    0x06: function (group_id, buf, time) {
        if (buf[5] !== 1) return;
        onGroupSetting.call(this, group_id, "enable_guest", buf[10] > 0, time);
    },
    0x0E: function (group_id, buf, time) {
        if (buf[5] !== 1) return;
        const duration = buf.readInt32BE(10);
        if (buf[14] === 0)
            onGroupSetting.call(this, group_id, "enable_anonymous", duration === 0, time);
        else {
            const nickname = String(buf.slice(15, 15 + buf[14]));
            const operator_id = buf.readUInt32BE(6);
            this.em("notice.group.ban", {
                group_id, operator_id,
                user_id: 80000000, nickname,
                duration, time
            });
        }
    },
    0x0F: function (group_id, buf, time) {
        if (buf[12] === 1)
            var field = "enable_upload_album";
        else if (buf[12] === 2)
            var field = "enable_upload_file";
        var enable = buf[8] === 0x0 || buf[8] === 0x20;
        onGroupSetting.call(this, group_id, field, enable, time);
    },
    0x10: function (group_id, buf, time) {
        if (buf[6] === 0x22) {
            let field;
            if (buf[buf.length - 2] === 0x08)
                field = "enable_show_honor";
            if (buf[buf.length - 2] === 0x10)
                field = "enable_show_level";
            let enable = buf[buf.length - 1] === 0;
            return onGroupSetting.call(this, group_id, field, enable, time);
        }
        if (buf[6] === 0x26) {
            // 改群分类 <Buffer 44 25 6e 9f 10 00 26 08 18 10 96 8a e3 fa 05 18 ff ff ff ff 0f 20 9f dd 95 a1 04 68 17 a8 01 f5 ef e8 b1 01 f2 01 06 18 8c 04 40 9a 4e>
        }
        const sub = pb.decode(buf.slice(7));
        if (sub[5] && sub[5][2]) {
            let str = String(sub[5][2]);
            if (str.includes("获得群主授予的")) {
                return; //有bug以后再研究
                // const user_id = sub[5][5];
                // str = str.substr(0, str.length - 2);
                // const title = str.substr(str.lastIndexOf("获得群主授予的") + 7);
                // str = str.substr(0, str.length - title.length - 7);
                // const nickname = str.substr(2);
                // try {
                //     this.gml.get(group_id).get(user_id).title = title;
                //     this.gml.get(group_id).get(user_id).title_expire_time = -1;
                // } catch (e) { }
                // return this.em("notice.group.title", {
                //     group_id, user_id,
                //     nickname, title
                // });
            }

            let field, enable;
            if (str.includes("坦白说")) {
                field = "enable_confess";
                enable = str.includes("开启");
            } else if (str.includes("临时会话")) {
                field = "enable_temp_chat";
                enable = str.includes("允许");
            } else if (str.includes("新的群聊")) {
                field = "enable_new_group";
                enable = str.includes("允许");
            } else {
                return;
            }
            return onGroupSetting.call(this, group_id, field, enable, time);
        }
    },
};

/**
 * @this {import("./ref").Client}
 */
function onOnlinePush(blob, seq) {
    const nested = jce.decode(blob);
    const list = nested[2];
    const rubbish = [];
    for (let v of list) {
        rubbish.push(jce.encodeNested([
            this.uin, v[1], v[3], v[8], 0, 0, 0, 0, 0, 0, 0
        ]));
        if (!this.sync_finished) continue;
        const time = v[5];
        if (v[2] === 528) {
            const decoded = jce.decodeNested(v[6]);
            const type = decoded[0], buf = decoded[10];
            if (typeof push528[type] === "function")
                push528[type].call(this, buf, time, v[0]);
        }
        if (v[2] === 732) {
            const group_id = v[6].readUInt32BE();
            const type = v[6][4];
            if (typeof push732[type] === "function")
                push732[type].call(this, group_id, v[6], time);
        }
    }
    handleOnlinePush.call(this, nested[3], seq, rubbish);
}

/**
 * @this {import("./ref").Client}
 */
function onOnlinePushTrans(blob, seq) {
    const push = pb.decode(blob);
    handleOnlinePush.call(this, push[11], seq);
    if (!this.sync_finished) return;
    const time = push[8];
    const buf = push[10].toBuffer();
    const group_id = buf.readUInt32BE();
    if (push[3] === 44) {
        if (buf[5] === 0 || buf[5] === 1) {
            const user_id = buf.readUInt32BE(6);
            const set = buf[10] > 0;
            (async () => {
                try {
                    (await this.getGroupMemberInfo(group_id, user_id)).data.role = (set ? "admin" : "member");
                } catch (e) { }
                this.em("notice.group.admin", {
                    group_id, user_id, set, time
                });
            })();
        } else if (buf[5] === 0xFF) {
            const operator_id = buf.readUInt32BE(6);
            const user_id = buf.readUInt32BE(10);
            (async () => {
                try {
                    this.gl.get(group_id).owner_id = user_id;
                    (await this.getGroupMemberInfo(group_id, operator_id)).data.role = "member";
                    (await this.getGroupMemberInfo(group_id, user_id)).data.role = "owner";
                } catch (e) { }
                this.em("notice.group.transfer", {
                    group_id, operator_id, user_id, time
                });
            })();
        }
    }
    if (push[3] === 34) {

        const user_id = buf.readUInt32BE(5);
        let operator_id, dismiss = false, member;
        try {
            member = this.gml.get(group_id).get(user_id);
        } catch { }
        if (buf[9] === 0x82 || buf[9] === 0x2) {
            operator_id = user_id;
            try {
                this.gml.get(group_id).delete(user_id);
            } catch { }
        } else {
            operator_id = buf.readUInt32BE(10);
            if (buf[9] === 0x01 || buf[9] === 0x81)
                dismiss = true;
            if (user_id === this.uin) {
                this.gl.delete(group_id);
                this.gml.delete(group_id);
                this.logger.info(`更新了群列表，删除了群：${group_id}`);
            } else {
                try {
                    this.gml.get(group_id).delete(user_id);
                } catch { }
                this.logger.info(`${user_id}离开了群${group_id}`);
            }
        }
        try {
            this.gl.get(group_id).member_count--;
        } catch { }
        this.em("notice.group.decrease", {
            group_id, user_id, operator_id, dismiss, member, time
        });
    }
}

/**
 * @this {import("./ref").Client}
 */
function onC2CMsgSync(blob, seq) {
    handleOnlinePush.call(this, pb.decode(blob)[2], seq);
}

/**
 * @this {import("./ref").Client}
 */
async function onGroupMsg(blob, seq) {
    if (!this.sync_finished)
        return;
    try {
        /**
         * @type {import("./ref").Msg}
         */
        let msg = pb.decode(blob)[1];

        //生成消息id
        const head = msg[1], content = msg[2], body = msg[3];
        const user_id = head[1], time = head[6], seq = head[5];
        const group_id = head[9][1], random = body[1][1][3];
        const message_id = genGroupMessageId(group_id, user_id, seq, random, time, content[1]);

        if (content[1] > 1) {
            //重组分片消息
            if (content[2] === 0)
                this.emit(`interval.${group_id}.${body[1][1][3]}`, message_id);
            msg = rebuildFragments(msg);
            if (!msg)
                return;
        } else {
            //非分片消息
            this.emit(`interval.${group_id}.${body[1][1][3]}`, message_id);
        }

        ++this.stat.recv_msg_cnt;

        //解析消息
        const data = await parseGroupMsg.call(this, msg, true);
        if (data && data.raw_message) {
            if (data.user_id === this.uin && this.config.ignore_self)
                return;
            data.reply = (message, auto_escape = false) => this.sendGroupMsg(data.group_id, message, auto_escape);
            data.message_id = message_id;
            const sender = data.sender;
            this.logger.info(`recv from: [Group: ${data.group_name}(${data.group_id}), Member: ${sender.card ? sender.card : sender.nickname}(${data.user_id})] ` + data.raw_message);
            this.em("message.group." + data.sub_type, data);
        }
    } catch (e) {
        this.logger.debug(e);
    }
}

/**
 * @this {import("./ref").Client}
 */
async function onDiscussMsg(blob, seq) {
    ++this.stat.recv_msg_cnt;
    const o = pb.decode(blob);
    handleOnlinePush.call(this, o[2], seq);
    if (!this.sync_finished)
        return;
    try {
        const data = await parseDiscussMsg.call(this, o[1]);
        if (data && data.raw_message) {
            if (data.user_id === this.uin && this.config.ignore_self)
                return;
            data.reply = (message, auto_escape = false) => this.sendDiscussMsg(data.discuss_id, message, auto_escape);
            const sender = data.sender;
            this.logger.info(`recv from: [Discuss: ${data.discuss_name}(${data.discuss_id}), Member: ${sender.card}(${data.user_id})] ` + data.raw_message);
            this.em("message.discuss", data);
        }
    } catch (e) {
        this.logger.debug(e);
    }
}

const FRAG = new Map;

/**
 * Fuck Tencent
 * 1.是最后一个分片，返回组装好的消息
 * 2.不是最后一个分片，返回空
 * @param {import("./ref").Msg} msg 
 * @returns {import("./ref").Msg}
 */
function rebuildFragments(msg) {
    const head = msg[1], content = msg[2], body = msg[3];
    const cnt = content[1], index = content[2], div = content[3];
    const id = head[1] + "-" + div;
    if (!FRAG.has(id)) {
        FRAG.set(id, {3: new Array(cnt)});
        setTimeout(() => {
            FRAG.delete(id);
        }, 5000);
    }
    const comb = FRAG.get(id);
    comb[3][index] = body;
    if (index === 0) {
        comb[1] = head;
        comb[2] = content;
    }
    if (!comb[3].includes(undefined)) {
        const new_body = {
            1: {
                1: body[1][1],
                2: []
            }
        };
        for (let v of comb[3]) {
            if (v[1][2])
                new_body[1][2].push(v[1][2]);
        }
        new_body[1][2] = new_body[1][2].flat();
        comb[3] = new_body;
        FRAG.delete(id);
        return comb;
    }
}

module.exports = {
    onOnlinePush, onOnlinePushTrans, onC2CMsgSync, onGroupMsg, onDiscussMsg
};