1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
|
use core::fmt;
use axum::{
http::StatusCode,
response::{IntoResponse, Response},
Json,
};
use serde_json::json;
/// All errors raised by the web app
#[derive(Debug)]
pub enum AppError {
/// Database error
Database,
/// Generic bad request. It is handled with a message value
BadRequest(String),
/// Not found error
NotFound(String),
/// Raised when a token is not good created
TokenCreation,
/// Raised when a passed token is not valid
InvalidToken,
/// Raised if an user wants to do something can't do
Unauthorized,
}
/// Use `AppError` as response for an endpoint
impl IntoResponse for AppError {
/// Matches `AppError` into a tuple of status and error message.
/// The response will be a JSON in the format of:
/// ```json
/// { "error": "<message>" }
/// ```
fn into_response(self) -> Response {
let (status, error_message) = match self {
AppError::Database => (
StatusCode::INTERNAL_SERVER_ERROR,
"Error with database connection".to_string(),
),
AppError::BadRequest(value) => (StatusCode::BAD_REQUEST, value),
AppError::NotFound(value) => (StatusCode::NOT_FOUND, value),
AppError::TokenCreation => (
StatusCode::INTERNAL_SERVER_ERROR,
"Token creation error".to_string(),
),
AppError::InvalidToken => (StatusCode::BAD_REQUEST, "Invalid token".to_string()),
AppError::Unauthorized => (StatusCode::UNAUTHORIZED, "Unauthorized".to_string()),
};
let body = Json(json!({
"error": error_message,
}));
(status, body).into_response()
}
}
/// Raise a generic error from a string
impl From<std::string::String> for AppError {
fn from(error: std::string::String) -> AppError {
AppError::BadRequest(error)
}
}
/// Raise a generic io error
impl From<std::io::Error> for AppError {
fn from(error: std::io::Error) -> Self {
AppError::BadRequest(error.to_string())
}
}
/// Implementation of the `{}` marker for AppError
impl fmt::Display for AppError {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
match self {
AppError::Database => write!(f, "Database"),
AppError::BadRequest(value) => write!(f, "BadRequest: {}", value),
AppError::NotFound(value) => write!(f, "Not found: {}", value),
AppError::TokenCreation => write!(f, "Token creation"),
AppError::InvalidToken => write!(f, "Invalid Token"),
AppError::Unauthorized => write!(f, "Unauthorized"),
}
}
}
/// A tokio_postgres error is mapped to an `AppError::Database`
impl From<tokio_postgres::Error> for AppError {
fn from(_: tokio_postgres::Error) -> Self {
AppError::Database
}
}
/// A async_graphql error is mapped to an `AppError::BadRequest`
impl From<async_graphql::Error> for AppError {
fn from(value: async_graphql::Error) -> Self {
AppError::BadRequest(value.message)
}
}
|
