feat: 集成 Baileys 和 HTTP 框架; 暂时去掉eslint

baileys-server/index.js

@@ -0,0 +1,193 @@
+/* eslint-disable no-undef */
+const {
+  makeWASocket,
+  WAProto,
+  DisconnectReason,
+  fetchLatestBaileysVersion,
+  makeCacheableSignalKeyStore,
+  makeInMemoryStore,
+  useMultiFileAuthState,
+  downloadMediaMessage
+} = require('@whiskeysockets/baileys');
+
+const { writeFile } = require('fs/promises');
+
+const NodeCache = require('node-cache');
+const P = require('pino');
+
+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 channelId = '创建时赋值，唯一标识'
+  const phone = '手机号'
+  const createTimestamp = '创建时间戳'
+  const status = 'close, open, connecting, online'
+  const waVersion = 'v2.3000.1017531287, latest(out)'
+
+  const sendTextMessage = (whatsAppNo, content) => {
+    sock.sendMessage(whatsAppNo + '@s.whatsapp.net', { text: content })
+  }
+
+  const sendImageMessage = (whatsAppNo, imageUrl) => {
+    sock.sendMessage(whatsAppNo + '@s.whatsapp.net', {
+      image: { url: imageUrl},
+    })
+  }
+
+  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,
+  })
+
+  sock.ev.on('connection.update', (update) => {
+    const { connection, lastDisconnect, qr } = update
+    if(connection === 'close') {
+        console.info('链接断了')
+    } else if(connection === 'open') {
+      console.info('扫码成功')
+    } else if(connection === 'connecting') {
+      console.info('二维码：', qr)
+    }
+  })
+
+  sock.ev.on('messages.upsert', async (upsert) => {
+    console.log('收到消息：', JSON.stringify(upsert, undefined, 2))
+
+    if (upsert.type === 'notify') {
+      for (const msg of upsert.messages) {
+
+        const messageType = Object.keys(msg.message)[0]
+
+        console.log('messageType', messageType)
+        if (messageType === 'imageMessage') {
+          // download the message
+          const buffer = await downloadMediaMessage(
+            msg,
+            'buffer',
+            { },
+            {
+                logger,
+                // pass this so that baileys can request a reupload of media
+                // that has been deleted
+                reuploadRequest: sock.updateMediaMessage
+            }
+          )
+          // save to file
+          await writeFile('d:/my-download.jpeg', buffer)
+
+          console.log('writeFile', messageType)
+        }
+
+        if (msg.message?.conversation || msg.message?.extendedTextMessage?.text) {
+          const text = msg.message?.conversation || msg.message?.extendedTextMessage?.text
+
+          if (text.indexOf('图片') > -1){
+            sendImageMessage('8617607730395', 'https://images.asiahighlights.com/allpicture/2022/07/8a7d9ced5936463bb904c82a_cut_750x850_349.webp')
+          } else if (text.indexOf('文本') > -1){
+            sendTextMessage('8617607730395', '文本消息' + new Date().toString())
+          }
+        }
+      }
+    }
+  })
+
+  // 不绑定不会影响扫码登录
+  store?.bind(sock.ev)
+
+  // 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, lastDisconnect, qr } = update
+        if (connection === 'close') {
+          console.log('链接断开：', lastDisconnect)
+          if (lastDisconnect?.error?.output?.statusCode !== DisconnectReason.loggedOut) {
+            startSock()
+          } else {
+            sock.end((error) => console.error('end.error: ', error))
+            sock.logout((msg) => console.error('logout.msg: ', msg))
+            console.log('Connection closed. You are logged out.')
+          }
+        }
+
+        // 扫码成功，可以发送消息
+        if (update.connection === 'open') {
+          await sock.sendMessage('8617607730395' + '@s.whatsapp.net', { text: 'OPEN: ' + new Date().toString() })
+        }
+
+        // WebSocket 创建成功等待扫码，如果没有扫码会更新 qr
+        if (update.connection === 'connecting') {
+          // qr
+        }
+
+        console.log('connection update', update)
+      }
+
+      // credentials updated -- save them
+      if (events['creds.update']) {
+        await saveCreds()
+      }
+
+      // history received
+      if (events['messaging-history.set']) {
+        const { chats, contacts, messages, isLatest, progress, syncType } = events['messaging-history.set']
+        if (syncType === WAProto.HistorySync.HistorySyncType.ON_DEMAND) {
+          console.log('received on-demand history sync, messages=', messages)
+        }
+        console.log(`recv ${chats.length} chats, ${contacts.length} contacts, ${messages.length} msgs (is latest: ${isLatest}, progress: ${progress}%), type: ${syncType}`)
+      }
+
+      // received a new message
+      if (events['messages.upsert']) {
+        const upsert = events['messages.upsert']
+        console.log('收到消息：', 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
+              console.log('收到 notify：', text)
+            }
+          }
+        }
+      }
+    },
+  )
+
+  return sock
+}
+
+startSock()

