From b3b0a6f7f3b9654ce44bb530026de75fca7372e7 Mon Sep 17 00:00:00 2001
From: Aldino Kemal <aldinokemal2104@gmail.com>
Date: Tue, 3 Jan 2023 08:15:28 +0700
Subject: [PATCH] feat: add image url in webhook

---
 src/config/settings.go       |  1 +
 src/pkg/whatsapp/whatsapp.go | 41 ++++++++++++++++++++++++++----------
 src/statics/media/.gitignore |  2 ++
 3 files changed, 33 insertions(+), 11 deletions(-)
 create mode 100644 src/statics/media/.gitignore

diff --git a/src/config/settings.go b/src/config/settings.go
index 19d9b41..c89e00c 100644
--- a/src/config/settings.go
+++ b/src/config/settings.go
@@ -15,6 +15,7 @@ var (
 
 	PathQrCode    = "statics/qrcode"
 	PathSendItems = "statics/senditems"
+	PathMedia     = "statics/media"
 	PathStorages  = "storages"
 
 	DBName = "whatsapp.db"
diff --git a/src/pkg/whatsapp/whatsapp.go b/src/pkg/whatsapp/whatsapp.go
index ad1f0a8..5f61dbf 100644
--- a/src/pkg/whatsapp/whatsapp.go
+++ b/src/pkg/whatsapp/whatsapp.go
@@ -198,19 +198,12 @@ func handler(rawEvt interface{}) {
 
 		img := evt.Message.GetImageMessage()
 		if img != nil {
-			data, err := cli.Download(img)
+			path, err := DownloadImage(config.PathStorages, img)
 			if err != nil {
 				log.Errorf("Failed to download image: %v", err)
-				return
-			}
-			exts, _ := mime.ExtensionsByType(img.GetMimetype())
-			path := fmt.Sprintf("%s/%s%s", config.PathStorages, evt.Info.ID, exts[0])
-			err = os.WriteFile(path, data, 0600)
-			if err != nil {
-				log.Errorf("Failed to save image: %v", err)
-				return
+			} else {
+				log.Infof("Image downloaded to %s", path)
 			}
-			log.Infof("Saved image in message to %s", path)
 		}
 
 		if config.WhatsappAutoReplyMessage != "" && !isGroupJid(evt.Info.Chat.String()) {
@@ -262,10 +255,11 @@ func handler(rawEvt interface{}) {
 
 func sendAutoReplyWebhook(evt *events.Message) error {
 	client := &http.Client{Timeout: 10 * time.Second}
+	imageMedia := evt.Message.GetImageMessage()
 	body := map[string]any{
 		"from":          evt.Info.SourceString(),
 		"message":       evt.Message.GetConversation(),
-		"image":         evt.Message.GetImageMessage(),
+		"image":         imageMedia,
 		"video":         evt.Message.GetVideoMessage(),
 		"audio":         evt.Message.GetAudioMessage(),
 		"document":      evt.Message.GetDocumentMessage(),
@@ -278,6 +272,16 @@ func sendAutoReplyWebhook(evt *events.Message) error {
 		"contact":       evt.Message.GetContactMessage(),
 		"forwarded":     evt.Message.GetGroupInviteMessage(),
 	}
+
+	if imageMedia != nil {
+		path, err := DownloadImage(config.PathMedia, imageMedia)
+		if err != nil {
+			return pkgError.WebhookError(fmt.Sprintf("Failed to download image: %v", err))
+		}
+
+		body["image"] = path
+	}
+
 	postBody, _ := json.Marshal(body)
 	req, err := http.NewRequestWithContext(context.Background(), http.MethodPost, config.WhatsappAutoReplyWebhook, bytes.NewBuffer(postBody))
 	if err != nil {
@@ -297,3 +301,18 @@ func sendAutoReplyWebhook(evt *events.Message) error {
 func isGroupJid(jid string) bool {
 	return strings.Contains(jid, "@g.us")
 }
+
+func DownloadImage(storageLocation string, image *waProto.ImageMessage) (path string, err error) {
+	data, err := cli.Download(image)
+	if err != nil {
+		return path, err
+	}
+
+	extensions, _ := mime.ExtensionsByType(image.GetMimetype())
+	path = fmt.Sprintf("%s/%d%s", storageLocation, time.Now().Unix(), extensions[0])
+	err = os.WriteFile(path, data, 0600)
+	if err != nil {
+		return path, err
+	}
+	return path, nil
+}
diff --git a/src/statics/media/.gitignore b/src/statics/media/.gitignore
new file mode 100644
index 0000000..c96a04f
--- /dev/null
+++ b/src/statics/media/.gitignore
@@ -0,0 +1,2 @@
+*
+!.gitignore
\ No newline at end of file