from db import cursor
from ruleset import resolve_ruleset


def outcome(outcome_str, player_1):
    outcome_int = {"w": 1, "l": -1, "d": 0}[outcome_str]
    if not player_1: outcome_int *= -1
    return {1: "w", -1: "l", 0: "d"}[outcome_int]


def rating_v1(won_games):  # python is a garbage language
    return abs(won_games)**(1 / 2.2) * 23 * (1, -1)[won_games < 0]


def get_all_games(user_id):
    return cursor.execute("select player_1_id, player_2_id, outcome, ruleset " + \
            "from games " + \
            "where (player_1_id = ? or player_2_id = ?) " + \
            "and status = \"finished\" or status = \"resign\"", [user_id, user_id]).fetchall()


    # simple rating function that doesn't use game analysis
def get_rating(user_id):
    score = 400
    games = get_all_games(user_id)
    # get all games for user_id and switch perspective in which user_id is player_2_id
    mapped_games = [
        game if game[0] == user_id else
        (game[1], game[0], outcome(game[2], False), game[3]) for game in games
    ]

    counted_opponents = {}
    for game in mapped_games:
        ruleset = resolve_ruleset(game[3])

        if ruleset["ranked"] == False:
            continue

        # calculate sum score against user (+1 for win, -1 for lose, 0 for draw game)
        counted_opponents[game[1]] = (counted_opponents.get(game[1]) or 0) + {
            "w": 1,
            "l": -1,
            "d": 0
        }[game[2]]

    for opponent in counted_opponents:
        # apply the cool curve to the sum score and add to the base score of 400
        score += rating_v1(counted_opponents.get(opponent))

    return int(score)