from fastapi import APIRouter
from core.config import engine
from services.datasets.delete import delete_dataset_from_partition  # import fungsi di atas
from response import successRes, errorRes

router = APIRouter()

@router.delete("/dataset/{user_id}/{metadata_id}")
async def delete_dataset(user_id: int, metadata_id: int, title: str):
    """
    Hapus dataset tertentu (berdasarkan user_id dan metadata_id)
    """
    try:
        async with engine.begin() as conn:
            await delete_dataset_from_partition(conn, user_id, metadata_id, title)
        return successRes(message=f"Dataset {title} berhasil dihapus.", data="")

    except Exception as e:
        print(f"[ERROR] Gagal hapus dataset: {e}")
        raise errorRes(status_code=500, details=str(e), message="Gagal hapus dataset")