from app.utils.downloader import HttpStreamingDownloader
from app.utils.uploader import DiskChunkUploadBackend
from app.utils.youtube import YtDlpInfo


def download(url: str):
    upload_backend = DiskChunkUploadBackend("trash_holder")
    downloader = HttpStreamingDownloader(upload_backend)
    youtube = YtDlpInfo(url)
    video_headers, video_url = youtube.get_video_url("480p")
    video_name = youtube.title + ".mp4"
    audio_headers, audio_url = youtube.get_audio_url()
    audio_name = youtube.title + ".m4a"
    
    downloader.download(video_url, video_name, video_headers)
    downloader.download(audio_url, audio_name, audio_headers)


def main():
    url = "https://youtu.be/OSAOh4L41Wg"
    download(url)


if __name__ == "__main__":
    main()