#!/usr/bin/env python3
# -*- coding: utf-8 -*-
# SPDX-FileCopyrightText: 2009,2010,2012,2019 kaliko <kaliko@azylum.org>
# SPDX-License-Identifier: GPL-3.0-or-later

import argparse
import sys

from random import choice
from os.path import basename

import musicpd


VERSION = '0.2'


class WakeUp(musicpd.MPDClient):
    script_info = dict({
        'prog': basename(__file__),
        'description': 'Queues and plays an album chosen at random from a list. Albums list provided from CLI or text file.',
        'epilog': 'Set MPD host/port in env. var'
    })

    def __init__(self):
        musicpd.MPDClient.__init__(self)
        self._run()

    def _get_args(self):
        parser = argparse.ArgumentParser(**self.__class__.script_info)
        parser.add_argument('--version', action='version',
                            version='v%s' % VERSION)
        parser.add_argument('-f', type=argparse.FileType('r'), dest='albumf',
                            help='text file, one album per line', nargs='?')
        parser.add_argument('album', type=str,
                            nargs='*', help='album name')
        args = parser.parse_args()
        if args.albumf:
            args.album.extend([line.rstrip(' \n')
                               for line in args.albumf.readlines()])
        if not args.album:
            # parser.print_help()
            print(
                'No album names found on CLI, run "{} -h" for usage'.format(sys.argv[0]))
            sys.exit(1)
        return args.album

    def _run(self):
        albums_list = self._get_args()
        album = choice(albums_list)
        print('Queueing "%s"' % album)
        self.connect()
        tracks = self.search('album', album)
        if not tracks:
            print('No tracks found for "%s"' % album)
            self.disconnect()
            sys.exit(1)
        self.searchadd('album', album)
        self.disconnect()
        sys.exit(0)


# Script starts here
if __name__ == '__main__':
    try:
        WakeUp()
    except KeyboardInterrupt:
        print('exit')

# VIM MODLINE
# vim: ai ts=4 sw=4 sts=4 expandtab