refactor: migrate exif support to helper

2026-07-15 12:13:26 +07:00
parent 9a73dc73c4
commit 1a279a89c0
2 changed files with 120 additions and 110 deletions
@@ -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
+}