funkwlmpv/src/fw_radios.py

from src.fw_api import s, get_radios, post_radio_session, get_track_radio, list_libraries, favorite_track, get_audio_file
from src.fw_libraries import libraries
from src.fw_tags import list_tags
from src.utils import download_track
from src.mpv_control import player, track_url_to_uuid
from src.settings import get_config
from pyfzf.pyfzf import FzfPrompt
from loguru import logger
import threading
import time

fzf = FzfPrompt()

audio_info = {}

@logger.catch
def list_radios():
    radios = get_radios()
    count = radios.get('count')

    results = radios.get('results')
    view = []
    for i in results: 
        index = results.index(i)
        id_radio = i.get('id')
        name = i.get('name')
        descr = i.get('description')
        radio_option = name
        if descr and descr != "":
            radio_option += f' | {descr}'
        view.append(f'{index}.{radio_option}')
    if s.headers.get('Authorization'): # Radios avalaible only for auth user
        view.append('Favourites')
        view.append('Less listened')
    view.extend(['Tag', 'Random', 'Libraries', 'Users', 'Recently Added'])

    selected = fzf.prompt(view, f'--header \'Found {count} radios\' --read0', delimiter="\0")[0].split('.', 1)
    if 'Favourites' in selected:
        radio_load(id_radio, 'favorites', name='your favorites tracks')
    elif 'Tag' in selected:
        tag = list_tags()
        radio_load(type_radio='tag', name=f'by tag: {tag}', related_object=tag)
    elif 'Random' in selected:
        radio_load(id_radio, 'random', name='totally random')
    elif 'Libraries' in selected:
        id_radio, type_radio, name_radio, related_obj = libraries()
        radio_load(id_radio, type_radio, name_radio, related_obj)
    elif 'Users' in selected:
        libs = list_libraries()['results']
        libraries_listing = []
        for lib_i in libs:
            lib_name = lib_i.get('actor').get('full_username')
            if lib_name not in libraries_listing:
                libraries_listing.append(lib_name)
        libraries_listing.append('Custom')
        lib_addr = fzf.prompt(libraries_listing)[0]
        if lib_addr == 'Custom':
            print('Input remote user library (ex. nick@funkwhale.domain.example: ')
            lib_addr = input()
        radio_load(None, 'actor-content', lib_addr, lib_addr)
    elif 'Recently Added' in selected:
        radio_load(id_radio, 'recently-added', name='Newest content on the network')
    elif 'Less listened' in selected:
        radio_load(id_radio, 'less-listened', name="Less listened tracks")
    else:
        id_selected = selected[0]
        id_radio = results[int(id_selected)].get('id')
        name_radio = results[int(id_selected)].get('name')
        radio_load(id_radio, name=name_radio)


def radio_generator(radio_session_id):
    count_t = 0
    while radio_session_id != '':
        time.sleep(1)
        if not radio_event_gen.wait(0): 
            break
        count_t += 1
        if count_t >= 60:
            count_t = 0
            playlist_remaining = len(player.playlist) - player.playlist_current_pos
            if playlist_remaining <= 2: 
                radio_get_track(radio_session_id)
    logger.info('Radio generator stopped')

radio_event_gen = threading.Event()


@logger.catch
def radio_load(id_radio=None, type_radio='custom', name=None, related_object=None):
    show_like_button = get_config('show_like_button')
    player.volume = get_config('mpv_volume')
    requested_radio = {
            'custom_radio': id_radio,
            'radio_type': type_radio,
            'related_object_id': related_object
            }
    radio_session_id = post_radio_session(requested_radio).get('id')
    for i in range(0, 2):
        radio_get_track(radio_session_id)

    radio_event_gen.set()
    radio_task = threading.Thread(target=radio_generator, args=(radio_session_id,), daemon=True)
    radio_task.start()
    player_items_menu = ['Next', 'Prev', 'Pause', 'Download', 'Info'] 

    if show_like_button:
        player_items_menu.append('Like')
    player_items_menu.extend(['Hide artist', 'Exit'])
    while True:
        try:
            if player.pause:
                player_items_menu[2] = 'Play'
            else:
                player_items_menu[2] = 'Pause'
            select = fzf.prompt(player_items_menu, f"--header=\'Radio {name} playing...\'")[0]
            if select == 'Next':
                playlist_remaining = len(player.playlist) - player.playlist_current_pos
                if playlist_remaining <= 2:
                    threading.Thread(target=radio_get_track, args=(radio_session_id,), daemon=True).start()
                player.playlist_next()
            elif select == 'Prev':
                player.playlist_prev()
            elif select in ('Pause', 'Play'):
                if player.pause:
                    player.pause = False
                else:
                    player.pause = True
            elif select == 'Download':
                print('Downloading...')
                name_downloaded = download_track(player.stream_open_filename)
                print(f'Downloaded: {name_downloaded}')
            elif select == 'Info':
                track = audio_info.get(track_url_to_uuid())
                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[i]['title']
                        print(i + ': '+ name_aa)
                    key = track.get(i)
                    if key and isinstance(key, str):
                        print(i + ': ' + key)
                input()
            elif select == 'Like':
                favorite_track(audio_info.get(track_url_to_uuid())['id'])
            elif select == 'Exit':
                try:
                    radio_event_gen.clear()
                except:
                    logger.exception('Error stopping Thread radio generator')
                    pass
                player.playlist_clear()
                player.stop()
                break
        except: 
            try:
                radio_event_gen.clear()
            except:
                logger.exception('Error stopping Thread radio generator')
                pass
            player.playlist_clear()
            player.stop()
            logger.exception('Radio force stopped')
            break


def radio_get_track(radio_session_id):
    radio_context = get_track_radio({'session': radio_session_id})
    if not radio_context:
        return
    if isinstance(radio_context, str):
        logger.error(radio_context)
        if radio_context == "Radio doesn't have more candidates":
            radio_event_gen.clear()
        return
    if radio_context.get('error'):
        logger.error(radio_context.get('error'))
        return
    else:
        track = radio_context.get('track')
        listen_url = track['listen_url']
        audio_info[track_url_to_uuid(listen_url)] = track
        player.loadfile(get_audio_file(listen_url, listen_url=True), 'append-play')