feat: 本地测试 WA

wai-server/core/baileys/local.js

@@ -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();