wire in reverse proxy request handlers

muxinc · andrewjl-mux · Feb 13, 2024 · Feb 9, 2024 · Jan 22, 2024 · Jan 31, 2024
commit aa97cdd750f998793282decb45259968a2e10255
diff --git a/Sources/MuxPlayerSwift/ResourceLoader/ReverseProxyServer.swift b/Sources/MuxPlayerSwift/ResourceLoader/ReverseProxyServer.swift
@@ -23,6 +23,9 @@ class ReverseProxyServer {
             return
         }
 
+        self.setupManifestRequestHandler()
+        self.setupSegmentHandler()
+
         webServer.start(
             withPort: port,
             bonjourName: nil
@@ -87,8 +90,22 @@ class ReverseProxyServer {
                         .map { line in self.processPlaylistLine(line, forOriginURL: origenURL) }
                         .joined(separator: "\n")
 
+                    let contentType = response.mimeType ?? "application/x-mpegurl"
+
+                    guard let serializedParsedManifest = parsedManifest?.data(using: .utf8) else {
+                        return completion(
+                            GCDWebServerErrorResponse(
+                                statusCode: 500
+                            )
+                        )
+                    }
 
-                    return completion(GCDWebServerErrorResponse(statusCode: 500))
+                    return completion(
+                        GCDWebServerDataResponse(
+                            data: serializedParsedManifest,
+                            contentType: contentType
+                        )
+                    )
                 }
 
                 task.resume()
@@ -99,6 +116,37 @@ class ReverseProxyServer {
 
     }
 
+    private func setupSegmentHandler() {
+        self.webServer.addHandler(
+            forMethod: "GET",
+            pathRegex: "^/.*\\.ts$",
+            request: GCDWebServerRequest.self
+        ) { [weak self] request, completion in
+
+            guard let self = self else {
+                return completion(GCDWebServerDataResponse(statusCode: 500))
+            }
+
+            guard let origenURL = self.origenURL(from: request) else {
+                return completion(GCDWebServerErrorResponse(statusCode: 400))
+            }
+
+            let task = self.session.dataTask(with: origenURL) { data, response, error in
+              guard let data = data, let response = response else {
+                return completion(GCDWebServerErrorResponse(statusCode: 500))
+              }
+
+              let contentType = response.mimeType ?? "video/mp2t"
+              completion(GCDWebServerDataResponse(data: data, contentType: contentType))
+
+
+            }
+
+            task.resume()
+
+        }
+    }
+
     private func processPlaylistLine(
         _ line: String,
         forOriginURL origenURL: URL