6 files changed, 205 insertions, 38 deletions

diff --git a/dichroism/src/repo/entities/mod.rs b/dichroism/src/repo/entities/mod.rs
index 2cff899..288f1e3 100644
--- a/dichroism/src/repo/entities/mod.rs
+++ b/dichroism/src/repo/entities/mod.rs
@@ -1,36 +1,9 @@
-use crate::models;
-use crate::schema::products;
+mod photo_set;
+mod photo_set_form;
+mod product;
+mod product_form;
 
-#[derive(Debug, Clone, Identifiable, Queryable, Serialize)]
-pub struct Product {
-    pub id: i32,
-    pub name: String,
-    pub description: String,
-    pub quantity: i32,
-    pub cents: i32,
-    pub featured: i32,
-    pub original: String,
-    pub fullsize: String,
-    pub base: String,
-    pub thumbnail: String,
-}
-
-impl Into<models::Product> for Product {
-    fn into(self) -> models::Product {
-        models::Product {
-            id: self.id as u32,
-            name: self.name,
-            description: self.description,
-            quantity: self.quantity as u32,
-            cents: self.cents as u32,
-            featured: self.featured != 0, // TODO: is this safe?
-            category: String::new(),      // TODO: real category
-            photo_set: models::PhotoSet::from_photos(
-                models::Photo::from_filename(self.original),
-                models::Photo::from_filename(self.fullsize),
-                models::Photo::from_filename(self.base),
-                models::Photo::from_filename(self.thumbnail),
-            ),
-        }
-    }
-}
+pub use photo_set::*;
+pub use photo_set_form::*;
+pub use product::*;
+pub use product_form::*;
diff --git a/dichroism/src/repo/entities/photo_set.rs b/dichroism/src/repo/entities/photo_set.rs
new file mode 100644
index 0000000..6e48e12
--- /dev/null
+++ b/dichroism/src/repo/entities/photo_set.rs
@@ -0,0 +1,22 @@
+use crate::models;
+
+#[derive(Debug, Clone, Queryable)]
+pub struct PhotoSet {
+    pub id: i32,
+    pub original: String,
+    pub fullsize: String,
+    pub base: String,
+    pub thumbnail: String,
+}
+
+impl Into<models::PhotoSet> for PhotoSet {
+    fn into(self) -> models::PhotoSet {
+        models::PhotoSet {
+            id: Some(self.id),
+            original: models::Photo::new(self.original),
+            fullsize: models::Photo::new(self.fullsize),
+            base: models::Photo::new(self.base),
+            thumbnail: models::Photo::new(self.thumbnail),
+        }
+    }
+}
diff --git a/dichroism/src/repo/entities/photo_set_form.rs b/dichroism/src/repo/entities/photo_set_form.rs
new file mode 100644
index 0000000..611c8f0
--- /dev/null
+++ b/dichroism/src/repo/entities/photo_set_form.rs
@@ -0,0 +1,24 @@
+use crate::models::PhotoSet;
+use crate::schema::photo_sets;
+
+#[derive(Insertable, AsChangeset)]
+#[table_name = "photo_sets"]
+pub struct PhotoSetForm {
+    pub id: i32,
+    pub original: String,
+    pub fullsize: String,
+    pub base: String,
+    pub thumbnail: String,
+}
+
+impl From<PhotoSet> for PhotoSetForm {
+    fn from(p: PhotoSet) -> Self {
+        Self {
+            id: p.id.unwrap_or(-1),
+            original: p.original.id,
+            fullsize: p.fullsize.id,
+            base: p.base.id,
+            thumbnail: p.thumbnail.id,
+        }
+    }
+}
diff --git a/dichroism/src/repo/entities/product.rs b/dichroism/src/repo/entities/product.rs
new file mode 100644
index 0000000..e6ba223
--- /dev/null
+++ b/dichroism/src/repo/entities/product.rs
@@ -0,0 +1,38 @@
+use crate::models;
+use crate::schema::products;
+
+#[derive(Debug, Clone, Identifiable, Queryable)]
+pub struct Product {
+    pub id: i32,
+    pub name: String,
+    pub description: String,
+    pub quantity: i32,
+    pub cents: i32,
+    pub featured: i32,
+    pub photo_set_id: i32,
+    pub original: String,
+    pub fullsize: String,
+    pub base: String,
+    pub thumbnail: String,
+}
+
+impl Into<models::Product> for Product {
+    fn into(self) -> models::Product {
+        models::Product {
+            id: Some(self.id),
+            name: self.name,
+            description: self.description,
+            quantity: self.quantity,
+            cents: self.cents,
+            featured: self.featured != 0, // TODO: is this safe?
+            category: String::new(),      // TODO: real category
+            photo_set: models::PhotoSet {
+                id: Some(self.photo_set_id),
+                original: models::Photo::new(self.original),
+                fullsize: models::Photo::new(self.fullsize),
+                base: models::Photo::new(self.base),
+                thumbnail: models::Photo::new(self.thumbnail),
+            },
+        }
+    }
+}
diff --git a/dichroism/src/repo/entities/product_form.rs b/dichroism/src/repo/entities/product_form.rs
new file mode 100644
index 0000000..0d6e452
--- /dev/null
+++ b/dichroism/src/repo/entities/product_form.rs
@@ -0,0 +1,28 @@
+use crate::models::*;
+use crate::schema::products;
+
+#[derive(Insertable, AsChangeset)]
+#[table_name = "products"]
+pub struct ProductForm {
+    pub id: i32,
+    pub name: String,
+    pub description: String,
+    pub quantity: i32,
+    pub cents: i32,
+    pub featured: i32,
+    pub photo_set: i32,
+}
+
+impl From<Product> for ProductForm {
+    fn from(p: Product) -> Self {
+        Self {
+            id: p.id.unwrap_or(-1),
+            name: p.name,
+            description: p.description,
+            quantity: p.quantity,
+            cents: p.cents,
+            featured: p.featured as i32,
+            photo_set: p.photo_set.id.unwrap_or(-1), // TODO: ?
+        }
+    }
+}
diff --git a/dichroism/src/repo/mod.rs b/dichroism/src/repo/mod.rs
index 0df8aad..6dc245a 100644
--- a/dichroism/src/repo/mod.rs
+++ b/dichroism/src/repo/mod.rs
@@ -1,13 +1,53 @@
 use crate::models;