wai-server/.gitignore

@@ -32,3 +32,8 @@ node_modules/
 .DS_Store
 
 *.bak
+
+wa-logs.txt
+baileys_store_multi.json
+
+baileys_auth_info_*/

wai-server/config/components/log4j.config.js

@@ -39,7 +39,7 @@ log4js.configure(logConfig);
 
 const logger = log4js.getLogger();
 
-console.log = logger.info.bind(logger);
+// console.log = logger.info.bind(logger);
-console.error = logger.error.bind(logger);
+// console.error = logger.error.bind(logger);
 
 // module.exports = logger;

wai-server/core/index.js

@@ -6,7 +6,166 @@ const websocketConnectionI = require('./websocket/connection');
 const websocketService = websocketServicesI();
 const websocketManager = websocketConnectionI(websocketService);
 
+
+const {
+  makeWASocket,
+  WAProto,
+  DisconnectReason,
+  fetchLatestBaileysVersion,
+  makeCacheableSignalKeyStore,
+  makeInMemoryStore,
+  useMultiFileAuthState,
+  downloadMediaMessage
+} = require('@whiskeysockets/baileys');
+
+const { writeFile } = require('fs/promises');
+
+const NodeCache = require('node-cache');
+const P = require('pino');
+
+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 createWhatsApp = async (channelId, phone) => {
+
+  const { state, saveCreds } = await useMultiFileAuthState('baileys_auth_info_' + channelId + '_' + phone)
+  // fetch latest version of WA Web
+  const { version, isLatest } = await fetchLatestBaileysVersion()
+  const waVersion = version.join('.') + ', ' + (isLatest ? 'latest' : 'out')
+
+  return new Promise((resolve, reject) => {
+
+    const sendTextMessage = (whatsAppNo, content) => {
+      sock.sendMessage(whatsAppNo + '@s.whatsapp.net', { text: content })
+    }
+
+    const sendImageMessage = (whatsAppNo, imageUrl) => {
+      sock.sendMessage(whatsAppNo + '@s.whatsapp.net', {
+        image: { url: imageUrl},
+      })
+    }
+
+    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,
+    })
+
+    // something about the connection changed
+    // maybe it closed, or we received all offline message or connection opened
+    sock.ev.on('connection.update', async (update) => {
+      console.log('connection update: ', update)
+      const { connection, lastDisconnect, qr } = update
+      if(connection === 'close') {
+        console.log('链接断开：', lastDisconnect)
+        if (lastDisconnect?.error?.output?.statusCode !== DisconnectReason.loggedOut) {
+          createWhatsApp(channelId, phone)
+        } else {
+          sock.end((error) => console.error('end.error: ', error))
+          sock.logout((msg) => console.error('logout.msg: ', msg))
+          console.log('Connection closed. You are logged out.')
+          reject('Connection closed. You are logged out.')
+        }
+      } else if(connection === 'open') {
+        console.info('链接成功')
+        await sock.sendMessage('8617607730395' + '@s.whatsapp.net', { text: 'OPEN: ' + new Date().toString() })
+      } else if(qr !== undefined) {
+        // WebSocket 创建成功等待扫码，如果没有扫码会更新 qr
+        resolve({
+          qrCode: qr,
+          version: waVersion,
+          createTimestamp: Date.now(),
+          status: 'connecting',
+          channelId : channelId,
+          phone: phone,
+          ...sendTextMessage,
+          ...sendImageMessage
+        })
+      }
+    })
+
+    sock.ev.on('messages.upsert', async (upsert) => {
+      console.log('收到消息：', JSON.stringify(upsert, undefined, 2))
+
+      if (upsert.type === 'notify') {
+        for (const msg of upsert.messages) {
+
+          const messageType = Object.keys(msg.message)[0]
+
+          console.log('messageType', messageType)
+          if (messageType === 'imageMessage') {
+            // download the message
+            const buffer = await downloadMediaMessage(
+              msg,
+              'buffer',
+              { },
+              {
+                  logger,
+                  // pass this so that baileys can request a reupload of media
+                  // that has been deleted
+                  reuploadRequest: sock.updateMediaMessage
+              }
+            )
+            // save to file
+            await writeFile('d:/my-download.jpeg', buffer)
+
+            console.log('writeFile', messageType)
+          }
+
+          const fromWhatsAppNo = msg.key.remoteJid.split('@')[0]
+
+          if (msg.message?.conversation || msg.message?.extendedTextMessage?.text) {
+            const text = msg.message?.conversation || msg.message?.extendedTextMessage?.text
+
+            if (text.indexOf('图片') > -1){
+              sendImageMessage(fromWhatsAppNo, 'https://images.asiahighlights.com/allpicture/2022/07/8a7d9ced5936463bb904c82a_cut_750x850_349.webp')
+            } else if (text.indexOf('文本') > -1){
+              sendTextMessage(fromWhatsAppNo, '文本消息' + new Date().toString())
+            }
+          }
+        }
+      }
+    })
+
+    // 不绑定不会影响扫码登录
+    store?.bind(sock.ev)
+
+    // 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) => {
+        // credentials updated -- save them
+        if (events['creds.update']) {
+          await saveCreds()
+        }
+      },
+    )
+  })
+}
+
 module.exports = {
   websocketService,
   websocketManager,
+  createWhatsApp,
 };

