added cors fix

main.go

@@ -1,17 +1,39 @@
 package main
 
 import (
+	"bytes"
+	"io"
+	"log"
+	"net/http"
+	"os"
+	"strings"
+
+	"github.com/labstack/echo/v5"
 	"github.com/pocketbase/pocketbase"
 	"github.com/pocketbase/pocketbase/apis"
 	"github.com/pocketbase/pocketbase/core"
-	"log"
-	"os"
-	"strings"
 )
 
 func main() {
 	app := pocketbase.New()
 
+	app.OnBeforeServe().Add(func(e *core.ServeEvent) error {
+		_, err := e.Router.AddRoute(echo.Route{
+			Method: http.MethodGet,
+			Path:   "/cors",
+			Handler: func(c echo.Context) error {
+				return corsProxyHandler(c)
+			},
+			Middlewares: []echo.MiddlewareFunc{
+				apis.ActivityLogger(app),
+			},
+		})
+		if err != nil {
+			return err
+		}
+		return nil
+	})
+
 	app.OnBeforeServe().Add(func(e *core.ServeEvent) error {
 		// add new "GET /hello" route to the app router (echo)
 		e.Router.GET("/*", apis.StaticDirectoryHandler(os.DirFS("./pb_public"), false))
@@ -33,3 +55,65 @@ func main() {
 		log.Fatal(err)
 	}
 }
+
+func corsProxyHandler(c echo.Context) error {
+	// Get the target URL from the request query string
+	url := c.QueryParam("url")
+	if url == "" {
+		return c.JSON(http.StatusBadRequest, "url param is required")
+	}
+
+	// Fetch the resource from the target URL
+	resp, err := http.Get(url)
+	if err != nil {
+		return c.JSON(http.StatusInternalServerError, "Failed to fetch the URL")
+	}
+	defer resp.Body.Close()
+
+	// If the content is a .m3u8 file, rewrite the URLs
+	if strings.HasSuffix(url, ".m3u8") {
+		bodyBytes, err := io.ReadAll(resp.Body)
+		if err != nil {
+			return c.JSON(http.StatusInternalServerError, "Failed to read response body")
+		}
+
+		proxyBaseURL := c.Scheme() + "://" + c.Request().Host + c.Path() + "?url=" + url[:strings.LastIndex(url, "/")+1]
+
+		// Rewrite the URLs in the .m3u8 file
+		lines := strings.Split(string(bodyBytes), "\n")
+		for i, line := range lines {
+			if strings.HasSuffix(line, ".ts") {
+				lines[i] = proxyBaseURL + line
+			}
+		}
+		modifiedContent := strings.Join(lines, "\n")
+
+		// Write the modified content to the response
+		c.Response().Header().Set("Content-Type", resp.Header.Get("Content-Type"))
+		c.Response().WriteHeader(resp.StatusCode)
+		_, err = io.Copy(c.Response().Writer, bytes.NewReader([]byte(modifiedContent)))
+		if err != nil {
+			return c.JSON(http.StatusInternalServerError, "Failed to copy modified response body")
+		}
+
+		return nil
+	}
+
+	// Copy the headers from the fetched response to the proxy response
+	for key, values := range resp.Header {
+		for _, value := range values {
+			c.Response().Header().Add(key, value)
+		}
+	}
+
+	// Copy the status code from the fetched response to the proxy response
+	c.Response().WriteHeader(resp.StatusCode)
+
+	// Copy the body from the fetched response to the proxy response
+	_, err = io.Copy(c.Response().Writer, resp.Body)
+	if err != nil {
+		return c.JSON(http.StatusInternalServerError, "Failed to copy response body")
+	}
+
+	return nil
+}