"""
Body Parts API Endpoints
GET /api/v1/body-parts - Returns body part categories for sideline entry picker
"""
from typing import List
from fastapi import APIRouter, Depends
from pydantic import BaseModel
from sqlalchemy.orm import Session

from app.database import get_db
from app.auth import get_current_user, require_role

router = APIRouter(tags=["body-parts"])

============== SCHEMAS ==============

class BodyPart(BaseModel):
"""Individual body part"""
id: str
label: str
icon: str

class BodyPartCategory(BaseModel):
"""Category of body parts"""
name: str
parts: List[BodyPart]

============== DATA ==============

Body parts hierarchy matching the frontend picker

BODY_PARTS_DATA: List[BodyPartCategory] = [
BodyPartCategory(
name="Head & Neck",
parts=[
BodyPart(id="head", label="Head", icon="🧠"),
BodyPart(id="face", label="Face", icon="😷"),
BodyPart(id="neck", label="Neck", icon="🦒"),
BodyPart(id="concussion", label="Concussion", icon="🧠"),
]
),
BodyPartCategory(
name="Upper Body",
parts=[
BodyPart(id="shoulder", label="Shoulder", icon="💪"),
BodyPart(id="upper_arm", label="Upper Arm", icon="💪"),
BodyPart(id="elbow", label="Elbow", icon="🦴"),
BodyPart(id="forearm", label="Forearm", icon="💪"),
BodyPart(id="wrist", label="Wrist", icon="⌚"),
BodyPart(id="hand", label="Hand", icon="✋"),
BodyPart(id="fingers", label="Fingers", icon="👆"),
BodyPart(id="chest", label="Chest", icon="🫁"),
BodyPart(id="upper_back", label="Upper Back", icon="↕️"),
]
),
BodyPartCategory(
name="Lower Body",
parts=[
BodyPart(id="hip", label="Hip", icon="🦴"),
BodyPart(id="groin", label="Groin", icon="🩳"),
BodyPart(id="thigh", label="Thigh", icon="🦵"),
BodyPart(id="knee", label="Knee", icon="🦵"),
BodyPart(id="shin", label="Shin", icon="🦵"),
BodyPart(id="calf", label="Calf", icon="🦵"),
BodyPart(id="ankle", label="Ankle", icon="🦶"),
BodyPart(id="foot", label="Foot", icon="🦶"),
BodyPart(id="toes", label="Toes", icon="👣"),
BodyPart(id="lower_back", label="Lower Back", icon="↕️"),
]
),
BodyPartCategory(
name="Other",
parts=[
BodyPart(id="abdomen", label="Abdomen", icon="🫃"),
BodyPart(id="ribs", label="Ribs", icon="🍖"),
BodyPart(id="general", label="General/Illness", icon="🤒"),
]
),
]

============== ROUTER ENDPOINTS ==============

@router.get("", response_model=List[BodyPartCategory])
async def get_body_parts(
db: Session = Depends(get_db),
current_user = Depends(require_role(["at", "coach", "admin"])),
):
"""
Get body part categories and parts for the sideline entry picker

Returns a hierarchical list of body part categories and their parts
with icons for the 3-tap sideline entry flow.
"""
return BODY_PARTS_DATA