backend/core/exceptions.py

from fastapi.responses import JSONResponse
from fastapi import Request


class BaseAppException(Exception):
    status_code = 500
    message = "An error occurred"

    def __init__(self, message: str = None):
        self.message = message or self.message
        super().__init__(self.message)


class InvalidFileTypeError(BaseAppException):
    status_code = 415
    message = "Invalid file type. Only PDF files are allowed."


class FileSizeExceededError(BaseAppException):
    status_code = 400
    message = "File size exceeds maximum allowed size."


class DuplicateDocumentError(BaseAppException):
    status_code = 409
    message = "Document with this file hash already exists."


class ProcessingError(BaseAppException):
    status_code = 500
    message = "Document processing failed."


class ParsingError(BaseAppException):
    status_code = 500
    message = "PDF parsing failed."


class IndexingError(BaseAppException):
    status_code = 500
    message = "Indexing to vector database failed."


class EmbeddingError(BaseAppException):
    status_code = 500
    message = "Embedding generation failed."


class DocumentNotFoundError(BaseAppException):
    status_code = 404
    message = "Document not found."


class PaperFetchError(BaseAppException):
    status_code = 502
    message = "Failed to fetch paper from external source."


async def global_exception_handler(request: Request, exc: Exception):
    if isinstance(exc, BaseAppException):
        return JSONResponse(
            status_code=exc.status_code,
            content={
                "error": exc.__class__.__name__,
                "message": exc.message,
            }
        )

    return JSONResponse(
        status_code=500,
        content={
            "error": "Internal Server Error",
            "details": str(exc) if request.app.debug else "An unexpected error occurred",
        }
    )