rowt/src/tera/log.rs

use rocket::{
    form::Form,
    get, post,
    request::{FlashMessage, FromRequest, self},
    response::{Flash, Redirect},
    routes, Route, State, Request, http::{CookieJar, Cookie}, time::{OffsetDateTime, Duration},
};
use rocket_dyn_templates::Template;
use sqlx::SqlitePool;
use tera::Context;

use crate::model::{
    boat::Boat,
    logbook::{LogToAdd, LogToFinalize, Logbook, LogbookCreateError},
    logtype::LogType,
    user::{AdminUser, User},
};

pub struct KioskCookie(String);

#[rocket::async_trait]
impl<'r> FromRequest<'r> for KioskCookie {
    type Error = std::convert::Infallible;

    async fn from_request(request: &'r Request<'_>) -> request::Outcome<KioskCookie, Self::Error> {
        match request.cookies().get_private("kiosk") {
            Some(cookie) => request::Outcome::Success(KioskCookie(cookie.value().to_string())),
            None => request::Outcome::Forward(()),
        }
    }
}

#[get("/")]
async fn index(
    db: &State<SqlitePool>,
    flash: Option<FlashMessage<'_>>,
    adminuser: AdminUser,
) -> Template {
    let boats = Boat::all(db).await;
    let coxes = User::cox(db).await;
    let users = User::all(db).await;
    let logtypes = LogType::all(db).await;
    let distances = Logbook::distances(db).await;

    let on_water = Logbook::on_water(db).await;
    let completed = Logbook::completed(db).await;

    let mut context = Context::new();
    if let Some(msg) = flash {
        context.insert("flash", &msg.into_inner());
    }

    context.insert("boats", &boats);
    context.insert("coxes", &coxes);
    context.insert("users", &users);
    context.insert("logtypes", &logtypes);
    context.insert("loggedin_user", &adminuser.user);
    context.insert("on_water", &on_water);
    context.insert("completed", &completed);
    context.insert("distances", &distances);

    Template::render("log", context.into_json())
}


#[get("/kiosk/ekrv2019")]
fn new_kiosk(
    flash: Option<FlashMessage<'_>>,
    cookies: &CookieJar<'_>,
) -> Redirect {
    let mut cookie = Cookie::new("kiosk", format!("yes"));
    cookie.set_expires(OffsetDateTime::now_utc() + Duration::weeks(12));
    cookies.add_private(cookie);
    Redirect::to("/log/kiosk")
}

#[get("/kiosk")]
async fn kiosk(db: &State<SqlitePool>, flash: Option<FlashMessage<'_>>, _kiosk: KioskCookie) -> Template {
    let boats = Boat::all(db).await;
    let coxes = User::cox(db).await;
    let users = User::all(db).await;
    let logtypes = LogType::all(db).await;
    let distances = Logbook::distances(db).await;

    let on_water = Logbook::on_water(db).await;
    let completed = Logbook::completed(db).await;

    let mut context = Context::new();
    if let Some(msg) = flash {
        context.insert("flash", &msg.into_inner());
    }

    context.insert("boats", &boats);
    context.insert("coxes", &coxes);
    context.insert("users", &users);
    context.insert("logtypes", &logtypes);
    context.insert("on_water", &on_water);
    context.insert("completed", &completed);
    context.insert("distances", &distances);

    Template::render("kiosk", context.into_json())
}

#[post("/", data = "<data>", rank=2)]
async fn create(
    db: &State<SqlitePool>,
    data: Form<LogToAdd>,
    _adminuser: AdminUser,
) -> Flash<Redirect> {
    match Logbook::create(
        db,
        data.into_inner()
    )
    .await
    {
        Ok(_) => Flash::success(Redirect::to("/log"), "Ausfahrt erfolgreich hinzugefügt"),
        Err(LogbookCreateError::BoatAlreadyOnWater) => Flash::error(Redirect::to("/log"), format!("Boot schon am Wasser")),
        Err(LogbookCreateError::BoatLocked) => Flash::error(Redirect::to("/log"), format!("Boot gesperrt")),
        Err(LogbookCreateError::BoatNotFound) => Flash::error(Redirect::to("/log"), format!("Boot gibt's ned")),
        Err(LogbookCreateError::TooManyRowers(expected, actual)) => Flash::error(Redirect::to("/log"), format!("Zu viele Ruderer (Boot fasst maximal {expected}, es wurden jedoch {actual} Ruderer ausgewählt)")),
    }
}

#[post("/", data = "<data>")]
async fn create_kiosk(
    db: &State<SqlitePool>,
    data: Form<LogToAdd>,
) -> Flash<Redirect> {
    match Logbook::create(
        db,
        data.into_inner()
    )
    .await
    {
        Ok(_) => Flash::success(Redirect::to("/log/kiosk"), "Ausfahrt erfolgreich hinzugefügt"),
        Err(LogbookCreateError::BoatAlreadyOnWater) => Flash::error(Redirect::to("/log/kiosk"), format!("Boot schon am Wasser")),
        Err(LogbookCreateError::BoatLocked) => Flash::error(Redirect::to("/log/kiosk"), format!("Boot gesperrt")),
        Err(LogbookCreateError::BoatNotFound) => Flash::error(Redirect::to("/log/kiosk"), format!("Boot gibt's ned")),
        Err(LogbookCreateError::TooManyRowers(expected, actual)) => Flash::error(Redirect::to("/log/kiosk"), format!("Zu viele Ruderer (Boot fasst maximal {expected}, es wurden jedoch {actual} Ruderer ausgewählt)")),
    }
}


#[post("/<logbook_id>", data = "<data>")]
async fn home_kiosk(
    db: &State<SqlitePool>,
    data: Form<LogToFinalize>,
    logbook_id: i32,
) -> Flash<Redirect> {
    let logbook = Logbook::find_by_id(db, logbook_id).await;
    let Some(logbook) = logbook else {
            return Flash::error(
                Redirect::to("/admin/log"),
                format!("Log with ID {} does not exist!", logbook_id),
            )
    };

    match logbook.home(db, &User::find_by_id(db, logbook.shipmaster as i32).await.unwrap(), data.into_inner()).await {
        Ok(_) => Flash::success(Redirect::to("/log"), "Successfully updated log"),
        Err(_) => Flash::error(
            Redirect::to("/log"),
            format!("Logbook with ID {} could not be updated!", logbook_id),
        ),
    }
}


#[post("/<logbook_id>", data = "<data>", rank=2)]
async fn home(
    db: &State<SqlitePool>,
    data: Form<LogToFinalize>,
    logbook_id: i32,
    adminuser: AdminUser,
) -> Flash<Redirect> {
    let logbook = Logbook::find_by_id(db, logbook_id).await;
    let Some(logbook) = logbook else {
            return Flash::error(
                Redirect::to("/admin/log"),
                format!("Log with ID {} does not exist!", logbook_id),
            )
    };

    match logbook.home(db, &adminuser.user, data.into_inner()).await {
        Ok(_) => Flash::success(Redirect::to("/log"), "Successfully updated log"),
        Err(_) => Flash::error(
            Redirect::to("/log"),
            format!("Logbook with ID {} could not be updated!", logbook_id),
        ),
    }
}

pub fn routes() -> Vec<Route> {
    routes![index, create, create_kiosk, home, kiosk, home_kiosk, new_kiosk]
}

#[cfg(test)]
mod test {}