class EveAIException(Exception):
    """Base exception class for EveAI API"""

    def __init__(self, message, status_code=400, payload=None):
        super().__init__()
        self.message = message
        self.status_code = status_code
        self.payload = payload

    def to_dict(self):
        rv = dict(self.payload or ())
        rv['message'] = self.message
        rv['error'] = self.__class__.__name__
        return rv

    def __str__(self):
        return self.message  # Return the message when the exception is converted to a string


class EveAIInvalidLanguageException(EveAIException):
    """Raised when an invalid language is provided"""

    def __init__(self, message="Langage is required", status_code=400, payload=None):
        super().__init__(message, status_code, payload)


class EveAIDoubleURLException(EveAIException):
    """Raised when an existing url is provided"""

    def __init__(self, message="URL already exists", status_code=400, payload=None):
        super().__init__(message, status_code, payload)


class EveAIUnsupportedFileType(EveAIException):
    """Raised when an invalid file type is provided"""

    def __init__(self, message="Filetype is not supported", status_code=400, payload=None):
        super().__init__(message, status_code, payload)


class EveAINoLicenseForTenant(EveAIException):
    """Raised when no active license for a tenant is provided"""

    def __init__(self, message="No license for tenant found", status_code=400, payload=None):
        super().__init__(message, status_code, payload)


class EveAITenantNotFound(EveAIException):
    """Raised when a tenant is not found"""

    def __init__(self, tenant_id, status_code=400, payload=None):
        self.tenant_id = tenant_id
        message = f"Tenant {tenant_id} not found"
        super().__init__(message, status_code, payload)


class EveAITenantInvalid(EveAIException):
    """Raised when a tenant is invalid"""

    def __init__(self, tenant_id, status_code=400, payload=None):
        self.tenant_id = tenant_id
        # Construct the message dynamically
        message = f"Tenant with ID '{tenant_id}' is not valid. Please contact the System Administrator."
        super().__init__(message, status_code, payload)


class EveAINoActiveLicense(EveAIException):
    """Raised when a tenant has no active licenses"""

    def __init__(self, tenant_id, status_code=400, payload=None):
        self.tenant_id = tenant_id
        # Construct the message dynamically
        message = f"Tenant with ID '{tenant_id}' has no active licenses. Please contact the System Administrator."
        super().__init__(message, status_code, payload)


class EveAIInvalidCatalog(EveAIException):
    """Raised when a catalog cannot be found"""

    def __init__(self, tenant_id, catalog_id, status_code=400, payload=None):
        self.tenant_id = tenant_id
        self.catalog_id = catalog_id
        # Construct the message dynamically
        message = f"Tenant with ID '{tenant_id}' has no valid catalog with ID {catalog_id}. Please contact the System Administrator."
        super().__init__(message, status_code, payload)


class EveAIInvalidProcessor(EveAIException):
    """Raised when no valid processor can be found for a given Catalog ID"""

    def __init__(self, tenant_id, catalog_id, file_type, status_code=400, payload=None):
        self.tenant_id = tenant_id
        self.catalog_id = catalog_id
        self.file_type = file_type
        # Construct the message dynamically
        message = (f"Tenant with ID '{tenant_id}' has no valid {file_type} processor for catalog with ID {catalog_id}. "
                   f"Please contact the System Administrator.")
        super().__init__(message, status_code, payload)


class EveAIInvalidDocument(EveAIException):
    """Raised when a tenant has no document with given ID"""

    def __init__(self, tenant_id, document_id, status_code=400, payload=None):
        self.tenant_id = tenant_id
        self.document_id = document_id
        # Construct the message dynamically
        message = f"Tenant with ID '{tenant_id}' has no document with ID {document_id}."
        super().__init__(message, status_code, payload)


class EveAIInvalidDocumentVersion(EveAIException):
    """Raised when a tenant has no document version with given ID"""

    def __init__(self, tenant_id, document_version_id, status_code=400, payload=None):
        self.tenant_id = tenant_id
        self.document_version_id = document_version_id
        # Construct the message dynamically
        message = f"Tenant with ID '{tenant_id}' has no document version with ID {document_version_id}."
        super().__init__(message, status_code, payload)


class EveAISocketInputException(EveAIException):
    """Raised when a socket call receives an invalid payload"""

    def __init__(self, message, status_code=400, payload=None):
        super.__init__(message, status_code, payload)


