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

perf: 增加群联系人更新；模拟 macOS；

Browse Source
dev/supplier-email-drawer
LiaoYijun 9 months ago
parent 24cb081232
commit 8f6d7e15a7
1 changed files with 174 additions and 146 deletions
Show Stats Download Patch File Download Diff File

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

@ -1,12 +1,12 @@
const { const {
makeWASocket, makeWASocket,
WAProto, Browsers,
DisconnectReason, DisconnectReason,
fetchLatestBaileysVersion, fetchLatestBaileysVersion,
makeCacheableSignalKeyStore, makeCacheableSignalKeyStore,
makeInMemoryStore, makeInMemoryStore,
useMultiFileAuthState, useMultiFileAuthState,
downloadMediaMessage downloadMediaMessage,
} = require('@whiskeysockets/baileys'); } = require('@whiskeysockets/baileys');
const { writeFile } = require('fs/promises'); const { writeFile } = require('fs/promises');
const waEmitter = require('../emitter'); const waEmitter = require('../emitter');
@ -15,254 +15,282 @@ const generateId = require('../../utils/generateId.util');
const NodeCache = require('node-cache'); const NodeCache = require('node-cache');
const P = require('pino'); const P = require('pino');
const logger = P({ timestamp: () => `,"time":"${new Date().toJSON()}"` }, P.destination('./wa-logs.txt')) const logger = P({ timestamp: () => `,"time":"${new Date().toJSON()}"` }, P.destination('./wa-logs.txt'));
logger.level = 'trace' logger.level = 'trace';
// external map to store retry counts of messages when decryption/encryption fails // external map to store retry counts of messages when decryption/encryption fails
// keep this out of the socket itself, so as to prevent a message decryption/encryption loop across socket restarts // keep this out of the socket itself, so as to prevent a message decryption/encryption loop across socket restarts
const msgRetryCounterCache = new NodeCache() const msgRetryCounterCache = new NodeCache();
// the store maintains the data of the WA connection in memory // the store maintains the data of the WA connection in memory
// can be written out to a file & read from it // can be written out to a file & read from it
const store = makeInMemoryStore({ logger }) const store = makeInMemoryStore({ logger });
store?.readFromFile('./baileys_store_multi.json') store?.readFromFile('./baileys_store_multi.json');
// save every 10s // save every 10s
setInterval(() => { setInterval(() => {
store?.writeToFile('./baileys_store_multi.json') store?.writeToFile('./baileys_store_multi.json');
}, 10_000) }, 10_000);
waEmitter.on('message:updated', event => { console.info('msg:event', event)}) waEmitter.on('message:updated', event => {
waEmitter.on('message:received', event => { console.info('msg:event', event)}) console.info('msg:event', event);
});
waEmitter.on('message:received', event => {
console.info('msg:event', event);
});
// start a connection // start a connection
const createWhatsApp = async (phone) => { const createWhatsApp = async phone => {
const channelId = generateId() const channelId = generateId();
const whatsAppNo = phone const whatsAppNo = phone;
const { state, saveCreds } = await useMultiFileAuthState('baileys_auth_info/' + phone) 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');
const formatPhoneNumber = (number) => { const formatPhoneNumber = number => {
if (number === null || number === undefined) return '';
if (number === null || number === undefined) return ''
if (number.indexOf('@g.us') > -1) { if (number.indexOf('@g.us') > -1) {
return number return number;
} else if (number.indexOf('@broadcast') > -1) { } else if (number.indexOf('@broadcast') > -1) {
return number return number;
} else { } else {
return number + '@s.whatsapp.net' return number + '@s.whatsapp.net';
} }
} };
const parsePhoneNumber = (number) => {
if (number === null || number === undefined) return '' const parsePhoneNumber = number => {
if (number === null || number === undefined) return '';
if (number.indexOf('@s.whatsapp.net') > -1) { if (number.indexOf('@s.whatsapp.net') > -1) {
return number.split('@')[0] return number.split('@')[0];
} else { } else {
return number return number;
}
}
// status: sent read delivered failed
// 2 sent, 3 delivered, 4 read, 0 error
// Time: 2008-07-07 15:37:07
const formatStatus = (number) => {
if (number === 2) return 'sent'
else if (number === 3) return 'delivered'
else if (number === 4) return 'read'
else if (number === 0) return 'error'
else return 'played'
} }
const formatTimestamp = (timestamp) => { };
if (timestamp === null) return ''
const datetime = new Date(timestamp * 1000) // status: sent read delivered failed
// 2 sent, 3 delivered, 4 read, 0 error
// Time: 2008-07-07 15:37:07
return datetime.getFullYear() + '-' + (datetime.getMonth() + 1) + '-' + datetime.getDay() + ' ' + datetime.getHours() + ':' + datetime.getMinutes() + ':' + datetime.getSeconds() const formatStatus = number => {
} if (number === 2) return 'sent';
else if (number === 3) return 'delivered';
else if (number === 4) return 'read';
else if (number === 0) return 'error';
else return 'played';
};
const formatTimestamp = timestamp => {
if (timestamp === null) return '';
const datetime = new Date(timestamp * 1000);
return datetime.getFullYear() + '-' + (datetime.getMonth() + 1) + '-' + datetime.getDay() + ' ' + datetime.getHours() + ':' + datetime.getMinutes() + ':' + datetime.getSeconds();
};
const sendTextMessage = (whatsAppNo, content) => { const sendTextMessage = (whatsAppNo, content) => {
const number = formatPhoneNumber(whatsAppNo) const number = formatPhoneNumber(whatsAppNo);
console.info('formatPhoneNumber: ', number) console.info('formatPhoneNumber: ', number);
waSocket.sendMessage(number, { text: content }) waSocket.sendMessage(number, { text: content });
} };
const sendImageMessage = (whatsAppNo, imageUrl) => { const sendImageMessage = (whatsAppNo, imageUrl) => {
const number = formatPhoneNumber(whatsAppNo) const number = formatPhoneNumber(whatsAppNo);
waSocket.sendMessage(number, { waSocket.sendMessage(number, {
image: { url: imageUrl}, image: { url: imageUrl },
}) });
} };
let waSocket = null let waSocket = null;
const start = () => { const start = () => {
return new Promise((resolve, reject) => { return new Promise((resolve, reject) => {
waSocket = makeWASocket({ waSocket = makeWASocket({
version, version,
logger, logger,
auth: { auth: {
creds: state.creds, creds: state.creds,
/** caching makes the store faster to send/recv messages */ /** caching makes the store faster to send/recv messages */
keys: makeCacheableSignalKeyStore(state.keys, logger), keys: makeCacheableSignalKeyStore(state.keys, logger),
}, },
msgRetryCounterCache, // https://github.com/WhiskeySockets/Baileys/blob/31bc8ab/src/Utils/generics.ts#L21
generateHighQualityLinkPreview: true, browser: Browsers.macOS('Desktop'),
}) msgRetryCounterCache,
generateHighQualityLinkPreview: false,
// something about the connection changed syncFullHistory: false,
});
// something about the connection changed
// maybe it closed, or we received all offline message or connection opened // maybe it closed, or we received all offline message or connection opened
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') {
console.log('链接断开：', lastDisconnect) console.log('链接断开：', lastDisconnect);
if (lastDisconnect?.error?.output?.statusCode !== DisconnectReason.loggedOut) { if (lastDisconnect?.error?.output?.statusCode !== DisconnectReason.loggedOut) {
start() start();
} else { } else {
waSocket.end((error) => console.error('end.error: ', error)) waSocket.end(error => console.error('end.error: ', error));
waSocket.logout((msg) => console.error('logout.msg: ', msg)) waSocket.logout(msg => console.error('logout.msg: ', msg));
console.log('Connection closed. You are logged out.') console.log('Connection closed. You are logged out.');
} }
} else if(connection === 'open') { } else if (connection === 'open') {
console.info('链接成功') console.info('链接成功');
waEmitter.emit('connection.open', waSocket); waEmitter.emit('connection.open', waSocket);
} else if(qr !== undefined) { } else if (qr !== undefined) {
// WebSocket 创建成功等待扫码，如果没有扫码会更新 qr // WebSocket 创建成功等待扫码，如果没有扫码会更新 qr
resolve(qr) resolve(qr);
} }
}) });
waSocket.ev.on('messages.upsert', async (upsert) => { waSocket.ev.on('messages.upsert', async upsert => {
console.log('收到消息：', JSON.stringify(upsert, undefined, 2)) console.log('收到消息：', JSON.stringify(upsert, undefined, 2));
if (upsert.type === 'notify') { if (upsert.type === 'notify') {
for (const msg of upsert.messages) { for (const msg of upsert.messages) {
const messageType = Object.keys(msg.message)[0];
const messageType = Object.keys(msg.message)[0] console.log('messageType', messageType);
console.log('messageType', messageType)
if (messageType === 'imageMessage') { if (messageType === 'imageMessage') {
// download the message // download the message
const buffer = await downloadMediaMessage( const buffer = await downloadMediaMessage(
msg, msg,
'buffer', 'buffer',
{ }, {},
{ {
logger, logger,
// pass this so that baileys can request a reupload of media // pass this so that baileys can request a reupload of media
// that has been deleted // that has been deleted
reuploadRequest: waSocket.updateMediaMessage reuploadRequest: waSocket.updateMediaMessage,
} },
) );
// save to file // save to file
await writeFile('d:/my-download.jpeg', buffer) await writeFile('d:/my-download.jpeg', buffer);
console.log('writeFile', messageType) console.log('writeFile', messageType);
} }
const fromWhatsAppNo = parsePhoneNumber(msg.key.remoteJid) const fromWhatsAppNo = parsePhoneNumber(msg.key.remoteJid);
if (msg.key.remoteJid.indexOf('@g.us') > -1) {
const metadata = await waSocket.groupMetadata(msg.key.remoteJid);
console.log('群: ' + metadata.subject + ', description: ' + metadata.desc);
try {
const ppUrl = await waSocket.profilePictureUrl(msg.key.remoteJid);
console.log('群头像: ' + ppUrl);
} catch (ex) {
console.error('群头像出错了: ', ex);
}
} else {
console.log('remoteJid: ', msg.key.remoteJid);
try {
const ppUrl = await waSocket.profilePictureUrl(msg.key.remoteJid);
console.log('个人头像: ' + ppUrl);
} catch (ex) {
console.error('个人头像出错了: ', ex);
}
}
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;
if (text.indexOf('图片') > -1){ if (text.indexOf('图片') > -1) {
sendImageMessage(fromWhatsAppNo, 'https://images.asiahighlights.com/allpicture/2022/07/8a7d9ced5936463bb904c82a_cut_750x850_349.webp') sendImageMessage(fromWhatsAppNo, 'https://images.asiahighlights.com/allpicture/2022/07/8a7d9ced5936463bb904c82a_cut_750x850_349.webp');
} else if (text.indexOf('文本') > -1){ } else if (text.indexOf('文本') > -1) {
sendTextMessage(fromWhatsAppNo, '文本消息' + new Date().toString()) sendTextMessage(fromWhatsAppNo, '这是文本消息：' + new Date().toString());
} else if (text.indexOf('发群') > -1){ } else if (text.indexOf('发群') > -1) {
sendTextMessage('120363335516526642@g.us', '这是群消息：' + new Date().toString()) sendTextMessage('120363335516526642@g.us', '这是群消息：' + new Date().toString());
} }
if (msg.key.fromMe) { if (msg.key.fromMe) {
waEmitter.emit('message:updated', { waEmitter.emit('message:updated', {
"id": msg.key.id, id: msg.key.id,
"status": formatStatus(msg.status), status: formatStatus(msg.status),
"from": whatsAppNo, from: whatsAppNo,
"to": fromWhatsAppNo, to: fromWhatsAppNo,
"type": "text", type: 'text',
"text": { text: {
"body": text body: text,
}, },
participant: { participant: {
id: parsePhoneNumber(msg.key.participant), id: parsePhoneNumber(msg.key.participant),
name: msg.pushName name: msg.pushName,
}, },
"updateTime": formatTimestamp(msg.messageTimestamp), updateTime: formatTimestamp(msg.messageTimestamp),
}); });
} else { } else {
waEmitter.emit('message:received', { waEmitter.emit('message:received', {
"id": msg.key.id, id: msg.key.id,
"status": "", status: '',
"from": fromWhatsAppNo, from: fromWhatsAppNo,
"to": whatsAppNo, to: whatsAppNo,
"type": "text", type: 'text',
"text": { text: {
"body": text body: text,
}, },
participant: { participant: {
id: parsePhoneNumber(msg.key.participant), id: parsePhoneNumber(msg.key.participant),
name: msg.pushName name: msg.pushName,
}, },
"createTime": formatTimestamp(msg.messageTimestamp), createTime: formatTimestamp(msg.messageTimestamp),
}); });
} }
} }
} }
} }
}) });
waSocket.ev.on('messages.update', async (messageUpdate) => { waSocket.ev.on('messages.update', async messageUpdate => {
console.info('messages.update: ', messageUpdate) console.info('messages.update: ', messageUpdate);
for (const msg of messageUpdate) { for (const msg of messageUpdate) {
waEmitter.emit('message:updated', { waEmitter.emit('message:updated', {
"id": msg.key.id, id: msg.key.id,
"status": formatStatus(msg.update.status), status: formatStatus(msg.update.status),
"from": msg.key.fromMe ? whatsAppNo : parsePhoneNumber(msg.key.remoteJid), from: msg.key.fromMe ? whatsAppNo : parsePhoneNumber(msg.key.remoteJid),
"to": msg.key.fromMe ? parsePhoneNumber(msg.key.remoteJid) : whatsAppNo, to: msg.key.fromMe ? parsePhoneNumber(msg.key.remoteJid) : whatsAppNo,
participant: { participant: {
id: parsePhoneNumber(msg.key.participant), id: parsePhoneNumber(msg.key.participant),
name: msg.pushName name: msg.pushName,
}, },
"updateTime": formatTimestamp(new Date().getTime() / 1000), updateTime: formatTimestamp(new Date().getTime() / 1000),
}); });
} }
}) });
waSocket.ev.on('group-participants.update', async GroupMetadata => {
console.info('group-participants.update: ', GroupMetadata);
});
// 不绑定不会影响扫码登录 // 不绑定不会影响扫码登录
store?.bind(waSocket.ev) store?.bind(waSocket.ev);
// the process function lets you process all events that just occurred // the process function lets you process all events that just occurred
// efficiently in a batch // efficiently in a batch
waSocket.ev.process( waSocket.ev.process(
// events is a map for event name => event data // events is a map for event name => event data
async (events) => { async events => {
// credentials updated -- save them // credentials updated -- save them
if (events['creds.update']) { if (events['creds.update']) {
await saveCreds() await saveCreds();
} }
}, },
) );
}) });
} };
return { return {
createTimestamp: Date.now(), createTimestamp: Date.now(),
status: 'offline', status: 'offline',
version: waVersion, version: waVersion,
channelId : channelId, channelId: channelId,
phone: phone, phone: phone,
start, sendTextMessage, sendImageMessage start,
} sendTextMessage,
} sendImageMessage,
};
};
module.exports = { module.exports = {
createWhatsApp, createWhatsApp,

Write Preview
Loading…
Cancel
Save