feat(plex.tv): log all POST and PUT requests

Author: RoyXiang

handler/main.go

@@ -6,7 +6,6 @@ import (
 	"net/http/httputil"
 	"net/url"
 	"os"
-	"strings"
 
 	"github.com/go-chi/chi/v5/middleware"
 	"github.com/go-redis/redis/v8"
@@ -54,18 +53,9 @@ func NewRouter() http.Handler {
 	plexTvUrl, _ := url.Parse("https://www." + domainPlexTv)
 	plexTvProxy := httputil.NewSingleHostReverseProxy(plexTvUrl)
 	plexTvRouter := r.Host(domainPlexTv).Subrouter()
-	sslRouter := plexTvRouter.MatcherFunc(func(r *http.Request, _ *mux.RouteMatch) bool {
-		path := r.URL.EscapedPath()
-		if r.Method == http.MethodPost && path == "/servers.xml" {
-			return true
-		}
-		if r.Method == http.MethodPut && strings.HasPrefix(path, "/devices/") {
-			return true
-		}
-		return false
-	}).Subrouter()
-	sslRouter.Use(sslMiddleware)
-	sslRouter.PathPrefix("/").Handler(plexTvProxy)
+	updateRouter := plexTvRouter.Methods(http.MethodPost, http.MethodPut).Subrouter()
+	updateRouter.Use(plexTvMiddleware)
+	updateRouter.PathPrefix("/").Handler(plexTvProxy)
 	plexTvRouter.PathPrefix("/").Handler(plexTvProxy)
 
 	pmsRouter := r.MatcherFunc(func(r *http.Request, match *mux.RouteMatch) bool {

handler/middleware.go

@@ -26,26 +26,40 @@ var (
 	reLocalAddresses = regexp.MustCompile(` localAddresses="[^"]*"`)
 )
 
-func sslMiddleware(next http.Handler) http.Handler {
+func plexTvMiddleware(next http.Handler) http.Handler {
 	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
-		if plexClient.sslHost != "" && r.Body != nil {
-			name := "Connection[][uri]"
-			query := r.URL.Query()
-			if query.Has(name) {
-				query.Set(name, "https://"+plexClient.sslHost)
-			}
-			nr := cloneRequest(r, r.Header, query)
+		if plexClient.sslHost == "" {
+			next.ServeHTTP(w, r)
+			return
+		}
+
+		var bodyBytes []byte
+		path := r.URL.EscapedPath()
+		query := r.URL.Query()
+		if r.Body != nil {
+			bodyBytes, _ = io.ReadAll(r.Body)
+			common.GetLogger().Printf("%s", string(bodyBytes))
+		}
 
-			bodyBytes, _ := io.ReadAll(r.Body)
+		switch path {
+		case "/servers.xml":
 			sslHost := fmt.Sprintf("address=\"%s\" scheme=\"https\"", plexClient.sslHost)
-			modifiedBody := bytes.ReplaceAll(bodyBytes, []byte("host=\"\""), []byte(sslHost))
-			modifiedBody = reLocalAddresses.ReplaceAll(modifiedBody, []byte(""))
-			nr.Body = io.NopCloser(bytes.NewReader(modifiedBody))
+			bodyBytes = bytes.ReplaceAll(bodyBytes, []byte("host=\"\""), []byte(sslHost))
+			bodyBytes = reLocalAddresses.ReplaceAll(bodyBytes, []byte(""))
+		default:
+			if strings.HasPrefix(path, "/devices/") {
+				name := "Connection[][uri]"
+				if query.Has(name) {
+					query.Set(name, "https://"+plexClient.sslHost)
+				}
+			}
+		}
 
-			next.ServeHTTP(w, nr)
-		} else {
-			next.ServeHTTP(w, r)
+		nr := cloneRequest(r, r.Header, query)
+		if bodyBytes != nil {
+			nr.Body = io.NopCloser(bytes.NewReader(bodyBytes))
 		}
+		next.ServeHTTP(w, nr)
 	})
 }