Compare commits
10
Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
5f16856959 | ||
|
|
10209a5258 | ||
|
|
cde38e4c2d | ||
|
|
199846857f | ||
|
|
2bb2463aa5 | ||
|
|
5439186bc6 | ||
|
|
8c255a7066 | ||
|
|
d93a40e531 | ||
|
|
68f964f810 | ||
|
|
6228cb2733 |
@@ -0,0 +1,6 @@
|
|||||||
|
*.db
|
||||||
|
.env
|
||||||
|
doc.json
|
||||||
|
neo
|
||||||
|
testing/
|
||||||
|
temp/
|
||||||
+1
-1
@@ -17,7 +17,7 @@ RUN CGO_ENABLED=1 GOOS=linux go build -trimpath -ldflags="-w -s" -o /app/neo .
|
|||||||
# Stage: prod
|
# Stage: prod
|
||||||
FROM alpine:latest AS prod
|
FROM alpine:latest AS prod
|
||||||
|
|
||||||
# Install required tools (webpmux via libwebp-tools, ffmpeg)
|
# Install required tools (cwebp/gif2webp via libwebp-tools, ffmpeg, imagemagick)
|
||||||
RUN apk add --no-cache \
|
RUN apk add --no-cache \
|
||||||
ffmpeg \
|
ffmpeg \
|
||||||
libwebp-tools \
|
libwebp-tools \
|
||||||
|
|||||||
@@ -0,0 +1,79 @@
|
|||||||
|
package general
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
hc "neo/context"
|
||||||
|
"neo/core"
|
||||||
|
)
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
core.Default.Register(Help)
|
||||||
|
}
|
||||||
|
|
||||||
|
var Help = &core.Command{
|
||||||
|
Name: "help",
|
||||||
|
Aliases: []string{"?", "h", "menu"},
|
||||||
|
Category: "general",
|
||||||
|
Description: "Shows the bot's features and commands",
|
||||||
|
Run: func(ctx *hc.Ctx) {
|
||||||
|
loc, err := time.LoadLocation("Asia/Jakarta")
|
||||||
|
if err != nil {
|
||||||
|
loc = time.FixedZone("UTC+7", 7*3600)
|
||||||
|
}
|
||||||
|
|
||||||
|
now := time.Now().In(loc)
|
||||||
|
hour := now.Hour()
|
||||||
|
|
||||||
|
var greeting string
|
||||||
|
switch {
|
||||||
|
case hour >= 4 && hour < 12:
|
||||||
|
greeting = "Ohayō"
|
||||||
|
case hour >= 12 && hour < 18:
|
||||||
|
greeting = "Konnichiwa"
|
||||||
|
default:
|
||||||
|
greeting = "Konbanwa"
|
||||||
|
}
|
||||||
|
|
||||||
|
pushName := ctx.Event().Info.PushName
|
||||||
|
if pushName == "" {
|
||||||
|
pushName = "User"
|
||||||
|
}
|
||||||
|
|
||||||
|
// Header
|
||||||
|
var sb strings.Builder
|
||||||
|
sb.WriteString(fmt.Sprintf("%s *%s*👋\n📬 Need help? Here are all of my commands\n", greeting, pushName))
|
||||||
|
|
||||||
|
// Group commands by category (simulate json object array appending)
|
||||||
|
cmdMap := make(map[string][]string)
|
||||||
|
for _, cmd := range core.Default.GetCommands() {
|
||||||
|
cat := cmd.Category
|
||||||
|
if cat == "" {
|
||||||
|
cat = "general"
|
||||||
|
}
|
||||||
|
cmdMap[cat] = append(cmdMap[cat], cmd.Name)
|
||||||
|
}
|
||||||
|
|
||||||
|
prefix := ctx.Prefix()
|
||||||
|
if prefix == "" {
|
||||||
|
// fallback if they called it somehow internally without prefix
|
||||||
|
prefix = "."
|
||||||
|
}
|
||||||
|
|
||||||
|
// Print categories
|
||||||
|
for cat, cmds := range cmdMap {
|
||||||
|
sb.WriteString(fmt.Sprintf("\n*📃%s*\n", strings.ToUpper(cat)))
|
||||||
|
for i, cmdName := range cmds {
|
||||||
|
branch := "├"
|
||||||
|
if i == len(cmds)-1 {
|
||||||
|
branch = "└"
|
||||||
|
}
|
||||||
|
sb.WriteString(fmt.Sprintf("%s %s%s\n", branch, prefix, cmdName))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
ctx.Reply(sb.String())
|
||||||
|
},
|
||||||
|
}
|
||||||
+23
-6
@@ -4,6 +4,7 @@ import (
|
|||||||
"context"
|
"context"
|
||||||
"fmt"
|
"fmt"
|
||||||
"os"
|
"os"
|
||||||
|
"strings"
|
||||||
|
|
||||||
hc "neo/context"
|
hc "neo/context"
|
||||||
"neo/core"
|
"neo/core"
|
||||||
@@ -20,12 +21,28 @@ var Sticker = &core.Command{
|
|||||||
Run: func(ctx *hc.Ctx) {
|
Run: func(ctx *hc.Ctx) {
|
||||||
// go ctx.React("👌")
|
// go ctx.React("👌")
|
||||||
|
|
||||||
exifPath, cleanupExif := helper.ResolveExif(ctx.Arguments())
|
var packName, publisher string
|
||||||
defer func() {
|
args := ctx.Arguments()
|
||||||
if cleanupExif {
|
|
||||||
os.Remove(exifPath)
|
packName = os.Getenv("STICKER_NAME")
|
||||||
|
if packName == "" {
|
||||||
|
packName = "Bot"
|
||||||
|
}
|
||||||
|
publisher = os.Getenv("STICKER_PUBLISHER")
|
||||||
|
if publisher == "" {
|
||||||
|
publisher = "Bot"
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(args) > 0 {
|
||||||
|
argsStr := strings.Join(args, " ")
|
||||||
|
parts := strings.Split(argsStr, "|")
|
||||||
|
packName = strings.TrimSpace(parts[0])
|
||||||
|
if len(parts) > 1 {
|
||||||
|
publisher = strings.TrimSpace(parts[1])
|
||||||
|
} else {
|
||||||
|
publisher = ""
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}()
|
|
||||||
|
|
||||||
msg := ctx.Message()
|
msg := ctx.Message()
|
||||||
quoted := hc.GetQuotedMessage(msg)
|
quoted := hc.GetQuotedMessage(msg)
|
||||||
@@ -66,7 +83,7 @@ var Sticker = &core.Command{
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
finalSticker, err := helper.InjectExif(rawWebp, exifPath)
|
finalSticker, err := helper.InjectExif(rawWebp, packName, publisher)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
ctx.Reply(fmt.Sprintf("error injecting EXIF: %v", err))
|
ctx.Reply(fmt.Sprintf("error injecting EXIF: %v", err))
|
||||||
return
|
return
|
||||||
|
|||||||
+1
-1
@@ -15,7 +15,7 @@ var Ping = &core.Command{
|
|||||||
Aliases: []string{"test", "tes", "p"},
|
Aliases: []string{"test", "tes", "p"},
|
||||||
Category: "general",
|
Category: "general",
|
||||||
Description: "Check bot response time",
|
Description: "Check bot response time",
|
||||||
Permissions: []core.Permission{core.PermissionGroupOnly},
|
// Permissions: []core.Permission{core.PermissionGroupOnly},
|
||||||
Run: func(ctx *hc.Ctx) {
|
Run: func(ctx *hc.Ctx) {
|
||||||
t := ctx.Event().Info.Timestamp
|
t := ctx.Event().Info.Timestamp
|
||||||
speed := time.Since(t).Milliseconds()
|
speed := time.Since(t).Milliseconds()
|
||||||
|
|||||||
@@ -2,6 +2,7 @@ package cmds
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
_ "neo/cmds/download"
|
_ "neo/cmds/download"
|
||||||
|
_ "neo/cmds/general"
|
||||||
_ "neo/cmds/media"
|
_ "neo/cmds/media"
|
||||||
_ "neo/cmds/misc"
|
_ "neo/cmds/misc"
|
||||||
)
|
)
|
||||||
|
|||||||
@@ -1,412 +0,0 @@
|
|||||||
# Pure-Go WebP EXIF RIFF Injector Implementation Plan
|
|
||||||
|
|
||||||
> **For agentic workers:** REQUIRED SUB-SKILL: Use superpowers:subagent-driven-development (recommended) or superpowers:executing-plans to implement this plan task-by-task. Steps use checkbox (`- [ ]`) syntax for tracking.
|
|
||||||
|
|
||||||
**Goal:** Modify `helper/exif.go` and `helper/media_convert.go` to inject EXIF payloads directly into WebP RIFF byte slices entirely in Go memory, replacing `webpmux`.
|
|
||||||
|
|
||||||
**Architecture:** We will implement an `InjectExif(webpBytes []byte, packName, publisher string) ([]byte, error)` function inside `helper/exif.go` to handle raw RIFF parsing in place, and strip out the old Temp File logic. Webpmux invocation in `helper/media_convert.go` will be removed.
|
|
||||||
|
|
||||||
**Tech Stack:** Go (encoding/binary)
|
|
||||||
|
|
||||||
## Global Constraints
|
|
||||||
- Target files: `helper/exif.go` and `helper/media_convert.go`
|
|
||||||
- Do not use CGO or any unstated 3rd party package for RIFF handling. Implement manually using `encoding/binary`.
|
|
||||||
- Format must be `VP8X` with the EXIF flag bit activated.
|
|
||||||
|
|
||||||
---
|
|
||||||
|
|
||||||
### Task 1: Clean Up media_convert.go
|
|
||||||
|
|
||||||
**Files:**
|
|
||||||
- Modify:`/Users/fd/Documents/perf/helper/media_convert.go`
|
|
||||||
|
|
||||||
**Interfaces:**
|
|
||||||
- Consumes: Nothing
|
|
||||||
- Produces: `ConvertImageToWebp`, `ConvertVideoToWebp`
|
|
||||||
|
|
||||||
- [ ] **Step 1: Simplify Convert methods**
|
|
||||||
Remove `InjectExif` from `media_convert.go` since it will be rewritten from scratch in `exif.go`.
|
|
||||||
|
|
||||||
```go
|
|
||||||
package helper
|
|
||||||
|
|
||||||
import (
|
|
||||||
"bytes"
|
|
||||||
"fmt"
|
|
||||||
"os"
|
|
||||||
"os/exec"
|
|
||||||
"path/filepath"
|
|
||||||
|
|
||||||
"github.com/google/uuid"
|
|
||||||
)
|
|
||||||
|
|
||||||
func ConvertImageToWebp(imgData []byte) ([]byte, error) {
|
|
||||||
convertCmd := exec.Command("convert", "-", "-resize", "512x512", "-background", "none", "-compose", "Copy", "-gravity", "center", "-extent", "512x512", "-quality", "100", "png:-")
|
|
||||||
convertCmd.Stdin = bytes.NewReader(imgData)
|
|
||||||
var convertBuf bytes.Buffer
|
|
||||||
convertCmd.Stdout = &convertBuf
|
|
||||||
if err := convertCmd.Run(); err != nil {
|
|
||||||
return nil, fmt.Errorf("convert failed: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
cwebpCmd := exec.Command("cwebp", "-quiet", "-mt", "-exact", "-q", "100", "-m", "6", "-alpha_q", "100", "-o", "-", "--", "-")
|
|
||||||
cwebpCmd.Stdin = &convertBuf
|
|
||||||
var cwebpBuf bytes.Buffer
|
|
||||||
cwebpCmd.Stdout = &cwebpBuf
|
|
||||||
if err := cwebpCmd.Run(); err != nil {
|
|
||||||
return nil, fmt.Errorf("cwebp failed: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
return cwebpBuf.Bytes(), nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func ConvertVideoToWebp(vidData []byte) ([]byte, error) {
|
|
||||||
webpPath := filepath.Join("temp", fmt.Sprintf("%s.webp", uuid.New().String()))
|
|
||||||
defer os.Remove(webpPath)
|
|
||||||
|
|
||||||
ffmpegCmd := exec.Command("ffmpeg",
|
|
||||||
"-y", "-hide_banner", "-loglevel", "error",
|
|
||||||
"-i", "pipe:0", "-f", "mp4",
|
|
||||||
"-ss", "00:00:00", "-t", "00:00:15",
|
|
||||||
"-vf", "fps=10,scale=720:-1:flags=lanczos:force_original_aspect_ratio=increase,crop=512:512,pad=512:512:(ow-iw)/2:(oh-ih)/2:color=#00000000,setsar=1",
|
|
||||||
"-compression_level", "6",
|
|
||||||
"-q:v", "60", "-loop", "0",
|
|
||||||
"-preset", "picture", "-an", "-fps_mode", "auto",
|
|
||||||
"-f", "webp", webpPath,
|
|
||||||
)
|
|
||||||
ffmpegCmd.Stdin = bytes.NewReader(vidData)
|
|
||||||
if err := ffmpegCmd.Run(); err != nil {
|
|
||||||
return nil, fmt.Errorf("ffmpeg failed: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
return os.ReadFile(webpPath)
|
|
||||||
}
|
|
||||||
```
|
|
||||||
|
|
||||||
- [ ] **Step 2: Check standard test suite**
|
|
||||||
Run: `go build ./...`
|
|
||||||
|
|
||||||
- [ ] **Step 3: Commit**
|
|
||||||
```bash
|
|
||||||
git add helper/media_convert.go
|
|
||||||
git commit -m "refactor: remove InjectExif shell wrapper from media_convert"
|
|
||||||
```
|
|
||||||
|
|
||||||
### Task 2: Implement Pure-Go RIFF Injector
|
|
||||||
|
|
||||||
**Files:**
|
|
||||||
- Modify:`/Users/fd/Documents/perf/helper/exif.go`
|
|
||||||
|
|
||||||
**Interfaces:**
|
|
||||||
- Produces: `InjectExif(webpData []byte, packName, publisher string) ([]byte, error)`
|
|
||||||
- Consumes: Nothing
|
|
||||||
|
|
||||||
- [ ] **Step 1: Write `helper/exif.go`**
|
|
||||||
Overwrite `/Users/fd/Documents/perf/helper/exif.go` with the completely new implementation. The previous dynamic `temp/default.exif` file logic is gone.
|
|
||||||
|
|
||||||
```go
|
|
||||||
package helper
|
|
||||||
|
|
||||||
import (
|
|
||||||
"bytes"
|
|
||||||
"encoding/binary"
|
|
||||||
"encoding/json"
|
|
||||||
"fmt"
|
|
||||||
"math/big"
|
|
||||||
)
|
|
||||||
|
|
||||||
type StickerMetadata struct {
|
|
||||||
PackId string `json:"sticker-pack-id"`
|
|
||||||
Name string `json:"sticker-pack-name"`
|
|
||||||
Publisher string `json:"sticker-pack-publisher"`
|
|
||||||
Emojis []string `json:"emojis,omitempty"`
|
|
||||||
}
|
|
||||||
|
|
||||||
func writeUIntLE(buffer []byte, value, offset, byteLength int64) {
|
|
||||||
slice := make([]byte, byteLength)
|
|
||||||
val := new(big.Int)
|
|
||||||
val.SetUint64(uint64(value))
|
|
||||||
valBytes := val.Bytes()
|
|
||||||
|
|
||||||
tmp := make([]byte, len(valBytes))
|
|
||||||
for i := range valBytes {
|
|
||||||
tmp[i] = valBytes[len(valBytes)-1-i]
|
|
||||||
}
|
|
||||||
copy(slice, tmp)
|
|
||||||
copy(buffer[offset:], slice)
|
|
||||||
}
|
|
||||||
|
|
||||||
func createMetadataBytes(packName, publisher string) []byte {
|
|
||||||
exifObj := StickerMetadata{
|
|
||||||
PackId: "com.wa.bot.sticker",
|
|
||||||
Name: packName,
|
|
||||||
Publisher: publisher,
|
|
||||||
Emojis: []string{"😀"},
|
|
||||||
}
|
|
||||||
|
|
||||||
exifJson, err := json.Marshal(exifObj)
|
|
||||||
if err != nil {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
bit := []byte{0x49, 0x49, 0x2A, 0x00, 0x08, 0x00, 0x00, 0x00, 0x01, 0x00, 0x41, 0x57, 0x07, 0x00, 0x00, 0x00, 0x00, 0x00, 0x16, 0x00, 0x00, 0x00}
|
|
||||||
bit = append(bit, exifJson...)
|
|
||||||
writeUIntLE(bit, int64(len(exifJson)), 14, 4)
|
|
||||||
|
|
||||||
return bit
|
|
||||||
}
|
|
||||||
|
|
||||||
// InjectExif takes raw WebP bytes and constructs a new WebP RIFF byte slice containing the EXIF metadata
|
|
||||||
func InjectExif(webpData []byte, packName, publisher string) ([]byte, error) {
|
|
||||||
if len(webpData) < 12 || string(webpData[0:4]) != "RIFF" || string(webpData[8:12]) != "WEBP" {
|
|
||||||
return nil, fmt.Errorf("not a valid WEBP file")
|
|
||||||
}
|
|
||||||
|
|
||||||
var width, height uint32
|
|
||||||
hasVP8X := false
|
|
||||||
var vp8xFlags byte
|
|
||||||
var vp8xBytes []byte
|
|
||||||
|
|
||||||
var chunks [][]byte
|
|
||||||
|
|
||||||
// Default defaults for VP8X canvas sizes
|
|
||||||
width = 512
|
|
||||||
height = 512
|
|
||||||
|
|
||||||
offset := 12
|
|
||||||
for offset < len(webpData) {
|
|
||||||
if offset+8 > len(webpData) {
|
|
||||||
break
|
|
||||||
}
|
|
||||||
chunkID := string(webpData[offset : offset+4])
|
|
||||||
chunkSize := binary.LittleEndian.Uint32(webpData[offset+4 : offset+8])
|
|
||||||
|
|
||||||
paddedSize := chunkSize
|
|
||||||
if paddedSize%2 != 0 {
|
|
||||||
paddedSize++
|
|
||||||
}
|
|
||||||
|
|
||||||
if offset+8+int(paddedSize) > len(webpData) {
|
|
||||||
break
|
|
||||||
}
|
|
||||||
|
|
||||||
chunkData := webpData[offset+8 : offset+8+int(paddedSize)]
|
|
||||||
|
|
||||||
switch chunkID {
|
|
||||||
case "VP8X":
|
|
||||||
hasVP8X = true
|
|
||||||
vp8xFlags = chunkData[0]
|
|
||||||
vp8xBytes = chunkData
|
|
||||||
// extract dimensions
|
|
||||||
wBytes := []byte{chunkData[4], chunkData[5], chunkData[6], 0}
|
|
||||||
hBytes := []byte{chunkData[7], chunkData[8], chunkData[9], 0}
|
|
||||||
width = binary.LittleEndian.Uint32(wBytes) + 1
|
|
||||||
height = binary.LittleEndian.Uint32(hBytes) + 1
|
|
||||||
case "VP8 ":
|
|
||||||
if !hasVP8X {
|
|
||||||
// extract 14 bit scale block
|
|
||||||
w := uint32(chunkData[6]) | (uint32(chunkData[7]) << 8)
|
|
||||||
width = w & 0x3FFF
|
|
||||||
h := uint32(chunkData[8]) | (uint32(chunkData[9]) << 8)
|
|
||||||
height = h & 0x3FFF
|
|
||||||
}
|
|
||||||
chunks = append(chunks, webpData[offset:offset+8+int(paddedSize)])
|
|
||||||
case "VP8L":
|
|
||||||
if !hasVP8X {
|
|
||||||
// Parse bits 1-28 for w/h
|
|
||||||
b0 := uint32(chunkData[1])
|
|
||||||
b1 := uint32(chunkData[2])
|
|
||||||
b2 := uint32(chunkData[3])
|
|
||||||
b3 := uint32(chunkData[4])
|
|
||||||
width = 1 + (((b1 & 0x3F) << 8) | b0)
|
|
||||||
height = 1 + (((b3 & 0xF) << 10) | (b2 << 2) | ((b1 & 0xC0) >> 6))
|
|
||||||
}
|
|
||||||
chunks = append(chunks, webpData[offset:offset+8+int(paddedSize)])
|
|
||||||
default:
|
|
||||||
if chunkID != "EXIF" {
|
|
||||||
chunks = append(chunks, webpData[offset:offset+8+int(paddedSize)])
|
|
||||||
}
|
|
||||||
}
|
|
||||||
offset += 8 + int(paddedSize)
|
|
||||||
}
|
|
||||||
|
|
||||||
exifMeta := createMetadataBytes(packName, publisher)
|
|
||||||
if exifMeta == nil {
|
|
||||||
return nil, fmt.Errorf("failed to create exif byte payload")
|
|
||||||
}
|
|
||||||
|
|
||||||
// Turn on EXIF flag (bit 3) in VP8X
|
|
||||||
if !hasVP8X {
|
|
||||||
vp8xBytes = make([]byte, 10)
|
|
||||||
vp8xFlags = 0x08 // EXIF flag is bit 3
|
|
||||||
vp8xBytes[0] = vp8xFlags
|
|
||||||
w := width - 1
|
|
||||||
h := height - 1
|
|
||||||
vp8xBytes[4] = byte(w)
|
|
||||||
vp8xBytes[5] = byte(w >> 8)
|
|
||||||
vp8xBytes[6] = byte(w >> 16)
|
|
||||||
vp8xBytes[7] = byte(h)
|
|
||||||
vp8xBytes[8] = byte(h >> 8)
|
|
||||||
vp8xBytes[9] = byte(h >> 16)
|
|
||||||
} else {
|
|
||||||
vp8xBytes[0] = vp8xFlags | 0x08
|
|
||||||
}
|
|
||||||
|
|
||||||
var buf bytes.Buffer
|
|
||||||
buf.WriteString("RIFF")
|
|
||||||
buf.Write([]byte{0, 0, 0, 0}) // Total size placeholder
|
|
||||||
buf.WriteString("WEBP")
|
|
||||||
|
|
||||||
// Write VP8X
|
|
||||||
buf.WriteString("VP8X")
|
|
||||||
vp8xSize := make([]byte, 4)
|
|
||||||
binary.LittleEndian.PutUint32(vp8xSize, 10)
|
|
||||||
buf.Write(vp8xSize)
|
|
||||||
buf.Write(vp8xBytes)
|
|
||||||
|
|
||||||
// Write existing chunks
|
|
||||||
for _, chunk := range chunks {
|
|
||||||
buf.Write(chunk)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Write EXIF
|
|
||||||
buf.WriteString("EXIF")
|
|
||||||
exifSize := uint32(len(exifMeta))
|
|
||||||
exifSizeSlice := make([]byte, 4)
|
|
||||||
binary.LittleEndian.PutUint32(exifSizeSlice, exifSize)
|
|
||||||
buf.Write(exifSizeSlice)
|
|
||||||
buf.Write(exifMeta)
|
|
||||||
if exifSize%2 != 0 {
|
|
||||||
buf.WriteByte(0)
|
|
||||||
}
|
|
||||||
|
|
||||||
result := buf.Bytes()
|
|
||||||
totalSize := uint32(len(result) - 8)
|
|
||||||
binary.LittleEndian.PutUint32(result[4:8], totalSize)
|
|
||||||
|
|
||||||
return result, nil
|
|
||||||
}
|
|
||||||
```
|
|
||||||
|
|
||||||
- [ ] **Step 2: Check test suite**
|
|
||||||
Run: `go build ./...`
|
|
||||||
|
|
||||||
- [ ] **Step 3: Commit**
|
|
||||||
```bash
|
|
||||||
git add helper/exif.go
|
|
||||||
git commit -m "feat: implement pure-go RIFF EXIF injector"
|
|
||||||
```
|
|
||||||
|
|
||||||
### Task 3: Update cmds/media/sticker.go Signature
|
|
||||||
|
|
||||||
**Files:**
|
|
||||||
- Modify:`/Users/fd/Documents/perf/cmds/media/sticker.go`
|
|
||||||
|
|
||||||
**Interfaces:**
|
|
||||||
- Consumes: `helper.InjectExif(rawWebp, packName, publisher)`
|
|
||||||
|
|
||||||
- [ ] **Step 1: Replace ResolveExif usages in `cmds/media/sticker.go`**
|
|
||||||
Read `ctx.Arguments()` directly to compute `packName` and `publisher` locally instead of calling `helper.ResolveExif`. Use the same ENV logic as fallback defaults if args are empty. Remove `helper.ResolveExif` code context cleanup.
|
|
||||||
|
|
||||||
```go
|
|
||||||
package media
|
|
||||||
|
|
||||||
import (
|
|
||||||
"context"
|
|
||||||
"fmt"
|
|
||||||
"os"
|
|
||||||
"strings"
|
|
||||||
|
|
||||||
hc "neo/context"
|
|
||||||
"neo/core"
|
|
||||||
"neo/helper"
|
|
||||||
)
|
|
||||||
|
|
||||||
func init() { core.Default.Register(Sticker) }
|
|
||||||
|
|
||||||
var Sticker = &core.Command{
|
|
||||||
Name: "sticker",
|
|
||||||
Aliases: []string{"s", "stiker"},
|
|
||||||
Category: "media",
|
|
||||||
Description: "Create sticker from image/video or rewrite existing sticker EXIF",
|
|
||||||
Run: func(ctx *hc.Ctx) {
|
|
||||||
ctx.React("👌")
|
|
||||||
|
|
||||||
args := ctx.Arguments()
|
|
||||||
packName := os.Getenv("STICKER_NAME")
|
|
||||||
if packName == "" {
|
|
||||||
packName = "Bot"
|
|
||||||
}
|
|
||||||
publisher := os.Getenv("STICKER_PUBLISHER")
|
|
||||||
if publisher == "" {
|
|
||||||
publisher = "Bot"
|
|
||||||
}
|
|
||||||
|
|
||||||
if len(args) > 0 {
|
|
||||||
argStr := strings.Join(args, " ")
|
|
||||||
parts := strings.Split(argStr, "|")
|
|
||||||
packName = strings.TrimSpace(parts[0])
|
|
||||||
publisher = "Bot"
|
|
||||||
if len(parts) > 1 {
|
|
||||||
publisher = strings.TrimSpace(parts[1])
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
msg := ctx.Message()
|
|
||||||
quoted := hc.GetQuotedMessage(msg)
|
|
||||||
|
|
||||||
var err error
|
|
||||||
var rawWebp []byte
|
|
||||||
var mediaData []byte
|
|
||||||
|
|
||||||
if img := msg.GetImageMessage(); img != nil {
|
|
||||||
mediaData, err = ctx.Client().Download(context.Background(), img)
|
|
||||||
if err == nil {
|
|
||||||
rawWebp, err = helper.ConvertImageToWebp(mediaData)
|
|
||||||
}
|
|
||||||
} else if img := quoted.GetImageMessage(); img != nil && quoted != nil {
|
|
||||||
mediaData, err = ctx.Client().Download(context.Background(), img)
|
|
||||||
if err == nil {
|
|
||||||
rawWebp, err = helper.ConvertImageToWebp(mediaData)
|
|
||||||
}
|
|
||||||
} else if vid := msg.GetVideoMessage(); vid != nil {
|
|
||||||
mediaData, err = ctx.Client().Download(context.Background(), vid)
|
|
||||||
if err == nil {
|
|
||||||
rawWebp, err = helper.ConvertVideoToWebp(mediaData)
|
|
||||||
}
|
|
||||||
} else if vid := quoted.GetVideoMessage(); vid != nil && quoted != nil {
|
|
||||||
mediaData, err = ctx.Client().Download(context.Background(), vid)
|
|
||||||
if err == nil {
|
|
||||||
rawWebp, err = helper.ConvertVideoToWebp(mediaData)
|
|
||||||
}
|
|
||||||
} else if smsg := quoted.GetStickerMessage(); smsg != nil && quoted != nil {
|
|
||||||
rawWebp, err = ctx.Client().Download(context.Background(), smsg)
|
|
||||||
} else {
|
|
||||||
ctx.Reply("error: reply to image/video/sticker or send with caption")
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
if err != nil {
|
|
||||||
ctx.Reply(fmt.Sprintf("error processing media: %v", err))
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
finalSticker, err := helper.InjectExif(rawWebp, packName, publisher)
|
|
||||||
if err != nil {
|
|
||||||
ctx.Reply(fmt.Sprintf("error injecting EXIF: %v", err))
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
ctx.SendSticker(finalSticker)
|
|
||||||
},
|
|
||||||
}
|
|
||||||
```
|
|
||||||
|
|
||||||
- [ ] **Step 2: Verify Compilation**
|
|
||||||
Run: `go build ./...`
|
|
||||||
|
|
||||||
- [ ] **Step 3: Commit**
|
|
||||||
```bash
|
|
||||||
git add cmds/media/sticker.go
|
|
||||||
git commit -m "feat: use pure-go exif injector in sticker command"
|
|
||||||
```
|
|
||||||
@@ -1,17 +0,0 @@
|
|||||||
# Pure-Go WebP EXIF RIFF Injector
|
|
||||||
|
|
||||||
## 1. Goal
|
|
||||||
Completely remove file-based IO and the `webpmux` dependency by weaving the EXIF chunk into the WebP RIFF byte struct physically in memory.
|
|
||||||
|
|
||||||
## 2. Architecture
|
|
||||||
Instead of using `webpmux` passing paths around, `helper/exif.go` will implement `InjectExif(webpBytes []byte, packName, publisher string) ([]byte, error)`.
|
|
||||||
|
|
||||||
## 3. RIFF Parsing Logic
|
|
||||||
1. Validate `RIFF` and `WEBP` signatures.
|
|
||||||
2. Read all chunks to extract Canvas Width/Height (from `VP8X`, `VP8L`, or `VP8 `).
|
|
||||||
3. Identify if `VP8X` is missing, and if so, construct it. We must ensure the `EXIF` bit flag is set.
|
|
||||||
4. Construct the custom `EXIF` chunk (which holds the exact binary array and JSON metadata we validated via F-BOT_GO earlier).
|
|
||||||
5. Append `EXIF` to the end of the WebP chunks, update the main `RIFF` file size pointer, and return the aggregated `[]byte` slice.
|
|
||||||
|
|
||||||
## 4. Main Command Refactor
|
|
||||||
`cmds/media/sticker.go` will stop requesting files via `ResolveExif()`. It will simply pass the raw Webp bytes received from `ConvertImageToWebp` or `ConvertVideoToWebp` into `InjectExif` and instantly upload the resulting byte array.
|
|
||||||
+115
-60
@@ -1,15 +1,11 @@
|
|||||||
package helper
|
package helper
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"bytes"
|
||||||
|
"encoding/binary"
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
"fmt"
|
"fmt"
|
||||||
"log"
|
|
||||||
"math/big"
|
"math/big"
|
||||||
"os"
|
|
||||||
"path/filepath"
|
|
||||||
"strings"
|
|
||||||
|
|
||||||
"github.com/google/uuid"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
type StickerMetadata struct {
|
type StickerMetadata struct {
|
||||||
@@ -19,7 +15,6 @@ type StickerMetadata struct {
|
|||||||
Emojis []string `json:"emojis,omitempty"`
|
Emojis []string `json:"emojis,omitempty"`
|
||||||
}
|
}
|
||||||
|
|
||||||
// Added matching writeUIntLE from F-BOT_GO reference
|
|
||||||
func writeUIntLE(buffer []byte, value, offset, byteLength int64) {
|
func writeUIntLE(buffer []byte, value, offset, byteLength int64) {
|
||||||
slice := make([]byte, byteLength)
|
slice := make([]byte, byteLength)
|
||||||
val := new(big.Int)
|
val := new(big.Int)
|
||||||
@@ -36,7 +31,7 @@ func writeUIntLE(buffer []byte, value, offset, byteLength int64) {
|
|||||||
|
|
||||||
func createMetadataBytes(packName, publisher string) []byte {
|
func createMetadataBytes(packName, publisher string) []byte {
|
||||||
exifObj := StickerMetadata{
|
exifObj := StickerMetadata{
|
||||||
PackId: "com.wa.bot.sticker", // or generic ID
|
PackId: "com.wa.bot.sticker",
|
||||||
Name: packName,
|
Name: packName,
|
||||||
Publisher: publisher,
|
Publisher: publisher,
|
||||||
Emojis: []string{"😀"},
|
Emojis: []string{"😀"},
|
||||||
@@ -44,83 +39,143 @@ func createMetadataBytes(packName, publisher string) []byte {
|
|||||||
|
|
||||||
exifJson, err := json.Marshal(exifObj)
|
exifJson, err := json.Marshal(exifObj)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Printf("Failed to marshal EXIF: %v", err)
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// WebP EXIF JSON string representation inside the file
|
|
||||||
bit := []byte{0x49, 0x49, 0x2A, 0x00, 0x08, 0x00, 0x00, 0x00, 0x01, 0x00, 0x41, 0x57, 0x07, 0x00, 0x00, 0x00, 0x00, 0x00, 0x16, 0x00, 0x00, 0x00}
|
bit := []byte{0x49, 0x49, 0x2A, 0x00, 0x08, 0x00, 0x00, 0x00, 0x01, 0x00, 0x41, 0x57, 0x07, 0x00, 0x00, 0x00, 0x00, 0x00, 0x16, 0x00, 0x00, 0x00}
|
||||||
bit = append(bit, exifJson...)
|
bit = append(bit, exifJson...)
|
||||||
|
|
||||||
// Crucial fix: Inject JSON string length at offset 14 (4 bytes)
|
|
||||||
writeUIntLE(bit, int64(len(exifJson)), 14, 4)
|
writeUIntLE(bit, int64(len(exifJson)), 14, 4)
|
||||||
|
|
||||||
return bit
|
return bit
|
||||||
}
|
}
|
||||||
|
|
||||||
func getOrCreateDefaultExif() string {
|
// InjectExif takes raw WebP bytes and constructs a new WebP RIFF byte slice containing the EXIF metadata
|
||||||
defaultExifPath := "temp/default.exif"
|
func InjectExif(webpData []byte, packName, publisher string) ([]byte, error) {
|
||||||
|
if len(webpData) < 12 || string(webpData[0:4]) != "RIFF" || string(webpData[8:12]) != "WEBP" {
|
||||||
if err := os.MkdirAll("temp", 0755); err != nil {
|
return nil, fmt.Errorf("not a valid WEBP file")
|
||||||
log.Printf("Failed to create temp directory: %v", err)
|
|
||||||
return ""
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// ALWAYS recreate temp/default.exif just in case previous boots were corrupted
|
var width, height uint32
|
||||||
// This corrects broken instances locally instantly
|
hasVP8X := false
|
||||||
packName := os.Getenv("STICKER_NAME")
|
var vp8xFlags byte
|
||||||
if packName == "" {
|
var vp8xBytes []byte
|
||||||
packName = "Bot"
|
|
||||||
|
var chunks [][]byte
|
||||||
|
|
||||||
|
// Default defaults for VP8X canvas sizes
|
||||||
|
width = 512
|
||||||
|
height = 512
|
||||||
|
|
||||||
|
offset := 12
|
||||||
|
for offset < len(webpData) {
|
||||||
|
if offset+8 > len(webpData) {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
chunkID := string(webpData[offset : offset+4])
|
||||||
|
chunkSize := binary.LittleEndian.Uint32(webpData[offset+4 : offset+8])
|
||||||
|
|
||||||
|
paddedSize := chunkSize
|
||||||
|
if paddedSize%2 != 0 {
|
||||||
|
paddedSize++
|
||||||
}
|
}
|
||||||
|
|
||||||
publisher := os.Getenv("STICKER_PUBLISHER")
|
if offset+8+int(paddedSize) > len(webpData) {
|
||||||
if publisher == "" {
|
break
|
||||||
publisher = "Bot"
|
|
||||||
}
|
}
|
||||||
|
|
||||||
exifContent := createMetadataBytes(packName, publisher)
|
chunkData := webpData[offset+8 : offset+8+int(paddedSize)]
|
||||||
if exifContent == nil {
|
|
||||||
return ""
|
switch chunkID {
|
||||||
|
case "VP8X":
|
||||||
|
hasVP8X = true
|
||||||
|
vp8xFlags = chunkData[0]
|
||||||
|
vp8xBytes = chunkData
|
||||||
|
// extract dimensions
|
||||||
|
wBytes := []byte{chunkData[4], chunkData[5], chunkData[6], 0}
|
||||||
|
hBytes := []byte{chunkData[7], chunkData[8], chunkData[9], 0}
|
||||||
|
width = binary.LittleEndian.Uint32(wBytes) + 1
|
||||||
|
height = binary.LittleEndian.Uint32(hBytes) + 1
|
||||||
|
case "VP8 ":
|
||||||
|
if !hasVP8X {
|
||||||
|
// extract 14 bit scale block
|
||||||
|
w := uint32(chunkData[6]) | (uint32(chunkData[7]) << 8)
|
||||||
|
width = w & 0x3FFF
|
||||||
|
h := uint32(chunkData[8]) | (uint32(chunkData[9]) << 8)
|
||||||
|
height = h & 0x3FFF
|
||||||
|
}
|
||||||
|
chunks = append(chunks, webpData[offset:offset+8+int(paddedSize)])
|
||||||
|
case "VP8L":
|
||||||
|
if !hasVP8X {
|
||||||
|
// Parse bits 1-28 for w/h
|
||||||
|
b0 := uint32(chunkData[1])
|
||||||
|
b1 := uint32(chunkData[2])
|
||||||
|
b2 := uint32(chunkData[3])
|
||||||
|
b3 := uint32(chunkData[4])
|
||||||
|
width = 1 + (((b1 & 0x3F) << 8) | b0)
|
||||||
|
height = 1 + (((b3 & 0xF) << 10) | (b2 << 2) | ((b1 & 0xC0) >> 6))
|
||||||
|
}
|
||||||
|
chunks = append(chunks, webpData[offset:offset+8+int(paddedSize)])
|
||||||
|
default:
|
||||||
|
if chunkID != "EXIF" {
|
||||||
|
chunks = append(chunks, webpData[offset:offset+8+int(paddedSize)])
|
||||||
|
}
|
||||||
|
}
|
||||||
|
offset += 8 + int(paddedSize)
|
||||||
}
|
}
|
||||||
|
|
||||||
if err := os.WriteFile(defaultExifPath, exifContent, 0644); err != nil {
|
exifMeta := createMetadataBytes(packName, publisher)
|
||||||
log.Printf("Failed to write default EXIF: %v", err)
|
if exifMeta == nil {
|
||||||
return ""
|
return nil, fmt.Errorf("failed to create exif byte payload")
|
||||||
}
|
}
|
||||||
|
|
||||||
return defaultExifPath
|
// Turn on EXIF flag (bit 3) in VP8X
|
||||||
}
|
if !hasVP8X {
|
||||||
|
vp8xBytes = make([]byte, 10)
|
||||||
func ResolveExif(args []string) (string, bool) {
|
vp8xFlags = 0x08 // EXIF flag is bit 3
|
||||||
if len(args) == 0 {
|
vp8xBytes[0] = vp8xFlags
|
||||||
return getOrCreateDefaultExif(), false
|
w := width - 1
|
||||||
}
|
h := height - 1
|
||||||
|
vp8xBytes[4] = byte(w)
|
||||||
argStr := strings.Join(args, " ")
|
vp8xBytes[5] = byte(w >> 8)
|
||||||
parts := strings.Split(argStr, "|")
|
vp8xBytes[6] = byte(w >> 16)
|
||||||
|
vp8xBytes[7] = byte(h)
|
||||||
packName := strings.TrimSpace(parts[0])
|
vp8xBytes[8] = byte(h >> 8)
|
||||||
publisher := "Bot"
|
vp8xBytes[9] = byte(h >> 16)
|
||||||
if len(parts) > 1 {
|
} else {
|
||||||
publisher = strings.TrimSpace(parts[1])
|
vp8xBytes[0] = vp8xFlags | 0x08
|
||||||
}
|
}
|
||||||
|
|
||||||
if packName == "" && publisher == "Bot" {
|
var buf bytes.Buffer
|
||||||
return getOrCreateDefaultExif(), false
|
buf.WriteString("RIFF")
|
||||||
|
buf.Write([]byte{0, 0, 0, 0}) // Total size placeholder
|
||||||
|
buf.WriteString("WEBP")
|
||||||
|
|
||||||
|
// Write VP8X
|
||||||
|
buf.WriteString("VP8X")
|
||||||
|
vp8xSize := make([]byte, 4)
|
||||||
|
binary.LittleEndian.PutUint32(vp8xSize, 10)
|
||||||
|
buf.Write(vp8xSize)
|
||||||
|
buf.Write(vp8xBytes)
|
||||||
|
|
||||||
|
// Write existing chunks
|
||||||
|
for _, chunk := range chunks {
|
||||||
|
buf.Write(chunk)
|
||||||
}
|
}
|
||||||
|
|
||||||
exifContent := createMetadataBytes(packName, publisher)
|
// Write EXIF
|
||||||
if exifContent == nil {
|
buf.WriteString("EXIF")
|
||||||
return getOrCreateDefaultExif(), false
|
exifSize := uint32(len(exifMeta))
|
||||||
|
exifSizeSlice := make([]byte, 4)
|
||||||
|
binary.LittleEndian.PutUint32(exifSizeSlice, exifSize)
|
||||||
|
buf.Write(exifSizeSlice)
|
||||||
|
buf.Write(exifMeta)
|
||||||
|
if exifSize%2 != 0 {
|
||||||
|
buf.WriteByte(0)
|
||||||
}
|
}
|
||||||
|
|
||||||
id := uuid.New().String()
|
|
||||||
tempPath := filepath.Join("temp", fmt.Sprintf("%s.exif", id))
|
|
||||||
|
|
||||||
if err := os.WriteFile(tempPath, exifContent, 0644); err != nil {
|
result := buf.Bytes()
|
||||||
log.Printf("Failed to write dynamic EXIF: %v", err)
|
totalSize := uint32(len(result) - 8)
|
||||||
return getOrCreateDefaultExif(), false
|
binary.LittleEndian.PutUint32(result[4:8], totalSize)
|
||||||
}
|
|
||||||
|
|
||||||
return tempPath, true
|
return result, nil
|
||||||
}
|
}
|
||||||
+23
-25
@@ -14,57 +14,55 @@ func ConvertImageToWebp(imgData []byte) ([]byte, error) {
|
|||||||
convertCmd := exec.Command("convert", "-", "-resize", "512x512", "-background", "none", "-compose", "Copy", "-gravity", "center", "-extent", "512x512", "-quality", "100", "png:-")
|
convertCmd := exec.Command("convert", "-", "-resize", "512x512", "-background", "none", "-compose", "Copy", "-gravity", "center", "-extent", "512x512", "-quality", "100", "png:-")
|
||||||
convertCmd.Stdin = bytes.NewReader(imgData)
|
convertCmd.Stdin = bytes.NewReader(imgData)
|
||||||
var convertBuf bytes.Buffer
|
var convertBuf bytes.Buffer
|
||||||
|
var stderr bytes.Buffer
|
||||||
convertCmd.Stdout = &convertBuf
|
convertCmd.Stdout = &convertBuf
|
||||||
|
convertCmd.Stderr = &stderr
|
||||||
if err := convertCmd.Run(); err != nil {
|
if err := convertCmd.Run(); err != nil {
|
||||||
return nil, fmt.Errorf("convert failed: %w", err)
|
return nil, fmt.Errorf("convert failed: %w, stderr: %s", err, stderr.String())
|
||||||
}
|
}
|
||||||
|
|
||||||
cwebpCmd := exec.Command("cwebp", "-quiet", "-mt", "-exact", "-q", "100", "-m", "6", "-alpha_q", "100", "-o", "-", "--", "-")
|
cwebpCmd := exec.Command("cwebp", "-quiet", "-mt", "-exact", "-q", "100", "-m", "6", "-alpha_q", "100", "-o", "-", "--", "-")
|
||||||
cwebpCmd.Stdin = &convertBuf
|
cwebpCmd.Stdin = &convertBuf
|
||||||
var cwebpBuf bytes.Buffer
|
var cwebpBuf bytes.Buffer
|
||||||
|
var cwebStderr bytes.Buffer
|
||||||
cwebpCmd.Stdout = &cwebpBuf
|
cwebpCmd.Stdout = &cwebpBuf
|
||||||
|
cwebpCmd.Stderr = &cwebStderr
|
||||||
if err := cwebpCmd.Run(); err != nil {
|
if err := cwebpCmd.Run(); err != nil {
|
||||||
return nil, fmt.Errorf("cwebp failed: %w", err)
|
return nil, fmt.Errorf("cwebp failed: %w, stderr: %s", err, cwebStderr.String())
|
||||||
}
|
}
|
||||||
|
|
||||||
return cwebpBuf.Bytes(), nil
|
return cwebpBuf.Bytes(), nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func ConvertVideoToWebp(vidData []byte) ([]byte, error) {
|
func ConvertVideoToWebp(vidData []byte) ([]byte, error) {
|
||||||
webpPath := filepath.Join("temp", fmt.Sprintf("%s.webp", uuid.New().String()))
|
id := uuid.New().String()
|
||||||
|
gifPath := filepath.Join("temp", fmt.Sprintf("%s.gif", id))
|
||||||
|
webpPath := filepath.Join("temp", fmt.Sprintf("%s.webp", id))
|
||||||
|
|
||||||
|
defer os.Remove(gifPath)
|
||||||
defer os.Remove(webpPath)
|
defer os.Remove(webpPath)
|
||||||
|
|
||||||
ffmpegCmd := exec.Command("ffmpeg",
|
ffmpegCmd := exec.Command("ffmpeg",
|
||||||
"-y", "-hide_banner", "-loglevel", "error",
|
"-y", "-hide_banner", "-loglevel", "error",
|
||||||
"-i", "pipe:0", "-f", "mp4",
|
"-i", "pipe:0",
|
||||||
"-ss", "00:00:00", "-t", "00:00:15",
|
"-ss", "00:00:00", "-t", "00:00:15",
|
||||||
"-vf", "fps=10,scale=720:-1:flags=lanczos:force_original_aspect_ratio=increase,crop=512:512,pad=512:512:(ow-iw)/2:(oh-ih)/2:color=#00000000,setsar=1",
|
"-vf", "fps=10,scale=720:-1:flags=lanczos:force_original_aspect_ratio=increase,crop=512:512,pad=512:512:(ow-iw)/2:(oh-ih)/2:color=#00000000,setsar=1",
|
||||||
"-compression_level", "6",
|
"-loop", "0",
|
||||||
"-q:v", "60", "-loop", "0",
|
"-an",
|
||||||
"-preset", "picture", "-an", "-fps_mode", "auto",
|
"-f", "gif", gifPath,
|
||||||
"-f", "webp", webpPath,
|
|
||||||
)
|
)
|
||||||
|
var stderr bytes.Buffer
|
||||||
|
ffmpegCmd.Stderr = &stderr
|
||||||
ffmpegCmd.Stdin = bytes.NewReader(vidData)
|
ffmpegCmd.Stdin = bytes.NewReader(vidData)
|
||||||
if err := ffmpegCmd.Run(); err != nil {
|
if err := ffmpegCmd.Run(); err != nil {
|
||||||
return nil, fmt.Errorf("ffmpeg failed: %w", err)
|
return nil, fmt.Errorf("ffmpeg failed: %w, stderr: %s", err, stderr.String())
|
||||||
}
|
}
|
||||||
|
|
||||||
return os.ReadFile(webpPath)
|
gif2webpCmd := exec.Command("gif2webp", "-quiet", "-q", "60", "-m", "6", gifPath, "-o", webpPath)
|
||||||
}
|
var g2wStderr bytes.Buffer
|
||||||
|
gif2webpCmd.Stderr = &g2wStderr
|
||||||
func InjectExif(webpData []byte, exifPath string) ([]byte, error) {
|
if err := gif2webpCmd.Run(); err != nil {
|
||||||
webpPath := filepath.Join("temp", fmt.Sprintf("%s.webp", uuid.New().String()))
|
return nil, fmt.Errorf("gif2webp failed: %w, stderr: %s", err, g2wStderr.String())
|
||||||
defer os.Remove(webpPath)
|
|
||||||
|
|
||||||
tmpWebp := webpPath + ".tmp.webp"
|
|
||||||
if err := os.WriteFile(tmpWebp, webpData, 0644); err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
defer os.Remove(tmpWebp)
|
|
||||||
|
|
||||||
webpmuxCmd := exec.Command("webpmux", "-set", "exif", exifPath, tmpWebp, "-o", webpPath)
|
|
||||||
if err := webpmuxCmd.Run(); err != nil {
|
|
||||||
return nil, fmt.Errorf("webpmux failed: %w", err)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
return os.ReadFile(webpPath)
|
return os.ReadFile(webpPath)
|
||||||
|
|||||||
Reference in New Issue
Block a user