From e4bae796c75366d6d231b19be59f34eeed178520 Mon Sep 17 00:00:00 2001
From: LiaoYijun <lyj@hainatravel.com>
Date: Fri, 20 Dec 2024 10:57:52 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20WA=20=E6=94=B6=E5=88=B0=E5=9B=BE?=
 =?UTF-8?q?=E7=89=87=E6=B6=88=E6=81=AF=E4=B8=8B=E8=BD=BD=E5=9B=BE=E7=89=87?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 baileys-server/main.js | 28 +++++++++++++++++++++++++++-
 1 file changed, 27 insertions(+), 1 deletion(-)

diff --git a/baileys-server/main.js b/baileys-server/main.js
index f998a88..be78754 100644
--- a/baileys-server/main.js
+++ b/baileys-server/main.js
@@ -6,8 +6,11 @@ import {
   makeCacheableSignalKeyStore,
   makeInMemoryStore,
   useMultiFileAuthState,
+  downloadMediaMessage
 } from '@whiskeysockets/baileys'
 
+import { writeFile } from 'fs/promises'
+
 import NodeCache from 'node-cache'
 import P from 'pino'
 
@@ -74,11 +77,34 @@ const startSock = async () => {
     }
   })
 
-  sock.ev.on('messages.upsert', (upsert) => {
+  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