feat(ssl): alter requests to https://plex.tv/devices/<device_id>

Author: RoyXiang

handler/main.go

@@ -6,6 +6,7 @@ import (
 	"net/http/httputil"
 	"net/url"
 	"os"
+	"strings"
 
 	"github.com/go-chi/chi/v5/middleware"
 	"github.com/go-redis/redis/v8"
@@ -53,7 +54,16 @@ func NewRouter() http.Handler {
 	plexTvUrl, _ := url.Parse("https://www." + domainPlexTv)
 	plexTvProxy := httputil.NewSingleHostReverseProxy(plexTvUrl)
 	plexTvRouter := r.Host(domainPlexTv).Subrouter()
-	sslRouter := plexTvRouter.Methods(http.MethodPost).Path("/servers.xml").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)
 	plexTvRouter.PathPrefix("/").Handler(plexTvProxy)

handler/middleware.go

@@ -29,12 +29,19 @@ var (
 func sslMiddleware(next http.Handler) http.Handler {
 	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
 		if plexClient.sslHost != "" && r.Body != nil {
-			nr := cloneRequest(r, r.Header, r.URL.Query())
+			name := "Connection[][uri]"
+			query := r.URL.Query()
+			if query.Has(name) {
+				query.Set(name, "https://"+plexClient.sslHost)
+			}
+			nr := cloneRequest(r, r.Header, query)
+
 			bodyBytes, _ := io.ReadAll(r.Body)
 			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))
+
 			next.ServeHTTP(w, nr)
 		} else {
 			next.ServeHTTP(w, r)