import os
from datetime import datetime

import pandas as pd
from pytest import fixture
from socceraction.data.opta import (
    OptaCompetitionSchema,
    OptaGameSchema,
    OptaPlayerSchema,
    OptaTeamSchema,
)
from socceraction.data.opta.parsers import MA1JSONParser


@fixture()
def ma1json_parser() -> MA1JSONParser:
    path = os.path.join(
        os.path.dirname(__file__),
        os.pardir,
        os.pardir,
        os.pardir,
        "datasets",
        "opta",
        "ma1_408bfjw6uz5k19zk4am50ykmh.json",
    )
    return MA1JSONParser(str(path))


def test_extract_competitions(ma1json_parser: MA1JSONParser) -> None:
    competitions = ma1json_parser.extract_competitions()
    assert len(competitions) == 1
    assert competitions[("722fdbecxzcq9788l6jqclzlw", "408bfjw6uz5k19zk4am50ykmh")] == {
        "competition_id": "722fdbecxzcq9788l6jqclzlw",
        "season_id": "408bfjw6uz5k19zk4am50ykmh",
        "competition_name": "2. Bundesliga",
        "season_name": "2015/2016",
    }
    OptaCompetitionSchema.validate(pd.DataFrame.from_dict(competitions, orient="index"))


def test_extract_games(ma1json_parser: MA1JSONParser) -> None:
    games = ma1json_parser.extract_games()
    assert len(games) == 1
    assert games["bsu6pjne1eqz2hs8r3685vbhl"] == {
        "game_id": "bsu6pjne1eqz2hs8r3685vbhl",
        "season_id": "408bfjw6uz5k19zk4am50ykmh",
        "competition_id": "722fdbecxzcq9788l6jqclzlw",
        "game_day": 22,
        "game_date": datetime(2016, 2, 20, 12, 0),
        "home_team_id": "aojwbjr39s1w2mcd9l2bf2dhk",
        "away_team_id": "kxpw3rqn4ukt7nqmtjj62lbn",
        "venue": "BBBank Wildpark",
        "away_score": 2,
        "home_score": 2,
        "duration": 93,
        "attendance": 12746,
        "referee": "Robert Kampka",
    }
    OptaGameSchema.validate(pd.DataFrame.from_dict(games, orient="index"))


def test_extract_teams(ma1json_parser: MA1JSONParser) -> None:
    teams = ma1json_parser.extract_teams()
    assert len(teams) == 2
    assert teams["aojwbjr39s1w2mcd9l2bf2dhk"] == {
        "team_id": "aojwbjr39s1w2mcd9l2bf2dhk",
        "team_name": "Karlsruher SC",
    }
    OptaTeamSchema.validate(pd.DataFrame.from_dict(teams, orient="index"))


def test_extract_players(ma1json_parser: MA1JSONParser) -> None:
    players = ma1json_parser.extract_players()
    assert len(players) == 36
    assert players[("bsu6pjne1eqz2hs8r3685vbhl", "b40xhpgxf8cvruo6vumzu3u1h")] == {
        "game_id": "bsu6pjne1eqz2hs8r3685vbhl",
        "player_id": "b40xhpgxf8cvruo6vumzu3u1h",
        "player_name": "Enrico Valentini",
        "team_id": "aojwbjr39s1w2mcd9l2bf2dhk",
        "jersey_number": 22,
        "minutes_played": 93,
        "starting_position": "Defender",
        "is_starter": True,
    }
    # substitute player on
    assert (
        players[("bsu6pjne1eqz2hs8r3685vbhl", "49797zk0b4wmp4tevwmaeeh91")]["minutes_played"] == 57
    )
    # substitute player off
    assert players[("bsu6pjne1eqz2hs8r3685vbhl", "yuw4a34cpasw5e4vqsg6ex1x")][
        "minutes_played"
    ] == (93 - 57)
    # red card
    assert (
        players[("bsu6pjne1eqz2hs8r3685vbhl", "2175hvbfk4jn4lnj3cetfpp1")]["minutes_played"] == 60
    )
    OptaPlayerSchema.validate(pd.DataFrame.from_dict(players, orient="index"))