From 7f2a3ae4c9e7e6e8d1d271a39cf0cbd753d50156 Mon Sep 17 00:00:00 2001
From: localhost_frssoft <userquicker@disroot.org>
Date: Tue, 8 Nov 2022 21:02:07 +0300
Subject: [PATCH] mpv player menu

---
 src/mpv_control.py | 47 ++++++++++++++++++++++++++++++++++++++++++++++
 src/utils.py       |  6 +++---
 2 files changed, 50 insertions(+), 3 deletions(-)

diff --git a/src/mpv_control.py b/src/mpv_control.py
index 79b585b..a0756f6 100644
--- a/src/mpv_control.py
+++ b/src/mpv_control.py
@@ -1,6 +1,11 @@
+from src.utils import download_track
 from loguru import logger
+from pyfzf.pyfzf import FzfPrompt
 import mpv
 
+fzf = FzfPrompt()
+
+
 def mpv_log(loglevel, component, message):
     if loglevel == 'info':
         logger.info(f'{component} {message}')
@@ -9,3 +14,45 @@ def mpv_log(loglevel, component, message):
 
 player = mpv.MPV(log_handler=mpv_log, ytdl=False,
                  prefetch_playlist=True)
+
+
+def set_http_header(headers=[]):
+    player.http_header_fields = headers
+
+
+def player_menu(header=None, storage={}):
+    while True:
+        try:
+            select = fzf.prompt(('Next', 'Prev', 'Pause', 'Download', 'Info', 'Exit'), f"--header=\'{header}\'")[0]
+            if select == 'Next':
+                player.playlist_next()
+            elif select == 'Prev':
+                player.playlist_prev()
+            elif select == 'Pause':
+                if player.pause:
+                    player.pause = False
+                else:
+                    player.pause = True
+            elif select == 'Download':
+                print('Downloading...')
+                name_downloaded = download_track(player.filename)
+                print(f'Downloaded: {name_downloaded}')
+            elif select == 'Info':
+                track = storage.get(player.filename)
+                for i in ('title', 'fid', 'license', 'album', 'artist'):
+                    if i in ('album', 'artist'):
+                        name_aa = track.get(i).get('name')
+                        if not name_aa:
+                            name_aa = track.get(i).get('title')
+                        print(i + ': '+ name_aa)
+                    key = track.get(i)
+                    if key and isinstance(key, str):
+                        print(i + ': ' + key)
+                input()
+            elif select == 'Exit':
+                player.playlist_clear()
+                player.stop()
+                break
+        except KeyboardInterrupt:
+            break
+
diff --git a/src/utils.py b/src/utils.py
index 3adcb6f..36227d7 100644
--- a/src/utils.py
+++ b/src/utils.py
@@ -1,17 +1,17 @@
-from src.fw_api import s
+import src.fw_api
 from urllib.parse import unquote
 
 
 def get_remote_file_name(url):
     '''This function return filename by content-disposition header'''
-    r = s.head(url)
+    r = src.fw_api.s.head(url)
     content_dispos = r.headers.get('content-disposition')
     if content_dispos.startswith('attachment; filename*=UTF-8\'\''):
         return unquote(content_dispos.split('attachment; filename*=UTF-8\'\'')[-1])
 
 
 def download_track(url, name=None):
-    r = s.get(url)
+    r = src.fw_api.s.get(url)
     if not name:
         name = get_remote_file_name(url)
         if not name: