#!/usr/bin/env python3
import argparse
import logging
import sys
import signal
import time
import gi
import json
gi.require_version('Playerctl', '2.0')
from gi.repository import Playerctl, GLib
import subprocess
from os.path import expanduser
home = expanduser("~")

logger = logging.getLogger(__name__)

playerVisible = False

def write_output(text, player_name):
    logger.info('Writing output')

    output = {'text': text.replace("&", "&"),
              'class': 'custom-' + player_name,
              'alt': player_name}

    sys.stdout.write(json.dumps(output) + '\n')
    sys.stdout.flush()


def on_play(player, status, manager):
    logger.info('Received new playback status')
    on_metadata(player, player.props.metadata, manager)

def on_metadata(player, metadata, manager):
    logger.info('Received new metadata')
    track_info = ''

    if player.props.player_name == 'spotify' and \
            'mpris:trackid' in metadata.keys() and \
            ':ad:' in player.props.metadata['mpris:trackid']:
        track_info = 'AD PLAYING'
    elif player.get_artist() != '' and player.get_title() != '':
        track_info = '{artist} - {title}'.format(artist=player.get_artist(),
                                                 title=player.get_title())
    else:
        track_info = player.get_title()

    if player.props.status != 'Playing' and track_info:
        track_info = ' ' + track_info
    write_output(track_info, player.props.player_name)


def update_song():
    global playerVisible
    while not playerVisible:
        result = subprocess.run("./capture_mp3.sh",stdout=subprocess.PIPE)
        resultString = result.stdout.decode("utf-8")
        logger.debug(resultString);
        write_output(resultString, "Spotify")
        time.sleep(6)

def on_player_appeared(manager, player, selected_player=None):
    global playerVisible
    if player is not None and (selected_player is None or player.name == selected_player):
        playerVisible = True
        init_player(manager, player)
    else:
        logger.debug("New player appeared, but it's not the selected player, skipping")


def on_player_vanished(manager, player):
    global playerVisible
    logger.info('Player has vanished')
    playerVisible = False
    update_song()
    sys.stdout.write('\n')
    sys.stdout.flush()


def init_player(manager, name):
    logger.debug('Initialize player: {player}'.format(player=name.name))
    player = Playerctl.Player.new_from_name(name)
    player.connect('playback-status', on_play, manager)
    player.connect('metadata', on_metadata, manager)
    update_song()
    manager.manage_player(player)
    on_metadata(player, player.props.metadata, manager)


def signal_handler(sig, frame):
    logger.debug('Received signal to stop, exiting')
    sys.stdout.write('\n')
    sys.stdout.flush()
    loop.quit()
    sys.exit(0)


def parse_arguments():
    parser = argparse.ArgumentParser()

    # Increase verbosity with every occurrence of -v
    parser.add_argument('-v', '--verbose', action='count', default=0)

    # Define for which player we're listening
    parser.add_argument('--player')

    return parser.parse_args()


def main():

    arguments = parse_arguments()

    # Initialize logging
    LOG = home+"/.dotfiles/configs/waybar/media.log"                                    logger.setLevel(max((3 - arguments.verbose) * 10, 0))
    logging.basicConfig(stream=sys.stderr, filename=LOG, level=logging.DEBUG,
                        format='%(name)s %(levelname)s %(message)s')

    logger.setLevel(0)

    # Log the sent command line arguments
    logger.debug('Arguments received {}'.format(vars(arguments)))

    manager = Playerctl.PlayerManager()
    loop = GLib.MainLoop()

    manager.connect('name-appeared', lambda *args: on_player_appeared(*args, arguments.player))
    manager.connect('player-vanished', on_player_vanished)

    signal.signal(signal.SIGINT, signal_handler)
    signal.signal(signal.SIGTERM, signal_handler)

    for player in manager.props.player_names:
        if arguments.player is not None and arguments.player != player.name:
            logger.debug('{player} is not the filtered player, skipping it'
                         .format(player=player.name)
                         )
            continue

        init_player(manager, player)

    update_song()
    loop.run()

if __name__ == '__main__':
    main()