Diffstat (limited to 'internal/converter/webp.go')
| -rw-r--r-- | internal/converter/webp.go | 79 |
1 files changed, 51 insertions, 28 deletions
diff --git a/internal/converter/webp.go b/internal/converter/webp.go index b7f1216..fbffd01 100644 --- a/internal/converter/webp.go +++ b/internal/converter/webp.go @@ -12,6 +12,8 @@ import ( "golang.org/x/image/draw" "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/encoder" "github.com/kolesa-team/go-webp/webp" ) @@ -19,37 +21,36 @@ import ( var _ Converter = (*WebpConverter)(nil) type WebpConverter struct { - maxWidth int - maxHeight int - quality int + maxWidth int + maxHeight int + quality int + outputClient output.OutputClient } -func NewWebpConverter(cfg *config.ConverterConfig) (*WebpConverter, error) { +func NewWebpConverter(cfg *config.ConverterConfig) (Converter, error) { if cfg.Type != "webp" { return nil, fmt.Errorf("invalid storage type for WebpConverter") } webpCfg := cfg.Config.(*config.WebpConfig) - return &WebpConverter{webpCfg.Size.MaxWidth, webpCfg.Size.MaxHeight, webpCfg.Quality}, nil -} - -func (p *WebpConverter) DeductOutputPath(inputPath string) string { - pathParts := strings.Split(inputPath, ".") - if len(pathParts) < 2 { - return inputPath + ".webp" + outputClient, err := output.NewOutputClientMap[cfg.Output.Storage.Type](&cfg.Output) + if err != nil { + return nil, fmt.Errorf("fail to initialize output client: %w", err) } - pathParts[len(pathParts)-1] = "webp" - return strings.Join(pathParts, ".") + + return &WebpConverter{webpCfg.Size.MaxWidth, webpCfg.Size.MaxHeight, webpCfg.Quality, outputClient}, nil } -func (p *WebpConverter) Process(contentType string, reader io.ReadCloser, writer io.WriteCloser) error { +func (p *WebpConverter) Process(inputMetadata *input.MetadataStruct, reader io.Reader, outputName string) error { var src image.Image - var err error - defer reader.Close() + writer, err := p.outputClient.GetWriter(outputName, inputMetadata, "image/webp") + if err != nil { + return fmt.Errorf("fail to initialize writer for output: %w", err) + } defer writer.Close() - switch contentType { + switch inputMetadata.ContentType { case "image/jpeg": src, err = jpeg.Decode(reader) if err != nil { @@ -60,8 +61,13 @@ func (p *WebpConverter) Process(contentType string, reader io.ReadCloser, writer 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", contentType) + return fmt.Errorf("unsupported content type: %s", inputMetadata.ContentType) } opts, err := encoder.NewLossyEncoderOptions(encoder.PresetDefault, float32(p.quality)) @@ -69,27 +75,44 @@ func (p *WebpConverter) Process(contentType string, reader io.ReadCloser, writer 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) return webp.Encode(writer, dst, opts) } + +func (p *WebpConverter) DeductOutputPath(inputPath string) string { + pathParts := strings.Split(inputPath, ".") + if len(pathParts) < 2 { + return inputPath + ".webp" + } + pathParts[len(pathParts)-1] = "webp" + return strings.Join(pathParts, ".") +} + +func (p *WebpConverter) ReadMetadata(path string) (*output.MetadataStruct, error) { + return p.outputClient.ReadMetadata(path) +} + +func (p *WebpConverter) IsMissing(path string) bool { + return p.outputClient.IsMissing(path) +} |