from sqlalchemy.ext.asyncio import AsyncSession

from app.models.augmented_part import AugmentedPart, AugmentedPartNote
from app.models.user import User


async def create(
    db: AsyncSession, *, augmented_part: AugmentedPart, author: User, body: str
) -> AugmentedPartNote:
    note = AugmentedPartNote(
        augmented_part_id=augmented_part.id,
        user_id=author.id,
        username=author.full_name or author.email,
        body=body,
    )
    db.add(note)
    await db.commit()
    await db.refresh(note)
    return note