# /// script
# dependencies = ["requests", "basic-svg-chart>=0.2.4"]
# ///

import sys
import statistics
import os

os.environ["SVG_CHART_DOMINANT_BASELINE_FIX"] = "True"

import requests
import basic_svg_chart as bsc

def moyenne(x:list[int], y: list[float], debut: int, fin: int) -> list[float]:
    valeurs = y[x.index(debut):x.index(fin) + 1]
    return [None] * x.index(debut) + [statistics.mean(valeurs)] * len(valeurs)

def get_data() -> (list[str], list[float]):
    # Source Insee : https://catalogue-donnees.insee.fr/fr/catalogue/recherche/DD_CNA_AGREGATS
    url = "https://api.insee.fr/melodi/data/DD_CNA_AGREGATS?UNIT_MEASURE=PT&STO=B1GQ&PRICES=L"
    r = requests.get(url)
    data = r.json()

    x = [int(d["dimensions"]["TIME_PERIOD"]) for d in data["observations"]]
    y = [d["measures"]["OBS_VALUE_NIVEAU"]["value"] for d in data["observations"]]

    # Tri par année croissante
    ordre = [x.index(annee) for annee in sorted(x)]
    x = [x[ordre[i]] for i in range(len(x))]
    y = [y[ordre[i]] for i in range(len(y))]

    # Moyennes
    valeurs = (y, moyenne(x, y, 1950, 1974), moyenne(x, y, 1975, 1989), moyenne(x, y, 1990, 2007), moyenne(x, y, 2008, 2025))
    names = ["Croissance du PIB en volume", "Moyenne 1950-1974", "Moyenne 1975-1989", "Moyenne 1990-2007", "Moyenne 2008-2025"]
    names = [names[i] if i == 0 else f"{names[i]} ({valeurs[i][-1]:+.1f} %)".replace(".", ",") for i in range(len(names))]

    for i in range(1, len(names)):
        names[i]

    return x, valeurs, names

def get_chart(output: str):
    x, y, names = get_data()

    bsc.chart(
        chart_type = ("bar", "line"),
        x = x,
        y = y,
        output = output,
        names = names,
        x_ticks = range(1950, 2026, 5),
        y_min = -8,
        y_max = 9,
        y_ticks_interval = 1,
        y_unit = "%",
        tooltip_type = None,
        legend_position = "bottom left",
        legend_background = "white",
        colors = ("dodgerblue", "red", "orange", "violet", "limegreen"),
        width = 1000,
        height = 700,
        font_size = 20,
        lang = "fr",
        title = "Croissance du PIB français",
        desc = "Taux d'évolution annuelle du PIB français en volume entre 1950 et 2025, et moyennes sur les périodes 1950-1974, 1975-1989, 1990-2007 et 2008-2025",
        notes = "Source : <a href='https://www.insee.fr/fr/statistiques/8988797?sommaire=8988934'>Insee, comptes nationaux annuels - base 2020</a>.",
        css = (
            "path {stroke-width: 4px;}",
            ":root {font-family: sans-serif;}",
            "text.title {font-size: 1.3em;}",
            "a {fill: dodgerblue;}",
        ),
        extra_lines = {"y": 0},
        pretty = True,
    )
    
if __name__ == "__main__":
    if len(sys.argv) == 2:
        output = sys.argv[1]
    else:
        output = "croissance_PIB.svg"
    
    get_chart(output)