#!python
import argparse

from wtm._wtm import wtm

NAME = 'wtm'
VERSION = '1.0.0'

parser = argparse.ArgumentParser(prog=NAME, description='Downloads songs from the internet.')

parser.add_argument('-s', dest='song', help='song search query', nargs='+')
parser.add_argument('-t', '--top', help='selects top search result', const=True, action='store_const')
parser.add_argument('-v', '--version', action='version', version='%(prog)s {VERSION}'.format(VERSION=VERSION))

args = parser.parse_args()

if __name__ == '__main__':

    query = ''

    try:
        query = ' '.join(args.song) if args.song is not None else input('Search for songs : ')
    except KeyboardInterrupt:
        print('Bye.')
        quit(-1)

    msg, success = wtm(query, args.top)

    if not success:
        print('ERROR : ' + msg)
    else:
        print('Done. Goodbye.')
