feat: add a reverse proxy to plex.tv

Author: RoyXiang

handler/const.go

@@ -5,6 +5,8 @@ import (
 )
 
 const (
+	domainPlexTv = "plex.tv"
+
 	headerPlexPrefix     = "X-Plex-"
 	headerCacheStatus    = "X-Plex-Cache-Status"
 	headerClientIdentity = "X-Plex-Client-Identifier"

handler/main.go

@@ -3,7 +3,10 @@ package handler
 import (
 	"log"
 	"net/http"
+	"net/http/httputil"
+	"net/url"
 	"os"
+	"strings"
 
 	"github.com/go-chi/chi/v5/middleware"
 	"github.com/go-redis/redis/v8"
@@ -21,6 +24,7 @@ var (
 func init() {
 	plexClient = NewPlexClient(PlexConfig{
 		BaseUrl:          os.Getenv("PLEX_BASEURL"),
+		SslHost:          os.Getenv("PLEX_SSL_HOST"),
 		Token:            os.Getenv("PLEX_TOKEN"),
 		PlaxtUrl:         os.Getenv("PLAXT_URL"),
 		RedirectWebApp:   os.Getenv("REDIRECT_WEB_APP"),
@@ -46,14 +50,27 @@ func NewRouter() http.Handler {
 	if !plexClient.NoRequestLogs {
 		r.Use(middleware.Logger)
 	}
-	r.Use(wrapMiddleware, middleware.Recoverer, trafficMiddleware)
 
+	plexTvUrl, _ := url.Parse("https://www." + domainPlexTv)
+	plexTvProxy := httputil.NewSingleHostReverseProxy(plexTvUrl)
+	plexTvRouter := r.Host(domainPlexTv).Subrouter()
+	sslRouter := plexTvRouter.MatcherFunc(func(r *http.Request, match *mux.RouteMatch) bool {
+		return strings.Index(r.URL.Path, "servers.xml") != -1
+	}).Subrouter()
+	sslRouter.Use(sslMiddleware)
+	sslRouter.PathPrefix("/").Handler(plexTvProxy)
+	plexTvRouter.PathPrefix("/").Handler(plexTvProxy)
+
+	pmsRouter := r.MatcherFunc(func(r *http.Request, match *mux.RouteMatch) bool {
+		return r.Host != domainPlexTv
+	}).Subrouter()
+	pmsRouter.Use(wrapMiddleware, middleware.Recoverer, trafficMiddleware)
 	if redisClient != nil {
 		// bypass cache
-		r.PathPrefix("/:/").Handler(plexClient)
-		r.PathPrefix("/library/parts/").Handler(plexClient)
+		pmsRouter.PathPrefix("/:/").Handler(plexClient)
+		pmsRouter.PathPrefix("/library/parts/").Handler(plexClient)
 
-		staticRouter := r.Methods(http.MethodGet).Subrouter()
+		staticRouter := pmsRouter.Methods(http.MethodGet).Subrouter()
 		staticRouter.Use(staticMiddleware)
 		staticRouter.Path("/library/media/{key}/chapterImages/{id}").Handler(plexClient)
 		staticRouter.Path("/library/metadata/{key}/art/{id}").Handler(plexClient)
@@ -62,11 +79,11 @@ func NewRouter() http.Handler {
 		staticRouter.PathPrefix("/web/js/").Handler(plexClient)
 		staticRouter.PathPrefix("/web/static/").Handler(plexClient)
 
-		dynamicRouter := r.Methods(http.MethodGet).Subrouter()
+		dynamicRouter := pmsRouter.Methods(http.MethodGet).Subrouter()
 		dynamicRouter.Use(dynamicMiddleware)
 		dynamicRouter.PathPrefix("/").Handler(plexClient)
 	}
+	pmsRouter.PathPrefix("/").Handler(plexClient)
 
-	r.PathPrefix("/").Handler(plexClient)
 	return r
 }

handler/middleware.go

@@ -23,6 +23,18 @@ var (
 	userCtxKey      = &ctxKeyType{"user"}
 )
 
+func sslMiddleware(next http.Handler) http.Handler {
+	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		if plexClient.sslHost != "" && r.Body != nil {
+			bodyBytes, _ := io.ReadAll(r.Body)
+			sslHost := fmt.Sprintf("address=\"%s\" scheme=\"https\"", plexClient.sslHost)
+			modifiedBody := strings.ReplaceAll(string(bodyBytes), "host=\"\"", sslHost)
+			r.Body = io.NopCloser(strings.NewReader(modifiedBody))
+		}
+		next.ServeHTTP(w, r)
+	})
+}
+
 func normalizeMiddleware(next http.Handler) http.Handler {
 	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
 		headers := http.Header{}

handler/plex.go

@@ -21,6 +21,7 @@ import (
 
 type PlexConfig struct {
 	BaseUrl          string
+	SslHost          string
 	Token            string
 	PlaxtUrl         string
 	RedirectWebApp   string
@@ -32,6 +33,7 @@ type PlexClient struct {
 	proxy  *httputil.ReverseProxy
 	client *plex.Plex
 
+	sslHost          string
 	plaxtUrl         string
 	redirectWebApp   bool
 	disableTranscode bool
@@ -92,6 +94,7 @@ func NewPlexClient(config PlexConfig) *PlexClient {
 	return &PlexClient{
 		proxy:            proxy,
 		client:           client,
+		sslHost:          config.SslHost,
 		plaxtUrl:         plaxtUrl,
 		redirectWebApp:   redirectWebApp,
 		disableTranscode: disableTranscode,