hacky-ruadat/src/model/logbook.rs

use chrono::NaiveDateTime;
use serde::{Deserialize, Serialize};
use sqlx::{FromRow, SqlitePool};

use super::user::User;

#[derive(FromRow, Debug, Serialize, Deserialize)]
pub struct Logbook {
    pub id: i64,
    pub boat_id: i64,
    pub shipmaster: i64,
    #[serde(default = "bool::default")]
    pub shipmaster_only_steering: bool,
    pub departure: String,       //TODO: Switch to chrono::nativedatetime
    pub arrival: Option<String>, //TODO: Switch to chrono::nativedatetime
    pub destination: Option<String>,
    pub distance_in_km: Option<i64>,
    pub comments: Option<String>,
    pub logtype: Option<i64>,
}

#[derive(Serialize, FromRow)]
pub struct LogbookWithBoatAndUsers {
    pub id: i64,
    pub boat_id: i64,
    pub shipmaster: i64,
    #[serde(default = "bool::default")]
    pub shipmaster_only_steering: bool,
    pub departure: String,       //TODO: Switch to chrono::nativedatetime
    pub arrival: Option<String>, //TODO: Switch to chrono::nativedatetime
    pub destination: Option<String>,
    pub distance_in_km: Option<i64>,
    pub comments: Option<String>,
    pub logtype: Option<i64>,
    pub boat: String,
    pub shipmaster_name: String,
}

pub enum LogbookUpdateError {
    NotYourEntry,
}

pub enum LogbookCreateError {
    BoatAlreadyOnWater,
    BoatLocked
}

impl Logbook {
    pub async fn find_by_id(db: &SqlitePool, id: i32) -> Option<Self> {
        sqlx::query_as!(
            Self,
            "
    SELECT id,boat_id,shipmaster,shipmaster_only_steering,departure,arrival,destination,distance_in_km,comments,logtype
    FROM logbook
    WHERE id like ?
            ",
            id
        )
        .fetch_one(db)
        .await
        .ok()
    }

    //    pub async fn find_by_name(db: &SqlitePool, name: &str) -> Option<Self> {
    //        sqlx::query_as!(
    //            User,
    //            "
    //SELECT id, name, pw, is_cox, is_admin, is_guest, deleted, last_access
    //FROM user
    //WHERE name like ?
    //        ",
    //            name
    //        )
    //        .fetch_one(db)
    //        .await
    //        .ok()
    //    }
    //
    pub async fn on_water(db: &SqlitePool) -> Vec<LogbookWithBoatAndUsers> {
        sqlx::query_as!(
                LogbookWithBoatAndUsers,
                "
    SELECT logbook.id, boat_id, shipmaster, shipmaster_only_steering, departure, arrival, destination, distance_in_km, comments, logtype, boat.name as boat, user.name as shipmaster_name
    FROM logbook
    INNER JOIN boat ON logbook.boat_id = boat.id
    INNER JOIN user ON shipmaster = user.id
    WHERE arrival is null
    ORDER BY departure DESC
            "
            )
            .fetch_all(db)
            .await
            .unwrap() //TODO: fixme
    }

    pub async fn completed(db: &SqlitePool) -> Vec<LogbookWithBoatAndUsers> {
        sqlx::query_as!(
                LogbookWithBoatAndUsers,
                "
    SELECT logbook.id, boat_id, shipmaster, shipmaster_only_steering, departure, arrival, destination, distance_in_km, comments, logtype, boat.name as boat, user.name as shipmaster_name
    FROM logbook
    INNER JOIN boat ON logbook.boat_id = boat.id
    INNER JOIN user ON shipmaster = user.id
    WHERE arrival is not null
    ORDER BY arrival DESC
            "
            )
            .fetch_all(db)
            .await
            .unwrap() //TODO: fixme
    }

    pub async fn create(
        db: &SqlitePool,
        boat_id: i64,
        shipmaster: i64,
        shipmaster_only_steering: bool,
        departure: NaiveDateTime,
        arrival: Option<NaiveDateTime>,
        destination: Option<String>,
        distance_in_km: Option<i64>,
        comments: Option<String>,
        logtype: Option<i64>,
    ) -> Result<(), LogbookCreateError> {
        //Check if boat is not locked
        //Check if boat is already on water
        sqlx::query!(
                "INSERT INTO logbook(boat_id, shipmaster, shipmaster_only_steering, departure, arrival, destination, distance_in_km, comments, logtype) VALUES (?,?,?,?,?,?,?,?,?)",
                boat_id, shipmaster, shipmaster_only_steering, departure, arrival, destination, distance_in_km, comments, logtype
            )
            .execute(db)
            .await;
        Ok(())
    }

    pub async fn home(
        &self,
        db: &SqlitePool,
        user: &User,
        destination: String,
        distance_in_km: i64,
        comments: Option<String>,
        logtype: Option<i64>,
    ) -> Result<(), LogbookUpdateError> {
        if user.id != self.shipmaster {
            return Err(LogbookUpdateError::NotYourEntry);
        }
        //TODO: check current date

        let arrival = format!("{}", chrono::offset::Local::now().format("%Y-%m-%d %H:%M"));

        sqlx::query!(
            "UPDATE logbook SET destination=?, distance_in_km=?, comments=?, logtype=?, arrival=? WHERE id=?",
            destination,
            distance_in_km,
            comments,
            logtype,
            arrival,
            self.id
        )
        .execute(db)
        .await.unwrap(); //TODO: fixme
        Ok(())
    }

    //    pub async fn delete(&self, db: &SqlitePool) {
    //        sqlx::query!("DELETE FROM boat WHERE id=?", self.id)
    //            .execute(db)
    //            .await
    //            .unwrap(); //Okay, because we can only create a User of a valid id
    //    }
}
//
//#[cfg(test)]
//mod test {
//    use crate::{model::boat::Boat, testdb};
//
//    use sqlx::SqlitePool;
//
//    #[sqlx::test]
//    fn test_find_correct_id() {
//        let pool = testdb!();
//        let boat = Boat::find_by_id(&pool, 1).await.unwrap();
//        assert_eq!(boat.id, 1);
//    }
//
//    #[sqlx::test]
//    fn test_find_wrong_id() {
//        let pool = testdb!();
//        let boat = Boat::find_by_id(&pool, 1337).await;
//        assert!(boat.is_none());
//    }
//
//    #[sqlx::test]
//    fn test_all() {
//        let pool = testdb!();
//        let res = Boat::all(&pool).await;
//        assert!(res.len() > 3);
//    }
//
//    #[sqlx::test]
//    fn test_succ_create() {
//        let pool = testdb!();
//
//        assert_eq!(
//            Boat::create(
//                &pool,
//                "new-boat-name".into(),
//                42,
//                None,
//                "Best Boatbuilder".into(),
//                true,
//                true,
//                false,
//                Some(1),
//                None
//            )
//            .await,
//            true
//        );
//    }
//
//    #[sqlx::test]
//    fn test_duplicate_name_create() {
//        let pool = testdb!();
//
//        assert_eq!(
//            Boat::create(
//                &pool,
//                "Haichenbach".into(),
//                42,
//                None,
//                "Best Boatbuilder".into(),
//                true,
//                true,
//                false,
//                Some(1),
//                None
//            )
//            .await,
//            false
//        );
//    }
//}