use super::entities::*;
use super::DBConn;
use crate::models;
use diesel::{insert_into, prelude::*, result::Error, update};

pub fn store(conn: &DBConn, mut photo_set: models::PhotoSet) -> Result<models::PhotoSet, Error> {
    use crate::schema::photo_sets::dsl::*;
    if let Some(photo_set_id) = photo_set.id {
        // update
        let form = PhotoSetForm::from(photo_set.clone());
        update(photo_sets.filter(id.eq(photo_set_id)))
            .set(&form)
            .execute(conn)?;
    } else {
        // insert
        photo_set.id = Some(find_next_id(conn));
        let form = PhotoSetForm::from(photo_set.clone());
        insert_into(photo_sets).values(&form).execute(conn)?;
    }
    Ok(photo_set)
}

pub fn find(conn: &DBConn, dbid: i32) -> Result<Option<models::PhotoSet>, Error> {
    use crate::schema::photo_sets::dsl::*;
    let query = photo_sets
        .filter(id.eq(dbid))
        .select((id, original, fullsize, base, thumbnail));
    let photo_set = query.first::<PhotoSet>(conn).map(|p| p.into());
    match photo_set {
        Ok(p) => Ok(Some(p)),
        Err(Error::NotFound) => Ok(None),
        Err(e) => Err(e),
    }
}

fn find_next_id(conn: &DBConn) -> i32 {
    use crate::schema::photo_sets::dsl::*;
    let last_id = photo_sets
        .select(id)
        .order(id.desc())
        .first::<i32>(conn)
        .unwrap_or(0);
    last_id + 1
}