from wikipedia import wikipedia
from FaustBot.Modules.PrivMsgObserverPrototype import PrivMsgObserverPrototype


class WikiObserver(PrivMsgObserverPrototype):
    @staticmethod
    def cmd():
        return [".w"]

    @staticmethod
    def help():
        return ".w <term> - fragt Wikipediaartikel zu <term> ab"

    def update_on_priv_msg(self, data, connection):

        if data['message'].find('.w ') == -1:
            return
        w = wikipedia.set_lang('de')
        q = data['message'].split(' ')
        query = ''
        for word in q:
            if word.strip() != '.w':
                query += word + ' '
        w = wikipedia.search(query)
        if w.__len__() == 0:
            connection.send_back(data['nick'] + ', ' +
                                'ich habe dazu keinen eintrag gefunden!',
                                 data)
            return
        try:
            page = wikipedia.WikipediaPage(w.pop(0))
        except wikipedia.DisambiguationError as error:
            print('disambiguation page')
            page = wikipedia.WikipediaPage(error.args[1][0])
        connection.send_back(data['nick'] + ' ' + page.url, data)
        index = 51 + page.summary[50:350].rfind('. ')
        if index == 50 or index > 230:
            index = page.summary[0:350].rfind(' ')
            connection.send_back(page.summary[0:index], data)
        else:
            connection.send_back(page.summary[0:index], data)