mirror of
https://gitea.phreedom.club/localhost_frssoft/funkwlmpv
synced 2024-04-29 09:09:19 +03:00
241 lines
8.6 KiB
Python
241 lines
8.6 KiB
Python
import src.fw_api
|
|
from src.utils import download_track, print_there, track_info_output, indices
|
|
from src.settings import get_config
|
|
from loguru import logger
|
|
from pyfzf.pyfzf import FzfPrompt
|
|
from shutil import get_terminal_size
|
|
from shlex import quote
|
|
from contextlib import suppress
|
|
import mpv
|
|
import time
|
|
import re
|
|
import requests
|
|
|
|
fzf = FzfPrompt()
|
|
|
|
if get_config('enable_persistent_cache'):
|
|
player = mpv.MPV(cache=True,
|
|
scripts='src/mpv_scripts/mpv_cache.lua:src/mpv_scripts/streamsave.lua',
|
|
script_opts='streamsave-save_directory=cache,streamsave-dump_mode=continuous,streansave-force_extension=.mkv,streamsave-autostart=no,output_label=overwrite')
|
|
player.command('script-message', 'streamsave-path', 'cache')
|
|
else:
|
|
player = mpv.MPV(cache=True, demuxer_max_bytes=25*1024*1024)
|
|
player.ytdl = False # Prevent attempts load track with yt-dlp
|
|
player.volume = get_config('mpv_volume')
|
|
player.prefetch_playlist = get_config('prefetch_playlist')
|
|
show_like_button = get_config('show_like_button')
|
|
share_to_fediverse_token = get_config('share_to_fediverse_token')
|
|
share_to_fediverse_instance = get_config('share_to_fediverse_instance')
|
|
shuffle = False
|
|
|
|
if get_config('termux_handle_track_switch_by_volume'):
|
|
import src.android_termux_api
|
|
|
|
|
|
class player_fw_storage:
|
|
storage = {}
|
|
menu_ctx = None
|
|
menu_ctx_args = None
|
|
|
|
|
|
@logger.catch
|
|
def track_url_to_uuid(listen_url=None):
|
|
'''Attempt get uuid from track listen url or current playing url'''
|
|
hex = '[0-9a-fA-F]+'
|
|
find_uuid = f'{hex}-{hex}-{hex}-{hex}-{hex}'
|
|
|
|
if listen_url:
|
|
uuid = re.findall(find_uuid, listen_url)
|
|
else:
|
|
uuid = re.findall(find_uuid, player.stream_open_filename)
|
|
|
|
return uuid[0]
|
|
|
|
|
|
def send_listen_activity():
|
|
try:
|
|
track = player_fw_storage.storage.get(track_url_to_uuid())
|
|
except:
|
|
return
|
|
if src.fw_api.current_instance.token is not None:
|
|
track_id = track.get('id')
|
|
|
|
if track_id:
|
|
src.fw_api.record_track_in_history(track_id)
|
|
else:
|
|
logger.error("Can't write track to history: No track id")
|
|
if share_to_fediverse_token != '':
|
|
fid = track.get('fid')
|
|
artist = track['artist'].get('name')
|
|
album = track['album'].get('title')
|
|
title = track.get('title')
|
|
tags = track.get('tags')
|
|
if tags:
|
|
tags = [f'#{tag}' for tag in tags]
|
|
tags = ' '.join(tags)
|
|
if tags == []:
|
|
tags = ''
|
|
status_obj = {'spoiler_text': 'funkwlmpv music share',
|
|
'visibility': 'unlisted',
|
|
'status': f'🎧 {artist} - {album} - {title}\n{fid}\n#NowPlaying {tags}'}
|
|
requests.post(f'https://{share_to_fediverse_instance}/api/v1/statuses',
|
|
json=status_obj,
|
|
headers={'Authorization': f'Bearer {share_to_fediverse_token}'})
|
|
|
|
|
|
def osd_observer(value):
|
|
'''Sumulate osd playing message in console'''
|
|
if value:
|
|
osd_message = []
|
|
for i in value.items():
|
|
if i[0] in ('Artist', 'Album', 'Title'):
|
|
osd_message.append(i[1])
|
|
osd_string = ' - '.join(osd_message)
|
|
term_len = get_terminal_size().columns
|
|
print_there(0, 0, '\r'+' '*term_len)
|
|
print_there(0, 0, '\r'+osd_string[:term_len])
|
|
else:
|
|
print_there(0, 0, '\rNo metadata...')
|
|
|
|
|
|
@player.event_callback('start-file')
|
|
@logger.catch
|
|
def starting_file_handler(value):
|
|
'''just show loading state'''
|
|
print_there(0, 0, '\rLoading track...')
|
|
|
|
|
|
@player.property_observer('percent-pos')
|
|
@logger.catch
|
|
def universal_observer(_name, value):
|
|
if value:
|
|
percent = int(value)
|
|
if player.audio_bitrate:
|
|
kbps = int(player.audio_bitrate/1024)
|
|
else:
|
|
kbps = '?'
|
|
if player.file_size:
|
|
track_size = round(player.file_size/1024/1024, 1)
|
|
else:
|
|
track_size = '?'
|
|
if player.cache_speed:
|
|
speed_load = player.cache_speed
|
|
if speed_load >= 3*1024*1024:
|
|
cache_speed = '| <<<'
|
|
elif speed_load >= 1*1024*1024:
|
|
cache_speed = '| <<*'
|
|
else:
|
|
cache_speed = '| <=>'
|
|
else:
|
|
cache_speed = ''
|
|
if player.playlist_count > -1:
|
|
player_pos = f'{player.playlist_pos_1}/{player.playlist_count}'
|
|
else:
|
|
player_pos = '-/-'
|
|
osd_observer(player.filtered_metadata)
|
|
print_there(2, 2, f'\r'+' '*get_terminal_size().columns)
|
|
print_there(2, 2, f'\r{player_pos} | {kbps} kbps | {percent}% | {track_size}MB {cache_speed}')
|
|
time.sleep(1)
|
|
|
|
|
|
def soft_volume_reduce():
|
|
while player.volume > 10:
|
|
player.volume = player.volume - 1
|
|
time.sleep(0.050)
|
|
|
|
|
|
@logger.catch
|
|
def player_menu(header='', storage={}):
|
|
for i in player.playlist_filenames:
|
|
count_same_tracks = indices(player.playlist_filenames, i)
|
|
while len(count_same_tracks) > 1:
|
|
with suppress(SystemError):
|
|
player.playlist_remove(count_same_tracks[-1])
|
|
count_same_tracks = indices(player.playlist_filenames, i)
|
|
player_fw_storage.storage.update(storage)
|
|
player.volume = get_config("mpv_volume")
|
|
global shuffle
|
|
while True:
|
|
try:
|
|
player_items_menu = ['Next', 'Prev', 'Pause',
|
|
'Shuffle', 'Download', 'Info', 'Share', 'Jump to']
|
|
if player.pause:
|
|
player_items_menu[2] = 'Play'
|
|
else:
|
|
player_items_menu[2] = 'Pause'
|
|
if shuffle:
|
|
player_items_menu[3] = 'Unshuffle'
|
|
else:
|
|
player_items_menu[3] = 'Shuffle'
|
|
if show_like_button:
|
|
player_items_menu.append('Like')
|
|
if player_fw_storage.menu_ctx:
|
|
player_items_menu.append('Add more tracks')
|
|
player_items_menu.extend(['Hide artist', 'Exit'])
|
|
|
|
select = fzf.prompt(player_items_menu, quote(f"--header=\'{header}\'"))
|
|
if select == []:
|
|
break
|
|
else:
|
|
select = select[0]
|
|
|
|
if select == 'Next':
|
|
try:
|
|
player.playlist_next()
|
|
except:
|
|
print('No more next tracks')
|
|
elif select == 'Prev':
|
|
player.playlist_prev()
|
|
elif select in ('Pause', 'Play'):
|
|
player.cycle('pause')
|
|
elif select in ('Shuffle', 'Unshuffle'):
|
|
if shuffle:
|
|
shuffle = False
|
|
player.playlist_unshuffle()
|
|
else:
|
|
shuffle = True
|
|
player.playlist_shuffle()
|
|
player.playlist_play_index(0)
|
|
elif select == 'Download':
|
|
name_downloaded = download_track(player.stream_open_filename)
|
|
elif select == 'Info':
|
|
track = player_fw_storage.storage.get(track_url_to_uuid())
|
|
track['direct_url'] = player.stream_open_filename
|
|
track_info_output(track)
|
|
elif select == 'Share':
|
|
send_listen_activity()
|
|
elif select == 'Jump to':
|
|
jump_to_idx = int(fzf.prompt(range(1, len(player.playlist_filenames)+1))[0])
|
|
jump_to_idx -= 1
|
|
player.playlist_play_index(jump_to_idx)
|
|
elif select == 'Like':
|
|
src.fw_api.favorite_track(
|
|
player_fw_storage.storage.get(track_url_to_uuid())['id'])
|
|
elif select == 'Add more tracks':
|
|
player_fw_storage.menu_ctx(*player_fw_storage.menu_ctx_args)
|
|
elif select == 'Hide artist':
|
|
track = player_fw_storage.storage.get(track_url_to_uuid())
|
|
player.playlist_remove('current')
|
|
src.fw_api.hide_content(
|
|
{'target': {'id': track.get('artist').get('id'), 'type': 'artist'}})
|
|
elif select == 'Exit':
|
|
shuffle = False
|
|
soft_volume_reduce()
|
|
player.playlist_clear()
|
|
player.stop()
|
|
player_fw_storage.storage = {}
|
|
break
|
|
except KeyboardInterrupt:
|
|
break
|
|
|
|
|
|
def play_track(track, multi=False):
|
|
listen_url = src.fw_api.get_audio_file(track['listen_url'], True)
|
|
player_fw_storage.storage[track_url_to_uuid(listen_url)] = track
|
|
if multi:
|
|
player.loadfile(listen_url, 'append-play')
|
|
else:
|
|
player.loadfile(listen_url, 'append-play')
|
|
track_name = track.get('title')
|
|
player_menu(f"{track_name} playing...", player_fw_storage.storage)
|