import sys
import re
import subprocess
import platform
import os
import time

from pypresence import Presence
import httpx
import fzf

headers = {
    "User-Agent": "Mozilla/5.0 (X11; Linux x86_64; rv:109.0) Gecko/20100101 Firefox/115.0"
}


start = int(time.time())
client_id = "1075791459409723472"
RPC = Presence(client_id)
RPC.connect()

client = httpx.Client(headers=headers, timeout=None)

base_url = "https://vid.puffyan.us"
pattern = r'<a.*?href="/watch\?v=(.*?)".*?><p.*?>(.*?)<\/p></a>'

MPV_EXECUTABLE = "mpv"

try:
    if len(sys.argv) == 1:
        query = input("Search: ")
        if query == "":
            print("ValueError: no query parameter provided")
            exit(1)
    else:
        query = " ".join(sys.argv[1:])
except KeyboardInterrupt:
    exit(0)

query = query.replace(' ', '+')
opts = []


def extract_video_id(video_title):
    match = re.search(r' - ([\w-]+)$', video_title)
    
    #match = re.search(pattern, video_title)
    
    if match:
        video_id = match.group(1)
        return video_id
    else:
        return None

def determine_path() -> str:
    
    plt = platform.system()

    if plt == "Windows":
        return f"C:\\Users\\{os.getenv('username')}\\Downloads"
    
    elif (plt == "Linux"):
        return f"/home/{os.getlogin()}/Downloads"
    
    elif (plt == "Darwin"):
        return f"/Users/{os.getlogin()}/Downloads"

    else:
        print("[!] Make an issue for your OS.")
        exit(0)

def download(video_id, video_title):
    
    path: str = determine_path()
    video_title = video_title.replace(' ', '_')

    subprocess.call(f"yt-dlp -x \"https://music.youtube.com/watch?v={video_id}\" -o \"{path}/{video_title}\"", shell=True)


def play_loop(video_id, video_title):
   
    RPC.update(
        large_image = f"http://img.youtube.com/vi/{video_id}/0.jpg",
        large_text = "haha checkmate spotify plebs",
        small_image = "youtube_music_icon_svg",
        small_text = "yt-music",
        start = start,
        details = f"{video_title} - loop",
        buttons = [{"label": "Play on YouTube Music", "url": f"https://music.youtube.com/watch?v={video_id}"}],
    )

    args = [
        MPV_EXECUTABLE,
        f"https://music.youtube.com/watch?v={video_id}",
        f"--force-media-title={video_title}",
        "--no-video",
        "--loop",
    ]

    mpv_process = subprocess.Popen(args, stdout=subprocess.DEVNULL)
    mpv_process.wait()



def play(video_id, video_title):
   
    RPC.update(
        large_image = f"http://img.youtube.com/vi/{video_id}/0.jpg",
        large_text = "haha checkmate spotify plebs",
        small_image = "youtube_music_icon_svg",
        small_text = "yt-music",
        start = start,
        details = video_title,
        buttons = [{"label": "Play on YouTube Music", "url": f"https://music.youtube.com/watch?v={video_id}"}],
    )

    args = [
        MPV_EXECUTABLE,
        f"https://music.youtube.com/watch?v={video_id}",
        f"--force-media-title={video_title}",
        "--no-video",
    ]

    mpv_process = subprocess.Popen(args, stdout=subprocess.DEVNULL)
    mpv_process.wait()


def main():
    fetch = client.get(f"{base_url}/search?q={query}")
    matches = re.findall(pattern, fetch.text)
    for match in matches:
        video_id,  title = match
        opt = f"{title} - {video_id}"
        opts.append(opt)
    ch = fzf.fzf_prompt(opts)
    print(ch)
    idx = extract_video_id(ch)
    play_ch = fzf.fzf_prompt(["play", "loop", "download"])
    try:
        if play_ch == "play":
            play(idx, ch)
        elif play_ch == "loop":
            play_loop(idx, ch)
        elif play_ch == "download":
            download(idx, ch)
        else:
            print("[!] Nothing selected.")
            exit(1)
    except KeyboardInterrupt:
        exit(0)

main()