refactor: migrate exif support to helper

cmds/media/sticker.go

@@ -3,127 +3,19 @@ package media
 import (
 	"bytes"
 	"context"
-	"encoding/json"
 	"fmt"
-	"log"
 	"os"
 	"os/exec"
 	"path/filepath"
-	"strings"
 
 	"github.com/google/uuid"
 	hc "neo/context"
 	"neo/core"
+	"neo/helper"
 
 	waProto "go.mau.fi/whatsmeow/binary/proto"
 )
 
-type exifData struct {
-	StickerPackId        string `json:"sticker-pack-id"`
-	StickerPackName      string `json:"sticker-pack-name"`
-	StickerPackPublisher string `json:"sticker-pack-publisher"`
-	IsAvatarSticker      int    `json:"is-avatar-sticker"`
-}
-
-type webpExif struct {
-	AppId string   `json:"app-id"`
-	Ext   exifData `json:"ext"`
-}
-
-func getOrCreateDefaultExif() string {
-	defaultExifPath := "temp/default.exif"
-
-	// Create temp directory if it doesn't exist
-	if err := os.MkdirAll("temp", 0755); err != nil {
-		log.Printf("Failed to create temp directory: %v", err)
-		return ""
-	}
-
-	if _, err := os.Stat(defaultExifPath); os.IsNotExist(err) {
-		packName := os.Getenv("STICKER_NAME")
-		if packName == "" {
-			packName = "Bot"
-		}
-
-		publisher := os.Getenv("STICKER_PUBLISHER")
-		if publisher == "" {
-			publisher = "Bot"
-		}
-
-		exifObj := webpExif{
-			AppId: "com.whatsapp.app",
-			Ext: exifData{
-				StickerPackId:        "com.whatsapp.app",
-				StickerPackName:      packName,
-				StickerPackPublisher: publisher,
-				IsAvatarSticker:      0,
-			},
-		}
-
-		exifJson, err := json.Marshal(exifObj)
-		if err != nil {
-			log.Printf("Failed to marshal default EXIF: %v", err)
-			return ""
-		}
-
-		// WebP EXIF JSON string representation inside the file
-		exifContent := append([]byte{0x49, 0x49, 0x2A, 0x00, 0x08, 0x00, 0x00, 0x00, 0x01, 0x00, 0x41, 0x57, 0x07, 0x00, 0x00, 0x00, 0x00, 0x00, 0x16, 0x00, 0x00, 0x00}, exifJson...)
-
-		if err := os.WriteFile(defaultExifPath, exifContent, 0644); err != nil {
-			log.Printf("Failed to write default EXIF: %v", err)
-			return ""
-		}
-	}
-	return defaultExifPath
-}
-
-func resolveExif(args []string) (string, bool) {
-	if len(args) == 0 {
-		return getOrCreateDefaultExif(), false
-	}
-
-	argStr := strings.Join(args, " ")
-	parts := strings.Split(argStr, "|")
-
-	packName := strings.TrimSpace(parts[0])
-	publisher := "Bot"
-	if len(parts) > 1 {
-		publisher = strings.TrimSpace(parts[1])
-	}
-
-	if packName == "" && publisher == "Bot" {
-		return getOrCreateDefaultExif(), false
-	}
-
-	exifObj := webpExif{
-		AppId: "com.whatsapp.app",
-		Ext: exifData{
-			StickerPackId:        "com.whatsapp.app",
-			StickerPackName:      packName,
-			StickerPackPublisher: publisher,
-			IsAvatarSticker:      0,
-		},
-	}
-
-	exifJson, err := json.Marshal(exifObj)
-	if err != nil {
-		log.Printf("Failed to marshal dynamic EXIF: %v", err)
-		return getOrCreateDefaultExif(), false
-	}
-
-	exifContent := append([]byte{0x49, 0x49, 0x2A, 0x00, 0x08, 0x00, 0x00, 0x00, 0x01, 0x00, 0x41, 0x57, 0x07, 0x00, 0x00, 0x00, 0x00, 0x00, 0x16, 0x00, 0x00, 0x00}, exifJson...)
-
-	id := uuid.New().String()
-	tempPath := filepath.Join("temp", fmt.Sprintf("%s.exif", id))
-
-	if err := os.WriteFile(tempPath, exifContent, 0644); err != nil {
-		log.Printf("Failed to write dynamic EXIF: %v", err)
-		return getOrCreateDefaultExif(), false
-	}
-
-	return tempPath, true
-}
-
 func cmdStickerImage(ctx *hc.Ctx, img *waProto.ImageMessage, exifPath string, webpPath string) error {
 	data, err := ctx.Client().Download(context.Background(), img)
 	if err != nil {
@@ -222,7 +114,7 @@ var Sticker = &core.Command{
 	Run: func(ctx *hc.Ctx) {
 		ctx.React("👌")
 
-		exifPath, cleanupExif := resolveExif(ctx.Arguments())
+		exifPath, cleanupExif := helper.ResolveExif(ctx.Arguments())
 		defer func() {
 			if cleanupExif {
 				os.Remove(exifPath)

helper/exif.go

@@ -0,0 +1,118 @@
+package helper
+
+import (
+	"encoding/json"
+	"fmt"
+	"log"
+	"os"
+	"path/filepath"
+	"strings"
+
+	"github.com/google/uuid"
+)
+
+type exifData struct {
+	StickerPackId        string `json:"sticker-pack-id"`
+	StickerPackName      string `json:"sticker-pack-name"`
+	StickerPackPublisher string `json:"sticker-pack-publisher"`
+	IsAvatarSticker      int    `json:"is-avatar-sticker"`
+}
+
+type webpExif struct {
+	AppId string   `json:"app-id"`
+	Ext   exifData `json:"ext"`
+}
+
+func getOrCreateDefaultExif() string {
+	defaultExifPath := "temp/default.exif"
+
+	// Create temp directory if it doesn't exist
+	if err := os.MkdirAll("temp", 0755); err != nil {
+		log.Printf("Failed to create temp directory: %v", err)
+		return ""
+	}
+
+	if _, err := os.Stat(defaultExifPath); os.IsNotExist(err) {
+		packName := os.Getenv("STICKER_NAME")
+		if packName == "" {
+			packName = "Bot"
+		}
+
+		publisher := os.Getenv("STICKER_PUBLISHER")
+		if publisher == "" {
+			publisher = "Bot"
+		}
+
+		exifObj := webpExif{
+			AppId: "com.whatsapp.app",
+			Ext: exifData{
+				StickerPackId:        "com.whatsapp.app",
+				StickerPackName:      packName,
+				StickerPackPublisher: publisher,
+				IsAvatarSticker:      0,
+			},
+		}
+
+		exifJson, err := json.Marshal(exifObj)
+		if err != nil {
+			log.Printf("Failed to marshal default EXIF: %v", err)
+			return ""
+		}
+
+		// WebP EXIF JSON string representation inside the file
+		exifContent := append([]byte{0x49, 0x49, 0x2A, 0x00, 0x08, 0x00, 0x00, 0x00, 0x01, 0x00, 0x41, 0x57, 0x07, 0x00, 0x00, 0x00, 0x00, 0x00, 0x16, 0x00, 0x00, 0x00}, exifJson...)
+
+		if err := os.WriteFile(defaultExifPath, exifContent, 0644); err != nil {
+			log.Printf("Failed to write default EXIF: %v", err)
+			return ""
+		}
+	}
+	return defaultExifPath
+}
+
+func ResolveExif(args []string) (string, bool) {
+	if len(args) == 0 {
+		return getOrCreateDefaultExif(), false
+	}
+
+	argStr := strings.Join(args, " ")
+	parts := strings.Split(argStr, "|")
+
+	packName := strings.TrimSpace(parts[0])
+	publisher := "Bot"
+	if len(parts) > 1 {
+		publisher = strings.TrimSpace(parts[1])
+	}
+
+	if packName == "" && publisher == "Bot" {
+		return getOrCreateDefaultExif(), false
+	}
+
+	exifObj := webpExif{
+		AppId: "com.whatsapp.app",
+		Ext: exifData{
+			StickerPackId:        "com.whatsapp.app",
+			StickerPackName:      packName,
+			StickerPackPublisher: publisher,
+			IsAvatarSticker:      0,
+		},
+	}
+
+	exifJson, err := json.Marshal(exifObj)
+	if err != nil {
+		log.Printf("Failed to marshal dynamic EXIF: %v", err)
+		return getOrCreateDefaultExif(), false
+	}
+
+	exifContent := append([]byte{0x49, 0x49, 0x2A, 0x00, 0x08, 0x00, 0x00, 0x00, 0x01, 0x00, 0x41, 0x57, 0x07, 0x00, 0x00, 0x00, 0x00, 0x00, 0x16, 0x00, 0x00, 0x00}, exifJson...)
+
+	id := uuid.New().String()
+	tempPath := filepath.Join("temp", fmt.Sprintf("%s.exif", id))
+
+	if err := os.WriteFile(tempPath, exifContent, 0644); err != nil {
+		log.Printf("Failed to write dynamic EXIF: %v", err)
+		return getOrCreateDefaultExif(), false
+	}
+
+	return tempPath, true
+}