compy/transcoder/jpeg.go
Andrew Gaul defbc287f1 Allow X-Compy-Quality override header
This allows clients to override the configured compression quality.
Fixes #14.
2017-08-19 11:29:31 -07:00

58 lines
1.2 KiB
Go

package transcoder
import (
"github.com/barnacs/compy/proxy"
"github.com/chai2010/webp"
"github.com/pixiv/go-libjpeg/jpeg"
"net/http"
"strconv"
)
type Jpeg struct {
decOptions *jpeg.DecoderOptions
encOptions *jpeg.EncoderOptions
}
func NewJpeg(quality int) *Jpeg {
return &Jpeg{
decOptions: &jpeg.DecoderOptions{},
encOptions: &jpeg.EncoderOptions{
Quality: quality,
OptimizeCoding: true,
},
}
}
func (t *Jpeg) Transcode(w *proxy.ResponseWriter, r *proxy.ResponseReader, headers http.Header) error {
img, err := jpeg.Decode(r, t.decOptions)
if err != nil {
return err
}
encOptions := t.encOptions
qualityString := headers.Get("X-Compy-Quality")
if qualityString != "" {
if quality, err := strconv.Atoi(qualityString); err != nil {
return err
} else {
encOptions.Quality = quality
}
}
if SupportsWebP(headers) {
w.Header().Set("Content-Type", "image/webp")
options := webp.Options{
Lossless: false,
Quality: float32(encOptions.Quality),
}
if err = webp.Encode(w, img, &options); err != nil {
return err
}
} else {
if err = jpeg.Encode(w, img, encOptions); err != nil {
return err
}
}
return nil
}