from urllib2 import urlopen
from urllib import urlencode
from json import loads
from subprocess import Popen
import click

@click.command()
@click.option('-g', '--game', default='', help='Load streams for specific game')
@click.option('-l', '--limit', type=click.IntRange(1,100), default=10, help='Number of streams to load')
@click.option('-t', '--top', is_flag=True, help='Open top stream right away')
@click.option('-q', '--quality', type=click.Choice(['worst', 'low', 'high', 'medium', 'high', 'source', 'best']), default='high', help='Stream quality')
def main(game, limit, top, quality):
    params = urlencode({'limit': limit, 'game': game})
    streams = loads(urlopen("https://api.twitch.tv/kraken/streams?{}".format(params)).read())["streams"]
    if top:
        streamNum = 1
    else:
        click.echo("Top twitch.tv streams right now:")
        click.echo("\n".join(map(makeStreamInfoString, enumerate(streams))))
        click.confirm('Want to watch something?', abort=True)
        streamNum = click.prompt('Type stream number to start livestreamer', type=click.IntRange(1, len(streams)))
    Popen(['livestreamer', streams[int(streamNum) - 1]["channel"]["url"], quality])

def makeStreamInfoString(streamEnum):
    (i, stream) = streamEnum
    channelName = stream["channel"]["name"]
    channelStatus = stream["channel"]["status"]
    game = stream["game"]
    viewers = str(stream["viewers"])
    return u"[{:02d}] {} || {} viewers on {} || {}".format(
        i + 1,
        channelStatus,
        click.style(viewers, bold=True),
        click.style(channelName, bold=True),
        click.style(game, bold=True)
    )

if __name__ == '__main__':
    main()