const { makeWASocket, Browsers, DisconnectReason, fetchLatestBaileysVersion, makeCacheableSignalKeyStore, makeInMemoryStore, useMultiFileAuthState, downloadMediaMessage, isJidUser } = require('@whiskeysockets/baileys'); const { writeFile } = require('fs/promises'); const waEmitter = require('../emitter'); const { formatPhoneNumber, parsePhoneNumber, formatStatus, formatTimestamp } = require('./helper'); const generateId = require('../../utils/generateId.util'); const NodeCache = require('node-cache'); const P = require('pino'); waEmitter.on('message:updated', event => { console.info('msg:evt:updated', event); }); waEmitter.on('message:received', event => { console.info('msg:evt:received', event); }); waEmitter.on('connection:open', event => { console.info('con:evt.open', event); }); waEmitter.on('connection:close', event => { console.info('con:evt.close', event); }); const createWhatsApp = async phone => { const channelId = generateId(); const whatsAppNo = phone; let socketRef = null; // 储存键值对 msgId-externalId // TODO 什么时候清理旧的? const msgIdMap = new Map(); const sendTextMessage = async (number, content, externalId) => { const jid = formatPhoneNumber(number); return new Promise(() => { socketRef.sendMessage(jid, { text: content }) .then(msg => { msgIdMap.set(msg.key.id, externalId); }) .catch(ex => { waEmitter.emit('message:updated', { id: generateId(), externalId, status: 'failed', direction: 'outbound', from: whatsAppNo, to: number, errro: `发送文本消息出错 ` + ex, eventSource: 'sendMessage.promise.catch', updateTime: formatTimestamp(new Date().getTime() / 1000), }); }); }); }; const sendImageMessage = async (number, imageUrl) => { const jid = formatPhoneNumber(number); try { const msgInfo = await waSocket.sendMessage(jid, { image: { url: imageUrl }, }); return { messageId: msgInfo?.key?.id ?? generateId() }; } catch (ex) { waEmitter.emit('message.error', { messge: `[${whatsAppNo}->${number}]发送图片消息出错`, from: whatsAppNo, to: number, error: ex }) console.error(`[${whatsAppNo}->${number}]发送图片消息出错: `, ex); } }; const getProfilePicture = async (whatsAppNo) => { const number = formatPhoneNumber(whatsAppNo); try { const ppUrl = await waSocket.profilePictureUrl(number); console.log('头像: ' + ppUrl); } catch (ex) { console.error('头像出错: ', ex); } } const handleMessagesUpsert = async upsert => { console.info('messages.upsert: ', JSON.stringify(upsert, undefined, 2)); if (upsert.type === 'notify') { for (const msg of upsert.messages) { // 没有类型的消息,先忽略 if (!msg.message) { continue; } const messageType = Object.keys(msg.message)[0]; console.log('messageType', messageType); const fromWhatsAppNo = parsePhoneNumber(msg.key.remoteJid); if (msg.message?.conversation || msg.message?.extendedTextMessage?.text) { const text = msg.message?.conversation || msg.message?.extendedTextMessage?.text; if (text.indexOf('「自~测」') > -1) { sendTextMessage(fromWhatsAppNo, '这是自测消息:' + new Date().toString()); } const externalId = msgIdMap.get(msg.key.id); if (msg.key.fromMe) { waEmitter.emit('message:updated', { id: msg.key.id, externalId, status: formatStatus(msg.status), direction: 'outbound', from: whatsAppNo, to: fromWhatsAppNo, type: 'text', text: { body: text, }, conversation: { type: isJidUser(msg.key.remoteJid) ? 'individual' : 'group', }, customerProfile: { id: parsePhoneNumber(msg.key.participant), name: msg.pushName, }, eventSource: 'messages.upsert.notify', updateTime: formatTimestamp(msg.messageTimestamp), }); } else { waEmitter.emit('message:received', { id: msg.key.id, externalId, status: '', direction: 'inbound', from: fromWhatsAppNo, to: whatsAppNo, type: 'text', text: { body: text, }, conversation: { type: isJidUser(msg.key.remoteJid) ? 'individual' : 'group', }, customerProfile: { id: parsePhoneNumber(msg.key.participant), name: msg.pushName, }, eventSource: 'messages.upsert.notify', createTime: formatTimestamp(msg.messageTimestamp), }); } } } } else if (upsert.type === 'append') { for (const msg of upsert.messages) { if (msg.message?.conversation || msg.message?.extendedTextMessage?.text) { const text = msg.message?.conversation || msg.message?.extendedTextMessage?.text; const fromWhatsAppNo = parsePhoneNumber(msg.key.remoteJid); const externalId = msgIdMap.get(msg.key.id); if (msg.key.fromMe) { waEmitter.emit('message:updated', { id: msg.key.id, externalId, status: formatStatus(msg.status), direction: 'outbound', from: whatsAppNo, to: fromWhatsAppNo, type: 'text', text: { body: text, }, conversation: { type: isJidUser(msg.key.remoteJid) ? 'individual' : 'group', }, customerProfile: { id: parsePhoneNumber(msg.participant), name: msg.pushName, }, eventSource: 'messages.upsert.append', updateTime: formatTimestamp(msg.messageTimestamp), }); } } } } } const handleMessagesUpdate = async messageUpdate => { console.info('messages.update: ', JSON.stringify(messageUpdate, undefined, 2)); for (const msg of messageUpdate) { const externalId = msgIdMap.get(msg.key.id); waEmitter.emit('message:updated', { id: msg.key.id, externalId, status: formatStatus(msg.update?.status), direction: msg.key.fromMe ? 'outbound' : 'inbound', from: msg.key.fromMe ? whatsAppNo : parsePhoneNumber(msg.key.remoteJid), to: msg.key.fromMe ? parsePhoneNumber(msg.key.remoteJid) : whatsAppNo, conversation: { type: isJidUser(msg.key.remoteJid) ? 'individual' : 'group', }, customerProfile: { id: parsePhoneNumber(msg.key.participant), name: msg.pushName, }, eventSource: 'messages.updated', updateTime: formatTimestamp(new Date().getTime() / 1000), }); } } const handleCredsUpdate = async () => { } const handleConnection = async (socket) => { socketRef = socket; socketRef.sendMessage('8613317835586@s.whatsapp.net', { text: '使用原始连接发消息:' + new Date().toString()}) } // const start = async () => { // } const start = async () => { let qrCode = null; const logger = P({ timestamp: () => `,"time":"${new Date().toJSON()}"` }, P.destination('./wa-logs-' + phone + '_' + channelId + '.txt')); logger.level = 'trace'; const msgRetryCounterCache = new NodeCache(); const storeFilename = './baileys_auth_info/baileys_store_' + phone + '_' + channelId + '.json' const store = makeInMemoryStore({ logger }); store?.readFromFile(storeFilename); // save every 10s setInterval(() => { store?.writeToFile(storeFilename); }, 10_000); const { state, saveCreds } = await useMultiFileAuthState('baileys_auth_info/' + phone + '_' + channelId); // fetch latest version of WA Web const { version, isLatest } = await fetchLatestBaileysVersion(); const waSocket = makeWASocket({ version, logger, auth: { creds: state.creds, keys: makeCacheableSignalKeyStore(state.keys, logger), }, // connectTimeoutMs: 1000*60*10, // defaultQueryTimeoutMs: 1000*60*1, // keepAliveIntervalMs: 1000*60*60, //retryRequestDelayMs: 1000*25, // https://github.com/WhiskeySockets/Baileys/blob/31bc8ab/src/Utils/generics.ts#L21 // https://github.com/WhiskeySockets/Baileys/blob/31bc8ab4e2c825c0d774875701ed07e20d05bdb6/WAProto/WAProto.proto browser: Browsers.ubuntu('IOS_PHONE'),//Browsers.macOS('SAFARI'),//Browsers.ubuntu('IOS_PHONE'),//Browsers.baileys('WEAR_OS'),// msgRetryCounterCache, generateHighQualityLinkPreview: false, syncFullHistory: false, }); store?.bind(waSocket.ev); waSocket.ev.process( // events is a map for event name => event data async(events) => { // something about the connection changed // maybe it closed, or we received all offline message or connection opened if(events['connection.update']) { const update = events['connection.update'] const { connection, qr, lastDisconnect } = update if (connection === 'close') { if((lastDisconnect?.error)?.output?.statusCode !== DisconnectReason.loggedOut) { start(); } else { waEmitter.emit('connection:close', { whatsAppNo, channelId, eventSource: 'connection.update.close', status: 'offline', }); } } else if (connection === 'open') { // if (connectionStatus === 'open') { // sendTextMessage('120363363417115199@g.us', whatsAppNo + ' 在线:' + new Date().toString()); // } else { // // 扫码成功后向这个群('120363363417115199@g.us')发消息,后续就能使用 API 发送了。 //sendTextMessage('86133l17835586', whatsAppNo + ' 登录成功:' + new Date().toString()); //waSocket.sendMessage('8613317835586@s.whatsapp.net', { text: "扫码成功 发送:" + new Date().toString() }) handleConnection(waSocket) // } connectionStatus = 'open'; waEmitter.emit('connection:open', { status: 'open', whatsAppNo, channelId, eventSource: 'connection.update.open', socket: waSocket, sendTextMessage, sendImageMessage, }); } else if (qr !== undefined) { // WebSocket 创建成功等待扫码,如果没有扫码会更新 qr // 第一次一分钟,后面是 20 秒更新一次 if (qrCode === null) { qrCode = qr; console.info('qr: ', qr) waEmitter.emit('connection:qr', { createTimestamp: Date.now(), status: 'offline', version: '0.111111', channelId: channelId, phone: phone, qrCode: qr, socket: waSocket }) } else { // 第一次二维码时效后退出,不需要等待更新二维码 waSocket.logout(() => '二维码已过期'); } } } if (events['creds.update']) { await saveCreds() } } ) waSocket.ev.on('messages.upsert', handleMessagesUpsert); waSocket.ev.on('messages.update', handleMessagesUpdate); }; return { createTimestamp: Date.now(), status: 'offline', version: '0.111111', channelId: channelId, phone: phone, start, sendTextMessage, sendImageMessage, getProfilePicture, }; }; module.exports = { createWhatsApp, };