xtream bundler

Dockers/xtream-ppv-bundler/test/hls.test.js

@@ -0,0 +1,46 @@
+'use strict';
+
+const test = require('node:test');
+const assert = require('node:assert/strict');
+const { rewritePlaylist, signUrl, verifyToken } = require('../src/hls');
+
+test('rewrites playlist URIs to signed proxy URLs', () => {
+  const base = 'http://upstream.example/live/user/pass/index.m3u8';
+  const body = [
+    '#EXTM3U',
+    '#EXT-X-KEY:METHOD=AES-128,URI="keys/key.bin"',
+    '#EXTINF:6,',
+    'segment1.ts',
+    '#EXTINF:6,',
+    'http://cdn.example/segment2.ts',
+    '#EXT-X-ENDLIST',
+  ].join('\n');
+
+  const rewritten = rewritePlaylist(body, base, 2, 'http://proxy.example');
+  const lines = rewritten.split('\n');
+
+  assert.equal(lines[0], '#EXTM3U');
+  assert.match(lines[1], /^#EXT-X-KEY:METHOD=AES-128,URI="http:\/\/proxy\.example\/hls\/2\/.+"/);
+  assert.match(lines[3], /^http:\/\/proxy\.example\/hls\/2\/.+/);
+  assert.match(lines[5], /^http:\/\/proxy\.example\/hls\/2\/.+/);
+
+  const token = lines[3].split('/hls/2/')[1];
+  assert.equal(
+    verifyToken(2, token),
+    'http://upstream.example/live/user/pass/segment1.ts'
+  );
+
+  const keyToken = lines[1].match(/URI="http:\/\/proxy\.example\/hls\/2\/([^"]+)"/)[1];
+  assert.equal(
+    verifyToken(2, keyToken),
+    'http://upstream.example/live/user/pass/keys/key.bin'
+  );
+});
+
+test('rejects tampered tokens', () => {
+  const token = signUrl('http://upstream.example/segment.ts', 1);
+  assert.equal(verifyToken(1, token), 'http://upstream.example/segment.ts');
+  assert.equal(verifyToken(2, token), null);
+  const tampered = token.slice(0, -1) + (token.endsWith('a') ? 'b' : 'a');
+  assert.equal(verifyToken(1, tampered), null);
+});