heif/heic

internal/builtins/heif.go

@@ -0,0 +1,8 @@
+//go:build heif || full
+// +build heif full
+
+package builtins
+
+import (
+	_ "github.com/coalaura/ffwebp/internal/codec/heif"
+)

internal/codec/heif/heif.go

@@ -0,0 +1,80 @@
+package heif
+
+import (
+	"bytes"
+	"fmt"
+	"image"
+	"io"
+
+	"github.com/gen2brain/heic"
+
+	"github.com/coalaura/ffwebp/internal/codec"
+	"github.com/coalaura/ffwebp/internal/opts"
+	"github.com/urfave/cli/v3"
+)
+
+func init() {
+	codec.Register(impl{})
+}
+
+type impl struct{}
+
+func (impl) String() string {
+	return "heif"
+}
+
+func (impl) Extensions() []string {
+	return []string{"heic", "heif"}
+}
+
+func (impl) CanEncode() bool {
+	return false
+}
+
+func (impl) Flags(flags []cli.Flag) []cli.Flag {
+	return flags
+}
+
+func (impl) Sniff(reader io.ReaderAt) (int, []byte, error) {
+	buf := make([]byte, 32)
+
+	n, err := reader.ReadAt(buf, 0)
+	if err != nil && err != io.EOF {
+		return 0, nil, err
+	}
+
+	buf = buf[:n]
+
+	if len(buf) < 12 {
+		return 0, nil, nil
+	}
+
+	if !bytes.Equal(buf[4:8], []byte("ftyp")) {
+		return 0, nil, nil
+	}
+
+	brands := [][]byte{
+		[]byte("heic"),
+		[]byte("heix"),
+		[]byte("hevc"),
+		[]byte("hevx"),
+		[]byte("mif1"),
+		[]byte("msf1"),
+	}
+
+	for _, b := range brands {
+		if bytes.Contains(buf[8:], b) {
+			return 90, buf[:n], nil
+		}
+	}
+
+	return 0, nil, nil
+}
+
+func (impl) Decode(reader io.Reader) (image.Image, error) {
+	return heic.Decode(reader)
+}
+
+func (impl) Encode(writer io.Writer, img image.Image, options opts.Common) error {
+	return fmt.Errorf("heif: encoding not supported")
+}