#IMPORTS
import os
import subprocess
import getpass
import keyboard

#VARIABLES
user_path = os.environ.get('USERPROFILE')
location = f"{user_path}\\Documents\\ffmpeg"
loopVar = True

#MAIN LOOP
while loopVar == True:
    print(f"---------- DOWNLOADER V 1.4 ----------\n\n1 - Video\n2 - Audio\n3 - Update\n4 - Exit\n")
    
    try:
        mode = int(input("Select Mode: "))
    except ValueError:
        print(f"\n----------------------------------------------------------------------------------------------------\n")
        continue

    if mode == 1:
        while True:
            videoLink = str(input("Enter video link: "))
            if videoLink == "Exit" or videoLink == "exit":
                break
            else:
                downloadVideoCommand = f"yt-dlp --ffmpeg {location}\\bin --merge-output-format mp4 {videoLink}"
                print(f"\nCommand: {downloadVideoCommand}\n")
                subprocess.run(downloadVideoCommand, shell=True, cwd=location)
    elif mode == 2:
        while True:
            videoLink = str(input("Enter video link: "))
            if videoLink == "Exit" or videoLink == "exit":
                break
            else:
                downloadAudioCommand = f"yt-dlp --ffmpeg {location}\\bin --extract-audio --audio-format mp3 --audio-quality 0 {videoLink}"
                print(f"\nCommand: {downloadAudioCommand}\n")
                subprocess.run(downloadAudioCommand, shell=True, cwd=location)
    elif mode == 3:
        downloadUpdateCommand = f"yt-dlp -U"
        print(f"\nCommand: {downloadUpdateCommand}\n")
        subprocess.run(downloadUpdateCommand, shell=True, cwd=location)
    elif mode == 4:
        subprocess.Popen(f'explorer "{location}"')
        loopVar = False
    else:
        loopVar = False
    
#VERSION 1.5 - Added a loop to allow multiple downloads without restarting the script.
#move [File_name] [Destination]
#Do --cookies-from-browser to use cookies from a browser option.
#Repair printing of commands.