init commit

app/utils/merging.py

@@ -0,0 +1,69 @@
+# merging.py
+from __future__ import annotations
+from typing import Iterable, Dict, List, Tuple
+
+from app.models.primitives import Attributes, Resources
+from app.models.races import RaceSheet
+from app.utils.hero_catalog import HeroArchetype, SkillDef, SpellDef
+
+
+def _add_attributes(a: Attributes, b: Attributes) -> Attributes:
+    return Attributes(
+        base_str=a.base_str + b.base_str,
+        base_dex=a.base_dex + b.base_dex,
+        base_int=a.base_int + b.base_int,
+        base_wis=a.base_wis + b.base_wis,
+        base_luk=a.base_luk + b.base_luk,
+        base_cha=a.base_cha + b.base_cha,
+    )
+
+def _add_resources(a: Resources, b: Resources) -> Resources:
+    return Resources(
+        maxhp=a.maxhp + b.maxhp,
+        hp=a.hp + b.hp,
+        maxmp=a.maxmp + b.maxmp,
+        mp=a.mp + b.mp,
+        gold=a.gold + b.gold if hasattr(a, "gold") else getattr(b, "gold", 0),
+    )
+
+def _unique_chain(strings: Iterable[str]) -> List[str]:
+    seen = set()
+    out: List[str] = []
+    for s in strings:
+        if s not in seen:
+            out.append(s)
+            seen.add(s)
+    return out
+
+def merge_catalogs(
+    race_skills: Dict[str, "SkillDef"],
+    race_spells: Dict[str, "SpellDef"],
+    class_skills: Dict[str, "SkillDef"],
+    class_spells: Dict[str, "SpellDef"],
+) -> Tuple[Dict[str, "SkillDef"], Dict[str, "SpellDef"]]:
+    """
+    Return merged skill and spell maps. By default, CLASS overrides RACE on conflicts.
+    """
+    skills = dict(race_skills)
+    skills.update(class_skills)
+
+    spells = dict(race_spells)
+    spells.update(class_spells)
+
+    return skills, spells
+
+def merge_sheets(race: RaceSheet, cls: HeroArchetype):
+    """
+    Compute final base Attributes/Resources and starting lists, plus merged catalogs.
+    """
+    attrs = _add_attributes(race.base_attributes, cls.base_attributes)
+    res = _add_resources(race.starting_resources, cls.starting_resources)
+
+    starting_skills = _unique_chain([*race.starting_skills, *cls.starting_skills])
+    starting_spells = _unique_chain([*race.starting_spells, *cls.starting_spells])
+
+    skills_catalog, spells_catalog = merge_catalogs(
+        race.skills, race.spells, cls.skills, cls.spells
+    )
+
+    return attrs, res, starting_skills, starting_spells, skills_catalog, spells_catalog