surrealdb-server 3.1.3

use std::error::Error as StdError;

use axum::Json;
use axum::response::{IntoResponse, Response};
use http::{HeaderName, HeaderValue, StatusCode};
use serde::{Serialize, Serializer};
use surrealdb_core::api::X_SURREAL_REQUEST_ID;
use surrealdb_core::api::err::ApiError;
use surrealdb_core::err::anyhow_to_types_error;
use surrealdb_types::{AuthError, NotAllowedError};
use thiserror::Error;

#[derive(Error, Debug)]
pub enum Error {
	#[error("The server is unable to handle the request")]
	ServerOverloaded,

	#[error("The request body contains invalid data")]
	Request,

	#[error("There was a problem with authentication")]
	InvalidAuth,

	#[error("The specified media type is unsupported")]
	InvalidType,

	#[error("There was a problem connecting with the storage engine")]
	InvalidStorage,

	#[error("There was an error with the remote request: {0}")]
	Remote(#[from] reqwest::Error),

	#[error("The HTTP route '{0}' is forbidden")]
	ForbiddenRoute(String),

	#[error("The HTTP route '{0}' is not found")]
	NotFound(String),

	#[error("There was a problem parsing the header {0}: {1}")]
	InvalidHeader(HeaderName, String),
}

impl IntoResponse for Error {
	fn into_response(self) -> Response {
		match self {
			Error::ForbiddenRoute(_) => {
				ErrorMessage{
					code: StatusCode::FORBIDDEN,
					details: Some("Forbidden".to_string()),
					description: Some("Not allowed to do this.".to_string()),
					information: Some(self.to_string())
				}.into_response()
			}
			Error::InvalidAuth => {
				ErrorMessage{
					code: StatusCode::UNAUTHORIZED,
					details: Some("Authentication failed".to_string()),
					description: Some("Your authentication details are invalid. Reauthenticate using valid authentication parameters.".to_string()),
					information: Some("There was a problem with authentication".to_string())
				}.into_response()
			}
			Error::InvalidType => {
				ErrorMessage {
					code: StatusCode::UNSUPPORTED_MEDIA_TYPE,
					details: Some("Unsupported media type".to_string()),
					description: Some("The request needs to adhere to certain constraints. Refer to the documentation for supported content types.".to_string()),
					information: None,
				}.into_response()
			}
			Error::NotFound(_) => {
				ErrorMessage {
					code: StatusCode::NOT_FOUND,
					details: Some("Not found".to_string()),
					description: Some("The request was made to an endpoint which does not exist.".to_string()),
					information: Some(self.to_string()),
				}.into_response()
			}
			Error::InvalidStorage =>
				ErrorMessage {
					code: StatusCode::INTERNAL_SERVER_ERROR,
					details: Some("Health check failed".to_string()),
					description: Some("The database health check for this instance failed. There was an issue with the underlying storage engine.".to_string()),
					information: Some(self.to_string()),
				}.into_response(),
			_ => ErrorMessage {
				code: StatusCode::BAD_REQUEST,
				details: Some("Request problems detected".to_string()),
				description: Some("There is a problem with your request. Refer to the documentation for further information.".to_string()),
				information: Some(format!("{self}")),
			}.into_response()
		}
	}
}

#[derive(Serialize)]
pub(super) struct ErrorMessage {
	#[serde(serialize_with = "serialize_status_code")]
	code: StatusCode,
	details: Option<String>,
	#[serde(skip_serializing_if = "Option::is_none")]
	description: Option<String>,
	#[serde(skip_serializing_if = "Option::is_none")]
	information: Option<String>,
}

/// Serde invokes `serialize_with` helpers with `&T` regardless of `Copy`.
#[allow(clippy::trivially_copy_pass_by_ref)]
fn serialize_status_code<S>(code: &StatusCode, s: S) -> Result<S::Ok, S::Error>
where
	S: Serializer,
{
	code.as_u16().serialize(s)
}

impl IntoResponse for ErrorMessage {
	fn into_response(self) -> Response {
		(self.code, Json(self)).into_response()
	}
}

/// Wrapper around anyhow error which implements [`IntoResponse`]
pub struct ResponseError(pub anyhow::Error);

impl<E: StdError + Send + Sync + 'static> From<E> for ResponseError {
	fn from(value: E) -> Self {
		ResponseError(anyhow::Error::new(value))
	}
}

