"""Стартовые nutrients для компонентов без EAV (демо-шаблоны, не лабораторный анализ).

При старте и пересчёте рациона заполняет только пустые `lab_component_nutrient_value`.
Маркер в data/ — журнал последнего прогона, не блокирует новые компоненты.
"""

from __future__ import annotations

import logging
from pathlib import Path
from typing import Any

from app import db
from config import Config
from app.lab.nutrient_schema import read_from_mapping
from app.lab.services.component_nutrients import nutrients_full_dict, nutrients_is_empty, save_component_nutrients
from app.models.component import Component

_log = logging.getLogger(__name__)

_MARKER_NAME = ".lab_demo_nutrients_seeded"

# Доп. поля для полной матрицы рациона (демо; заменить лабораторными при наличии).
_MINERAL_SUPPLEMENT: dict[str, float] = {
    "Ca": 8.0,
    "P": 5.5,
    "Mg": 3.2,
    "Na": 1.2,
    "K": 6.5,
    "DCAB Форм": 45.0,
    "Сырая зола": 55.0,
    "Каротин": 5.0,
}

_CARB_SUPPLEMENT: dict[str, float] = {
    "Сахар": 80.0,
    "Крахмал": 280.0,
    "Сахар и Крохм": 420.0,
    "Нераств. Крохмал": 35.0,
}

_ROUGHAGE_TYPES = frozenset({"объемные корма", "грубые корма", "сочные корма"})

_SUPPLEMENT_KEYS: tuple[str, ...] = tuple({**_MINERAL_SUPPLEMENT, **_CARB_SUPPLEMENT}.keys())

_NAMED_SEEDS: dict[str, dict[str, Any]] = {
    "комбикорм": {
        "dry_matter": 88.0,
        "nutrients": {
            "Сыр. Протеин": 170,
            "уСП": 142,
            "БРА": 36,
            "ОЭ-КРС": 12.5,
            "ЧЭЛ- КРС": 7.8,
            "Сырая клетчатка": 95,
            "Структур. клетчатка": 28,
            "Сырой жир": 38,
        },
    },
    "жом свекловичный гранулы": {
        "dry_matter": 90.0,
        "nutrients": {
            "Сыр. Протеин": 105,
            "уСП": 88,
            "ОЭ-КРС": 11.8,
            "ЧЭЛ- КРС": 7.2,
            "Сырая клетчатка": 220,
            "Структур. клетчатка": 48,
            "Сырой жир": 12,
        },
    },
    "жом": {
        "dry_matter": 90.0,
        "nutrients": {
            "Сыр. Протеин": 105,
            "уСП": 88,
            "ОЭ-КРС": 11.8,
            "ЧЭЛ- КРС": 7.2,
            "Сырая клетчатка": 220,
            "Структур. клетчатка": 48,
            "Сырой жир": 12,
        },
    },
    "дробина": {
        "dry_matter": 91.0,
        "nutrients": {
            "Сыр. Протеин": 100,
            "уСП": 88,
            "ОЭ-КРС": 11.5,
            "ЧЭЛ- КРС": 7.0,
            "Сырая клетчатка": 200,
            "Структур. клетчатка": 40,
            "Сырой жир": 12,
        },
    },
    "добавки": {
        "dry_matter": 99.0,
        "nutrients": {
            "Сыр. Протеин": 8,
            "ОЭ-КРС": 2.0,
            "ЧЭЛ- КРС": 1.2,
            "Сырая клетчатка": 5,
            "Сырой жир": 3,
        },
    },
    "сено": {
        "dry_matter": 85.0,
        "nutrients": {
            "Сыр. Протеин": 78,
            "уСП": 62,
            "ОЭ-КРС": 5.9,
            "ЧЭЛ- КРС": 3.4,
            "Сырая клетчатка": 328,
            "Структур. клетчатка": 265,
            "Сырой жир": 22,
        },
    },
    "анионовые соли": {
        "dry_matter": 95.0,
        "nutrients": {
            "Сыр. Протеин": 12,
            "ОЭ-КРС": 1.2,
            "ЧЭЛ- КРС": 0.8,
            "Сырая клетчатка": 8,
            "Сырой жир": 2,
        },
    },
}

_TYPE_DEFAULTS: dict[str, dict[str, Any]] = {
    "зерновые": {
        "dry_matter": 88.0,
        "nutrients": {
            "Сыр. Протеин": 120,
            "уСП": 105,
            "БРА": 28,
            "ОЭ-КРС": 11.0,
            "ЧЭЛ- КРС": 6.8,
            "Сырая клетчатка": 85,
            "Структур. клетчатка": 22,
            "Сырой жир": 35,
        },
    },
    "белковые": {
        "dry_matter": 90.0,
        "nutrients": {
            "Сыр. Протеин": 380,
            "уСП": 300,
            "БРА": 80,
            "ОЭ-КРС": 11.5,
            "ЧЭЛ- КРС": 7.0,
            "Сырая клетчатка": 120,
            "Сырой жир": 45,
        },
    },
    "минеральные": {
        "dry_matter": 95.0,
        "nutrients": {
            "Сыр. Протеин": 12,
            "ОЭ-КРС": 1.2,
            "ЧЭЛ- КРС": 0.8,
            "Сырая клетчатка": 8,
            "Сырой жир": 2,
        },
    },
    "витаминные": {
        "dry_matter": 99.0,
        "nutrients": {
            "Сыр. Протеин": 8,
            "ОЭ-КРС": 2.0,
            "ЧЭЛ- КРС": 1.2,
            "Сырая клетчатка": 5,
            "Сырой жир": 3,
        },
    },
    "энергетические": {
        "dry_matter": 90.0,
        "nutrients": {
            "Сыр. Протеин": 100,
            "уСП": 88,
            "ОЭ-КРС": 11.5,
            "ЧЭЛ- КРС": 7.0,
            "Сырая клетчатка": 200,
            "Структур. клетчатка": 40,
            "Сырой жир": 12,
        },
    },
    "объемные корма": {
        "dry_matter": 85.0,
        "nutrients": {
            "Сыр. Протеин": 78,
            "уСП": 62,
            "ОЭ-КРС": 5.9,
            "ЧЭЛ- КРС": 3.4,
            "Сырая клетчатка": 328,
            "Структур. клетчатка": 265,
            "Сырой жир": 22,
        },
    },
}

