Decompress more bodies (#140)

* Decompress more bodies

* Fix linting

* Bump version

configuration.go

@@ -3,7 +3,7 @@ package main
 import "time"
 
 const (
-	version     = "2.3.1"
+	version     = "2.3.2"
 	agentName   = "muffet"
 	concurrency = 1024
 	tcpTimeout  = 5 * time.Second

fasthttp_http_response.go

@@ -29,6 +29,10 @@ func (r fasthttpHTTPResponse) Body() ([]byte, error) {
 	switch string(r.response.Header.Peek("Content-Encoding")) {
 	case "gzip":
 		return r.response.BodyGunzip()
+	case "deflate":
+		return r.response.BodyInflate()
+	case "br":
+		return r.response.BodyUnbrotli()
 	}
 
 	return r.response.Body(), nil

fasthttp_http_response_test.go

@@ -0,0 +1,66 @@
+package main
+
+import (
+	"bytes"
+	"compress/gzip"
+	"compress/zlib"
+	"testing"
+
+	"github.com/andybalholm/brotli"
+	"github.com/stretchr/testify/assert"
+	"github.com/valyala/fasthttp"
+)
+
+func TestFastHTTPResponseDecodeGzipBody(t *testing.T) {
+	b := bytes.Buffer{}
+	w := gzip.NewWriter(&b)
+	_, err := w.Write([]byte("foo"))
+	assert.Nil(t, err)
+	err = w.Close()
+	assert.Nil(t, err)
+
+	r := fasthttp.Response{}
+	r.Header.Add("Content-Encoding", "gzip")
+	r.SetBody(b.Bytes())
+
+	bs, err := newFasthttpHTTPResponse(nil, &r).Body()
+
+	assert.Nil(t, err)
+	assert.Equal(t, "foo", string(bs))
+}
+
+func TestFastHTTPResponseDecodeDeflateBody(t *testing.T) {
+	b := bytes.Buffer{}
+	w := zlib.NewWriter(&b)
+	_, err := w.Write([]byte("foo"))
+	assert.Nil(t, err)
+	err = w.Close()
+	assert.Nil(t, err)
+
+	r := fasthttp.Response{}
+	r.Header.Add("Content-Encoding", "deflate")
+	r.SetBody(b.Bytes())
+
+	bs, err := newFasthttpHTTPResponse(nil, &r).Body()
+
+	assert.Nil(t, err)
+	assert.Equal(t, "foo", string(bs))
+}
+
+func TestFastHTTPResponseDecodeBrotliBody(t *testing.T) {
+	b := bytes.Buffer{}
+	w := brotli.NewWriter(&b)
+	_, err := w.Write([]byte("foo"))
+	assert.Nil(t, err)
+	err = w.Close()
+	assert.Nil(t, err)
+
+	r := fasthttp.Response{}
+	r.Header.Add("Content-Encoding", "br")
+	r.SetBody(b.Bytes())
+
+	bs, err := newFasthttpHTTPResponse(nil, &r).Body()
+
+	assert.Nil(t, err)
+	assert.Equal(t, "foo", string(bs))
+}

go.mod

@@ -3,6 +3,7 @@ module github.com/raviqqe/muffet/v2
 go 1.12
 
 require (
+	github.com/andybalholm/brotli v1.0.0
 	github.com/bradleyjkemp/cupaloy v2.3.0+incompatible
 	github.com/davecgh/go-spew v1.1.1 // indirect
 	github.com/jessevdk/go-flags v1.4.0