impl IntoResponse for ResponseError {
	fn into_response(self) -> Response {
		// Check for ApiError first since it's public
		if let Some(e) = self.0.downcast_ref::<ApiError>() {
			return ErrorMessage {
				code: e.status_code(),
				details: Some("An error occurred while processing this API request".to_string()),
				description: Some(e.to_string()),
				information: None,
			}
			.into_response();
		}

		// Check for our local Error type (consumes on success)
		let err = match self.0.downcast::<Error>() {
			Ok(e) => return e.into_response(),
			Err(e) => e,
		};

		// Check for surrealdb_types::Error
		let err = match err.downcast::<surrealdb_types::Error>() {
			Ok(e) => return types_error_into_response(&e),
			Err(e) => e,
		};

		// Convert via core downcast (handles core::err::Error internally)
		// or fall back to anyhow chain preservation.
		types_error_into_response(&anyhow_to_types_error(err))
	}
}

/// Map a structured [`surrealdb_types::Error`] to an HTTP response with the appropriate status
/// code based on the error kind and details.
fn types_error_into_response(e: &surrealdb_types::Error) -> Response {
	if e.is_not_allowed() {
		let (code, details, description, information) = match e.not_allowed_details() {
			Some(NotAllowedError::Auth(AuthError::InvalidAuth))
			| Some(NotAllowedError::Auth(AuthError::TokenExpired))
			| Some(NotAllowedError::Auth(AuthError::SessionExpired))
			| Some(NotAllowedError::Auth(AuthError::UnexpectedAuth))
			| Some(NotAllowedError::Auth(AuthError::MissingUserOrPass))
			| Some(NotAllowedError::Auth(AuthError::NoSigninTarget))
			| Some(NotAllowedError::Auth(AuthError::InvalidPass))
			| Some(NotAllowedError::Auth(AuthError::TokenMakingFailed))
			| Some(NotAllowedError::Auth(AuthError::InvalidSignup)) => (
				StatusCode::UNAUTHORIZED,
				Some("Authentication failed".to_string()),
				Some("Your authentication details are invalid. Reauthenticate using valid authentication parameters.".to_string()),
				Some("There was a problem with authentication".to_string()),
			),
			_ => (
				StatusCode::FORBIDDEN,
				Some("Forbidden".to_string()),
				Some("Not allowed to do this.".to_string()),
				Some(e.message().to_string()),
			),
		};
		return ErrorMessage {
			code,
			details,
			description,
			information,
		}
		.into_response();
	}
	if e.is_not_found() {
		return ErrorMessage {
			code: StatusCode::NOT_FOUND,
			details: Some("Not found".to_string()),
			description: Some("The requested resource was not found.".to_string()),
			information: Some(e.message().to_string()),
		}
		.into_response();
	}
	ErrorMessage {
		code: StatusCode::BAD_REQUEST,
		details: Some("Request problems detected".to_string()),
		description: Some("There is a problem with your request. Refer to the documentation for further information.".to_string()),
		information: Some(e.message().to_string()),
	}
	.into_response()
}

/// Error wrapper for the API HTTP handler that attaches a request ID to the response
/// so all errors (including those before invocation) can be traced.
pub(super) struct ApiHandlerError(pub ResponseError, pub String);

impl IntoResponse for ApiHandlerError {
	fn into_response(self) -> Response {
		let mut response = self.0.into_response();
		if !self.1.is_empty()
			&& let Ok(value) = HeaderValue::from_str(&self.1)
		{
			response.headers_mut().insert(X_SURREAL_REQUEST_ID, value);
		}
		response
	}
}