BinaryNinja/tvapp2
4
1
Fork 0
mirror of https://github.com/TheBinaryNinja/tvapp2.git synced 2026-06-04 08:15:42 -04:00
Code Issues Packages Projects Releases Wiki Activity

feat: add multiple url subpath keywords for assets

Browse Source
This commit is contained in:
Aetherx
2025-03-25 00:27:23 -07:00
parent fdd9093f49
commit f086e806df
1 changed files with 55 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

82
tvapp2/index.js
View File

@@ -6,8 +6,8 @@
import fs from 'fs'; import fs from 'fs';
import path from 'path'; import path from 'path';
import https from 'https';
import http from 'http'; import http from 'http';
import https from 'https';
import zlib from 'zlib'; import zlib from 'zlib';
import chalk from 'chalk'; import chalk from 'chalk';
import ejs from 'ejs'; import ejs from 'ejs';
@@ -46,6 +46,9 @@ chalk.level = 3;
/* /*
Define > General Define > General
@note if you change `FOLDER_WWW`; ensure you re-name the folder where the
website assets are stored.
*/ */
let FILE_URL; let FILE_URL;
@@ -58,14 +61,15 @@ let FILE_TAR_SIZE = 0;
let FILE_M3U_MODIFIED = 0; let FILE_M3U_MODIFIED = 0;
let FILE_XML_MODIFIED = 0; let FILE_XML_MODIFIED = 0;
let FILE_TAR_MODIFIED = 0; let FILE_TAR_MODIFIED = 0;
const FOLDER_WWW = 'www';
/* /*
Define > Environment Variables || Defaults Define > Environment Variables || Defaults
*/ */
const envUrlRepo = process.env.URL_REPO || `https://git.binaryninja.net/binaryninja`; const envUrlRepo = process.env.URL_REPO || 'https://git.binaryninja.net/binaryninja';
const envStreamQuality = process.env.STREAM_QUALITY || `hd`; const envStreamQuality = process.env.STREAM_QUALITY || 'hd';
const envFileM3U = process.env.FILE_PLAYLIST || `playlist.m3u8`; const envFileM3U = process.env.FILE_PLAYLIST || 'playlist.m3u8';
const envFileXML = process.env.FILE_EPG || 'xmltv.xml'; const envFileXML = process.env.FILE_EPG || 'xmltv.xml';
const envFileTAR = process.env.FILE_TAR || 'xmltv.xml.gz'; const envFileTAR = process.env.FILE_TAR || 'xmltv.xml.gz';
const LOG_LEVEL = process.env.LOG_LEVEL || 10; const LOG_LEVEL = process.env.LOG_LEVEL || 10;
@@ -76,7 +80,7 @@ const LOG_LEVEL = process.env.LOG_LEVEL || 10;
const extURL = `${ envUrlRepo }/tvapp2-externals/raw/branch/main/urls.txt`; const extURL = `${ envUrlRepo }/tvapp2-externals/raw/branch/main/urls.txt`;
const extXML = `${ envUrlRepo }/XMLTV-EPG/raw/branch/main/xmltv.1.xml`; const extXML = `${ envUrlRepo }/XMLTV-EPG/raw/branch/main/xmltv.1.xml`;
const extFormatted = `${ envUrlRepo }/tvapp2-externals/raw/branch/main/formatted.dat`; const extM3U = `${ envUrlRepo }/tvapp2-externals/raw/branch/main/formatted.dat`;
/* /*
Define > Defaults Define > Defaults
@@ -86,6 +90,24 @@ let urls = [];
const gCookies = {}; const gCookies = {};
const USERAGENT = 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/131.0.0.0 Safari/537.36'; const USERAGENT = 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/131.0.0.0 Safari/537.36';
/*
Web url shortcuts
using any of the following subdomains / subpaths will trigger the download for that specific file
@example http://127.0.0.1:4124/gzip
http://127.0.0.1:4124/gz
http://127.0.0.1:4124/playlist
http://127.0.0.1:4124/key
http://127.0.0.1:4124/channel
*/
const subdomainGZ = [ 'gzip', 'gz' ];
const subdomainM3U = [ 'playlist', 'm3u', 'm3u8' ];
const subdomainEPG = [ 'guide', 'epg', 'xml' ];
const subdomainKey = [ 'key', 'keys' ];
const subdomainChan = [ 'channels', 'channel' ];
/* /*
Define > Logs Define > Logs
@@ -860,7 +882,7 @@ async function serveXML( response, req )
}; };
/* /*
Serves IPTV .tar.gz guide data Serves IPTV .gz guide data
*/ */
async function serveTAR( response, req ) async function serveTAR( response, req )
@@ -887,7 +909,7 @@ async function serveTAR( response, req )
'Content-Type': 'text/plain' 'Content-Type': 'text/plain'
}); });
response.end( `Error serving tar.gz: ${ err.message }` ); response.end( `Error serving gzip: ${ err.message }` );
} }
}; };
@@ -933,7 +955,7 @@ async function initialize()
await getFile( extURL, FILE_URL ); await getFile( extURL, FILE_URL );
await getFile( extXML, FILE_XML ); await getFile( extXML, FILE_XML );
await getFile( extFormatted, FILE_M3U ); await getFile( extM3U, FILE_M3U );
await prepareGzip(); await prepareGzip();
urls = fs.readFileSync( FILE_URL, 'utf-8' ).split( '\n' ).filter( Boolean ); urls = fs.readFileSync( FILE_URL, 'utf-8' ).split( '\n' ).filter( Boolean );
@@ -988,11 +1010,17 @@ const server = http.createServer( ( request, response ) =>
*/ */
const method = request.method || 'GET'; const method = request.method || 'GET';
let loadAsset = request.url; let reqUrl = request.url;
if ( loadAsset === '/' ) if ( reqUrl === '/' )
loadAsset = 'index.html'; reqUrl = 'index.html';
Log.debug( `www`, chalk.yellow( ` [GET] ` ), chalk.white( `→` ), chalk.grey( `${ loadAsset }` ) ); /*
Remove leading forward slash
*/
const loadFile = reqUrl.replace( /^\/+/, '' );
Log.debug( `www`, chalk.yellow( ` [GET] ` ), chalk.white( `→` ), chalk.grey( `${ loadFile }` ) );
const handleRequest = async() => const handleRequest = async() =>
{ {
@@ -1001,41 +1029,41 @@ const server = http.createServer( ( request, response ) =>
Place the template system last. Getting TVApp data should take priority. Place the template system last. Getting TVApp data should take priority.
*/ */
if ( loadAsset === '/playlist' && method === 'GET' ) if ( subdomainM3U.includes( `${ loadFile }` ) && method === 'GET' )
{ {
Log.info( `Received request for playlist data`, chalk.white( `→` ), chalk.grey( `/playlist` ) ); Log.info( `Received request for m3u playlist data`, chalk.white( `→` ), chalk.grey( `${ loadFile }` ) );
await serveM3U( response, request ); await serveM3U( response, request );
return; return;
} }
if ( loadAsset.startsWith( '/channel' ) && method === 'GET' ) if ( subdomainChan.includes( `${ loadFile }` ) && method === 'GET' )
{ {
Log.info( `Received request for channel data`, chalk.white( `→` ), chalk.grey( `/channel` ) ); Log.info( `Received request for channel data`, chalk.white( `→` ), chalk.grey( `${ loadFile }` ) );
await serveM3UPlaylist( request, response ); await serveM3UPlaylist( request, response );
return; return;
} }
if ( loadAsset.startsWith( '/key' ) && method === 'GET' ) if ( subdomainKey.includes( `${ loadFile }` ) && method === 'GET' )
{ {
Log.info( `Received request for key data`, chalk.white( `→` ), chalk.grey( `/key` ) ); Log.info( `Received request for key data`, chalk.white( `→` ), chalk.grey( `${ loadFile }` ) );
await serveKey( request, response ); await serveKey( request, response );
return; return;
} }
if ( loadAsset === '/epg' && method === 'GET' ) if ( subdomainEPG.includes( `${ loadFile }` ) && method === 'GET' )
{ {
Log.info( `Received request for EPG data`, chalk.white( `→` ), chalk.grey( `/epg` ) ); Log.info( `Received request for raw EPG data`, chalk.white( `→` ), chalk.grey( `${ loadFile }` ) );
await serveXML( response, request ); await serveXML( response, request );
return; return;
} }
if ( loadAsset === '/tar' && method === 'GET' ) if ( subdomainGZ.includes( `${ loadFile }` ) && method === 'GET' )
{ {
Log.info( `Received request for EPG data`, chalk.white( `→` ), chalk.grey( `/epg` ) ); Log.info( `Received request for compressed EPG data`, chalk.white( `→` ), chalk.grey( `${ loadFile }` ) );
await serveTAR( response, request ); await serveTAR( response, request );
return; return;
@@ -1046,7 +1074,7 @@ const server = http.createServer( ( request, response ) =>
read the loaded asset file read the loaded asset file
*/ */
ejs.renderFile( './www/' + loadAsset, ejs.renderFile( `./${ FOLDER_WWW }/${ loadFile }`,
{ {
fileM3U: envFileM3U, fileM3U: envFileM3U,
sizeM3U: FILE_M3U_SIZE, sizeM3U: FILE_M3U_SIZE,
@@ -1074,7 +1102,7 @@ const server = http.createServer( ( request, response ) =>
the file loaded is dependent on what comes to the right of the period. the file loaded is dependent on what comes to the right of the period.
*/ */
const fileExt = loadAsset.lastIndexOf( '.' ); const fileExt = loadFile.lastIndexOf( '.' );
const fileMime = fileExt === -1 const fileMime = fileExt === -1
? 'text/plain' ? 'text/plain'
: { : {
@@ -1086,19 +1114,19 @@ const server = http.createServer( ( request, response ) =>
'.css' : 'text/css', '.css' : 'text/css',
'.gz' : 'application/gzip', '.gz' : 'application/gzip',
'.js' : 'text/javascript' '.js' : 'text/javascript'
}[loadAsset.substring( fileExt )]; }[loadFile.substring( fileExt )];
/* /*
ejs is only for templates; if we want to load an binary data (like images); we must use fs.readFile ejs is only for templates; if we want to load an binary data (like images); we must use fs.readFile
*/ */
if ( fileMime !== 'text/html' ) if ( fileMime !== 'text/html' )
data = fs.readFileSync( './www/' + loadAsset ); data = fs.readFileSync( `./${ FOLDER_WWW }/${ loadFile }` );
response.setHeader( 'Content-type', fileMime ); response.setHeader( 'Content-type', fileMime );
response.end( data ); response.end( data );
Log.debug( `www`, chalk.green( ` [LOAD] ` ), chalk.white( `→` ), chalk.grey( `asset:${ loadAsset } mime:${ fileMime }` ) ); Log.debug( `www`, chalk.green( ` [LOAD] ` ), chalk.white( `→` ), chalk.grey( `asset:${ loadFile } mime:${ fileMime }` ) );
} }
else else
{ {