import os
import time
import subprocess
import random
from multiprocessing import Process
import uvicorn
from fastapi import FastAPI, Response
import py7zr

app = FastAPI()

# Carpeta de videos y rutas de salida
video_folder = "Los Simpsons"
hls_output = "stream.m3u8"
hls_segment_prefix = "stream"
concat_file = "concat.txt"

# Función para descargar y descomprimir el archivo 7z
def download_and_extract_videos():
    video_file = "Los Simpsons.7z"
    download_path = os.path.join(video_folder, video_file)
    extract_path = os.path.join(video_folder)
    subprocess.run(["wget", "https://huggingface.co/spaces/lozanogamer/lozanogamers/resolve/main/Los%20Simpsons.7z?download=true", "-P", video_folder])
    with py7zr.SevenZipFile(download_path, 'r') as archive:
        archive.extractall(path=extract_path)
    os.remove(download_path)

# Función para generar un archivo de concatenación de videos
def generate_concat_file():
    video_files = os.listdir(video_folder)
    with open(concat_file, "w") as f:
        for video_file in video_files:
            if video_file.endswith(".mp4"):
                f.write(f"file '{os.path.join(video_folder, video_file)}'\n")

# Comando FFmpeg para transmitir video en HLS
def build_ffmpeg_command():
    return [
        'ffmpeg',
        '-re',
        '-f', 'concat',
        '-safe', '0',
        '-i', concat_file,
        '-c:v', 'libx264',
        '-c:a', 'aac',
        '-b:v', '2000k',
        '-b:a', '128k',
        '-f', 'hls',
        '-hls_time', '10',
        '-hls_list_size', '0',
        '-hls_flags', 'delete_segments',
        '-hls_segment_filename', f'{hls_segment_prefix}%d.ts',
        hls_output
    ]

def stream_video():
    # Descarga y descomprime el archivo de video
    download_and_extract_videos()

    # Ejecuta el comando FFmpeg para iniciar la transmisión
    while True:
        generate_concat_file()
        ffmpeg_command = build_ffmpeg_command()
        process = subprocess.Popen(ffmpeg_command)
        process.wait()

def start_video_streaming():
    video_process = Process(target=stream_video)
    video_process.start()

@app.on_event("startup")
async def startup_event():
    # Espera activa para verificar que el archivo M3U8 esté generado
    while not os.path.exists(hls_output):
        print("Esperando a que se genere el archivo M3U8...")
        time.sleep(1)
    print("Archivo M3U8 generado. Servidor listo para recibir solicitudes.")

@app.get("/")
async def read_root():
    return {"message": "Servidor de Streaming de Videos M3U8 Activo"}

@app.get("/stream.m3u8")
def get_m3u8():
    # Devuelve el archivo M3U8 generado
    with open(hls_output, "rb") as f:
        m3u8_content = f.read()
    return Response(content=m3u8_content, media_type="application/vnd.apple.mpegurl")

@app.get("/stream{segment}.ts")
def get_segment(segment: int):
    # Devuelve los segmentos TS generados
    segment_file = f"{hls_segment_prefix}{segment}.ts"
    if os.path.exists(segment_file):
        with open(segment_file, "rb") as f:
            segment_content = f.read()
        return Response(content=segment_content, media_type="video/mp2t")
    else:
        return Response(status_code=404, content="Segment not found")

if __name__ == "__main__":
    start_video_streaming()
    uvicorn.run(app, host="0.0.0.0", port=7860)