summaryrefslogtreecommitdiff
path: root/internal/converter
diff options
from:
to:
context:
space:
mode:
Diffstat (limited to 'internal/converter')
-rw-r--r--internal/converter/converter_interface.go1
-rw-r--r--internal/converter/jpeg.go114
-rw-r--r--internal/converter/webp.go27
3 files changed, 131 insertions, 11 deletions
diff --git a/internal/converter/converter_interface.go b/internal/converter/converter_interface.go
index fda23d1..abff99d 100644
--- a/internal/converter/converter_interface.go
+++ b/internal/converter/converter_interface.go
@@ -17,4 +17,5 @@ type Converter interface {
var NewConverterMap = map[string]func(cfg *config.ConverterConfig) (Converter, error){
"webp": NewWebpConverter,
+ "jpeg": NewJpegConverter,
}
diff --git a/internal/converter/jpeg.go b/internal/converter/jpeg.go
new file mode 100644
index 0000000..a35a47c
--- /dev/null
+++ b/internal/converter/jpeg.go
@@ -0,0 +1,114 @@
+package converter
+
+import (
+ "fmt"
+ "image"
+ "image/jpeg"
+ "image/png"
+ "io"
+ "log/slog"
+ "path/filepath"
+ "strings"
+
+ "github.com/SayaAndy/saya-today-thumbnail-generator/config"
+ "github.com/SayaAndy/saya-today-thumbnail-generator/internal/client/input"
+ "github.com/SayaAndy/saya-today-thumbnail-generator/internal/client/output"
+ "github.com/kolesa-team/go-webp/webp"
+ "golang.org/x/image/draw"
+)
+
+var _ Converter = (*JpegConverter)(nil)
+
+type JpegConverter struct {
+ maxWidth int
+ maxHeight int
+ extensionName string
+ quality int
+ outputClient output.OutputClient
+}
+
+func NewJpegConverter(cfg *config.ConverterConfig) (Converter, error) {
+ if cfg.Type != "jpeg" {
+ return nil, fmt.Errorf("invalid storage type for JpegConverter")
+ }
+ jpegCfg := cfg.Config.(*config.JpegConfig)
+
+ outputClient, err := output.NewOutputClientMap[cfg.Output.Storage.Type](&cfg.Output)
+ if err != nil {
+ return nil, fmt.Errorf("fail to initialize output client: %w", err)
+ }
+
+ extensionName := ".jpg"
+ if jpegCfg.ExtensionName != "" {
+ extensionName = "." + jpegCfg.ExtensionName
+ }
+
+ return &JpegConverter{jpegCfg.Size.MaxWidth, jpegCfg.Size.MaxHeight, extensionName, jpegCfg.Quality, outputClient}, nil
+}
+
+func (p *JpegConverter) Process(inputMetadata *input.MetadataStruct, reader io.Reader, outputName string) error {
+ var src image.Image
+
+ writer, err := p.outputClient.GetWriter(outputName, inputMetadata, "image/jpeg")
+ if err != nil {
+ return fmt.Errorf("fail to initialize writer for output: %w", err)
+ }
+ defer writer.Close()
+
+ switch inputMetadata.ContentType {
+ case "image/jpeg":
+ src, err = jpeg.Decode(reader)
+ if err != nil {
+ return fmt.Errorf("decode jpeg: %w", err)
+ }
+ case "image/png":
+ src, err = png.Decode(reader)
+ if err != nil {
+ return fmt.Errorf("decode png: %w", err)
+ }
+ case "image/webp":
+ src, err = webp.Decode(reader, nil)
+ if err != nil {
+ return fmt.Errorf("decode webp: %w", err)
+ }
+ default:
+ return fmt.Errorf("unsupported content type: %s", inputMetadata.ContentType)
+ }
+
+ xCoef := 1.0
+ if p.maxWidth > 0 {
+ xCoef = float64(p.maxWidth) / float64(src.Bounds().Max.X)
+ }
+ yCoef := 1.0
+ if p.maxHeight > 0 {
+ yCoef = float64(p.maxHeight) / float64(src.Bounds().Max.Y)
+ }
+ slog.Debug("calculated coefficients", slog.Float64("x_coef", xCoef), slog.Float64("y_coef", yCoef))
+
+ minCoef := xCoef
+ if yCoef < minCoef {
+ minCoef = yCoef
+ }
+
+ if minCoef >= 1.0 {
+ return jpeg.Encode(writer, src, &jpeg.Options{Quality: p.quality})
+ }
+
+ dst := image.NewRGBA(image.Rect(0, 0, int(float64(src.Bounds().Max.X)*minCoef+0.5), int(float64(src.Bounds().Max.Y)*minCoef+0.5)))
+ draw.CatmullRom.Scale(dst, dst.Rect, src, src.Bounds(), draw.Over, nil)
+
+ return jpeg.Encode(writer, dst, &jpeg.Options{Quality: p.quality})
+}
+
+func (p *JpegConverter) DeductOutputPath(inputPath string) string {
+ withoutExt, _ := strings.CutSuffix(inputPath, filepath.Ext(inputPath))
+ return withoutExt + p.extensionName
+}
+
+func (p *JpegConverter) ReadMetadata(path string) (*output.MetadataStruct, error) {
+ return p.outputClient.ReadMetadata(path)
+}
+
+func (p *JpegConverter) IsMissing(path string) bool {
+ return p.outputClient.IsMissing(path)
+}
diff --git a/internal/converter/webp.go b/internal/converter/webp.go
index 8cfe5a7..fbffd01 100644
--- a/internal/converter/webp.go
+++ b/internal/converter/webp.go
@@ -44,7 +44,7 @@ func NewWebpConverter(cfg *config.ConverterConfig) (Converter, error) {
func (p *WebpConverter) Process(inputMetadata *input.MetadataStruct, reader io.Reader, outputName string) error {
var src image.Image
- writer, err := p.outputClient.GetWriter(outputName, inputMetadata)
+ writer, err := p.outputClient.GetWriter(outputName, inputMetadata, "image/webp")
if err != nil {
return fmt.Errorf("fail to initialize writer for output: %w", err)
}
@@ -61,6 +61,11 @@ func (p *WebpConverter) Process(inputMetadata *input.MetadataStruct, reader io.R
if err != nil {
return fmt.Errorf("decode png: %w", err)
}
+ case "image/webp":
+ src, err = webp.Decode(reader, nil)
+ if err != nil {
+ return fmt.Errorf("decode webp: %w", err)
+ }
default:
return fmt.Errorf("unsupported content type: %s", inputMetadata.ContentType)
}
@@ -70,25 +75,25 @@ func (p *WebpConverter) Process(inputMetadata *input.MetadataStruct, reader io.R
return fmt.Errorf("create webp encoder options: %w", err)
}
- xCoef := float64(p.maxWidth) / float64(src.Bounds().Max.X)
- if p.maxWidth == 0 {
- xCoef = 1
+ xCoef := 1.0
+ if p.maxWidth > 0 {
+ xCoef = float64(p.maxWidth) / float64(src.Bounds().Max.X)
}
- yCoef := float64(p.maxHeight) / float64(src.Bounds().Max.Y)
- if p.maxHeight == 0 {
- yCoef = 1
+ yCoef := 1.0
+ if p.maxHeight > 0 {
+ yCoef = float64(p.maxHeight) / float64(src.Bounds().Max.Y)
}
slog.Debug("calculated coefficients", slog.Float64("x_coef", xCoef), slog.Float64("y_coef", yCoef))
- if xCoef > 1 && yCoef > 1 {
- return webp.Encode(writer, src, opts)
- }
-
minCoef := xCoef
if yCoef < minCoef {
minCoef = yCoef
}
+ if minCoef >= 1.0 {
+ return webp.Encode(writer, src, opts)
+ }
+
dst := image.NewRGBA(image.Rect(0, 0, int(float64(src.Bounds().Max.X)*minCoef+0.5), int(float64(src.Bounds().Max.Y)*minCoef+0.5)))
draw.CatmullRom.Scale(dst, dst.Rect, src, src.Bounds(), draw.Over, nil)