from fuzzywuzzy import process

import channel_util
import google_api
import slack_util
import identifier
import time

SHEET_ID = "1lPj9GjB00BuIq9GelOWh5GmiGsheLlowPnHLnWBvMOM"

# Note: These ranges use named range feature of google sheets.
# To edit range of jobs, edit the named range in Data -> Named Ranges in the Goole Sheets page
output_range = "JobScoreTracker"

MIN_RATIO = 0.9
SIGNOFF_REWARD = 0.1


# Used to track a sufficiently shitty typed name
class BadName(Exception):
    def __init__(self, name, score):
        self.name = name
        self.score = score

    def as_response(self):
        return "Unable to perform operation. Best name match {} had a match score of {}, falling short of match ratio " \
               "{}. Please type name better.".format(self.name, self.score, MIN_RATIO)


def get_curr_points():
    # Get the current stuff
    curr_output = google_api.get_sheet_range(SHEET_ID, output_range)

    # Each element: force length of 2.
    def row_fixer(r):
        if len(r) == 0:
            return ["", 0]
        elif len(r) == 1:
            return [r[0], 0]
        else:
            try:
                v = int(r[1])
            except ValueError:
                v = 0
            return [r[0], v]

    # Fix each row
    curr_output = [row_fixer(x) for x in curr_output]

    # Cut off the chaff - doesn't seem to be necessary.
    return curr_output


def put_points(vals):
    google_api.set_sheet_range(SHEET_ID, output_range, vals)


def signoff_callback(slack, msg, match):
    # Find the index of our person.
    name = match.group(1)

    # Also, who just signed us off?
    signer = identifier.lookup_msg_brother(msg)["name"]

    # Try giving the person a point
    try:
        bro_name, bro_total = adjust_score(name, 1)
        time.sleep(1)
        ass_name, ass_total = adjust_score(signer, SIGNOFF_REWARD)
        slack_util.reply(slack, msg, "Gave {} one housejob point.\n"
                                     "They now have {} for this period.\n"
                                     "You ({}) were credited with the signoff".format(bro_name, bro_total, ass_name))
    except BadName as e:
        # We didn't find a name - no action was performed.
        slack_util.reply(slack, msg, e.as_response())


def punish_callback(slack, msg, match):
    # Find the index of our person.
    name = match.group(2)

    # Also, who just signed us off?
    signer = identifier.lookup_msg_brother(msg)["name"]

    # Try giving the person a point
    try:
        bro_name, bro_total = adjust_score(name, -1)
        time.sleep(1)
        ass_name, ass_total = adjust_score(signer, -SIGNOFF_REWARD)
        slack_util.reply(slack, msg, "Took one housejob point from {}.\n"
                                     "They now have {} for this period.\n"
                                     "Under the assumption that this was to undo a mistake, we have deducted the "
                                     "usual signoff reward from you, ({}).\n "
                                     "You can easily earn it back by signing off the right person ;).".format(bro_name,
                                                                                                              bro_total,
                                                                                                              ass_name))
    except BadName as e:
        # We didn't find a name - no action was performed.
        slack_util.reply(slack, msg, e.as_response())


def adjust_score(name, delta):
    # Get the current stuff
    points = get_curr_points()
    names = [p[0] for p in points]

    # Find our guy
    target_name, ratio = process.extractOne(name, names)
    ratio = ratio / 100.0

    # If bad ratio, error
    if ratio < MIN_RATIO:
        raise BadName(target_name, ratio)

    # Where is he in the list?
    target_index = names.index(target_name)

    # Get his current score
    curr_score = points[target_index][1]

    # target should be in the form index, (name, score)
    target_new = [target_name, curr_score + delta]

    # Put it back
    points[target_index] = target_new
    put_points(points)

    # Return the adjusted name
    return target_new


def reset_callback(slack, msg, match):
    pass
    # reset_callback()


signoff_hook = slack_util.Hook(signoff_callback, pattern=r"signoff\s+(.*)",
                               channel_whitelist=[channel_util.HOUSEJOBS])
undosignoff_hook = slack_util.Hook(punish_callback, pattern=r"(unsignoff|undosignoff|undo)\s+(.*)",
                                   channel_whitelist=[channel_util.HOUSEJOBS])
# reset_hook = slack_util.Hook(reset_callback, pattern=r"reset_job_scores")