feat: DOWNLOADER

1 files changed, 35 insertions, 2 deletions

diff --git a/yt_music/__yt_music__.py b/yt_music/__yt_music__.py
index 3f0d6ef..5e19c2d 100644
--- a/yt_music/__yt_music__.py
+++ b/yt_music/__yt_music__.py
@@ -1,6 +1,8 @@
 import sys
 import re
 import subprocess
+import platform
+import os
 
 import httpx
 import fzf
@@ -9,6 +11,7 @@ headers = {
     "User-Agent": "Mozilla/5.0 (X11; Linux x86_64; rv:109.0) Gecko/20100101 Firefox/115.0"
 }
 
+
 client = httpx.Client(headers=headers, timeout=None)
 
 base_url = "https://vid.puffyan.us"
@@ -42,6 +45,31 @@ def extract_video_id(video_title):
     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):
    
     args = [
@@ -80,12 +108,17 @@ def main():
     ch = fzf.fzf_prompt(opts)
     print(ch)
     idx = extract_video_id(ch)
-    play_ch = fzf.fzf_prompt(["play", "loop"])
+    play_ch = fzf.fzf_prompt(["play", "loop", "download"])
     try:
         if play_ch == "play":
             play(idx, ch)
-        else:
+        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)