Home/streamlink-server
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Added support for proxy for each channel. Added rai 4,rai 5 and rai premium

Browse Source
This commit is contained in:
Stefano Fiorini
2025-02-28 21:03:26 -06:00
parent b333851e37
commit 3d1e65f17a
2 changed files with 31 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
src/stream_link_server.py
View File

@@ -35,7 +35,7 @@ def load_channels():
with open(channels_path, 'r') as f:
return {str(c['id']): c for c in json.load(f)['channels']}
async def generate_streamlink_process(url, headers=None) -> Process:
async def generate_streamlink_process(url, headers=None, proxy=None) -> Process:
"""
Run Streamlink as an async subprocess and pipe its output to the response.
Args:
@@ -53,6 +53,10 @@ async def generate_streamlink_process(url, headers=None) -> Process:
'--ringbuffer-size', '32M',
]
# Add proxy if specified
if proxy:
cmd.extend(['--http-proxy', proxy])
# Add headers if specified
if headers:
for key, value in headers.items():
@@ -67,14 +71,14 @@ async def generate_streamlink_process(url, headers=None) -> Process:
)
return process
async def stream_generator(process: Process, url: str, headers=None):
async def stream_generator(process: Process, url: str, headers=None, proxy=None):
"""Generate streaming content asynchronously"""
CHUNK_SIZE = 32768
try:
while True:
if process.returncode is not None:
# Process has terminated, restart it
process = await generate_streamlink_process(url, headers)
process = await generate_streamlink_process(url, headers, proxy)
continue
try:
@@ -90,7 +94,7 @@ async def stream_generator(process: Process, url: str, headers=None):
process.terminate()
except:
pass
process = await generate_streamlink_process(url, headers)
process = await generate_streamlink_process(url, headers, proxy)
finally:
try:
process.terminate()
@@ -123,10 +127,13 @@ async def stream_channel(channel_id: str, auth: bool = Depends(verify_credential
if 'referer' in channel:
headers['Referer'] = channel['referer']
# Get proxy if specified
proxy = channel.get('proxy')
try:
process = await generate_streamlink_process(url, headers if headers else None)
process = await generate_streamlink_process(url, headers if headers else None, proxy if proxy else None)
return StreamingResponse(
stream_generator(process, url, headers if headers else None),
stream_generator(process, url, headers if headers else None, proxy if proxy else None),
media_type='video/mp2t'
)
except Exception as e: