Merge remote-tracking branch 'origin/main'

wai-server/core/baileys/index.js

@@ -14,6 +14,7 @@ const { writeFile } = require('fs/promises');
 const generateId = require('../../utils/generateId.util');
 const NodeCache = require('node-cache');
 const P = require('pino');
+const { format } = require('path');
 
 const logger = P({ timestamp: () => `,"time":"${new Date().toJSON()}"` }, P.destination('./wa-logs.txt'))
 logger.level = 'trace'
@@ -34,25 +35,44 @@ setInterval(() => {
 // start a connection
 const createWhatsApp = async (phone) => {
   const channelId = generateId()
+  const whatsAppNo = phone
   const { state, saveCreds } = await useMultiFileAuthState('baileys_auth_info_' + phone + '_' + channelId)
   // fetch latest version of WA Web
   const { version, isLatest } = await fetchLatestBaileysVersion()
   const waVersion = version.join('.') + ', ' + (isLatest ? 'latest' : 'out')
-  const myEmitter = new EventEmitter();
-  myEmitter.on('connection.open', (p) => {
-    console.info('EventEmitter: connection.open')
-  });
+  const waEmitter = new EventEmitter();
+
+  const formatPhoneNumber = (number) => {
+    if (number.indexOf('@g.us') > -1) {
+      return number
+    } else if (number.indexOf('@broadcast') > -1) {
+      return number
+    } else {
+      return number + '@s.whatsapp.net'
+    }
+  }
+
+  const parsePhoneNumber = (number) => {
+    if (number.indexOf('@s.whatsapp.net') > -1) {
+      return number.split('@')[0]
+    } else {
+      return number
+    }
+  }
 
   const on = (eventName, listener) => {
-    myEmitter.on(eventName, listener);
+    waEmitter.on(eventName, listener);
   }
 
   const sendTextMessage = (whatsAppNo, content) => {
-    waSocket.sendMessage(whatsAppNo + '@s.whatsapp.net', { text: content })
+    const number = formatPhoneNumber(whatsAppNo)
+    console.info('formatPhoneNumber: ', number)
+    waSocket.sendMessage(number, { text: content })
   }
 
   const sendImageMessage = (whatsAppNo, imageUrl) => {
-    waSocket.sendMessage(whatsAppNo + '@s.whatsapp.net', {
+    const number = formatPhoneNumber(whatsAppNo)
+    waSocket.sendMessage(number, {
       image: { url: imageUrl},
     })
   }
@@ -91,8 +111,7 @@ const createWhatsApp = async (phone) => {
           }
         } else if(connection === 'open') {
           console.info('链接成功')
-          myEmitter.emit('connection.open', waSocket);
-          await waSocket.sendMessage('8617607730395' + '@s.whatsapp.net', { text: 'OPEN: ' + new Date().toString() })
+          waEmitter.emit('connection.open', waSocket);
         } else if(qr !== undefined) {
           // WebSocket 创建成功等待扫码，如果没有扫码会更新 qr
           resolve(qr)
@@ -127,7 +146,7 @@ const createWhatsApp = async (phone) => {
               console.log('writeFile', messageType)
             }
 
-            const fromWhatsAppNo = msg.key.remoteJid.split('@')[0]
+            const fromWhatsAppNo = parsePhoneNumber(msg.key.remoteJid)
 
             if (msg.message?.conversation || msg.message?.extendedTextMessage?.text) {
               const text = msg.message?.conversation || msg.message?.extendedTextMessage?.text
@@ -136,6 +155,73 @@ const createWhatsApp = async (phone) => {
                 sendImageMessage(fromWhatsAppNo, 'https://images.asiahighlights.com/allpicture/2022/07/8a7d9ced5936463bb904c82a_cut_750x850_349.webp')
               } else if (text.indexOf('文本') > -1){
                 sendTextMessage(fromWhatsAppNo, '文本消息' + new Date().toString())
+              } else if (text.indexOf('发群') > -1){
+                sendTextMessage('120363335516526642@g.us', '这是群消息：' + new Date().toString())
+              }
+
+              // 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(msg.messageTimestamp * 1000)
+
+                return datetime.getFullYear() + '-' + (datetime.getMonth() + 1) + '-' + datetime.getDay() + ' ' + datetime.getHours() + ':' + datetime.getMinutes() + ':'  + datetime.getSeconds()
+              }
+
+              if (msg.key.fromMe) {
+                waEmitter.emit('message.updated', {
+                  "id":  msg.key.id,
+                  "wamid": msg.key.id,
+                  "status": formatStatus(msg.status),
+                  "from":  whatsAppNo,
+                  "to": fromWhatsAppNo,
+                  "type": "text",
+                  "text": {
+                      "body": text
+                  },
+                  conversation: {
+                    id: msg.key.participant,
+                    type: '不区分',
+                    name: msg.pushName
+                  },
+                  "createTime": formatTimestamp(msg.messageTimestamp),
+                  "updateTime": formatTimestamp(new Date().getTime()),
+                  "sendTime": formatTimestamp(msg.messageTimestamp),
+                  "deliverTime": formatTimestamp(msg.messageTimestamp),
+                  "readTime": formatTimestamp(msg.messageTimestamp),
+                });
+              } else {
+                waEmitter.emit('message.received', {
+                  "id":  msg.key.id,
+                  "wamid": msg.key.id,
+                  "status": "",
+                  "from":  fromWhatsAppNo,
+                  "to": whatsAppNo,
+                  "type": "text",
+                  "text": {
+                      "body": text
+                  },
+                  conversation: {
+                    id: msg.key.participant,
+                    type: '不区分',
+                    name: msg.pushName
+                  },
+                  "createTime": formatTimestamp(msg.messageTimestamp),
+                  "updateTime": formatTimestamp(new Date().getTime()),
+                  "sendTime": formatTimestamp(msg.messageTimestamp),
+                  "deliverTime": formatTimestamp(msg.messageTimestamp),
+                  "readTime": formatTimestamp(msg.messageTimestamp),
+                });
               }
             }
           }