Sumner Evans
103 lines
2.9 KiB
Go
103 lines
2.9 KiB
Go
package media
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
"path/filepath"
|
|
"strconv"
|
|
|
|
"github.com/rs/zerolog"
|
|
"go.mau.fi/util/ffmpeg"
|
|
"go.mau.fi/util/lottie"
|
|
"go.mau.fi/util/random"
|
|
)
|
|
|
|
type AnimatedStickerConfig struct {
|
|
Target string `yaml:"target"`
|
|
ConvertFromWebm bool `yaml:"convert_from_webm"`
|
|
Args struct {
|
|
Width int `yaml:"width"`
|
|
Height int `yaml:"height"`
|
|
FPS int `yaml:"fps"`
|
|
} `yaml:"args"`
|
|
}
|
|
|
|
type ConvertedSticker struct {
|
|
DataWriter io.Reader
|
|
MIMEType string
|
|
ThumbnailData []byte
|
|
ThumbnailMIMEType string
|
|
Width int
|
|
Height int
|
|
Size int
|
|
}
|
|
|
|
func (c AnimatedStickerConfig) convert(ctx context.Context, data []byte) ConvertedSticker {
|
|
input := bytes.NewBuffer(data)
|
|
if c.Target == "disable" {
|
|
return ConvertedSticker{DataWriter: input, MIMEType: "application/x-tgsticker"}
|
|
}
|
|
|
|
log := zerolog.Ctx(ctx).With().Str("animated_sticker_target", c.Target).Logger()
|
|
|
|
if !lottie.Supported() {
|
|
log.Warn().Msg("lottie not supported, cannot convert animated stickers")
|
|
return ConvertedSticker{DataWriter: input, MIMEType: "application/x-tgsticker"}
|
|
} else if (c.Target == "webp" || c.Target == "webm") && !ffmpeg.Supported() {
|
|
log.Warn().Msg("ffmpeg not supported, cannot convert animated stickers")
|
|
return ConvertedSticker{DataWriter: input, MIMEType: "application/x-tgsticker"}
|
|
}
|
|
|
|
dataWriter := new(bytes.Buffer)
|
|
var thumbnailData []byte
|
|
var mimeType, thumbnailMIMEType string
|
|
|
|
var err error
|
|
switch c.Target {
|
|
case "png":
|
|
mimeType = "image/png"
|
|
err = lottie.Convert(ctx, input, "", dataWriter, c.Target, c.Args.Width, c.Args.Height, "1")
|
|
case "gif":
|
|
mimeType = "image/gif"
|
|
err = lottie.Convert(ctx, input, "", dataWriter, c.Target, c.Args.Width, c.Args.Height, strconv.Itoa(c.Args.FPS))
|
|
case "webm", "webp":
|
|
tmpFile := filepath.Join(os.TempDir(), fmt.Sprintf("mautrix-telegram-lottieconverter-%s.%s", random.String(10), c.Target))
|
|
defer func() {
|
|
_ = os.Remove(tmpFile)
|
|
}()
|
|
thumbnailMIMEType = "image/png"
|
|
mimeType = "image/" + c.Target
|
|
thumbnailData, err = lottie.FFmpegConvert(ctx, input, tmpFile, c.Args.Width, c.Args.Height, c.Args.FPS)
|
|
if err != nil {
|
|
break
|
|
}
|
|
var convertedData []byte
|
|
convertedData, err = os.ReadFile(tmpFile)
|
|
dataWriter = bytes.NewBuffer(convertedData)
|
|
default:
|
|
err = fmt.Errorf("unsupported target format %s", c.Target)
|
|
}
|
|
if err != nil {
|
|
log.Err(err).
|
|
Str("target", c.Target).
|
|
Msg("failed to convert animated sticker to target format")
|
|
|
|
// Fallback to original data
|
|
return ConvertedSticker{DataWriter: input, MIMEType: "application/x-tgsticker"}
|
|
}
|
|
|
|
return ConvertedSticker{
|
|
DataWriter: dataWriter,
|
|
MIMEType: mimeType,
|
|
ThumbnailData: thumbnailData,
|
|
ThumbnailMIMEType: thumbnailMIMEType,
|
|
Width: c.Args.Width,
|
|
Height: c.Args.Height,
|
|
Size: dataWriter.Len(),
|
|
}
|
|
|
|
}
|