use std::fmt; use super::{ consts, error::{DatabaseError, Error, QueryError}, inventory, }; use crate::Context; use futures::{TryFutureExt, TryStreamExt}; use serde::{Deserialize, Serialize}; use serde_variant::to_variant_name; use time; use uuid::Uuid; #[derive(sqlx::Type, PartialEq, PartialOrd, Deserialize)] pub enum TripState { Init, Planning, Planned, Active, Review, Done, } impl TripState { pub fn new() -> Self { TripState::Init } pub fn next(&self) -> Option { match self { Self::Init => Some(Self::Planning), Self::Planning => Some(Self::Planned), Self::Planned => Some(Self::Active), Self::Active => Some(Self::Review), Self::Review => Some(Self::Done), Self::Done => None, } } pub fn prev(&self) -> Option { match self { Self::Init => None, Self::Planning => Some(Self::Init), Self::Planned => Some(Self::Planning), Self::Active => Some(Self::Planned), Self::Review => Some(Self::Active), Self::Done => Some(Self::Review), } } } impl fmt::Display for TripState { fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result { write!( f, "{}", match self { Self::Init => "Init", Self::Planning => "Planning", Self::Planned => "Planned", Self::Active => "Active", Self::Review => "Review", Self::Done => "Done", }, ) } } impl std::convert::TryFrom<&str> for TripState { type Error = Error; fn try_from(value: &str) -> Result { Ok(match value { "Init" => Self::Init, "Planning" => Self::Planning, "Planned" => Self::Planned, "Active" => Self::Active, "Review" => Self::Review, "Done" => Self::Done, _ => { return Err(Error::Database(DatabaseError::Enum { description: format!("{value} is not a valid value for TripState"), })) } }) } } #[derive(Serialize, Debug)] pub enum TripItemStateKey { Pick, Pack, Ready, } impl fmt::Display for TripItemStateKey { fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result { write!( f, "{}", match self { Self::Pick => "pick", Self::Pack => "pack", Self::Ready => "ready", }, ) } } #[derive(Debug)] pub struct TripCategory { pub category: inventory::Category, pub items: Option>, } impl TripCategory { pub fn total_picked_weight(&self) -> i64 { self.items .as_ref() .unwrap() .iter() .filter(|item| item.picked) .map(|item| item.item.weight) .sum() } pub async fn find( ctx: &Context, pool: &sqlx::Pool, trip_id: Uuid, category_id: Uuid, ) -> Result, Error> { let mut category: Option = None; let user_id = ctx.user.id.to_string(); let trip_id_param = trip_id.to_string(); let category_id_param = category_id.to_string(); sqlx::query!( " SELECT category.id as category_id, category.name as category_name, category.description AS category_description, inner.trip_id AS trip_id, inner.item_id AS item_id, inner.item_name AS item_name, inner.item_description AS item_description, inner.item_weight AS item_weight, inner.item_is_picked AS item_is_picked, inner.item_is_packed AS item_is_packed, inner.item_is_ready AS item_is_ready, inner.item_is_new AS item_is_new FROM inventory_items_categories AS category LEFT JOIN ( SELECT trip.trip_id AS trip_id, category.id as category_id, category.name as category_name, category.description as category_description, item.id as item_id, item.name as item_name, item.description as item_description, item.weight as item_weight, trip.pick as item_is_picked, trip.pack as item_is_packed, trip.ready as item_is_ready, trip.new as item_is_new FROM trips_items as trip INNER JOIN inventory_items as item ON item.id = trip.item_id INNER JOIN inventory_items_categories as category ON category.id = item.category_id WHERE trip.trip_id = ? AND trip.user_id = ? ) AS inner ON inner.category_id = category.id WHERE category.id = ? ", trip_id_param, user_id, category_id_param ) .fetch(pool) .map_ok(|row| -> Result<(), Error> { match &category { Some(_) => (), None => { category = Some(TripCategory { category: inventory::Category { id: Uuid::try_parse(&row.category_id)?, name: row.category_name, description: row.category_description, items: None, }, items: None, }) } }; match row.item_id { None => { // we have an empty (unused) category which has NULL values // for the item_id column category.as_mut().unwrap().items = Some(vec![]); category.as_mut().unwrap().category.items = Some(vec![]); } Some(item_id) => { let item = TripItem { item: inventory::Item { id: Uuid::try_parse(&item_id)?, name: row.item_name.unwrap(), description: row.item_description, weight: row.item_weight.unwrap(), category_id: category.as_ref().unwrap().category.id, }, picked: row.item_is_picked.unwrap(), packed: row.item_is_packed.unwrap(), ready: row.item_is_ready.unwrap(), new: row.item_is_new.unwrap(), }; match &mut category.as_mut().unwrap().items { None => category.as_mut().unwrap().items = Some(vec![item]), Some(ref mut items) => items.push(item), } } } Ok(()) }) .try_collect::>>() .await? .into_iter() .collect::>()?; // this may be None if there are no results (which // means that the category was not found) Ok(category) } } #[derive(Debug)] pub struct TripItem { pub item: inventory::Item, pub picked: bool, pub packed: bool, pub ready: bool, pub new: bool, } pub struct DbTripsItemsRow { pub picked: bool, pub packed: bool, pub ready: bool, pub new: bool, pub id: String, pub name: String, pub weight: i64, pub description: Option, pub category_id: String, } impl TryFrom for TripItem { type Error = Error; fn try_from(row: DbTripsItemsRow) -> Result { Ok(TripItem { picked: row.picked, packed: row.packed, ready: row.ready, new: row.new, item: inventory::Item { id: Uuid::try_parse(&row.id)?, name: row.name, description: row.description, weight: row.weight, category_id: Uuid::try_parse(&row.category_id)?, }, }) } } impl TripItem { pub async fn find( ctx: &Context, pool: &sqlx::Pool, trip_id: Uuid, item_id: Uuid, ) -> Result, Error> { let user_id = ctx.user.id.to_string(); let item_id_param = item_id.to_string(); let trip_id_param = trip_id.to_string(); sqlx::query_as!( DbTripsItemsRow, " SELECT t_item.item_id AS id, t_item.pick AS picked, t_item.pack AS packed, t_item.ready AS ready, t_item.new AS new, i_item.name AS name, i_item.description AS description, i_item.weight AS weight, i_item.category_id AS category_id FROM trips_items AS t_item INNER JOIN inventory_items AS i_item ON i_item.id = t_item.item_id WHERE t_item.item_id = ? AND t_item.trip_id = ? AND t_item.user_id = ? ", item_id_param, trip_id_param, user_id, ) .fetch_optional(pool) .await? .map(|row| row.try_into()) .transpose() } pub async fn set_state( ctx: &Context, pool: &sqlx::Pool, trip_id: Uuid, item_id: Uuid, key: TripItemStateKey, value: bool, ) -> Result<(), Error> { let user_id = ctx.user.id.to_string(); let result = sqlx::query(&format!( "UPDATE trips_items SET {key} = ? WHERE trip_id = ? AND item_id = ? AND user_id = ?", key = to_variant_name(&key).unwrap() )) .bind(value) .bind(trip_id.to_string()) .bind(item_id.to_string()) .bind(user_id) .execute(pool) .await?; (result.rows_affected() != 0).then_some(()).ok_or_else(|| { Error::Query(QueryError::NotFound { description: format!("item {item_id} not found for trip {trip_id}"), }) }) } } pub struct DbTripRow { pub id: String, pub name: String, pub date_start: String, pub date_end: String, pub state: String, pub location: Option, pub temp_min: Option, pub temp_max: Option, pub comment: Option, } impl TryFrom for Trip { type Error = Error; fn try_from(row: DbTripRow) -> Result { Ok(Trip { id: Uuid::try_parse(&row.id)?, name: row.name, date_start: time::Date::parse(&row.date_start, consts::DATE_FORMAT)?, date_end: time::Date::parse(&row.date_end, consts::DATE_FORMAT)?, state: row.state.as_str().try_into()?, location: row.location, temp_min: row.temp_min, temp_max: row.temp_max, comment: row.comment, types: None, categories: None, }) } } pub struct Trip { pub id: Uuid, pub name: String, pub date_start: time::Date, pub date_end: time::Date, pub state: TripState, pub location: Option, pub temp_min: Option, pub temp_max: Option, pub comment: Option, pub(crate) types: Option>, pub(crate) categories: Option>, } #[derive(Clone, Debug, Serialize, Deserialize, PartialEq)] pub enum TripAttribute { #[serde(rename = "name")] Name, #[serde(rename = "date_start")] DateStart, #[serde(rename = "date_end")] DateEnd, #[serde(rename = "location")] Location, #[serde(rename = "temp_min")] TempMin, #[serde(rename = "temp_max")] TempMax, } pub(crate) struct DbTripWeightRow { pub total_weight: Option, } impl Trip { pub async fn all(ctx: &Context, pool: &sqlx::Pool) -> Result, Error> { let user_id = ctx.user.id.to_string(); sqlx::query_as!( DbTripRow, "SELECT id, name, CAST (date_start AS TEXT) date_start, CAST (date_end AS TEXT) date_end, state, location, temp_min, temp_max, comment FROM trips WHERE user_id = ?", user_id ) .fetch(pool) .map_ok(|row| row.try_into()) .try_collect::>>() .await? .into_iter() .collect::, Error>>() } pub async fn find( ctx: &Context, pool: &sqlx::Pool, trip_id: Uuid, ) -> Result, Error> { let trip_id_param = trip_id.to_string(); let user_id = ctx.user.id.to_string(); sqlx::query_as!( DbTripRow, "SELECT id, name, CAST (date_start AS TEXT) date_start, CAST (date_end AS TEXT) date_end, state, location, temp_min, temp_max, comment FROM trips WHERE id = ? and user_id = ?", trip_id_param, user_id, ) .fetch_optional(pool) .await? .map(|row| row.try_into()) .transpose() } pub async fn trip_type_remove( ctx: &Context, pool: &sqlx::Pool, id: Uuid, type_id: Uuid, ) -> Result { let user_id = ctx.user.id.to_string(); let id_param = id.to_string(); let type_id_param = type_id.to_string(); let results = sqlx::query!( "DELETE FROM trips_to_trips_types AS ttt WHERE ttt.trip_id = ? AND ttt.trip_type_id = ? AND EXISTS(SELECT * FROM trips WHERE id = ? AND user_id = ?) AND EXISTS(SELECT * FROM trips_types WHERE id = ? AND user_id = ?) ", id_param, type_id_param, id_param, user_id, type_id_param, user_id, ) .execute(pool) .await?; Ok(results.rows_affected() != 0) } pub async fn trip_type_add( ctx: &Context, pool: &sqlx::Pool, id: Uuid, type_id: Uuid, ) -> Result<(), Error> { let user_id = ctx.user.id.to_string(); // TODO user handling? let trip_id_param = id.to_string(); let type_id_param = type_id.to_string(); sqlx::query!( "INSERT INTO trips_to_trips_types (trip_id, trip_type_id) SELECT trips.id as trip_id, trips_types.id as trip_type_id FROM trips INNER JOIN trips_types WHERE trips.id = ? AND trips.user_id = ? AND trips_types.id = ? AND trips_types.user_id = ?", trip_id_param, user_id, type_id_param, user_id, ) .execute(pool) .await?; Ok(()) } pub async fn set_state( ctx: &Context, pool: &sqlx::Pool, id: Uuid, new_state: &TripState, ) -> Result { let user_id = ctx.user.id.to_string(); let trip_id_param = id.to_string(); let result = sqlx::query!( "UPDATE trips SET state = ? WHERE id = ? and user_id = ?", new_state, trip_id_param, user_id, ) .execute(pool) .await?; Ok(result.rows_affected() != 0) } pub async fn set_comment( ctx: &Context, pool: &sqlx::Pool, id: Uuid, new_comment: &str, ) -> Result { let user_id = ctx.user.id.to_string(); let trip_id_param = id.to_string(); let result = sqlx::query!( "UPDATE trips SET comment = ? WHERE id = ? AND user_id = ?", new_comment, trip_id_param, user_id, ) .execute(pool) .await?; Ok(result.rows_affected() != 0) } pub async fn set_attribute( ctx: &Context, pool: &sqlx::Pool, trip_id: Uuid, attribute: TripAttribute, value: &str, ) -> Result<(), Error> { let user_id = ctx.user.id.to_string(); let result = sqlx::query(&format!( "UPDATE trips SET {attribute} = ? WHERE id = ? AND user_id = ?", attribute = to_variant_name(&attribute).unwrap() )) .bind(value) .bind(trip_id.to_string()) .bind(user_id) .execute(pool) .await?; (result.rows_affected() != 0).then_some(()).ok_or_else(|| { Error::Query(QueryError::NotFound { description: format!("trip {trip_id} not found"), }) }) } pub async fn save( ctx: &Context, pool: &sqlx::Pool, name: &str, date_start: time::Date, date_end: time::Date, ) -> Result { let user_id = ctx.user.id.to_string(); let id = Uuid::new_v4(); let id_param = id.to_string(); let date_start = date_start.format(consts::DATE_FORMAT)?; let date_end = date_end.format(consts::DATE_FORMAT)?; let trip_state = TripState::new(); sqlx::query!( "INSERT INTO trips (id, name, date_start, date_end, state, user_id) VALUES (?, ?, ?, ?, ?, ?)", id_param, name, date_start, date_end, trip_state, user_id, ) .execute(pool) .await?; Ok(id) } pub async fn find_total_picked_weight( ctx: &Context, pool: &sqlx::Pool, trip_id: Uuid, ) -> Result { let user_id = ctx.user.id.to_string(); let trip_id_param = trip_id.to_string(); let weight = sqlx::query_as!( DbTripWeightRow, " SELECT CAST(IFNULL(SUM(i_item.weight), 0) AS INTEGER) AS total_weight FROM trips AS trip INNER JOIN trips_items AS t_item ON t_item.trip_id = trip.id INNER JOIN inventory_items AS i_item ON t_item.item_id = i_item.id WHERE trip.id = ? AND trip.user_id = ? AND t_item.pick = true ", trip_id_param, user_id, ) .fetch_one(pool) .map_ok(|row| row.total_weight.unwrap() as i64) .await?; Ok(weight) } pub fn types(&self) -> &Vec { self.types .as_ref() .expect("you need to call load_trips_types()") } pub fn categories(&self) -> &Vec { self.categories .as_ref() .expect("you need to call load_trips_types()") } pub fn total_picked_weight(&self) -> i64 { self.categories() .iter() .map(|category| -> i64 { category .items .as_ref() .unwrap() .iter() .filter_map(|item| Some(item.item.weight).filter(|_| item.picked)) .sum::() }) .sum::() } pub async fn load_trips_types( &mut self, ctx: &Context, pool: &sqlx::Pool, ) -> Result<(), Error> { let user_id = ctx.user.id.to_string(); let id = self.id.to_string(); let types = sqlx::query!( " SELECT type.id as id, type.name as name, inner.id IS NOT NULL AS active FROM trips_types AS type LEFT JOIN ( SELECT type.id as id, trip.user_id as user_id FROM trips as trip INNER JOIN trips_to_trips_types as ttt ON ttt.trip_id = trip.id INNER JOIN trips_types AS type ON type.id == ttt.trip_type_id WHERE trip.id = ? AND trip.user_id = ? ) AS inner ON inner.id = type.id WHERE type.user_id = ? ", id, user_id, user_id, ) .fetch(pool) .map_ok(|row| -> Result { Ok(TripType { id: Uuid::try_parse(&row.id)?, name: row.name, active: match row.active { 0 => false, 1 => true, _ => unreachable!(), }, }) }) .try_collect::>>() .await? .into_iter() .collect::, Error>>()?; self.types = Some(types); Ok(()) } pub async fn sync_trip_items_with_inventory( &mut self, ctx: &Context, pool: &sqlx::Pool, ) -> Result<(), Error> { // we need to get all items that are part of the inventory but not // part of the trip items // // then, we know which items we need to sync. there are different // states for them: // // * if the trip is new (it's state is INITIAL), we can just forward // as-is // * if the trip is new, we have to make these new items prominently // visible so the user knows that there might be new items to // consider let user_id = ctx.user.id.to_string(); let trip_id = self.id.to_string(); let unsynced_items: Vec = sqlx::query!( " SELECT i_item.id AS item_id FROM inventory_items AS i_item LEFT JOIN ( SELECT t_item.item_id AS item_id, t_item.user_id AS user_id FROM trips_items AS t_item WHERE t_item.trip_id = ? AND t_item.user_id = ? ) AS t_item ON t_item.item_id = i_item.id WHERE t_item.item_id IS NULL AND i_item.user_id = ?", trip_id, user_id, user_id, ) .fetch(pool) .map_ok(|row| -> Result { Ok(Uuid::try_parse(&row.item_id)?) }) .try_collect::>>() .await? .into_iter() .collect::, Error>>()?; // looks like there is currently no nice way to do multiple inserts // with sqlx. whatever, this won't matter // only mark as new when the trip is already underway let mark_as_new = self.state != TripState::new(); for unsynced_item in &unsynced_items { let item_id = unsynced_item.to_string(); sqlx::query!( " INSERT INTO trips_items ( item_id, trip_id, pick, pack, ready, new, user_id ) VALUES (?, ?, ?, ?, ?, ?, ?) ", item_id, trip_id, false, false, false, mark_as_new, user_id, ) .execute(pool) .await?; } tracing::info!("unsynced items: {:?}", &unsynced_items); Ok(()) } pub async fn load_categories( &mut self, ctx: &Context, pool: &sqlx::Pool, ) -> Result<(), Error> { let mut categories: Vec = vec![]; // we can ignore the return type as we collect into `categories` // in the `map_ok()` closure let id = self.id.to_string(); let user_id = ctx.user.id.to_string(); sqlx::query!( " SELECT category.id as category_id, category.name as category_name, category.description AS category_description, inner.trip_id AS trip_id, inner.item_id AS item_id, inner.item_name AS item_name, inner.item_description AS item_description, inner.item_weight AS item_weight, inner.item_is_picked AS item_is_picked, inner.item_is_packed AS item_is_packed, inner.item_is_ready AS item_is_ready, inner.item_is_new AS item_is_new FROM inventory_items_categories AS category LEFT JOIN ( SELECT trip.trip_id AS trip_id, category.id as category_id, category.name as category_name, category.description as category_description, item.id as item_id, item.name as item_name, item.description as item_description, item.weight as item_weight, trip.pick as item_is_picked, trip.pack as item_is_packed, trip.ready as item_is_ready, trip.new as item_is_new, trip.user_id as user_id FROM trips_items as trip INNER JOIN inventory_items as item ON item.id = trip.item_id INNER JOIN inventory_items_categories as category ON category.id = item.category_id WHERE trip.trip_id = ? AND trip.user_id = ? ) AS inner ON inner.category_id = category.id WHERE category.user_id = ? ", id, user_id, user_id, ) .fetch(pool) .map_ok(|row| -> Result<(), Error> { let mut category = TripCategory { category: inventory::Category { id: Uuid::try_parse(&row.category_id)?, name: row.category_name, description: row.category_description, items: None, }, items: None, }; match row.item_id { None => { // we have an empty (unused) category which has NULL values // for the item_id column category.items = Some(vec![]); categories.push(category); } Some(item_id) => { let item = TripItem { item: inventory::Item { id: Uuid::try_parse(&item_id)?, name: row.item_name.unwrap(), description: row.item_description, weight: row.item_weight.unwrap(), category_id: category.category.id, }, picked: row.item_is_picked.unwrap(), packed: row.item_is_packed.unwrap(), ready: row.item_is_ready.unwrap(), new: row.item_is_new.unwrap(), }; if let Some(&mut ref mut c) = categories .iter_mut() .find(|c| c.category.id == category.category.id) { // we always populate c.items when we add a new category, so // it's safe to unwrap here c.items.as_mut().unwrap().push(item); } else { category.items = Some(vec![item]); categories.push(category); } } } Ok(()) }) .try_collect::>>() .await? .into_iter() .collect::>()?; self.categories = Some(categories); Ok(()) } } pub struct TripType { pub id: Uuid, pub name: String, pub active: bool, } impl TripsType { pub async fn all(ctx: &Context, pool: &sqlx::Pool) -> Result, Error> { let user_id = ctx.user.id.to_string(); sqlx::query_as!( DbTripsTypesRow, "SELECT id, name FROM trips_types WHERE user_id = ?", user_id, ) .fetch(pool) .map_ok(|row| row.try_into()) .try_collect::>>() .await? .into_iter() .collect::, Error>>() } pub async fn save( ctx: &Context, pool: &sqlx::Pool, name: &str, ) -> Result { let user_id = ctx.user.id.to_string(); let id = Uuid::new_v4(); let id_param = id.to_string(); sqlx::query!( "INSERT INTO trips_types (id, name, user_id) VALUES (?, ?, ?)", id_param, name, user_id, ) .execute(pool) .await?; Ok(id) } pub async fn set_name( ctx: &Context, pool: &sqlx::Pool, id: Uuid, new_name: &str, ) -> Result { let user_id = ctx.user.id.to_string(); let id_param = id.to_string(); let result = sqlx::query!( "UPDATE trips_types SET name = ? WHERE id = ? and user_id = ?", new_name, id_param, user_id, ) .execute(pool) .await?; Ok(result.rows_affected() != 0) } } pub(crate) struct DbTripsTypesRow { pub id: String, pub name: String, } #[derive(Clone, Debug, Serialize, Deserialize, PartialEq)] pub enum TripTypeAttribute { #[serde(rename = "name")] Name, } pub struct TripsType { pub id: Uuid, pub name: String, } impl TryFrom for TripsType { type Error = Error; fn try_from(row: DbTripsTypesRow) -> Result { Ok(TripsType { id: Uuid::try_parse(&row.id)?, name: row.name, }) } }