Haina
/
Global-sales
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

perf: Emitter+Handler 处理发消息

Browse Source
dev/supplier-email-drawer
LiaoYijun 9 months ago
parent c1d3f14f42
commit f9bd3141d5
2 changed files with 67 additions and 93 deletions
Show Stats Download Patch File Download Diff File

16
wai-server/core/baileys/helper.js
Unescape Escape View File

@ -1,18 +1,16 @@
const { isJidUser, isJidGroup, isJidBroadcast, jidEncode, jidDecode, S_WHATSAPP_NET } = require('@whiskeysockets/baileys'); const { isJidUser, isJidGroup, isJidBroadcast, jidDecode, S_WHATSAPP_NET } = require('@whiskeysockets/baileys');
// -> encodePhoneNo const encodeJid = number => {
const formatPhoneNumber = number => {
if (number === null || number === undefined) return ''; if (number === null || number === undefined) return '';
if (isJidGroup(number) || isJidBroadcast(number)) { if (number.indexOf('@') > -1) {
return number; return number;
} else { } else {
return jidEncode(number, S_WHATSAPP_NET); return number + S_WHATSAPP_NET;
} }
}; };
// -> decodePhoneNo const decodeJid = number => {
const parsePhoneNumber = number => {
if (number === null || number === undefined) return ''; if (number === null || number === undefined) return '';
if (isJidUser(number)) { if (isJidUser(number)) {
@ -55,6 +53,6 @@ const formatTimestamp = timestamp => {
module.exports = { module.exports = {
formatTimestamp, formatTimestamp,
formatStatus, formatStatus,
parsePhoneNumber, encodeJid,
formatPhoneNumber, decodeJid,
}; };

144
wai-server/core/baileys/index.js
Unescape Escape View File

@ -13,7 +13,7 @@ const { writeFile } = require('fs/promises');
const waEmitter = require('../emitter'); const waEmitter = require('../emitter');
const serverConfig = require('../../config').server; const serverConfig = require('../../config').server;
const { formatPhoneNumber, parsePhoneNumber, formatStatus, formatTimestamp } = require('./helper'); const { encodeJid, decodeJid, formatStatus, formatTimestamp } = require('./helper');
const generateId = require('../../utils/generateId.util'); const generateId = require('../../utils/generateId.util');
const NodeCache = require('node-cache'); const NodeCache = require('node-cache');
const P = require('pino'); const P = require('pino');
@ -34,7 +34,7 @@ const createWhatsApp = async phone => {
setInterval(() => { setInterval(() => {
store?.writeToFile(storeFilename); store?.writeToFile(storeFilename);
}, 10_000); }, 10_000);
const { state, saveCreds } = await useMultiFileAuthState('baileys_auth_info/' + phone + '_' + channelId); const { state, saveCreds } = await useMultiFileAuthState('baileys_auth_info/' + phone);
// fetch latest version of WA Web // fetch latest version of WA Web
const { version, isLatest } = await fetchLatestBaileysVersion(); const { version, isLatest } = await fetchLatestBaileysVersion();
const waVersion = version.join('.') + ', ' + (isLatest ? 'latest' : 'out'); const waVersion = version.join('.') + ', ' + (isLatest ? 'latest' : 'out');
@ -53,7 +53,7 @@ const createWhatsApp = async phone => {
const messageType = Object.keys(msg.message)[0]; const messageType = Object.keys(msg.message)[0];
console.log('messageType', messageType); console.log('messageType', messageType);
const fromWhatsAppNo = parsePhoneNumber(msg.key.remoteJid); const fromWhatsAppNo = decodeJid(msg.key.remoteJid);
if (msg.message?.conversation || msg.message?.extendedTextMessage?.text) { if (msg.message?.conversation || msg.message?.extendedTextMessage?.text) {
@ -76,7 +76,7 @@ const createWhatsApp = async phone => {
type: isJidUser(msg.key.remoteJid) ? 'individual' : 'group', type: isJidUser(msg.key.remoteJid) ? 'individual' : 'group',
}, },
customerProfile: { customerProfile: {
id: parsePhoneNumber(msg.key.participant), id: decodeJid(msg.key.participant),
name: msg.pushName, name: msg.pushName,
}, },
whatsAppNo, whatsAppNo,
@ -100,7 +100,7 @@ const createWhatsApp = async phone => {
type: isJidUser(msg.key.remoteJid) ? 'individual' : 'group', type: isJidUser(msg.key.remoteJid) ? 'individual' : 'group',
}, },
customerProfile: { customerProfile: {
id: parsePhoneNumber(msg.key.participant), id: decodeJid(msg.key.participant),
name: msg.pushName, name: msg.pushName,
}, },
whatsAppNo, whatsAppNo,
@ -115,7 +115,7 @@ const createWhatsApp = async phone => {
for (const msg of upsert.messages) { for (const msg of upsert.messages) {
if (msg.message?.conversation || msg.message?.extendedTextMessage?.text) { if (msg.message?.conversation || msg.message?.extendedTextMessage?.text) {
const text = msg.message?.conversation || msg.message?.extendedTextMessage?.text; const text = msg.message?.conversation || msg.message?.extendedTextMessage?.text;
const fromWhatsAppNo = parsePhoneNumber(msg.key.remoteJid); const fromWhatsAppNo = decodeJid(msg.key.remoteJid);
const externalId = externalIdCache.get(msg.key.id); const externalId = externalIdCache.get(msg.key.id);
if (msg.key.fromMe) { if (msg.key.fromMe) {
@ -134,7 +134,7 @@ const createWhatsApp = async phone => {
type: isJidUser(msg.key.remoteJid) ? 'individual' : 'group', type: isJidUser(msg.key.remoteJid) ? 'individual' : 'group',
}, },
customerProfile: { customerProfile: {
id: parsePhoneNumber(msg.participant), id: decodeJid(msg.participant),
name: msg.pushName, name: msg.pushName,
}, },
whatsAppNo, whatsAppNo,
@ -164,13 +164,13 @@ const createWhatsApp = async phone => {
externalId, externalId,
status: formatStatus(msg.update?.status), status: formatStatus(msg.update?.status),
direction: msg.key.fromMe ? 'outbound' : 'inbound', direction: msg.key.fromMe ? 'outbound' : 'inbound',
from: msg.key.fromMe ? whatsAppNo : parsePhoneNumber(msg.key.remoteJid), from: msg.key.fromMe ? whatsAppNo : decodeJid(msg.key.remoteJid),
to: msg.key.fromMe ? parsePhoneNumber(msg.key.remoteJid) : whatsAppNo, to: msg.key.fromMe ? decodeJid(msg.key.remoteJid) : whatsAppNo,
conversation: { conversation: {
type: isJidUser(msg.key.remoteJid) ? 'individual' : 'group', type: isJidUser(msg.key.remoteJid) ? 'individual' : 'group',
}, },
customerProfile: { customerProfile: {
id: parsePhoneNumber(msg.key.participant), id: decodeJid(msg.key.participant),
name: msg.pushName, name: msg.pushName,
}, },
whatsAppNo, whatsAppNo,
@ -199,70 +199,44 @@ const createWhatsApp = async phone => {
//retryRequestDelayMs: 1000*25, //retryRequestDelayMs: 1000*25,
// https://github.com/WhiskeySockets/Baileys/blob/31bc8ab/src/Utils/generics.ts#L21 // https://github.com/WhiskeySockets/Baileys/blob/31bc8ab/src/Utils/generics.ts#L21
// https://github.com/WhiskeySockets/Baileys/blob/31bc8ab4e2c825c0d774875701ed07e20d05bdb6/WAProto/WAProto.proto // https://github.com/WhiskeySockets/Baileys/blob/31bc8ab4e2c825c0d774875701ed07e20d05bdb6/WAProto/WAProto.proto
browser: Browsers.macOS('SAFARI'),//Browsers.macOS('SAFARI'),//Browsers.ubuntu('IOS_PHONE'),//Browsers.baileys('WEAR_OS'),// browser: Browsers.baileys('WEAR_OS'),//Browsers.macOS('SAFARI'),//Browsers.ubuntu('IOS_PHONE'),//Browsers.baileys('WEAR_OS'),//
msgRetryCounterCache, msgRetryCounterCache,
generateHighQualityLinkPreview: false, generateHighQualityLinkPreview: false,
syncFullHistory: false, syncFullHistory: false,
}); });
store?.bind(waSocket.ev); store?.bind(waSocket.ev);
const sendTextMessage = async (number, content, externalId) => { const sendTextMessageHandler = (event) => {
const jid = formatPhoneNumber(number); const { to: number, externalId, content } = event;
waSocket.sendMessage(jid, { text: content }) const jid = encodeJid(number);
.then(msg => { console.info('number: %s; jid: %s: ', number, jid)
externalIdCache.set(msg.key.id, externalId); waSocket.sendMessage(
}) jid, { text: content }
.catch(ex => { ).then(msg => {
console.error('sendTextMessage.error: ', ex) externalIdCache.set(msg.key.id, externalId)
waEmitter.emit('message:updated', { }).catch(ex => {
id: generateId(), console.error('sendMessage.error: ', ex)
externalId, waEmitter.emit('message:updated', {
status: 'failed', id: generateId(),
direction: 'outbound', externalId,
from: whatsAppNo, status: 'failed',
to: number, direction: 'outbound',
error: `发送文本消息出错 ` + ex,
eventSource: serverConfig.name + '.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, from: whatsAppNo,
to: number, to: number,
error: ex error: `发送文本消息出错 ` + ex,
}) eventSource: serverConfig.name + '.sendMessage.catch',
console.error(`[${whatsAppNo}->${number}]发送图片消息出错: `, ex); updateTime: formatTimestamp(new Date().getTime() / 1000),
} });
}; });
const getProfilePicture = async (whatsAppNo) => {
const number = formatPhoneNumber(whatsAppNo);
try {
const ppUrl = await waSocket.profilePictureUrl(number);
console.log('头像: ' + ppUrl);
} catch (ex) {
console.error('头像出错: ', ex);
}
} }
waSocket.ev.on('connection.update', async update => { waSocket.ev.on('connection.update', async update => {
console.log('connection update: ', update); console.log('connection update: ', update);
const { connection, lastDisconnect, qr } = update; const { connection, lastDisconnect, qr } = update;
if (connection === 'close') { if (connection === 'close') {
waEmitter.off('request.' + whatsAppNo + '.send.text', sendTextMessageHandler);
if((lastDisconnect?.error)?.output?.statusCode !== DisconnectReason.loggedOut) { if((lastDisconnect?.error)?.output?.statusCode !== DisconnectReason.loggedOut) {
start(); start();
} else { } else {
@ -277,34 +251,36 @@ const createWhatsApp = async phone => {
status: 'open', whatsAppNo, channelId, status: 'open', whatsAppNo, channelId,
eventSource: serverConfig.name + '.connection.update.open', eventSource: serverConfig.name + '.connection.update.open',
}); });
waEmitter.on('request.' + whatsAppNo + '.send.text', sendTextMessageHandler);
// 考虑迁移到 connection.update 事件之外，测试看是否能发送消息？ // 考虑迁移到 connection.update 事件之外，测试看是否能发送消息？
waEmitter.on('request.' + whatsAppNo + '.send.text', event => { // waEmitter.on('request.' + whatsAppNo + '.send.text', event => {
const {to: number, externalId, content} = event; // const { to: number, externalId, content } = event;
console.info('request.' + whatsAppNo + '.send.text:', event) // const jid = encodeJid(number);
// const jid = formatPhoneNumber(event.to); // console.info('number: %s; jid: %s: ', number, jid)
waSocket.sendMessage( // waSocket.sendMessage(
number + '@s.whatsapp.net', { text: content } // jid, { text: content }
).then(msg => { // ).then(msg => {
externalIdCache.set(msg.key.id, externalId) // externalIdCache.set(msg.key.id, externalId)
}).catch(ex => { // }).catch(ex => {
console.error('sendMessage.error: ', ex) // console.error('sendMessage.error: ', ex)
waEmitter.emit('message:updated', { // waEmitter.emit('message:updated', {
id: generateId(), // id: generateId(),
externalId, // externalId,
status: 'failed', // status: 'failed',
direction: 'outbound', // direction: 'outbound',
from: whatsAppNo, // from: whatsAppNo,
to: number, // to: number,
error: `发送文本消息出错 ` + ex, // error: `发送文本消息出错 ` + ex,
eventSource: serverConfig.name + '.sendMessage.catch', // eventSource: serverConfig.name + '.sendMessage.catch',
updateTime: formatTimestamp(new Date().getTime() / 1000), // updateTime: formatTimestamp(new Date().getTime() / 1000),
}); // });
}); // });
}); // });
waEmitter.on('request.' + whatsAppNo + '.send.image', event => { waEmitter.on('request.' + whatsAppNo + '.send.image', event => {
const {to: number, externalId, imageUrl} = event; const {to: number, externalId, imageUrl} = event;
const jid = formatPhoneNumber(event.to); const jid = encodeJid(number);
waSocket.sendMessage( waSocket.sendMessage(
jid, {image: { url: imageUrl }} jid, {image: { url: imageUrl }}
).then(msg => { ).then(msg => {

Write Preview
Loading…
Cancel
Save