from typing import List

from fastapi import APIRouter, Depends
from sqlalchemy.ext.asyncio import AsyncSession

from app.core.database import get_db
from app.core.dependencies import get_current_user
from app.models import User
from app.services.key_service import KeyService
from app.datamodels.schemas import CreateKeyRequest, ApiKeyResponse, ApiKeyCreatedResponse, MessageResponse

router = APIRouter(prefix="/keys", tags=["keys"])


@router.get("", response_model=List[ApiKeyResponse])
async def list_keys(
    user: User = Depends(get_current_user),
    db: AsyncSession = Depends(get_db),
):
    return await KeyService.list_keys(db, user.id)


@router.post("", response_model=ApiKeyCreatedResponse)
async def create_key(
    body: CreateKeyRequest,
    user: User = Depends(get_current_user),
    db: AsyncSession = Depends(get_db),
):
    return await KeyService.create_key(db, user.id, body.name)


@router.delete("/{key_id}", response_model=MessageResponse)
async def delete_key(
    key_id: str,
    user: User = Depends(get_current_user),
    db: AsyncSession = Depends(get_db),
):
    await KeyService.delete_key(db, user.id, key_id)
    return {"message": "Key deleted"}