wai-server/index.js

@@ -4,9 +4,13 @@ const http = require('http');
 const server = require('./server');
 
 const { port } = require('./config').server;
-const { websocketService, websocketManager } = require('./core');
+const { websocketService, websocketManager, createWhatsApp } = require('./core');
 
 async function bootstrap() {
+  const whatsApp1 = await createWhatsApp('channel_001', '8618777396951')
+  console.info(whatsApp1)
+  const whatsApp2 = await createWhatsApp('channel_002', '8613557032060')
+  console.info(whatsApp2)
   /**
    * Add external services init as async operations (db, redis, etc...)
    * e.g.
@@ -18,7 +22,7 @@ async function bootstrap() {
 bootstrap()
   .then(server => {
     console.log(`🚀 Server listening on port ${server.address().port}!`);
-    websocketManager.connect(); // connect websocket after server start
+    // websocketManager.connect(); // connect websocket after server start
   })
   .catch(err => {
     setImmediate(() => {

wai-server/package.json

@@ -5,7 +5,7 @@
   "main": "index.js",
   "scripts": {
     "check": "npm run format && npm run lint",
-    "dev": "nodemon index.js",
+    "dev": "node index.js",
     "format": "prettier --write \"**/*.+(js|yml|yaml|json)\"",
     "lint": "eslint .",
     "start": "node index.js",
@@ -39,6 +39,7 @@
     "mysql2": "^3.11.5",
     "sequelize": "^6.37.5",
     "uuid": "3.3.2",
-    "ws": "^8.18.0"
+    "ws": "^8.18.0",
+    "@whiskeysockets/baileys": "^6.7.9"
   }
 }