import re
from common import get_requests_session
from pathlib import Path

OVERALL_LEADERBOARD_RANKS = [
    58, # Day 1
    16, # Day 2
    6, # Day 3
    16, # Day 4
    23, # Day 5
    29, # Day 6
    35, # Day 7
    49, # Day 8
    57, # Day 9
    65, # Day 10
    69, # Day 11
    78, # Day 12
    89, # Day 13
    78, # Day 14
    73, # Day 15
    84, # Day 16
    91, # Day 17
    98, # Day 18
    100, # Day 19
    None, # Day 20
    98, # Day 21
    None, # Day 22
    84, # Day 23
    75, # Day 24
    63, # Day 25
]

def main() -> None:
    self_leaderboard_response = get_requests_session().get("https://adventofcode.com/2023/leaderboard/self")
    self_leaderboard_response.raise_for_status()

    rows = ""
    overall_score = 0

    matches = re.search(r"<span class=\"leaderboard-daydesc-both\">    Time  Rank  Score</span>(.*)</pre>", self_leaderboard_response.text, flags=re.DOTALL)
    for line in reversed(matches.group(1).splitlines()):
        line = line.strip()
        if len(line) == 0:
            continue

        parts = [part for part in line.split(" ") if len(part) > 0]
        day, part1_time, part1_rank, part1_score, part2_time, part2_rank, part2_score = parts

        day = int(parts[0])
        part1_time = parts[1]
        part1_rank = int(parts[2])
        part1_score = int(parts[3])
        part2_time = parts[4]
        part2_rank = int(parts[5])
        part2_score = int(parts[6])

        overall_rank = OVERALL_LEADERBOARD_RANKS[day - 1] or ">100"
        overall_score += part1_score + part2_score

        rows += f"""
        <tr>
            <td>
                <a href="https://adventofcode.com/2023/day/{day}">Day {day}</a>
                (<a href="https://github.com/jmerle/advent-of-code-2023/tree/master/src/day{day:02}">code</a>)
            </td>
            <td>{part1_time}</td>
            <td>{part1_rank}</td>
            <td>{part1_score}</td>
            <td>{part2_time}</td>
            <td>{part2_rank}</td>
            <td>{part2_score}</td>
            <td>{overall_rank}</td>
            <td>{overall_score}</td>
        </tr>"""

    table = f"""
<!-- results-start -->
<table>
    <thead>
        <tr>
            <th></th>
            <th colspan="3">Part 1</th>
            <th colspan="3">Part 2</th>
            <th colspan="2">Overall leaderboard</th>
        </tr>
        <tr>
            <th></th>
            <th>Time</th>
            <th>Rank</th>
            <th>Score</th>
            <th>Time</th>
            <th>Rank</th>
            <th>Score</th>
            <th>Rank</th>
            <th>Score</th>
        </tr>
    </thead>
    <tbody>
        {rows.lstrip()}
    </tbody>
</table>
<!-- results-end -->
    """.strip()

    readme_file = Path(__file__).parent.parent / "README.md"
    readme_content = readme_file.read_text(encoding="utf-8")

    readme_content = re.sub(r"<!-- results-start -->(.*)<!-- results-end -->", table, readme_content, flags=re.DOTALL)

    readme_file.write_text(readme_content, encoding="utf-8")

    print(f"Successfully updated the results table in the readme")

if __name__ == "__main__":
    main()