bot.js

const { Wechaty } = require('wechaty');
const { PuppetPadplus } = require('wechaty-puppet-padplus');
const QRCode = require('qrcode-terminal');
const Parser = require('./msg-parser');
const GitterUtils = require('./gitter-utils');
const CommandUtils = require('./command-utils');
const Dialog = require('./dialog');
const DBUtils = require('./db-utils');
const RoomID = require('./roomid.json');
const Util = require('util');

const puppet = new PuppetPadplus();

const bot = new Wechaty({
    name: "kaiyuanshe",
    puppet: puppet
});

function onScan(qrcode, status) {
    QRCode.generate(qrcode, { small: true });
    const qrcodeImageUrl = [
        'https://api.qrserver.com/v1/create-qr-code/?data=',
        encodeURIComponent(qrcode),
    ].join('');
    console.log(status);
    console.log(qrcodeImageUrl);
}

function onLogin(user) {
    console.log(`${user} login`);
}

function onLogout(user) {
    console.log(`${user} logout`);
}

async function onMessage(msg) {
    if (msg.age() > 60) {
        return;
    }
    if (msg.payload) {
        if (msg.room() != null && msg.payload.type != bot.Message.Type.Unknown) {
            var room = await msg.room();
            console.log(await room.topic() + ":" + room.id);
            GitterUtils.sendMsgToGitter(bot, msg);
            CommandUtils.do_room_command(bot, msg);
        } else if (msg.payload.type != bot.Message.Type.Unknown) {
            CommandUtils.do_user_command(bot, msg);
        }
        // DBUtils.save_msg(msg);
    }
}

async function onFriendship(friendship) {
    console.log(Util.inspect(friendship));
    if (friendship.type() == bot.Friendship.Type.Receive) {
        await friendship.accept();
    } else if (friendship.type() == bot.Friendship.Type.Confirm) {
        var contact = await friendship.contact();
        await contact.sync();
        contact.say(Dialog.greeting);
	console.log("add new friend to db");
        DBUtils.save_wechat_friend(contact);
    }
}

bot.on('scan', onScan);
bot.on('login', onLogin);
bot.on('logout', onLogout);
bot.on('message', onMessage);
bot.on('friendship', onFriendship);

bot.start()
    .then(() => console.log('Starter Bot Started.'))
    .catch(e => console.error(e));

const kue = require('kue');
const queue = kue.createQueue();
queue.process("UserApply", 1, async function (job, done) {
    var work_group = job.data.work_group;
    var room_id = RoomID[work_group];
    var room = await bot.Room.load(room_id);
    if (room) {
        room.sync();
        var text = "有新人申请加入：" + job.data.nick_name + "\n" + "申请加入的小组：" + job.data.work_group + "\n" + "申请理由与自我介绍：" + job.data.introduce + "\n";
        text = text + "组长可以@机器人，并发布命令：“接纳@" + job.data.nick_name + "” 或 “同意@" + job.data.nick_name + "” ，机器人将会把他拉入本群。\n";
        text = text + "如果组长@机器人，并发布命令：“正式@" + job.data.nick_name + "”，机器人将会同时把他拉入本群与开源社正式成员群。";
        await room.say(text);
    }
    if (job.data.referee1 && job.data.referee2) {
        room_id = RoomID["成员发展工作组"];
        room = await bot.Room.load(room_id);
        var text = "有新人申请加入：" + job.data.nick_name + "\n" +
            "申请加入的小组：" + job.data.work_group + "\n" +
            "申请理由与自我介绍：" + job.data.introduce + "\n" +
            "推荐人：" + job.data.referee1 + " 和 " + job.data.referee2;
        if (room) {
            await room.say(text);
        }
    }
    done();
});

queue.process("AddFriend", 1, async function (job, done) {
    var wechat_id = job.data;
    console.log(wechat_id);
    var contact = bot.Contact.load(wechat_id);
    await bot.Friendship.add(contact, "test add friend");
    done();
});

queue.process("AllFriend", 1, async function (job, done) {
    const list = await bot.Contact.findAll();
    list.forEach(function (item, index) {
        DBUtils.save_wechat_friend(item);
    });
    done();
});