From 7381a7033231e6454a37fd64b1f3de4e8d59355f Mon Sep 17 00:00:00 2001 From: "Adam T. Carpenter" Date: Wed, 11 Nov 2020 20:07:12 -0500 Subject: Kind of flailing with the UI; lots of API bugfixes though. --- dichroism/src/handlers.rs | 6 +-- dichroism/src/repo/entities/product.rs | 4 +- dichroism/src/repo/entities/product_form.rs | 6 ++- dichroism/src/repo/photo_set_repo.rs | 26 +++++++------ dichroism/src/repo/product_repo.rs | 60 ++++++++++++++--------------- dichroism/src/schema.rs | 11 +----- 6 files changed, 51 insertions(+), 62 deletions(-) (limited to 'dichroism/src') diff --git a/dichroism/src/handlers.rs b/dichroism/src/handlers.rs index ccaf347..8dd0f3b 100644 --- a/dichroism/src/handlers.rs +++ b/dichroism/src/handlers.rs @@ -8,7 +8,6 @@ use actix_web::{get, patch, post, web, Error, HttpResponse, Responder}; use futures::{StreamExt, TryStreamExt}; fn to_internal_error(e: impl std::error::Error) -> HttpResponse { - eprintln!("{}", e); HttpResponse::InternalServerError().body(e.to_string()) } @@ -120,10 +119,7 @@ async fn post_photo( // create new photo_set let photo_set = web::block(move || image_service::generate_photo_set(&data)) .await - .map_err(|e| { - eprintln!("{}", e.to_string()); - HttpResponse::InternalServerError().body(e.to_string()) - })?; + .map_err(to_internal_error)?; let conn = pool.get().map_err(to_internal_error)?; let photo_set = web::block(move || photo_set_repo::store(&conn, photo_set)) .await diff --git a/dichroism/src/repo/entities/product.rs b/dichroism/src/repo/entities/product.rs index 81f3d9e..c89495a 100644 --- a/dichroism/src/repo/entities/product.rs +++ b/dichroism/src/repo/entities/product.rs @@ -9,12 +9,12 @@ pub struct Product { pub quantity: i32, pub cents: i32, pub featured: i32, + pub category: String, pub photo_set_id: i32, pub original: String, pub fullsize: String, pub base: String, pub thumbnail: String, - pub category: String, } impl Into for Product { @@ -25,7 +25,7 @@ impl Into for Product { description: self.description, quantity: self.quantity, cents: self.cents, - featured: self.featured != 0, // TODO: is this safe? + featured: self.featured != 0, category: self.category, photo_set: models::PhotoSet { id: Some(self.photo_set_id), diff --git a/dichroism/src/repo/entities/product_form.rs b/dichroism/src/repo/entities/product_form.rs index 0d6e452..b3447eb 100644 --- a/dichroism/src/repo/entities/product_form.rs +++ b/dichroism/src/repo/entities/product_form.rs @@ -1,7 +1,7 @@ use crate::models::*; use crate::schema::products; -#[derive(Insertable, AsChangeset)] +#[derive(Debug, Insertable, AsChangeset)] #[table_name = "products"] pub struct ProductForm { pub id: i32, @@ -11,6 +11,7 @@ pub struct ProductForm { pub cents: i32, pub featured: i32, pub photo_set: i32, + pub category: String, } impl From for ProductForm { @@ -22,7 +23,8 @@ impl From for ProductForm { quantity: p.quantity, cents: p.cents, featured: p.featured as i32, - photo_set: p.photo_set.id.unwrap_or(-1), // TODO: ? + photo_set: p.photo_set.id.unwrap_or(-1), + category: p.category, } } } diff --git a/dichroism/src/repo/photo_set_repo.rs b/dichroism/src/repo/photo_set_repo.rs index 339ea91..56ca962 100644 --- a/dichroism/src/repo/photo_set_repo.rs +++ b/dichroism/src/repo/photo_set_repo.rs @@ -5,13 +5,15 @@ use diesel::{insert_into, prelude::*, result::Error, update}; pub fn store(conn: &DBConn, mut photo_set: models::PhotoSet) -> Result { use crate::schema::photo_sets::dsl::*; - if photo_set.id.is_some() { + if let Some(photo_set_id) = photo_set.id { // update let form = PhotoSetForm::from(photo_set.clone()); - update(photo_sets).set(&form).execute(conn)?; + update(photo_sets.filter(id.eq(photo_set_id))) + .set(&form) + .execute(conn)?; } else { // insert - photo_set.id = Some(find_next_id(conn)?); + photo_set.id = Some(find_next_id(conn)); let form = PhotoSetForm::from(photo_set.clone()); insert_into(photo_sets).values(&form).execute(conn)?; } @@ -26,17 +28,17 @@ pub fn find(conn: &DBConn, dbid: i32) -> Result, Error> let photo_set = query.first::(conn).map(|p| p.into()); match photo_set { Ok(p) => Ok(Some(p)), - Err(e) => { - if e == Error::NotFound { - Ok(None) - } else { - Err(e) - } - } + Err(Error::NotFound) => Ok(None), + Err(e) => Err(e), } } -fn find_next_id(conn: &DBConn) -> Result { +fn find_next_id(conn: &DBConn) -> i32 { use crate::schema::photo_sets::dsl::*; - Ok(photo_sets.select(id).order(id.desc()).first::(conn)? + 1) + let last_id = photo_sets + .select(id) + .order(id.desc()) + .first::(conn) + .unwrap_or(0); + last_id + 1 } diff --git a/dichroism/src/repo/product_repo.rs b/dichroism/src/repo/product_repo.rs index d7dff65..487f96e 100644 --- a/dichroism/src/repo/product_repo.rs +++ b/dichroism/src/repo/product_repo.rs @@ -5,13 +5,15 @@ use diesel::{insert_into, prelude::*, result::Error, update}; pub fn store(conn: &DBConn, mut product: models::Product) -> Result { use crate::schema::products::dsl::*; - if product.id.is_some() { + if let Some(product_id) = product.id { // update let form = ProductForm::from(product.clone()); - update(products).set(&form).execute(conn)?; + update(products.filter(id.eq(product_id))) + .set(&form) + .execute(conn)?; } else { // insert - product.id = Some(find_next_id(conn)?); + product.id = Some(find_next_id(conn)); let form = ProductForm::from(product.clone()); insert_into(products).values(&form).execute(conn)?; } @@ -21,23 +23,20 @@ pub fn store(conn: &DBConn, mut product: models::Product) -> Result Result, Error> { use crate::schema::*; - let query = products::table - .inner_join(photo_sets::table) - .inner_join(categories::table) - .select(( - products::id, - products::name, - products::description, - products::quantity, - products::cents, - products::featured, - photo_sets::id, - photo_sets::original, - photo_sets::fullsize, - photo_sets::base, - photo_sets::thumbnail, - categories::path, - )); + let query = products::table.inner_join(photo_sets::table).select(( + products::id, + products::name, + products::description, + products::quantity, + products::cents, + products::featured, + products::category, + photo_sets::id, + photo_sets::original, + photo_sets::fullsize, + photo_sets::base, + photo_sets::thumbnail, + )); Ok(query .load::(conn)? .into_iter() @@ -49,7 +48,6 @@ pub fn find(conn: &DBConn, dbid: i32) -> Result, Error> use crate::schema::*; let query = products::table .inner_join(photo_sets::table) - .inner_join(categories::table) .filter(products::id.eq(dbid)) .select(( products::id, @@ -58,27 +56,27 @@ pub fn find(conn: &DBConn, dbid: i32) -> Result, Error> products::quantity, products::cents, products::featured, + products::category, photo_sets::id, photo_sets::original, photo_sets::fullsize, photo_sets::base, photo_sets::thumbnail, - categories::path, )); let product = query.first::(conn).map(|p| p.into()); match product { Ok(p) => Ok(Some(p)), - Err(e) => { - if e == Error::NotFound { - Ok(None) - } else { - Err(e) - } - } + Err(Error::NotFound) => Ok(None), + Err(e) => Err(e), } } -fn find_next_id(conn: &DBConn) -> Result { +fn find_next_id(conn: &DBConn) -> i32 { use crate::schema::products::dsl::*; - Ok(products.select(id).order(id.desc()).first::(conn)? + 1) + let last_id = products + .select(id) + .order(id.desc()) + .first::(conn) + .unwrap_or(0); + last_id + 1 } diff --git a/dichroism/src/schema.rs b/dichroism/src/schema.rs index 45501cf..2d9e14a 100644 --- a/dichroism/src/schema.rs +++ b/dichroism/src/schema.rs @@ -1,10 +1,3 @@ -table! { - categories (id) { - id -> Integer, - path -> Text, - } -} - table! { photo_sets (id) { id -> Integer, @@ -24,15 +17,13 @@ table! { name -> Text, description -> Text, featured -> Integer, - category -> Integer, + category -> Text, } } -joinable!(products -> categories (category)); joinable!(products -> photo_sets (photo_set)); allow_tables_to_appear_in_same_query!( - categories, photo_sets, products, ); -- cgit v1.2.3