-use crate::schema::*;
+use diesel::insert_into;
 use diesel::prelude::*;
 use diesel::result::Error;
+use diesel::update;
+use entities::*;
 
-mod entities;
+pub mod entities;
 
 type DBConn = SqliteConnection;
 
-pub fn read_products(conn: &DBConn) -> Result<Vec<models::Product>, Error> {
+pub fn store_photo_set(
+    conn: &DBConn,
+    mut photo_set: models::PhotoSet,
+) -> Result<models::PhotoSet, Error> {
+    use crate::schema::photo_sets::dsl::*;
+    if photo_set.id.is_some() {
+        // update
+        let form = PhotoSetForm::from(photo_set.clone());
+        update(photo_sets).set(&form).execute(conn)?;
+    } else {
+        // insert
+        photo_set.id = Some(find_next_photo_set_id(conn)?);
+        let form = PhotoSetForm::from(photo_set.clone());
+        insert_into(photo_sets).values(&form).execute(conn)?;
+    }
+    Ok(photo_set)
+}
+
+pub fn store_product(
+    conn: &DBConn,
+    mut product: models::Product,
+) -> Result<models::Product, Error> {
+    use crate::schema::products::dsl::*;
+    if product.id.is_some() {
+        // update
+        let form = ProductForm::from(product.clone());
+        update(products).set(&form).execute(conn)?;
+    } else {
+        // insert
+        product.id = Some(find_next_product_id(conn)?);
+        let form = ProductForm::from(product.clone());
+        insert_into(products).values(&form).execute(conn)?;
+    }
+
+    Ok(product)
+}
+
+pub fn find_all(conn: &DBConn) -> Result<Vec<models::Product>, Error> {
+    use crate::schema::*;
     let query = products::table.inner_join(photo_sets::table).select((
         products::id,
         products::name,
@@ -15,6 +55,7 @@ pub fn read_products(conn: &DBConn) -> Result<Vec<models::Product>, Error> {
         products::quantity,
         products::cents,
         products::featured,
+        photo_sets::id,
         photo_sets::original,
         photo_sets::fullsize,
         photo_sets::base,
@@ -26,3 +67,44 @@ pub fn read_products(conn: &DBConn) -> Result<Vec<models::Product>, Error> {
         .map(|p| p.into())
         .collect::<Vec<models::Product>>())
 }
+
+pub fn find(conn: &DBConn, id: i32) -> Result<Option<models::Product>, Error> {
+    use crate::schema::*;
+    let query = products::table
+        .inner_join(photo_sets::table)
+        .filter(products::id.eq(id))
+        .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,
+        ));
+    let product = query.first::<Product>(conn).map(|p| p.into());
+    match product {
+        Ok(p) => Ok(Some(p)),
+        Err(e) => {
+            if e == Error::NotFound {
+                Ok(None)
+            } else {
+                Err(e)
+            }
+        }
+    }
+}
+
+fn find_next_product_id(conn: &DBConn) -> Result<i32, Error> {
+    use crate::schema::products::dsl::*;
+    Ok(products.select(id).order(id.desc()).first::<i32>(conn)? + 1)
+}
+
+fn find_next_photo_set_id(conn: &DBConn) -> Result<i32, Error> {
+    use crate::schema::photo_sets::dsl::*;
+    Ok(photo_sets.select(id).order(id.desc()).first::<i32>(conn)? + 1)
+}