#!/usr/bin/env python3
"""Telegram bot that can search in the KJV"""
import asyncio
from subprocess import check_output
import sys

import telepot
from telepot.async.delegate import per_inline_from_id, create_open


def get_verses(ref):
    """Get text from a bible reference"""
    verses = check_output(['bible', '-f', ref]).decode('utf-8').splitlines()
    return [s.split(' ', 1) for s in verses]


def search(query):
    """Search for a query"""
    query = query.split()
    try:
        ctx = [s for s in query if s[0] == ':'][0][1:]
    except IndexError:
        ctx = ''
    query = [s for s in query if s[0] != ':']
    print(ctx, query)
    refs = check_output(['bible', '-f']+query).decode('utf-8').splitlines()[2:]
    return [get_verses(r) for r in refs if r.find(ctx) == 0]


class KJVInlineHandler(telepot.async.helper.UserHandler):
    """Inline handler for the KJV bot"""
    def __init__(self, seed_tuple, timeout):
        super(KJVInlineHandler, self).__init__(
            seed_tuple, timeout,
            flavors=['inline_query', 'chosen_inline_result'])

    @asyncio.coroutine
    def on_inline_query(self, msg):
        """Handle a query"""
        query_id, _, query_string = telepot.glance(msg, flavor='inline_query')

        query_string = query_string.strip()

        articles = []
        if not query_string == '':
            if not query_string[0] == '?':
                for [ref, txt] in get_verses(query_string):
                    articles.append({
                        'type': 'article',
                        'id': ref,
                        'title': ref,
                        'message_text': '%s: %s' % (ref, txt),
                        'description': txt
                    })
            else:
                for [[ref, txt]] in search(query_string):
                    articles.append({
                        'type': 'article',
                        'id': ref,
                        'title': ref,
                        'message_text': '%s: %s' % (ref, txt),
                        'description': txt
                    })

        yield from self.bot.answerInlineQuery(query_id, articles[:20], 300)

    def on_chosen_inline_result(self, msg):
        """We don't care about chosen inline results"""
        pass

    def on_close(self, *args, **kwargs):
        """Tear-down"""
        pass


def main():
    """Start the KJV bot using a token from the command line"""
    token = sys.argv[1]

    bot = telepot.async.DelegatorBot(token, [
        (per_inline_from_id(), create_open(KJVInlineHandler, timeout=10)),
    ])
    loop = asyncio.get_event_loop()

    loop.create_task(bot.message_loop())
    print('Listening...')

    loop.run_forever()

if __name__ == '__main__':
    main()