From b8e7453fef3dac6036403bc384eec96becff5114 Mon Sep 17 00:00:00 2001
From: Matthew Holt <mholt@users.noreply.github.com>
Date: Thu, 13 Jun 2019 11:20:43 -0600
Subject: Implement brotli encoder; improve validation of other encoders

---
 modules/caddyhttp/encode/brotli/brotli.go | 51 +++++++++++++++++++++++++++++++
 1 file changed, 51 insertions(+)
 create mode 100644 modules/caddyhttp/encode/brotli/brotli.go

(limited to 'modules/caddyhttp/encode/brotli')

diff --git a/modules/caddyhttp/encode/brotli/brotli.go b/modules/caddyhttp/encode/brotli/brotli.go
new file mode 100644
index 0000000..3289f0c
--- /dev/null
+++ b/modules/caddyhttp/encode/brotli/brotli.go
@@ -0,0 +1,51 @@
+package caddybrotli
+
+import (
+	"fmt"
+
+	"github.com/andybalholm/brotli"
+	"github.com/caddyserver/caddy2"
+	"github.com/caddyserver/caddy2/modules/caddyhttp/encode"
+)
+
+func init() {
+	caddy2.RegisterModule(caddy2.Module{
+		Name: "http.encoders.brotli",
+		New:  func() interface{} { return new(Brotli) },
+	})
+}
+
+// Brotli can create brotli encoders. Note that brotli
+// is not known for great encoding performance.
+type Brotli struct {
+	Quality *int `json:"quality,omitempty"`
+}
+
+// Validate validates b's configuration.
+func (b Brotli) Validate() error {
+	if b.Quality != nil {
+		quality := *b.Quality
+		if quality < brotli.BestSpeed {
+			return fmt.Errorf("quality too low; must be >= %d", brotli.BestSpeed)
+		}
+		if quality > brotli.BestCompression {
+			return fmt.Errorf("quality too high; must be <= %d", brotli.BestCompression)
+		}
+	}
+	return nil
+}
+
+// NewEncoder returns a new brotli writer.
+func (b Brotli) NewEncoder() encode.Encoder {
+	quality := brotli.DefaultCompression
+	if b.Quality != nil {
+		quality = *b.Quality
+	}
+	return brotli.NewWriterLevel(nil, quality)
+}
+
+// Interface guards
+var (
+	_ encode.Encoding  = (*Brotli)(nil)
+	_ caddy2.Validator = (*Brotli)(nil)
+)
-- 
cgit v1.2.3