from util import hook, timesince

import time
import re

db_ready = False

CAN_DOWNVOTE = False


def db_init(db):
    db.execute("""CREATE TABLE if not exists karma(
               nick_vote TEXT PRIMARY KEY,
               up_karma INTEGER,
               down_karma INTEGER,
               total_karma INTEGER)""")

    db.execute("""CREATE TABLE if not exists karma_voters(
               voter TEXT,
               votee TEXT,
               epoch FLOAT,
               PRIMARY KEY(voter, votee))""")
    db_ready = True


def up(db, nick_vote):
    db.execute("""UPDATE karma SET
               up_karma = up_karma + 1,
               total_karma = total_karma + 1 WHERE nick_vote=?""", (nick_vote.lower(),))
    db.commit()


def down(db, nick_vote):
    db.execute("""UPDATE karma SET
               down_karma = down_karma + 1,
               total_karma = total_karma + 1 WHERE nick_vote=?""", (nick_vote.lower(),))
    db.commit()


def allowed(db, nick, nick_vote):
    time_restriction = 3600
    db.execute("""DELETE FROM karma_voters WHERE ? - epoch >= 3600""",
            (time.time(),))
    db.commit()
    check = db.execute("""SELECT epoch FROM karma_voters WHERE voter=? AND votee=?""",
            (nick.lower(), nick_vote.lower())).fetchone()

    if check:
        check = check[0]
        if time.time() - check >= time_restriction:
            db.execute("""INSERT OR REPLACE INTO karma_voters(
                       voter,
                       votee,
                       epoch) values(?,?,?)""", (nick.lower(), nick_vote.lower(), time.time()))
            db.commit()
            return True, 0
        else:
            return False, timesince.timeuntil(check, now=time.time() - time_restriction)
    else:
        db.execute("""INSERT OR REPLACE INTO karma_voters(
                   voter,
                   votee,
                   epoch) values(?,?,?)""", (nick.lower(), nick_vote.lower(), time.time()))
        db.commit()
        return True, 0


# TODO Make this work on multiple matches in a string, right now it'll only
# work on one match.
# karma_re = ('((\S+)(\+\+|\-\-))+', re.I)
karma_re = ('(.+)(\+\+|\-\-)$', re.I)


@hook.regex(*karma_re)
def karma_add(match, nick='', chan='', db=None, notice=None):

    if not db_ready:
        db_init(db)

    nick_vote = match.group(1).strip().replace("+", "")
    if nick.lower() == nick_vote.lower():
        notice("You can't vote on yourself!")
        return
    if len(nick_vote) < 3 or " " in nick_vote:
        return  # ignore anything below 3 chars in length or with spaces

    vote_allowed, when = allowed(db, nick, nick_vote)
    if vote_allowed:
        if match.group(2) == '++':
            db.execute("""INSERT or IGNORE INTO karma(
                       nick_vote,
                       up_karma,
                       down_karma,
                       total_karma) values(?,?,?,?)""", (nick_vote.lower(), 0, 0, 0))
            up(db, nick_vote)
            notice("Gave {} 1 karma!".format(nick_vote))
        if match.group(2) == '--' and CAN_DOWNVOTE:
            db.execute("""INSERT or IGNORE INTO karma(
                       nick_vote,
                       up_karma,
                       down_karma,
                       total_karma) values(?,?,?,?)""", (nick_vote.lower(), 0, 0, 0))
            down(db, nick_vote)
            notice("Took away 1 karma from {}.".format(nick_vote))
        else:
            return
    else:
        notice("You are trying to vote too often. You can vote again in {}!".format(when))

    return


@hook.command('k')
@hook.command
def karma(inp, nick='', chan='', db=None):
    """k/karma <nick> -- returns karma stats for <nick>"""

    if not db_ready:
        db_init(db)

    if not chan.startswith('#'):
        return

    nick_vote = inp
    out = db.execute("""SELECT * FROM karma WHERE nick_vote=?""",
            (nick_vote.lower(),)).fetchall()

    if not out:
        return "That user has no karma."
    else:
        out = out[0]
        return "{} has {} karma points.".format(nick_vote, out[1] - out[2])

    return