fix issues

demo/config.yml

@@ -24,9 +24,13 @@ http:
       headers:
         customResponseHeaders:
           Content-Type: "text/html"
+
+    compress:
+      compress: true
 
     plain-to-html:
       chain:
         middlewares:
+          - plain-to-html-header
           - plain-to-html-body
-          - plain-to-html-header
+          - compress

plugin.go

@@ -150,14 +150,17 @@ func (h *PluginHandler) ServeHTTP(rw http.ResponseWriter, req *http.Request) {
 		myReq.SetSupportedEncoding()
 		h.next.ServeHTTP(myRw, &myReq.Request)
 
+		// check if response is injectable
 		if myRw.IsInjectable() {
 			// h.log(fmt.Sprintf("Inject %s", req.URL.EscapedPath()))
 			body, err := myRw.ReadDecoded()
 			if err != nil {
-				newBody := InsertAtBodyEnd(body, h.scriptHtml)
-				myRw.WriteEncoded(newBody, encoding)
-				injected = true
+				h.log(fmt.Sprintf("Error: %s", err))
 			}
+			newBody := InsertAtBodyEnd(body, h.scriptHtml)
+			myRw.WriteEncoded(newBody, encoding)
+			rw.Write(myRw.Read())
+			injected = true
 		}
 	}
 

response_writer.go

@@ -2,6 +2,7 @@ package traefik_umami_plugin
 
 import (
 	"bytes"
+	"fmt"
 	"net/http"
 )
 
@@ -47,12 +48,14 @@ func (w *ResponseWriter) Write(p []byte) (int, error) {
 // Write body bytes
 // Compresses the body to the target encoding.
 func (w *ResponseWriter) WriteEncoded(plain []byte, encoding *Encoding) (int, error) {
+	size := len(plain)
 	encoded, err := Encode(plain, encoding)
 	if err != nil {
 		return 0, err
 	}
 	w.Write(encoded)
 	w.SetContentEncoding(encoding)
+	w.SetContentLength(size)
 	return len(plain), nil
 }
 
@@ -66,3 +69,8 @@ func (w *ResponseWriter) GetContentEncoding() (*Encoding, error) {
 func (w *ResponseWriter) SetContentEncoding(encoding *Encoding) {
 	w.Header().Set("Content-Encoding", encoding.name)
 }
+
+// set content size.
+func (w *ResponseWriter) SetContentLength(size int) {
+	w.Header().Set("Content-Length", fmt.Sprintf("%d", size))
+}