error refactor

This commit is contained in:
2023-08-29 21:34:00 +02:00
parent e7d85b60c7
commit ccb666f000
3 changed files with 340 additions and 722 deletions

View File

@@ -20,10 +20,8 @@ use std::str::FromStr;
use serde_variant::to_variant_name; use serde_variant::to_variant_name;
use sqlx::{ use sqlx::{
error::DatabaseError, sqlite::{SqliteConnectOptions, SqlitePoolOptions},
query, Pool, Sqlite,
sqlite::{SqliteConnectOptions, SqliteError, SqlitePoolOptions, SqliteRow},
Pool, Row, Sqlite,
}; };
use maud::Markup; use maud::Markup;
@@ -250,14 +248,7 @@ async fn inventory_active(
state.client_state.edit_item = inventory_query.edit_item; state.client_state.edit_item = inventory_query.edit_item;
state.client_state.active_category_id = Some(id); state.client_state.active_category_id = Some(id);
let inventory = models::Inventory::load(&state.database_pool) let inventory = models::Inventory::load(&state.database_pool).await?;
.await
.map_err(|error| {
(
StatusCode::INTERNAL_SERVER_ERROR,
components::ErrorPage::build(&error.to_string()),
)
})?;
let active_category: Option<&models::Category> = state let active_category: Option<&models::Category> = state
.client_state .client_state
@@ -296,14 +287,7 @@ async fn inventory_inactive(
state.client_state.edit_item = inventory_query.edit_item; state.client_state.edit_item = inventory_query.edit_item;
state.client_state.active_category_id = None; state.client_state.active_category_id = None;
let inventory = models::Inventory::load(&state.database_pool) let inventory = models::Inventory::load(&state.database_pool).await?;
.await
.map_err(|error| {
(
StatusCode::INTERNAL_SERVER_ERROR,
components::ErrorPage::build(&error.to_string()),
)
})?;
Ok(( Ok((
StatusCode::OK, StatusCode::OK,
@@ -318,29 +302,6 @@ async fn inventory_inactive(
)) ))
} }
// async fn inventory(
// mut state: AppState,
// active_id: Option<Uuid>,
// ) -> Result<(StatusCode, Markup), (StatusCode, Markup)> {
// state.client_state.active_category_id = active_id;
// Ok((
// StatusCode::OK,
// components::Root::build(
// &components::inventory::Inventory::build(state.client_state, categories).map_err(|e| match e {
// Error::NotFound { description } => {
// (StatusCode::NOT_FOUND, components::ErrorPage::build(&description))
// }
// _ => (
// StatusCode::INTERNAL_SERVER_ERROR,
// components::ErrorPage::build(&e.to_string()),
// ),
// })?,
// &TopLevelPage::Inventory,
// ),
// ))
// }
#[derive(Deserialize)] #[derive(Deserialize)]
struct NewItem { struct NewItem {
#[serde(rename = "new-item-name")] #[serde(rename = "new-item-name")]
@@ -363,14 +324,7 @@ async fn inventory_item_validate_name(
State(state): State<AppState>, State(state): State<AppState>,
Form(new_item): Form<NewItemName>, Form(new_item): Form<NewItemName>,
) -> Result<(StatusCode, Markup), (StatusCode, Markup)> { ) -> Result<(StatusCode, Markup), (StatusCode, Markup)> {
let exists = models::InventoryItem::name_exists(&state.database_pool, &new_item.name) let exists = models::InventoryItem::name_exists(&state.database_pool, &new_item.name).await?;
.map_err(|error| {
(
StatusCode::INTERNAL_SERVER_ERROR,
components::ErrorPage::build(&error.to_string()),
)
})
.await?;
Ok(( Ok((
StatusCode::OK, StatusCode::OK,
@@ -378,6 +332,27 @@ async fn inventory_item_validate_name(
)) ))
} }
impl From<models::Error> for (StatusCode, Markup) {
fn from(value: models::Error) -> (StatusCode, Markup) {
match value {
models::Error::Database(_) => (
StatusCode::INTERNAL_SERVER_ERROR,
components::ErrorPage::build(&value.to_string()),
),
models::Error::Query(error) => match error {
models::QueryError::NotFound { description } => (
StatusCode::NOT_FOUND,
components::ErrorPage::build(&description),
),
_ => (
StatusCode::BAD_REQUEST,
components::ErrorPage::build(&error.to_string()),
),
},
}
}
}
async fn inventory_item_create( async fn inventory_item_create(
State(state): State<AppState>, State(state): State<AppState>,
headers: HeaderMap, headers: HeaderMap,
@@ -396,27 +371,10 @@ async fn inventory_item_create(
new_item.category_id, new_item.category_id,
new_item.weight, new_item.weight,
) )
.map_err(|error| match error {
models::Error::Constraint { description } => (
StatusCode::BAD_REQUEST,
components::ErrorPage::build(&description),
),
_ => (
StatusCode::INTERNAL_SERVER_ERROR,
components::ErrorPage::build(&error.to_string()),
),
})
.await?; .await?;
if is_htmx(&headers) { if is_htmx(&headers) {
let inventory = models::Inventory::load(&state.database_pool) let inventory = models::Inventory::load(&state.database_pool).await?;
.await
.map_err(|error| {
(
StatusCode::INTERNAL_SERVER_ERROR,
components::ErrorPage::build(&error.to_string()),
)
})?;
// it's impossible to NOT find the item here, as we literally just added // it's impossible to NOT find the item here, as we literally just added
// it. but good error handling never hurts // it. but good error handling never hurts
@@ -456,18 +414,7 @@ async fn inventory_item_delete(
headers: HeaderMap, headers: HeaderMap,
Path(id): Path<Uuid>, Path(id): Path<Uuid>,
) -> Result<Redirect, (StatusCode, Markup)> { ) -> Result<Redirect, (StatusCode, Markup)> {
let deleted = models::InventoryItem::delete(&state.database_pool, id) let deleted = models::InventoryItem::delete(&state.database_pool, id).await?;
.map_err(|error| match error {
models::Error::Constraint { ref description } => (
StatusCode::NOT_IMPLEMENTED,
components::ErrorPage::build(description),
),
_ => (
StatusCode::INTERNAL_SERVER_ERROR,
components::ErrorPage::build(&error.to_string()),
),
})
.await?;
if !deleted { if !deleted {
Err(( Err((
@@ -520,20 +467,14 @@ async fn inventory_item_edit(
&state.database_pool, &state.database_pool,
id, id,
&edit_item.name, &edit_item.name,
i64::try_from(edit_item.weight).map_err(|e| { i64::try_from(edit_item.weight).map_err(|error| {
( (
StatusCode::UNPROCESSABLE_ENTITY, StatusCode::UNPROCESSABLE_ENTITY,
components::ErrorPage::build(&e.to_string()), components::ErrorPage::build(&error.to_string()),
) )
})?, })?,
) )
.await .await?
.map_err(|e| {
(
StatusCode::INTERNAL_SERVER_ERROR,
components::ErrorPage::build(&e.to_string()),
)
})?
.ok_or(( .ok_or((
StatusCode::NOT_FOUND, StatusCode::NOT_FOUND,
components::ErrorPage::build(&format!("item with id {id} not found", id = id)), components::ErrorPage::build(&format!("item with id {id} not found", id = id)),
@@ -545,14 +486,11 @@ async fn inventory_item_edit(
async fn inventory_item_cancel( async fn inventory_item_cancel(
State(state): State<AppState>, State(state): State<AppState>,
Path(id): Path<Uuid>, Path(id): Path<Uuid>,
) -> Result<Redirect, (StatusCode, String)> { ) -> Result<Redirect, (StatusCode, Markup)> {
let id = models::Item::find(&state.database_pool, id) let id = models::Item::find(&state.database_pool, id).await?.ok_or((
.await StatusCode::NOT_FOUND,
.map_err(|e| (StatusCode::INTERNAL_SERVER_ERROR, e.to_string()))? components::ErrorPage::build(&format!("item with id {id} not found")),
.ok_or(( ))?;
StatusCode::NOT_FOUND,
format!("item with id {id} not found", id = id),
))?;
Ok(Redirect::to(&format!( Ok(Redirect::to(&format!(
"/inventory/category/{id}/", "/inventory/category/{id}/",
@@ -587,16 +525,6 @@ async fn trip_create(
new_trip.date_start, new_trip.date_start,
new_trip.date_end, new_trip.date_end,
) )
.map_err(|error| match error {
models::Error::TimeParse { description } => (
StatusCode::BAD_REQUEST,
components::ErrorPage::build(&description),
),
_ => (
StatusCode::BAD_REQUEST,
components::ErrorPage::build(&error.to_string()),
),
})
.await?; .await?;
Ok(Redirect::to(&format!("/trips/{new_id}/"))) Ok(Redirect::to(&format!("/trips/{new_id}/")))
@@ -605,14 +533,7 @@ async fn trip_create(
async fn trips( async fn trips(
State(state): State<AppState>, State(state): State<AppState>,
) -> Result<(StatusCode, Markup), (StatusCode, Markup)> { ) -> Result<(StatusCode, Markup), (StatusCode, Markup)> {
let trips = models::Trip::all(&state.database_pool) let trips = models::Trip::all(&state.database_pool).await?;
.map_err(|error| {
(
StatusCode::INTERNAL_SERVER_ERROR,
components::ErrorPage::build(&error.to_string()),
)
})
.await?;
Ok(( Ok((
StatusCode::OK, StatusCode::OK,
@@ -637,45 +558,17 @@ async fn trip(
state.client_state.trip_edit_attribute = trip_query.edit; state.client_state.trip_edit_attribute = trip_query.edit;
state.client_state.active_category_id = trip_query.category; state.client_state.active_category_id = trip_query.category;
let mut trip: models::Trip = models::Trip::find(&state.database_pool, id) let mut trip: models::Trip = models::Trip::find(&state.database_pool, id).await?.ok_or((
.map_err(|error| { StatusCode::NOT_FOUND,
( components::ErrorPage::build(&format!("trip with id {} not found", id)),
StatusCode::INTERNAL_SERVER_ERROR, ))?;
components::ErrorPage::build(&error.to_string()),
)
})
.await?
.ok_or((
StatusCode::NOT_FOUND,
components::ErrorPage::build(&format!("trip with id {} not found", id)),
))?;
trip.load_trips_types(&state.database_pool) trip.load_trips_types(&state.database_pool).await?;
.await
.map_err(|e| {
(
StatusCode::INTERNAL_SERVER_ERROR,
components::ErrorPage::build(&e.to_string()),
)
})?;
trip.sync_trip_items_with_inventory(&state.database_pool) trip.sync_trip_items_with_inventory(&state.database_pool)
.await .await?;
.map_err(|e| {
(
StatusCode::INTERNAL_SERVER_ERROR,
components::ErrorPage::build(&e.to_string()),
)
})?;
trip.load_categories(&state.database_pool) trip.load_categories(&state.database_pool).await?;
.await
.map_err(|e| {
(
StatusCode::INTERNAL_SERVER_ERROR,
components::ErrorPage::build(&e.to_string()),
)
})?;
let active_category: Option<&models::TripCategory> = state let active_category: Option<&models::TripCategory> = state
.client_state .client_state
@@ -710,14 +603,7 @@ async fn trip_type_remove(
State(state): State<AppState>, State(state): State<AppState>,
Path((trip_id, type_id)): Path<(Uuid, Uuid)>, Path((trip_id, type_id)): Path<(Uuid, Uuid)>,
) -> Result<Redirect, (StatusCode, Markup)> { ) -> Result<Redirect, (StatusCode, Markup)> {
let found = models::Trip::trip_type_remove(&state.database_pool, trip_id, type_id) let found = models::Trip::trip_type_remove(&state.database_pool, trip_id, type_id).await?;
.map_err(|error| {
(
StatusCode::INTERNAL_SERVER_ERROR,
components::ErrorPage::build(&error.to_string()),
)
})
.await?;
if !found { if !found {
Err(( Err((
@@ -735,22 +621,7 @@ async fn trip_type_add(
State(state): State<AppState>, State(state): State<AppState>,
Path((trip_id, type_id)): Path<(Uuid, Uuid)>, Path((trip_id, type_id)): Path<(Uuid, Uuid)>,
) -> Result<Redirect, (StatusCode, Markup)> { ) -> Result<Redirect, (StatusCode, Markup)> {
models::Trip::trip_type_add(&state.database_pool, trip_id, type_id) models::Trip::trip_type_add(&state.database_pool, trip_id, type_id).await?;
.map_err(|error| match error {
models::Error::ReferenceNotFound { description } => (
StatusCode::BAD_REQUEST,
components::ErrorPage::build(&description),
),
models::Error::Duplicate { description } => (
StatusCode::BAD_REQUEST,
components::ErrorPage::build(&description),
),
_ => (
StatusCode::INTERNAL_SERVER_ERROR,
components::ErrorPage::build(&error.to_string()),
),
})
.await?;
Ok(Redirect::to(&format!("/trips/{trip_id}/"))) Ok(Redirect::to(&format!("/trips/{trip_id}/")))
} }
@@ -768,12 +639,6 @@ async fn trip_comment_set(
) -> Result<Redirect, (StatusCode, Markup)> { ) -> Result<Redirect, (StatusCode, Markup)> {
let found = let found =
models::Trip::set_comment(&state.database_pool, trip_id, &comment_update.new_comment) models::Trip::set_comment(&state.database_pool, trip_id, &comment_update.new_comment)
.map_err(|error| {
(
StatusCode::BAD_REQUEST,
components::ErrorPage::build(&error.to_string()),
)
})
.await?; .await?;
if !found { if !found {
@@ -811,16 +676,6 @@ async fn trip_edit_attribute(
attribute, attribute,
&trip_update.new_value, &trip_update.new_value,
) )
.map_err(|error| match error {
models::Error::NotFound { description } => (
StatusCode::NOT_FOUND,
components::ErrorPage::build(&description),
),
_ => (
StatusCode::BAD_REQUEST,
components::ErrorPage::build(&error.to_string()),
),
})
.await?; .await?;
Ok(Redirect::to(&format!("/trips/{trip_id}/"))) Ok(Redirect::to(&format!("/trips/{trip_id}/")))
@@ -833,19 +688,7 @@ async fn trip_item_set_state(
key: models::TripItemStateKey, key: models::TripItemStateKey,
value: bool, value: bool,
) -> Result<(), (StatusCode, Markup)> { ) -> Result<(), (StatusCode, Markup)> {
models::TripItem::set_state(&state.database_pool, trip_id, item_id, key, value) models::TripItem::set_state(&state.database_pool, trip_id, item_id, key, value).await?;
.map_err(|error| match error {
models::Error::NotFound { description } => (
StatusCode::NOT_FOUND,
components::ErrorPage::build(&description),
),
_ => (
StatusCode::INTERNAL_SERVER_ERROR,
components::ErrorPage::build(&error.to_string()),
),
})
.await?;
Ok(()) Ok(())
} }
@@ -855,13 +698,7 @@ async fn trip_row(
item_id: Uuid, item_id: Uuid,
) -> Result<Markup, (StatusCode, Markup)> { ) -> Result<Markup, (StatusCode, Markup)> {
let item = models::TripItem::find(&state.database_pool, trip_id, item_id) let item = models::TripItem::find(&state.database_pool, trip_id, item_id)
.await .await?
.map_err(|error| {
(
StatusCode::BAD_REQUEST,
components::ErrorPage::build(&error.to_string()),
)
})?
.ok_or_else(|| { .ok_or_else(|| {
( (
StatusCode::NOT_FOUND, StatusCode::NOT_FOUND,
@@ -875,23 +712,10 @@ async fn trip_row(
let item_row = components::trip::TripItemListRow::build( let item_row = components::trip::TripItemListRow::build(
trip_id, trip_id,
&item, &item,
models::Item::get_category_max_weight(&state.database_pool, item.item.category_id) models::Item::get_category_max_weight(&state.database_pool, item.item.category_id).await?,
.await
.map_err(|error| {
(
StatusCode::BAD_REQUEST,
components::ErrorPage::build(&error.to_string()),
)
})?,
); );
let category = models::TripCategory::find(&state.database_pool, trip_id, item.item.category_id) let category = models::TripCategory::find(&state.database_pool, trip_id, item.item.category_id)
.map_err(|error| {
(
StatusCode::BAD_REQUEST,
components::ErrorPage::build(&error.to_string()),
)
})
.await? .await?
.ok_or_else(|| { .ok_or_else(|| {
( (
@@ -915,14 +739,16 @@ async fn trip_item_set_pick(
Path((trip_id, item_id)): Path<(Uuid, Uuid)>, Path((trip_id, item_id)): Path<(Uuid, Uuid)>,
headers: HeaderMap, headers: HeaderMap,
) -> Result<Redirect, (StatusCode, Markup)> { ) -> Result<Redirect, (StatusCode, Markup)> {
Ok(trip_item_set_state( Ok::<_, models::Error>(
&state, trip_item_set_state(
trip_id, &state,
item_id, trip_id,
models::TripItemStateKey::Pick, item_id,
true, models::TripItemStateKey::Pick,
true,
)
.await?,
) )
.await?)
.map(|_| -> Result<Redirect, (StatusCode, Markup)> { .map(|_| -> Result<Redirect, (StatusCode, Markup)> {
Ok(Redirect::to( Ok(Redirect::to(
headers headers
@@ -932,12 +758,12 @@ async fn trip_item_set_pick(
components::ErrorPage::build("no referer header found"), components::ErrorPage::build("no referer header found"),
))? ))?
.to_str() .to_str()
.map_err(|e| { .map_err(|error| {
( (
StatusCode::BAD_REQUEST, StatusCode::BAD_REQUEST,
components::ErrorPage::build(&format!( components::ErrorPage::build(&format!(
"referer could not be converted: {}", "referer could not be converted: {}",
e error
)), )),
) )
})?, })?,
@@ -974,14 +800,16 @@ async fn trip_item_set_unpick(
Path((trip_id, item_id)): Path<(Uuid, Uuid)>, Path((trip_id, item_id)): Path<(Uuid, Uuid)>,
headers: HeaderMap, headers: HeaderMap,
) -> Result<Redirect, (StatusCode, Markup)> { ) -> Result<Redirect, (StatusCode, Markup)> {
Ok(trip_item_set_state( Ok::<_, models::Error>(
&state, trip_item_set_state(
trip_id, &state,
item_id, trip_id,
models::TripItemStateKey::Pick, item_id,
false, models::TripItemStateKey::Pick,
false,
)
.await?,
) )
.await?)
.map(|_| -> Result<Redirect, (StatusCode, Markup)> { .map(|_| -> Result<Redirect, (StatusCode, Markup)> {
Ok(Redirect::to( Ok(Redirect::to(
headers headers
@@ -991,12 +819,12 @@ async fn trip_item_set_unpick(
components::ErrorPage::build("no referer header found"), components::ErrorPage::build("no referer header found"),
))? ))?
.to_str() .to_str()
.map_err(|e| { .map_err(|error| {
( (
StatusCode::BAD_REQUEST, StatusCode::BAD_REQUEST,
components::ErrorPage::build(&format!( components::ErrorPage::build(&format!(
"referer could not be converted: {}", "referer could not be converted: {}",
e error
)), )),
) )
})?, })?,
@@ -1033,14 +861,16 @@ async fn trip_item_set_pack(
Path((trip_id, item_id)): Path<(Uuid, Uuid)>, Path((trip_id, item_id)): Path<(Uuid, Uuid)>,
headers: HeaderMap, headers: HeaderMap,
) -> Result<Redirect, (StatusCode, Markup)> { ) -> Result<Redirect, (StatusCode, Markup)> {
Ok(trip_item_set_state( Ok::<_, models::Error>(
&state, trip_item_set_state(
trip_id, &state,
item_id, trip_id,
models::TripItemStateKey::Pack, item_id,
true, models::TripItemStateKey::Pack,
true,
)
.await?,
) )
.await?)
.map(|_| -> Result<Redirect, (StatusCode, Markup)> { .map(|_| -> Result<Redirect, (StatusCode, Markup)> {
Ok(Redirect::to( Ok(Redirect::to(
headers headers
@@ -1050,12 +880,12 @@ async fn trip_item_set_pack(
components::ErrorPage::build("no referer header found"), components::ErrorPage::build("no referer header found"),
))? ))?
.to_str() .to_str()
.map_err(|e| { .map_err(|error| {
( (
StatusCode::BAD_REQUEST, StatusCode::BAD_REQUEST,
components::ErrorPage::build(&format!( components::ErrorPage::build(&format!(
"referer could not be converted: {}", "referer could not be converted: {}",
e error
)), )),
) )
})?, })?,
@@ -1092,14 +922,16 @@ async fn trip_item_set_unpack(
Path((trip_id, item_id)): Path<(Uuid, Uuid)>, Path((trip_id, item_id)): Path<(Uuid, Uuid)>,
headers: HeaderMap, headers: HeaderMap,
) -> Result<Redirect, (StatusCode, Markup)> { ) -> Result<Redirect, (StatusCode, Markup)> {
Ok(trip_item_set_state( Ok::<_, models::Error>(
&state, trip_item_set_state(
trip_id, &state,
item_id, trip_id,
models::TripItemStateKey::Pack, item_id,
false, models::TripItemStateKey::Pack,
false,
)
.await?,
) )
.await?)
.map(|_| -> Result<Redirect, (StatusCode, Markup)> { .map(|_| -> Result<Redirect, (StatusCode, Markup)> {
Ok(Redirect::to( Ok(Redirect::to(
headers headers
@@ -1109,12 +941,12 @@ async fn trip_item_set_unpack(
components::ErrorPage::build("no referer header found"), components::ErrorPage::build("no referer header found"),
))? ))?
.to_str() .to_str()
.map_err(|e| { .map_err(|error| {
( (
StatusCode::BAD_REQUEST, StatusCode::BAD_REQUEST,
components::ErrorPage::build(&format!( components::ErrorPage::build(&format!(
"referer could not be converted: {}", "referer could not be converted: {}",
e error
)), )),
) )
})?, })?,
@@ -1151,13 +983,7 @@ async fn trip_total_weight_htmx(
Path(trip_id): Path<Uuid>, Path(trip_id): Path<Uuid>,
) -> Result<(StatusCode, Markup), (StatusCode, Markup)> { ) -> Result<(StatusCode, Markup), (StatusCode, Markup)> {
let total_weight = models::Trip::find_total_picked_weight(&state.database_pool, trip_id) let total_weight = models::Trip::find_total_picked_weight(&state.database_pool, trip_id)
.await .await?
.map_err(|error| {
(
StatusCode::BAD_REQUEST,
components::ErrorPage::build(&error.to_string()),
)
})?
.ok_or(( .ok_or((
StatusCode::NOT_FOUND, StatusCode::NOT_FOUND,
components::ErrorPage::build(&format!("trip with id {trip_id} not found")), components::ErrorPage::build(&format!("trip with id {trip_id} not found")),
@@ -1185,18 +1011,7 @@ async fn inventory_category_create(
)); ));
} }
let _new_id = models::Category::save(&state.database_pool, &new_category.name) let _new_id = models::Category::save(&state.database_pool, &new_category.name).await?;
.map_err(|error| match error {
models::Error::Duplicate { description } => (
StatusCode::BAD_REQUEST,
components::ErrorPage::build(&description),
),
_ => (
StatusCode::INTERNAL_SERVER_ERROR,
components::ErrorPage::build(&error.to_string()),
),
})
.await?;
Ok(Redirect::to("/inventory/")) Ok(Redirect::to("/inventory/"))
} }
@@ -1206,14 +1021,7 @@ async fn trip_state_set(
headers: HeaderMap, headers: HeaderMap,
Path((trip_id, new_state)): Path<(Uuid, models::TripState)>, Path((trip_id, new_state)): Path<(Uuid, models::TripState)>,
) -> Result<impl IntoResponse, (StatusCode, Markup)> { ) -> Result<impl IntoResponse, (StatusCode, Markup)> {
let exists = models::Trip::set_state(&state.database_pool, trip_id, &new_state) let exists = models::Trip::set_state(&state.database_pool, trip_id, &new_state).await?;
.map_err(|e| {
(
StatusCode::BAD_REQUEST,
components::ErrorPage::build(&e.to_string()),
)
})
.await?;
if !exists { if !exists {
return Err(( return Err((
@@ -1251,14 +1059,7 @@ async fn trips_types(
) -> Result<(StatusCode, Markup), (StatusCode, Markup)> { ) -> Result<(StatusCode, Markup), (StatusCode, Markup)> {
state.client_state.trip_type_edit = trip_type_query.edit; state.client_state.trip_type_edit = trip_type_query.edit;
let trip_types: Vec<models::TripsType> = models::TripsType::all(&state.database_pool) let trip_types: Vec<models::TripsType> = models::TripsType::all(&state.database_pool).await?;
.map_err(|error| {
(
StatusCode::INTERNAL_SERVER_ERROR,
components::ErrorPage::build(&error.to_string()),
)
})
.await?;
Ok(( Ok((
StatusCode::OK, StatusCode::OK,
@@ -1286,18 +1087,7 @@ async fn trip_type_create(
)); ));
} }
let _new_id = models::TripsType::save(&state.database_pool, &new_trip_type.name) let _new_id = models::TripsType::save(&state.database_pool, &new_trip_type.name).await?;
.map_err(|error| match error {
models::Error::Duplicate { description } => (
StatusCode::BAD_REQUEST,
components::ErrorPage::build(&description),
),
_ => (
StatusCode::INTERNAL_SERVER_ERROR,
components::ErrorPage::build(&error.to_string()),
),
})
.await?;
Ok(Redirect::to("/trips/types/")) Ok(Redirect::to("/trips/types/"))
} }
@@ -1322,12 +1112,6 @@ async fn trips_types_edit_name(
let exists = let exists =
models::TripsType::set_name(&state.database_pool, trip_type_id, &trip_update.new_value) models::TripsType::set_name(&state.database_pool, trip_type_id, &trip_update.new_value)
.map_err(|error| {
(
StatusCode::INTERNAL_SERVER_ERROR,
components::ErrorPage::build(&error.to_string()),
)
})
.await?; .await?;
if !exists { if !exists {
@@ -1348,12 +1132,6 @@ async fn inventory_item(
Path(id): Path<Uuid>, Path(id): Path<Uuid>,
) -> Result<(StatusCode, Markup), (StatusCode, Markup)> { ) -> Result<(StatusCode, Markup), (StatusCode, Markup)> {
let item = models::InventoryItem::find(&state.database_pool, id) let item = models::InventoryItem::find(&state.database_pool, id)
.map_err(|e| {
(
StatusCode::INTERNAL_SERVER_ERROR,
components::ErrorPage::build(&e.to_string()),
)
})
.await? .await?
.ok_or(( .ok_or((
StatusCode::NOT_FOUND, StatusCode::NOT_FOUND,
@@ -1374,26 +1152,13 @@ async fn trip_category_select(
Path((trip_id, category_id)): Path<(Uuid, Uuid)>, Path((trip_id, category_id)): Path<(Uuid, Uuid)>,
) -> Result<(StatusCode, HeaderMap, Markup), (StatusCode, Markup)> { ) -> Result<(StatusCode, HeaderMap, Markup), (StatusCode, Markup)> {
let mut trip = models::Trip::find(&state.database_pool, trip_id) let mut trip = models::Trip::find(&state.database_pool, trip_id)
.await .await?
.map_err(|e| {
(
StatusCode::INTERNAL_SERVER_ERROR,
components::ErrorPage::build(&e.to_string()),
)
})?
.ok_or(( .ok_or((
StatusCode::NOT_FOUND, StatusCode::NOT_FOUND,
components::ErrorPage::build(&format!("trip with id {trip_id} not found")), components::ErrorPage::build(&format!("trip with id {trip_id} not found")),
))?; ))?;
trip.load_categories(&state.database_pool) trip.load_categories(&state.database_pool).await?;
.await
.map_err(|e| {
(
StatusCode::INTERNAL_SERVER_ERROR,
components::ErrorPage::build(&e.to_string()),
)
})?;
let active_category = trip let active_category = trip
.categories() .categories()
@@ -1421,14 +1186,7 @@ async fn inventory_category_select(
State(state): State<AppState>, State(state): State<AppState>,
Path(category_id): Path<Uuid>, Path(category_id): Path<Uuid>,
) -> Result<(StatusCode, HeaderMap, Markup), (StatusCode, Markup)> { ) -> Result<(StatusCode, HeaderMap, Markup), (StatusCode, Markup)> {
let inventory = models::Inventory::load(&state.database_pool) let inventory = models::Inventory::load(&state.database_pool).await?;
.await
.map_err(|error| {
(
StatusCode::INTERNAL_SERVER_ERROR,
components::ErrorPage::build(&error.to_string()),
)
})?;
let active_category: Option<&models::Category> = Some( let active_category: Option<&models::Category> = Some(
inventory inventory
@@ -1467,26 +1225,13 @@ async fn trip_packagelist(
Path(trip_id): Path<Uuid>, Path(trip_id): Path<Uuid>,
) -> Result<(StatusCode, Markup), (StatusCode, Markup)> { ) -> Result<(StatusCode, Markup), (StatusCode, Markup)> {
let mut trip = models::Trip::find(&state.database_pool, trip_id) let mut trip = models::Trip::find(&state.database_pool, trip_id)
.await .await?
.map_err(|e| {
(
StatusCode::INTERNAL_SERVER_ERROR,
components::ErrorPage::build(&e.to_string()),
)
})?
.ok_or(( .ok_or((
StatusCode::NOT_FOUND, StatusCode::NOT_FOUND,
components::ErrorPage::build(&format!("trip with id {trip_id} not found")), components::ErrorPage::build(&format!("trip with id {trip_id} not found")),
))?; ))?;
trip.load_categories(&state.database_pool) trip.load_categories(&state.database_pool).await?;
.await
.map_err(|e| {
(
StatusCode::INTERNAL_SERVER_ERROR,
components::ErrorPage::build(&e.to_string()),
)
})?;
Ok(( Ok((
StatusCode::OK, StatusCode::OK,
@@ -1511,13 +1256,7 @@ async fn trip_item_packagelist_set_pack_htmx(
.await?; .await?;
let item = models::TripItem::find(&state.database_pool, trip_id, item_id) let item = models::TripItem::find(&state.database_pool, trip_id, item_id)
.await .await?
.map_err(|error| {
(
StatusCode::INTERNAL_SERVER_ERROR,
components::ErrorPage::build(&error.to_string()),
)
})?
.ok_or(( .ok_or((
StatusCode::NOT_FOUND, StatusCode::NOT_FOUND,
components::ErrorPage::build(&format!("an item with id {item_id} does not exist")), components::ErrorPage::build(&format!("an item with id {item_id} does not exist")),
@@ -1545,13 +1284,7 @@ async fn trip_item_packagelist_set_unpack_htmx(
// note that this cannot fail due to a missing item, as trip_item_set_state would already // note that this cannot fail due to a missing item, as trip_item_set_state would already
// return 404. but error handling cannot hurt ;) // return 404. but error handling cannot hurt ;)
let item = models::TripItem::find(&state.database_pool, trip_id, item_id) let item = models::TripItem::find(&state.database_pool, trip_id, item_id)
.await .await?
.map_err(|error| {
(
StatusCode::INTERNAL_SERVER_ERROR,
components::ErrorPage::build(&error.to_string()),
)
})?
.ok_or(( .ok_or((
StatusCode::NOT_FOUND, StatusCode::NOT_FOUND,
components::ErrorPage::build(&format!("an item with id {item_id} does not exist")), components::ErrorPage::build(&format!("an item with id {item_id} does not exist")),

View File

@@ -7,108 +7,21 @@ use sqlx::{
Decode, Row, Decode, Row,
}; };
use std::convert; use std::convert;
use std::error;
use std::fmt; use std::fmt;
use std::num::TryFromIntError; use std::num::TryFromIntError;
use std::str::FromStr; use std::str::FromStr;
use uuid::Uuid; use uuid::Uuid;
use sqlx::{error::DatabaseError, sqlite::SqlitePoolOptions};
use futures::TryFutureExt; use futures::TryFutureExt;
use futures::TryStreamExt; use futures::TryStreamExt;
use time::{error::Parse as TimeParse, format_description::FormatItem, macros::format_description}; use time::{format_description::FormatItem, macros::format_description};
mod error;
pub use error::{DatabaseError, Error, QueryError};
pub const DATE_FORMAT: &[FormatItem<'static>] = format_description!("[year]-[month]-[day]"); pub const DATE_FORMAT: &[FormatItem<'static>] = format_description!("[year]-[month]-[day]");
pub enum Error {
Sql { description: String },
Uuid { description: String },
Enum { description: String },
Int { description: String },
Constraint { description: String },
TimeParse { description: String },
Duplicate { description: String },
NotFound { description: String },
ReferenceNotFound { description: String },
}
impl fmt::Display for Error {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
match self {
Self::Sql { description } => {
write!(f, "SQL error: {description}")
}
Self::Uuid { description } => {
write!(f, "UUID error: {description}")
}
Self::Int { description } => {
write!(f, "Integer error: {description}")
}
Self::Enum { description } => {
write!(f, "Enum error: {description}")
}
Self::TimeParse { description } => {
write!(f, "Date parse error: {description}")
}
Self::Constraint { description } => {
write!(f, "SQL constraint error: {description}")
}
Self::Duplicate { description } => {
write!(f, "Duplicate data entry: {description}")
}
Self::NotFound { description } => {
write!(f, "not found: {description}")
}
Self::ReferenceNotFound { description } => {
write!(f, "SQL foreign key reference was not found: {description}")
}
}
}
}
impl fmt::Debug for Error {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
// defer to Display
write!(f, "SQL error: {self}")
}
}
impl convert::From<uuid::Error> for Error {
fn from(value: uuid::Error) -> Self {
Error::Uuid {
description: value.to_string(),
}
}
}
impl convert::From<sqlx::Error> for Error {
fn from(value: sqlx::Error) -> Self {
Error::Sql {
description: value.to_string(),
}
}
}
impl convert::From<TryFromIntError> for Error {
fn from(value: TryFromIntError) -> Self {
Error::Int {
description: value.to_string(),
}
}
}
impl convert::From<TimeParse> for Error {
fn from(value: TimeParse) -> Self {
Error::TimeParse {
description: value.to_string(),
}
}
}
impl error::Error for Error {}
#[derive(sqlx::Type, PartialEq, PartialOrd, Deserialize)] #[derive(sqlx::Type, PartialEq, PartialOrd, Deserialize)]
pub enum TripState { pub enum TripState {
Init, Init,
@@ -176,9 +89,9 @@ impl std::convert::TryFrom<&str> for TripState {
"Review" => Self::Review, "Review" => Self::Review,
"Done" => Self::Done, "Done" => Self::Done,
_ => { _ => {
return Err(Error::Enum { return Err(Error::Database(DatabaseError::Enum {
description: format!("{value} is not a valid value for TripState"), description: format!("{value} is not a valid value for TripState"),
}) }))
} }
}) })
} }
@@ -375,7 +288,7 @@ impl TripItem {
) -> Result<Option<Self>, Error> { ) -> Result<Option<Self>, Error> {
let item_id_param = item_id.to_string(); let item_id_param = item_id.to_string();
let trip_id_param = trip_id.to_string(); let trip_id_param = trip_id.to_string();
let item: Result<Result<TripItem, Error>, sqlx::Error> = sqlx::query_as!( let item: Result<Result<TripItem, Error>, Error> = sqlx::query_as!(
DbTripsItemsRow, DbTripsItemsRow,
" "
SELECT SELECT
@@ -398,12 +311,13 @@ impl TripItem {
) )
.fetch_one(pool) .fetch_one(pool)
.map_ok(|row| row.try_into()) .map_ok(|row| row.try_into())
.await; .await
.map_err(|error| error.into());
match item { match item {
Err(e) => match e { Err(error) => match error {
sqlx::Error::RowNotFound => Ok(None), Error::Query(QueryError::NotFound { description: _ }) => Ok(None),
_ => Err(e.into()), _ => Err(error),
}, },
Ok(v) => Ok(Some(v?)), Ok(v) => Ok(Some(v?)),
} }
@@ -427,16 +341,13 @@ impl TripItem {
.bind(trip_id.to_string()) .bind(trip_id.to_string())
.bind(item_id.to_string()) .bind(item_id.to_string())
.execute(pool) .execute(pool)
.map_err(|error| Error::Sql {
description: error.to_string(),
})
.await?; .await?;
(result.rows_affected() != 0) (result.rows_affected() != 0).then_some(()).ok_or_else(|| {
.then_some(()) Error::Query(QueryError::NotFound {
.ok_or_else(|| Error::NotFound {
description: format!("item {item_id} not found for trip {trip_id}"), description: format!("item {item_id} not found for trip {trip_id}"),
}) })
})
} }
} }
@@ -553,12 +464,13 @@ impl Trip {
) )
.fetch_one(pool) .fetch_one(pool)
.map_ok(|row| row.try_into()) .map_ok(|row| row.try_into())
.map_err(|error| error.into())
.await; .await;
match trip { match trip {
Err(e) => match e { Err(error) => match error {
sqlx::Error::RowNotFound => Ok(None), Error::Query(QueryError::NotFound { description: _ }) => Ok(None),
_ => Err(e.into()), _ => Err(error),
}, },
Ok(v) => Ok(Some(v?)), Ok(v) => Ok(Some(v?)),
} }
@@ -603,45 +515,6 @@ impl Trip {
type_id_param, type_id_param,
) )
.execute(pool) .execute(pool)
.map_err(|error| match error {
sqlx::Error::Database(ref error) => {
let sqlite_error = error.downcast_ref::<sqlx::sqlite::SqliteError>();
if let Some(code) = sqlite_error.code() {
match &*code {
"787" => {
// SQLITE_CONSTRAINT_FOREIGNKEY
Error::ReferenceNotFound {
description: format!("invalid id: {}", code.to_string()),
}
}
"2067" => {
// SQLITE_CONSTRAINT_UNIQUE
Error::Duplicate {
description: format!(
"type {type_id} is already active for trip {id}"
),
}
}
_ => Error::Sql {
description: format!(
"got error with unknown code: {}",
sqlite_error.to_string()
),
},
}
} else {
Error::Sql {
description: format!(
"got error without code: {}",
sqlite_error.to_string()
),
}
}
}
_ => Error::Sql {
description: format!("got unknown error: {}", error.to_string()),
},
})
.await?; .await?;
Ok(()) Ok(())
@@ -700,16 +573,13 @@ impl Trip {
.bind(value) .bind(value)
.bind(trip_id.to_string()) .bind(trip_id.to_string())
.execute(pool) .execute(pool)
.map_err(|error| Error::Sql {
description: error.to_string(),
})
.await?; .await?;
(result.rows_affected() != 0) (result.rows_affected() != 0).then_some(()).ok_or_else(|| {
.then_some(()) Error::Query(QueryError::NotFound {
.ok_or_else(|| Error::NotFound {
description: format!("trip {trip_id} not found"), description: format!("trip {trip_id} not found"),
}) })
})
} }
pub async fn save( pub async fn save(
@@ -720,14 +590,8 @@ impl Trip {
) -> Result<Uuid, Error> { ) -> Result<Uuid, Error> {
let id = Uuid::new_v4(); let id = Uuid::new_v4();
let id_param = id.to_string(); let id_param = id.to_string();
let date_start = date_start let date_start = date_start.format(DATE_FORMAT)?;
.format(DATE_FORMAT) let date_end = date_end.format(DATE_FORMAT)?;
.map_err(|e| Error::TimeParse {
description: e.to_string(),
})?;
let date_end = date_end.format(DATE_FORMAT).map_err(|e| Error::TimeParse {
description: e.to_string(),
})?;
let trip_state = TripState::new(); let trip_state = TripState::new();
@@ -743,42 +607,7 @@ impl Trip {
trip_state, trip_state,
) )
.execute(pool) .execute(pool)
.await .await?;
.map_err(|e| match e {
sqlx::Error::Database(ref error) => {
let sqlite_error = error.downcast_ref::<sqlx::sqlite::SqliteError>();
if let Some(code) = sqlite_error.code() {
match &*code {
// SQLITE_CONSTRAINT_FOREIGNKEY
"787" => Error::Constraint {
description: format!(
"SQLITE_CONSTRAINT_FOREIGNKEY on table without foreignkey?",
),
},
// SQLITE_CONSTRAINT_UNIQUE
"2067" => Error::Constraint {
description: format!("trip with name \"{name}\" already exists",),
},
_ => Error::Sql {
description: format!(
"got error with unknown code: {}",
sqlite_error.to_string()
),
},
}
} else {
Error::Sql {
description: format!(
"got error without code: {}",
sqlite_error.to_string()
),
}
}
}
_ => Error::Sql {
description: format!("got unknown error: {}", e.to_string()),
},
})?;
Ok(id) Ok(id)
} }
@@ -806,12 +635,13 @@ impl Trip {
) )
.fetch_one(pool) .fetch_one(pool)
.map_ok(|row| row.total_weight.map(|weight| weight as i64)) .map_ok(|row| row.total_weight.map(|weight| weight as i64))
.map_err(|error| error.into())
.await; .await;
match weight { match weight {
Err(e) => match e { Err(error) => match error {
sqlx::Error::RowNotFound => Ok(None), Error::Query(QueryError::NotFound { description: _ }) => Ok(None),
_ => Err(e.into()), _ => Err(error.into()),
}, },
Ok(v) => Ok(v), Ok(v) => Ok(v),
} }
@@ -1100,39 +930,6 @@ impl TripsType {
name, name,
) )
.execute(pool) .execute(pool)
.map_err(|error| match error {
sqlx::Error::Database(ref error) => {
let sqlite_error = error.downcast_ref::<sqlx::sqlite::SqliteError>();
if let Some(code) = sqlite_error.code() {
match &*code {
"2067" => {
// SQLITE_CONSTRAINT_UNIQUE
Error::Duplicate {
description: format!(
"trip type with name \"{name}\" already exists"
),
}
}
_ => Error::Sql {
description: format!(
"got error with unknown code: {}",
sqlite_error.to_string()
),
},
}
} else {
Error::Sql {
description: format!(
"got error without code: {}",
sqlite_error.to_string()
),
}
}
}
_ => Error::Sql {
description: format!("got unknown error: {}", error.to_string()),
},
})
.await?; .await?;
Ok(id) Ok(id)
@@ -1200,7 +997,7 @@ impl Category {
id: Uuid, id: Uuid,
) -> Result<Option<Category>, Error> { ) -> Result<Option<Category>, Error> {
let id_param = id.to_string(); let id_param = id.to_string();
let item: Result<Result<Category, Error>, sqlx::Error> = sqlx::query_as!( let item = sqlx::query_as!(
DbCategoryRow, DbCategoryRow,
"SELECT "SELECT
id, id,
@@ -1212,12 +1009,13 @@ impl Category {
) )
.fetch_one(pool) .fetch_one(pool)
.map_ok(|row| row.try_into()) .map_ok(|row| row.try_into())
.map_err(|error| error.into())
.await; .await;
match item { match item {
Err(e) => match e { Err(error) => match error {
sqlx::Error::RowNotFound => Ok(None), Error::Query(QueryError::NotFound { description: _ }) => Ok(None),
_ => Err(e.into()), _ => Err(error),
}, },
Ok(v) => Ok(Some(v?)), Ok(v) => Ok(Some(v?)),
} }
@@ -1235,39 +1033,6 @@ impl Category {
name, name,
) )
.execute(pool) .execute(pool)
.map_err(|error| match error {
sqlx::Error::Database(ref error) => {
let sqlite_error = error.downcast_ref::<sqlx::sqlite::SqliteError>();
if let Some(code) = sqlite_error.code() {
match &*code {
"2067" => {
// SQLITE_CONSTRAINT_UNIQUE
Error::Duplicate {
description: format!(
"inventory item category with name \"{name}\" already exists"
),
}
}
_ => Error::Sql {
description: format!(
"got error with unknown code: {}",
sqlite_error.to_string()
),
},
}
} else {
Error::Sql {
description: format!(
"got error without code: {}",
sqlite_error.to_string()
),
}
}
}
_ => Error::Sql {
description: format!("got unknown error: {}", error.to_string()),
},
})
.await?; .await?;
Ok(id) Ok(id)
@@ -1335,7 +1100,7 @@ impl TryFrom<DbInventoryItemsRow> for Item {
impl Item { impl Item {
pub async fn find(pool: &sqlx::Pool<sqlx::Sqlite>, id: Uuid) -> Result<Option<Item>, Error> { pub async fn find(pool: &sqlx::Pool<sqlx::Sqlite>, id: Uuid) -> Result<Option<Item>, Error> {
let id_param = id.to_string(); let id_param = id.to_string();
let item: Result<Result<Item, Error>, sqlx::Error> = sqlx::query_as!( let item = sqlx::query_as!(
DbInventoryItemsRow, DbInventoryItemsRow,
"SELECT "SELECT
id, id,
@@ -1348,13 +1113,14 @@ impl Item {
id_param, id_param,
) )
.fetch_one(pool) .fetch_one(pool)
.map_err(|error| error.into())
.map_ok(|row| row.try_into()) .map_ok(|row| row.try_into())
.await; .await;
match item { match item {
Err(e) => match e { Err(error) => match error {
sqlx::Error::RowNotFound => Ok(None), Error::Query(QueryError::NotFound { description: _ }) => Ok(None),
_ => Err(e.into()), _ => Err(error),
}, },
Ok(v) => Ok(Some(v?)), Ok(v) => Ok(Some(v?)),
} }
@@ -1367,7 +1133,7 @@ impl Item {
weight: i64, weight: i64,
) -> Result<Option<Uuid>, Error> { ) -> Result<Option<Uuid>, Error> {
let id_param = id.to_string(); let id_param = id.to_string();
let id: Result<Result<Uuid, Error>, sqlx::Error> = sqlx::query!( let id = sqlx::query!(
"UPDATE inventory_items AS item "UPDATE inventory_items AS item
SET SET
name = ?, name = ?,
@@ -1383,15 +1149,16 @@ impl Item {
.map_ok(|row| { .map_ok(|row| {
let id: &str = &row.id.unwrap(); // TODO let id: &str = &row.id.unwrap(); // TODO
let uuid: Result<Uuid, uuid::Error> = Uuid::try_parse(id); let uuid: Result<Uuid, uuid::Error> = Uuid::try_parse(id);
let uuid: Result<Uuid, Error> = uuid.map_err(|e| e.into()); let uuid: Result<Uuid, Error> = uuid.map_err(|error| error.into());
uuid uuid
}) })
.map_err(|error| error.into())
.await; .await;
match id { match id {
Err(e) => match e { Err(error) => match error {
sqlx::Error::RowNotFound => Ok(None), Error::Query(QueryError::NotFound { description: _ }) => Ok(None),
_ => Err(e.into()), _ => Err(error.into()),
}, },
Ok(v) => Ok(Some(v?)), Ok(v) => Ok(Some(v?)),
} }
@@ -1402,7 +1169,7 @@ impl Item {
category_id: Uuid, category_id: Uuid,
) -> Result<i64, Error> { ) -> Result<i64, Error> {
let category_id_param = category_id.to_string(); let category_id_param = category_id.to_string();
let weight: Result<i64, sqlx::Error> = sqlx::query!( let weight = sqlx::query!(
" "
SELECT COALESCE(MAX(i_item.weight), 0) as weight SELECT COALESCE(MAX(i_item.weight), 0) as weight
FROM inventory_items_categories as category FROM inventory_items_categories as category
@@ -1420,9 +1187,9 @@ impl Item {
// We can be certain that the row exists, as we COALESCE it // We can be certain that the row exists, as we COALESCE it
row.weight.unwrap() as i64 row.weight.unwrap() as i64
}) })
.await; .await?;
Ok(weight?) Ok(weight)
} }
pub async fn _get_category_total_picked_weight( pub async fn _get_category_total_picked_weight(
@@ -1430,7 +1197,7 @@ impl Item {
category_id: Uuid, category_id: Uuid,
) -> Result<i64, Error> { ) -> Result<i64, Error> {
let category_id_param = category_id.to_string(); let category_id_param = category_id.to_string();
let weight: Result<i64, sqlx::Error> = sqlx::query!( let weight: Result<i64, Error> = sqlx::query!(
" "
SELECT COALESCE(SUM(i_item.weight), 0) as weight SELECT COALESCE(SUM(i_item.weight), 0) as weight
FROM inventory_items_categories as category FROM inventory_items_categories as category
@@ -1451,6 +1218,7 @@ impl Item {
// We can be certain that the row exists, as we COALESCE it // We can be certain that the row exists, as we COALESCE it
row.weight.unwrap() as i64 row.weight.unwrap() as i64
}) })
.map_err(|error| error.into())
.await; .await;
Ok(weight?) Ok(weight?)
@@ -1548,7 +1316,7 @@ impl InventoryItem {
pub async fn find(pool: &sqlx::Pool<sqlx::Sqlite>, id: Uuid) -> Result<Option<Self>, Error> { pub async fn find(pool: &sqlx::Pool<sqlx::Sqlite>, id: Uuid) -> Result<Option<Self>, Error> {
let id_param = id.to_string(); let id_param = id.to_string();
let item: Result<Result<InventoryItem, Error>, sqlx::Error> = sqlx::query_as!( let item = sqlx::query_as!(
DbInventoryItemRow, DbInventoryItemRow,
"SELECT "SELECT
item.id AS id, item.id AS id,
@@ -1572,19 +1340,20 @@ impl InventoryItem {
) )
.fetch_one(pool) .fetch_one(pool)
.map_ok(|row| row.try_into()) .map_ok(|row| row.try_into())
.map_err(|error| error.into())
.await; .await;
match item { match item {
Err(e) => match e { Err(error) => match error {
sqlx::Error::RowNotFound => Ok(None), Error::Query(QueryError::NotFound { description: _ }) => Ok(None),
_ => Err(e.into()), _ => Err(error.into()),
}, },
Ok(v) => Ok(Some(v?)), Ok(v) => Ok(Some(v?)),
} }
} }
pub async fn name_exists(pool: &sqlx::Pool<sqlx::Sqlite>, name: &str) -> Result<bool, Error> { pub async fn name_exists(pool: &sqlx::Pool<sqlx::Sqlite>, name: &str) -> Result<bool, Error> {
let item: Result<(), sqlx::Error> = sqlx::query!( let item = sqlx::query!(
"SELECT id "SELECT id
FROM inventory_items FROM inventory_items
WHERE name = ?", WHERE name = ?",
@@ -1592,12 +1361,13 @@ impl InventoryItem {
) )
.fetch_one(pool) .fetch_one(pool)
.map_ok(|_row| ()) .map_ok(|_row| ())
.map_err(|error| error.into())
.await; .await;
match item { match item {
Err(e) => match e { Err(error) => match error {
sqlx::Error::RowNotFound => Ok(false), Error::Query(QueryError::NotFound { description: _ }) => Ok(false),
_ => Err(e.into()), _ => Err(error.into()),
}, },
Ok(_) => Ok(true), Ok(_) => Ok(true),
} }
@@ -1611,24 +1381,7 @@ impl InventoryItem {
id_param id_param
) )
.execute(pool) .execute(pool)
.map_err(|error| match error { .await?;
sqlx::Error::Database(ref error) => {
let sqlite_error = error.downcast_ref::<sqlx::sqlite::SqliteError>();
if let Some(code) = sqlite_error.code() {
match &*code {
"787" => {
// SQLITE_CONSTRAINT_FOREIGNKEY
Error::Constraint { description: format!("item {} cannot be deleted because it's on use in trips. instead, archive it", code.to_string()) }
}
_ =>
Error::Sql { description: format!("got error with unknown code: {}", sqlite_error.to_string()) }
}
} else {
Error::Constraint { description: format!("got error without code: {}", sqlite_error.to_string()) }
}
}
_ => Error::Constraint { description: format!("got unknown error: {}", error.to_string()) }
}).await?;
Ok(results.rows_affected() != 0) Ok(results.rows_affected() != 0)
} }
@@ -1655,44 +1408,7 @@ impl InventoryItem {
category_id_param category_id_param
) )
.execute(pool) .execute(pool)
.await .await?;
.map_err(|e| match e {
sqlx::Error::Database(ref error) => {
let sqlite_error = error.downcast_ref::<sqlx::sqlite::SqliteError>();
if let Some(code) = sqlite_error.code() {
match &*code {
// SQLITE_CONSTRAINT_FOREIGNKEY
"787" => Error::Constraint {
description: format!(
"category {category_id} not found",
),
},
// SQLITE_CONSTRAINT_UNIQUE
"2067" => Error::Constraint {
description: format!(
"item with name \"{name}\" already exists in category {category_id}",
),
},
_ => Error::Sql {
description: format!(
"got error with unknown code: {}",
sqlite_error.to_string()
),
},
}
} else {
Error::Sql {
description: format!(
"got error without code: {}",
sqlite_error.to_string()
),
}
}
}
_ => Error::Sql {
description: format!("got unknown error: {}", e.to_string()),
},
})?;
Ok(id) Ok(id)
} }
@@ -1714,15 +1430,19 @@ impl Inventory {
.await .await
// we have two error handling lines here. these are distinct errors // we have two error handling lines here. these are distinct errors
// this one is the SQL error that may arise during the query // this one is the SQL error that may arise during the query
.map_err(|e| Error::Sql { .map_err(|error| {
description: e.to_string(), Error::Database(DatabaseError::Sql {
description: error.to_string(),
})
})? })?
.into_iter() .into_iter()
.collect::<Result<Vec<Category>, Error>>() .collect::<Result<Vec<Category>, Error>>()
// and this one is the model mapping error that may arise e.g. during // and this one is the model mapping error that may arise e.g. during
// reading of the rows // reading of the rows
.map_err(|e| Error::Sql { .map_err(|error| {
description: e.to_string(), Error::Database(DatabaseError::Sql {
description: error.to_string(),
})
})?; })?;
for category in &mut categories { for category in &mut categories {

165
rust/src/models/error.rs Normal file
View File

@@ -0,0 +1,165 @@
use std::convert;
use std::fmt;
use sqlx::error::DatabaseError as _;
pub enum DatabaseError {
/// Errors we can receive **from** the database that are caused by connection
/// problems or schema problems (e.g. we get a return value that does not fit our enum,
/// or a wrongly formatted date)
Sql {
description: String,
},
Uuid {
description: String,
},
Enum {
description: String,
},
TimeParse {
description: String,
},
}
impl fmt::Display for DatabaseError {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
match self {
Self::Sql { description } => {
write!(f, "SQL error: {description}")
}
Self::Uuid { description } => {
write!(f, "UUID error: {description}")
}
Self::Enum { description } => {
write!(f, "Enum error: {description}")
}
Self::TimeParse { description } => {
write!(f, "Date parse error: {description}")
}
}
}
}
pub enum QueryError {
/// Errors that are caused by wrong input data, e.g. ids that cannot be found, or
/// inserts that violate unique constraints
Constraint {
description: String,
},
Duplicate {
description: String,
},
NotFound {
description: String,
},
ReferenceNotFound {
description: String,
},
}
impl fmt::Display for QueryError {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
match self {
Self::Constraint { description } => {
write!(f, "SQL constraint error: {description}")
}
Self::Duplicate { description } => {
write!(f, "Duplicate data entry: {description}")
}
Self::NotFound { description } => {
write!(f, "not found: {description}")
}
Self::ReferenceNotFound { description } => {
write!(f, "SQL foreign key reference was not found: {description}")
}
}
}
}
pub enum Error {
Database(DatabaseError),
Query(QueryError),
}
impl fmt::Display for Error {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
match self {
Self::Database(error) => write!(f, "{}", error),
Self::Query(error) => write!(f, "{}", error),
}
}
}
impl fmt::Debug for Error {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
// defer to Display
write!(f, "SQL error: {self}")
}
}
impl convert::From<uuid::Error> for Error {
fn from(value: uuid::Error) -> Self {
Error::Database(DatabaseError::Uuid {
description: value.to_string(),
})
}
}
impl convert::From<time::error::Format> for Error {
fn from(value: time::error::Format) -> Self {
Error::Database(DatabaseError::TimeParse {
description: value.to_string(),
})
}
}
impl convert::From<sqlx::Error> for Error {
fn from(value: sqlx::Error) -> Self {
match value {
sqlx::Error::RowNotFound => Error::Query(QueryError::NotFound {
description: value.to_string(),
}),
sqlx::Error::Database(ref error) => {
let sqlite_error = error.downcast_ref::<sqlx::sqlite::SqliteError>();
if let Some(code) = sqlite_error.code() {
match &*code {
// SQLITE_CONSTRAINT_FOREIGNKEY
"787" => Error::Query(QueryError::Constraint {
description: format!("foreign key reference not found"),
}),
// SQLITE_CONSTRAINT_UNIQUE
"2067" => Error::Query(QueryError::Constraint {
description: format!("item with unique constraint already exists",),
}),
_ => Error::Database(DatabaseError::Sql {
description: format!(
"got error with unknown code: {}",
sqlite_error.to_string()
),
}),
}
} else {
Error::Database(DatabaseError::Sql {
description: format!(
"got error without code: {}",
sqlite_error.to_string()
),
})
}
}
_ => Error::Database(DatabaseError::Sql {
description: format!("got unknown error: {}", value.to_string()),
}),
}
}
}
impl convert::From<time::error::Parse> for Error {
fn from(value: time::error::Parse) -> Self {
Error::Database(DatabaseError::TimeParse {
description: value.to_string(),
})
}
}
impl std::error::Error for Error {}