_FALLBACK = {
    "dry_matter": 88.0,
    "nutrients": {
        "Сыр. Протеин": 100,
        "уСП": 85,
        "ОЭ-КРС": 10.0,
        "ЧЭЛ- КРС": 6.2,
        "Сырая клетчатка": 150,
        "Сырой жир": 25,
    },
}


def _normalize_name(name: str | None) -> str:
    return (name or "").strip().lower()


def _merge_supplement(nutrients: dict[str, Any], component: Component) -> dict[str, float]:
    merged = dict(_MINERAL_SUPPLEMENT)
    type_key = _normalize_name(component.type)
    if type_key not in _ROUGHAGE_TYPES:
        merged.update(_CARB_SUPPLEMENT)
    merged.update(nutrients)
    return {k: float(v) for k, v in merged.items() if v is not None}


def _payload_for_component(component: Component) -> dict[str, Any]:
    name_key = _normalize_name(component.name)
    if name_key in _NAMED_SEEDS:
        payload = _NAMED_SEEDS[name_key]
    else:
        payload = None
        for key in sorted(_NAMED_SEEDS, key=len, reverse=True):
            if key in name_key:
                payload = _NAMED_SEEDS[key]
                break
        if payload is None:
            type_key = _normalize_name(component.type)
            payload = _TYPE_DEFAULTS.get(type_key, _FALLBACK)
    return {
        "dry_matter": payload.get("dry_matter"),
        "nutrients": _merge_supplement(payload.get("nutrients") or {}, component),
    }


def _apply_demo_payload(component: Component, payload: dict[str, Any], *, user_id: str) -> None:
    if payload.get("dry_matter") is not None:
        dm = float(payload["dry_matter"])
        current_dm = float(component.dry_matter or 0)
        if dm > 0 and (current_dm <= 0 or current_dm > 100):
            component.dry_matter = dm
    save_component_nutrients(
        component.id,
        payload["nutrients"],
        user_id=user_id,
    )
    component.protein = 0.0
    component.energy = 0.0
    component.updated_by = "demo-nutrients-seed"


def supplement_component_nutrients_if_sparse(
    component_id: str | None,
    *,
    user_id: str = "system",
) -> bool:
    """Добавляет Ca/P/сахара и т.д., если EAV уже есть, но без минералов."""
    if not component_id or nutrients_is_empty(component_id):
        return False
    component = Component.query.filter_by(id=component_id, is_deleted=False).first()
    if component is None:
        return False
    full = nutrients_full_dict(component_id)
    template = _payload_for_component(component)["nutrients"]
    allowed = set(_MINERAL_SUPPLEMENT)
    if _normalize_name(component.type) not in _ROUGHAGE_TYPES:
        allowed.update(_CARB_SUPPLEMENT)
    missing = {
        key: float(template[key])
        for key in allowed
        if key in template and read_from_mapping(full, (key,)) is None
    }
    if not missing:
        return False
    save_component_nutrients(component_id, missing, user_id=user_id, pin=True)
    return True


def ensure_component_nutrients_if_empty(
    component_id: str | None,
    *,
    user_id: str = "system",
) -> bool:
    """Заполняет EAV демо-шаблоном, если у компонента нет nutrients. Возвращает True при записи."""
    if not component_id or not nutrients_is_empty(component_id):
        return False
    component = Component.query.filter_by(id=component_id, is_deleted=False).first()
    if component is None:
        return False
    _apply_demo_payload(component, _payload_for_component(component), user_id=user_id)
    return True


def _marker_path() -> Path:
    return Path(Config.DATA_DIR) / _MARKER_NAME


def _already_seeded() -> bool:
    return _marker_path().is_file()


def _mark_seeded() -> None:
    path = _marker_path()
    path.parent.mkdir(parents=True, exist_ok=True)
    path.write_text("1\n", encoding="utf-8")


def seed_demo_component_nutrients_once(*, user_id: str = "system", force: bool = False) -> dict:
    """Заполняет lab_component_nutrient_value у всех компонентов с пустой матрицей."""
    updated = 0
    skipped = 0
    for component in Component.query.filter_by(is_deleted=False).all():
        if not force and not nutrients_is_empty(component.id):
            skipped += 1
            continue
        if force and not nutrients_is_empty(component.id):
            skipped += 1
            continue
        _apply_demo_payload(component, _payload_for_component(component), user_id=user_id)
        updated += 1

    if updated:
        db.session.commit()
        _mark_seeded()
        _log.info(
            "demo component nutrients seed user=%s updated=%s skipped=%s",
            user_id,
            updated,
            skipped,
        )
        return {"seeded": True, "updated": updated, "skipped": skipped}

    return {"seeded": False, "reason": "nothing_to_update", "updated": 0, "skipped": skipped}