Merge remote-tracking branch 'origin/main'
Lei OT
9 months ago
commit
87b53da7e7
@ -0,0 +1,367 @@
|
||||
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,
|
||||
};
|
||||
@ -1,9 +1,164 @@
|
||||
const { createWhatsApp } = require('./index');
|
||||
const {
|
||||
makeWASocket,
|
||||
Browsers,
|
||||
DisconnectReason,
|
||||
fetchLatestBaileysVersion,
|
||||
makeCacheableSignalKeyStore,
|
||||
makeInMemoryStore,
|
||||
useMultiFileAuthState,
|
||||
delay,
|
||||
downloadMediaMessage,
|
||||
isJidUser
|
||||
} = require('@whiskeysockets/baileys');
|
||||
|
||||
const run = async () => {
|
||||
const whatsApp = await createWhatsApp('18777396951');
|
||||
const qr = await whatsApp.start();
|
||||
console.info('qr: ', qr);
|
||||
};
|
||||
const { formatPhoneNumber, parsePhoneNumber, formatStatus, formatTimestamp } = require('./helper');
|
||||
const NodeCache = require('node-cache');
|
||||
const P = require('pino');
|
||||
|
||||
run();
|
||||
const logger = P({ timestamp: () => `,"time":"${new Date().toJSON()}"` }, P.destination('./wa-logs.txt'))
|
||||
logger.level = 'trace'
|
||||
|
||||
// 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
|
||||
const msgRetryCounterCache = new NodeCache()
|
||||
|
||||
// the store maintains the data of the WA connection in memory
|
||||
// can be written out to a file & read from it
|
||||
const store = makeInMemoryStore({ logger })
|
||||
store?.readFromFile('./baileys_store_multi.json')
|
||||
// save every 10s
|
||||
setInterval(() => {
|
||||
store?.writeToFile('./baileys_store_multi.json')
|
||||
}, 10_000)
|
||||
|
||||
// start a connection
|
||||
const startSock = async() => {
|
||||
const { state, saveCreds } = await useMultiFileAuthState('baileys_auth_info')
|
||||
// fetch latest version of WA Web
|
||||
const { version, isLatest } = await fetchLatestBaileysVersion()
|
||||
console.log(`using WA v${version.join('.')}, isLatest: ${isLatest}`)
|
||||
|
||||
const sock = makeWASocket({
|
||||
version,
|
||||
logger,
|
||||
auth: {
|
||||
creds: state.creds,
|
||||
/** caching makes the store faster to send/recv messages */
|
||||
keys: makeCacheableSignalKeyStore(state.keys, logger),
|
||||
},
|
||||
msgRetryCounterCache,
|
||||
generateHighQualityLinkPreview: true,
|
||||
})
|
||||
|
||||
store?.bind(sock.ev)
|
||||
|
||||
|
||||
const sendMessageWTyping = async(msg, jid) => {
|
||||
await sock.presenceSubscribe(jid)
|
||||
await delay(500)
|
||||
|
||||
await sock.sendPresenceUpdate('composing', jid)
|
||||
await delay(2000)
|
||||
|
||||
await sock.sendPresenceUpdate('paused', jid)
|
||||
|
||||
await sock.sendMessage(jid, msg)
|
||||
}
|
||||
|
||||
|
||||
return new Promise((resolve) => {
|
||||
// the process function lets you process all events that just occurred
|
||||
// efficiently in a batch
|
||||
sock.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') {
|
||||
// reconnect if not logged out
|
||||
if((lastDisconnect?.error)?.output?.statusCode !== DisconnectReason.loggedOut) {
|
||||
startSock()
|
||||
} else {
|
||||
console.log('Connection closed. You are logged out.')
|
||||
}
|
||||
}
|
||||
if (connection === 'open') {
|
||||
console.log('Connection open.')
|
||||
|
||||
// setInterval(async() => {
|
||||
|
||||
const characters = 'ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789';
|
||||
let result = '';
|
||||
const charactersLength = characters.length;
|
||||
for (let i = 0; i < 10; i++) {
|
||||
result += characters.charAt(Math.floor(Math.random() * charactersLength));
|
||||
}
|
||||
sendMessageWTyping({ text: result + "-local.Connection open 发送：" + new Date().toString() }, '8613317835586@s.whatsapp.net')
|
||||
// }, 1000*60*5)
|
||||
|
||||
|
||||
|
||||
setInterval(async() => {
|
||||
|
||||
const characters = 'ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789';
|
||||
let result = '';
|
||||
const charactersLength = characters.length;
|
||||
for (let i = 0; i < 10; i++) {
|
||||
result += characters.charAt(Math.floor(Math.random() * charactersLength));
|
||||
}
|
||||
sendMessageWTyping({ text: result + "-setInterval(5min) 发送：" + new Date().toString() }, '8613317835586@s.whatsapp.net')
|
||||
}, 1000*60*5)
|
||||
}
|
||||
|
||||
if (qr !== undefined) {
|
||||
resolve(qr);
|
||||
}
|
||||
|
||||
console.log('connection update', update)
|
||||
}
|
||||
|
||||
if(events['creds.update']) {
|
||||
await saveCreds()
|
||||
}
|
||||
|
||||
// received a new message
|
||||
if(events['messages.upsert']) {
|
||||
const upsert = events['messages.upsert']
|
||||
console.log('recv messages ', JSON.stringify(upsert, undefined, 2))
|
||||
|
||||
if(upsert.type === 'notify') {
|
||||
for (const msg of upsert.messages) {
|
||||
|
||||
if (msg.message?.conversation || msg.message?.extendedTextMessage?.text) {
|
||||
const text = msg.message?.conversation || msg.message?.extendedTextMessage?.text
|
||||
if (text == "requestPlaceholder" && !upsert.requestId) {
|
||||
const messageId = await sock.requestPlaceholderResend(msg.key)
|
||||
console.log('requested placeholder resync, id=', messageId)
|
||||
} else if (upsert.requestId) {
|
||||
console.log('Message received from phone, id=', upsert.requestId, msg)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
}
|
||||
)
|
||||
|
||||
|
||||
|
||||
})
|
||||
|
||||
//return sock
|
||||
}
|
||||
|
||||
async function go() {
|
||||
const qrCode = await startSock();
|
||||
console.info('qr code: ', qrCode);
|
||||
}
|
||||
|
||||
go();
|
||||
Loading…
Reference in New Issue