class EveAIInvalidEmbeddingModel(EveAIException):
    """Raised when no or an invalid embedding model is provided in the catalog"""

    def __init__(self, tenant_id, catalog_id, status_code=400, payload=None):
        self.tenant_id = tenant_id
        self.catalog_id = catalog_id
        # Construct the message dynamically
        message = f"Tenant with ID '{tenant_id}' has no or an invalid embedding model in Catalog {catalog_id}."
        super().__init__(message, status_code, payload)


class EveAIDoublePartner(EveAIException):
    """Raised when there is already a partner defined for a given tenant (while registering a partner)"""

    def __init__(self, tenant_id, status_code=400, payload=None):
        self.tenant_id = tenant_id
        # Construct the message dynamically
        message = f"Tenant with ID '{tenant_id}' is already defined as a Partner."
        super().__init__(message, status_code, payload)


class EveAIRoleAssignmentException(EveAIException):
    """Exception raised when a role cannot be assigned due to business rules"""

    def __init__(self, message, status_code=403, payload=None):
        super().__init__(message, status_code, payload)


class EveAINoManagementPartnerService(EveAIException):
    """Exception raised when the operation requires the logged in partner (or selected parter by Super User)
    does not have a MANAGEMENT_SERVICE"""

    def __init__(self, message="No Management Service defined for partner", status_code=403, payload=None):
        super().__init__(message, status_code, payload)


class EveAINoSessionTenant(EveAIException):
    """Exception raised when no session tenant is set"""

    def __init__(self, message="No Session Tenant selected. Cannot perform requested action.", status_code=403,
                 payload=None):
        super().__init__(message, status_code, payload)


class EveAINoSessionPartner(EveAIException):
    """Exception raised when no session partner is set"""

    def __init__(self, message="No Session Partner selected. Cannot perform requested action.", status_code=403,
                 payload=None):
        super().__init__(message, status_code, payload)


class EveAINoManagementPartnerForTenant(EveAIException):
    """Exception raised when the selected partner is no management partner for tenant"""

    def __init__(self, message="No Management Partner for Tenant", status_code=403, payload=None):
        super().__init__(message, status_code, payload)


class EveAIQuotaExceeded(EveAIException):
    """Base exception for quota-related errors"""

    def __init__(self, message, quota_type, current_usage, limit, additional=0, status_code=400, payload=None):
        super().__init__(message, status_code, payload)
        self.quota_type = quota_type
        self.current_usage = current_usage
        self.limit = limit
        self.additional = additional


class EveAIStorageQuotaExceeded(EveAIQuotaExceeded):
    """Raised when storage quota is exceeded"""

    def __init__(self, current_usage, limit, additional, status_code=400, payload=None):
        message = (f"Storage quota exceeded. Current: {current_usage:.1f}MB, "
                   f"Additional: {additional:.1f}MB, Limit: {limit}MB")
        super().__init__(message, "storage", current_usage, limit, additional, status_code, payload)


class EveAIEmbeddingQuotaExceeded(EveAIQuotaExceeded):
    """Raised when embedding quota is exceeded"""

    def __init__(self, current_usage, limit, additional, status_code=400, payload=None):
        message = (f"Embedding quota exceeded. Current: {current_usage:.1f}MB, "
                   f"Additional: {additional:.1f}MB, Limit: {limit}MB")
        super().__init__(message, "embedding", current_usage, limit, additional, status_code, payload)


class EveAIInteractionQuotaExceeded(EveAIQuotaExceeded):
    """Raised when the interaction token quota is exceeded"""

    def __init__(self, current_usage, limit, status_code=400, payload=None):
        message = (f"Interaction token quota exceeded. Current: {current_usage:.2f}M tokens, "
                   f"Limit: {limit:.2f}M tokens")
        super().__init__(message, "interaction", current_usage, limit, 0, status_code, payload)


class EveAIQuotaWarning(EveAIException):
    """Warning when approaching quota limits (not blocking)"""

    def __init__(self, message, quota_type, usage_percentage, status_code=200, payload=None):
        super().__init__(message, status_code, payload)
        self.quota_type = quota_type
        self.usage_percentage = usage_percentage


class EveAILicensePeriodsExceeded(EveAIException):
    """Raised when no more license periods can be created for a given license"""

    def __init__(self, license_id, status_code=400, payload=None):
        message = f"No more license periods can be created for license with ID {license_id}. "
        super().__init__(message, status_code, payload)


class EveAIPendingLicensePeriod(EveAIException):
    """Raised when a license period is pending"""

    def __init__(self, status_code=400, payload=None):
        message = f"Basic Fee Payment has not been received yet. Please ensure payment has been made, and please wait for payment to be processed."
        super().__init__(message, status_code, payload)