###
# Copyright (c) 2019, Pedro de Oliveira
# All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions are met:
#
#   * Redistributions of source code must retain the above copyright notice,
#     this list of conditions, and the following disclaimer.
#   * Redistributions in binary form must reproduce the above copyright notice,
#     this list of conditions, and the following disclaimer in the
#     documentation and/or other materials provided with the distribution.
#   * Neither the name of the author of this software nor the name of
#     contributors to this software may be used to endorse or promote products
#     derived from this software without specific prior written consent.
#
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
# ARE DISCLAIMED.  IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
# LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
# SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
# CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
# POSSIBILITY OF SUCH DAMAGE.

###

from supybot import utils, plugins, ircutils, callbacks
from supybot.commands import *
try:
    from supybot.i18n import PluginInternationalization
    _ = PluginInternationalization('Calao')
except ImportError:
    # Placeholder that allows to run the plugin on a bot
    # without the i18n module
    _ = lambda x: x
import sqlite3
import json
import os
import re

class Calao(callbacks.Plugin):
    """Dicionario de Calão"""
    threaded = True

    def __init__(self, irc):
        self.__parent = super(Calao, self)
        self.__parent.__init__(irc)
        self.prepend = "1,13CALÃO"

    def __highlight(self, term, highlight):
        regex = r"(" + highlight + ")"
        subst = "\\1"
        result = re.sub(regex, subst, term, 0)
        if result:
            return result
        else:
            return term

    def __format_message(self, record, position = False, total = False, highlight = False):
        term = record[1]
        message = "{} ".format(self.prepend)
        if position is not False and total is not False:
            message += '[{}/{}] '.format(position, total)
        if highlight:
            message += "13{}".format(self.__highlight(term, highlight))
        else:
            message += "13{}".format(term)
        definition = record[2]
        if definition:
            definition = json.loads(definition)
            if 'definition' in definition.keys():
                message += ' {}'.format(definition['definition'])
                del definition['definition']
            for d in definition:
                message += ' {}: {}'.format(d, definition[d])
        return message

    def __get_connection(self):
        dir = os.path.dirname(__file__)
        calaodb = os.path.join(dir, "./calao.db")
        return sqlite3.connect(calaodb)
    
    def random(self, irc, msg, args):
        """takes no arguments

        Returns a random definition.
        """
        conn = self.__get_connection()
        cursor = conn.cursor()
        cursor.execute("select * from dicionario order by random() limit 1")
        result = cursor.fetchone()
        conn.close()
        irc.reply(self.__format_message(result), prefixNick=False)
    random = wrap(random)

    def term(self, irc, msg, args, term):
        """<term>

        Returns the definition of <term>.
        Requires you to use "" if the term has spaces.
        """
        message = 'Not found'
        conn = self.__get_connection()
        cursor = conn.cursor()
        cursor.execute("select * from dicionario where term = ?", (term,))
        result = cursor.fetchone()
        conn.close()
        if result is None:
            irc.reply(message, prefixNick=False)
        else:
            irc.reply(self.__format_message(result), prefixNick=False)
    term = wrap(term, ['anything'])


    def find(self, irc, msg, args, text, position):
        """<text> [position]

        Returns the matched term with <text>, at [position].
        Requires you to use "" if the term has spaces.
        """
        message = 'Not found'
        conn = self.__get_connection()
        cursor = conn.cursor()
        cursor.execute("select count(1) from dicionario where term like ?", ('%'+text+'%',))
        total = cursor.fetchone()[0]

        if position < 1:
            position = 1

        if total > 0:
            total + 1
            if position > total:
                position = total
        cursor.execute("select * from dicionario where term like ? order by term limit ?,1", ('%'+text+'%', position - 1))
        result = cursor.fetchone()
        conn.close()
        if result is None:
            irc.reply(message, prefixNick=False)
        else:
            irc.reply(self.__format_message(result, position, total, text), prefixNick=False)
    find = wrap(find, ['anything', optional('int', default=1)])

Class = Calao


# vim:set shiftwidth=4 softtabstop=4 expandtab textwidth=79: