@warren-bank/hls-proxy 3.4.2 → 3.4.4

package/README.md

@@ -13,7 +13,8 @@
 * inject custom HTTP headers in all outbound proxied requests
 * prefetch video segments (.ts files)
 * use a hook function to conditionally decide which video segments to prefetch
-* use a hook function to conditionally redirect URLs in the playlist (before they're modified to pass through the proxy)
+* use a hook function to conditionally redirect URLs in the playlist (_before_ and/or _after_ they're modified to pass through the proxy)
+* use a hook function to conditionally rewrite URLs after they're received by the proxy
 
 #### Benefits:
 
@@ -231,6 +232,10 @@ options:
       * conditionally modify the content of .m3u8 files __before__ they are parsed to extract URLs
     * `"redirect": (url) => new_url`
       * conditionally redirect the URLs encountered in .m3u8 files __before__ they are modified to pass through the proxy
+    * `"redirect_final": (url) => new_url`
+      * conditionally redirect the URLs encountered in .m3u8 files __after__ they are modified to pass through the proxy
+    * `"rewrite": (url) => new_url`
+      * conditionally rewrite the URLs requested by clients __before__ they are proxied
     * `"prefetch": (url) => boolean`
       * conditionally decide whether to prefetch video segments on a per-URL basis
       * return value must be a strict boolean type (ie: `true` or `false`)

package/hls-proxy/manifest_parser.js

@@ -123,7 +123,7 @@ const parse_manifest = function(m3u8_content, m3u8_url, referer_url, hooks, cach
       redirect_embedded_url(embedded_url, hooks, m3u8_url, debug)
       if (validate_embedded_url(embedded_url)) {
         finalize_embedded_url(embedded_url, vod_start_at_ms, debug)
-        encode_embedded_url(embedded_url, redirected_base_url, debug, manifest_extension, segment_extension)
+        encode_embedded_url(embedded_url, hooks, redirected_base_url, debug, manifest_extension, segment_extension)
         get_prefetch_url(embedded_url, should_prefetch_url, prefetch_urls)
         modify_m3u8_line(embedded_url, m3u8_lines)
       }
@@ -330,7 +330,7 @@ const finalize_embedded_url = function(embedded_url, vod_start_at_ms, debug) {
   }
 }
 
-const encode_embedded_url = function(embedded_url, redirected_base_url, debug, manifest_extension, segment_extension) {
+const encode_embedded_url = function(embedded_url, hooks, redirected_base_url, debug, manifest_extension, segment_extension) {
   if (embedded_url.unencoded_url) {
     let file_extension = embedded_url.url_type
     if (file_extension) {
@@ -343,6 +343,12 @@ const encode_embedded_url = function(embedded_url, redirected_base_url, debug, m
     embedded_url.encoded_url = `${redirected_base_url}/${ utils.base64_encode(embedded_url.unencoded_url) }.${file_extension || 'other'}`
 
     debug(3, 'redirecting (proxied):', embedded_url.encoded_url)
+
+    if (hooks && (hooks instanceof Object) && hooks.redirect_final && (typeof hooks.redirect_final === 'function')) {
+      embedded_url.encoded_url = hooks.redirect_final(embedded_url.encoded_url)
+
+      debug(3, 'redirecting (proxied, post-hook):', embedded_url.encoded_url)
+    }
   }
   else {
     embedded_url.encoded_url = ''

package/hls-proxy/utils.js

@@ -17,7 +17,7 @@ const base64_decode = function(str) {
 }
 
 const parse_req_url = function(params, req) {
-  const {is_secure, host, manifest_extension, segment_extension} = params
+  const {is_secure, host, manifest_extension, segment_extension, hooks} = params
 
   const result = {redirected_base_url: '', url_type: '', url: '', referer_url: ''}
 
@@ -38,7 +38,11 @@ const parse_req_url = function(params, req) {
 
     let url, url_lc, index
 
-    url    = base64_decode( decodeURIComponent( matches[2] ) ).trim()
+    url = base64_decode( decodeURIComponent( matches[2] ) ).trim()
+
+    if (hooks && (hooks instanceof Object) && hooks.rewrite && (typeof hooks.rewrite === 'function'))
+      url = hooks.rewrite(url)
+
     url_lc = url.toLowerCase()
     index  = url_lc.indexOf('http')
 

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@warren-bank/hls-proxy",
   "description": "Node.js server to proxy HLS video streams",
-  "version":     "3.4.2",
+  "version":     "3.4.4",
   "scripts": {
     "start":     "node hls-proxy/bin/hlsd.js",
     "sudo": "sudo node hls-proxy/bin/hlsd.js"