from rights_registry import CapabilityRegistration, FeatureRegistration, register_capability, register_feature

_PLANNING_ROLES = ("club_admin", "trainer", "division_lead")

register_capability(
    CapabilityRegistration(
        id="planning.ai.suggest",
        name="Planungs-KI Suggest",
        domain="planning",
        module="planning_exercise_suggest",
        linked_feature_id="ai_calls",
        default_club_grants=tuple((r, "planning.ai.suggest") for r in _PLANNING_ROLES),
    )
)
register_capability(
    CapabilityRegistration(
        id="planning.ai.progression_path",
        name="Planungs-KI Progressionspfad",
        domain="planning",
        module="planning_exercise_suggest",
        linked_feature_id="ai_calls",
        default_club_grants=tuple((r, "planning.ai.progression_path") for r in _PLANNING_ROLES),
    )
)