1 files changed, 25 insertions, 6 deletions
diff --git a/pkg/ext/compression.go b/pkg/ext/compression.go
index 6c7a219..d3a3df1 100644
--- a/pkg/ext/compression.go
+++ b/pkg/ext/compression.go
@@ -15,18 +15,37 @@ import (
 	"github.com/klauspost/compress/zstd"
 )
 
-var (
-	errInvalidParam = errors.New("Invalid weighted param")
-)
+var errInvalidParam = errors.New("Invalid weighted param")
 
 type CompressionResponseWriter struct {
 	innerWriter    http.ResponseWriter
 	compressWriter io.Writer
 }
 
-func Compress(next http.HandlerFunc) http.HandlerFunc {
-	return func(w http.ResponseWriter, r *http.Request) {
+func Compress(next HandlerFunc) HandlerFunc {
+	return func(w http.ResponseWriter, r *Request) {
+		// TODO: hand this better
+		if strings.HasSuffix(r.URL.Path, ".tar.gz") {
+			next(w, r)
+			return
+		}
+
+		if accept, ok := r.Header["Accept-Encoding"]; ok {
+			if compress, algo := GetCompressionWriter(u.FirstOrZero(accept), w); algo != "" {
+				defer compress.Close()
+				w.Header().Add("Content-Encoding", algo)
+				w = &CompressionResponseWriter{
+					innerWriter:    w,
+					compressWriter: compress,
+				}
+			}
+		}
+		next(w, r)
+	}
+}
 
+func Decompress(next http.HandlerFunc) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
 		// TODO: hand this better
 		if strings.HasSuffix(r.URL.Path, ".tar.gz") {
 			next(w, r)
@@ -61,12 +80,12 @@ func GetCompressionWriter(header string, inner io.Writer) (io.WriteCloser, strin
 	default:
 		return nil, ""
 	}
-
 }
 
 func (c *CompressionResponseWriter) Header() http.Header {
 	return c.innerWriter.Header()
 }
+
 func (c *CompressionResponseWriter) Write(b []byte) (int, error) {
 	return c.compressWriter.Write(b)
 }