Make bump session button pretty

server/src/routes/pages.rs

@@ -3,9 +3,11 @@ pub mod stats;
 use crate::auth::Authorized;
 use crate::database::latest::trees::{category, session};
 use crate::status_json::StatusJson;
+use crate::util::EventNotifier;
 use bincode::{deserialize, serialize};
 use rocket::http::Status;
-use rocket::{get, State};
+use rocket::response::Redirect;
+use rocket::{get, post, uri, State};
 use rocket_contrib::templates::Template;
 use rocket_contrib::uuid::Uuid;
 use serde_derive::{Deserialize, Serialize};
@@ -89,6 +91,72 @@ pub fn index(_auth: Authorized, db: State<'_, sled::Db>) -> Result<Template, Sta
     Ok(Template::render("index", &context))
 }
 
+#[post("/category/<category_uuid>/start_session")]
+pub fn start_session(
+    _auth: Authorized,
+    category_uuid: Uuid,
+    event_notifier: State<'_, EventNotifier>,
+    db: State<'_, sled::Db>,
+) -> Result<Redirect, StatusJson> {
+    super::api::toggle_category_session(category_uuid, true, event_notifier, db)?;
+    Ok(Redirect::to(uri!(index)))
+}
+
+#[post("/category/<category_uuid>/end_session")]
+pub fn end_session(
+    _auth: Authorized,
+    category_uuid: Uuid,
+    event_notifier: State<'_, EventNotifier>,
+    db: State<'_, sled::Db>,
+) -> Result<Redirect, StatusJson> {
+    super::api::toggle_category_session(category_uuid, false, event_notifier, db)?;
+    Ok(Redirect::to(uri!(index)))
+}
+
+#[post("/category/<category_uuid>/bump_session/minutes/<minutes>")]
+pub fn bump_session(
+    _auth: Authorized,
+    category_uuid: Uuid,
+    minutes: i64,
+    db: State<'_, sled::Db>,
+) -> Result<Redirect, StatusJson> {
+    use crate::database::latest::trees::category;
+
+    let duration = chrono::Duration::minutes(minutes);
+    let category_uuid_s = sled::IVec::from(serialize(&category_uuid.into_inner())?);
+
+    let categories_tree = db.open_tree(category::NAME)?;
+
+    (&categories_tree).transaction(|tx_categories| {
+        match tx_categories.get(&category_uuid_s)? {
+            None => Ok(Err(Status::NotFound.into())),
+            Some(data) => {
+                let mut category: category::V = deserialize(&data).unwrap();
+                match category.started.as_mut() {
+                    Some(started) => {
+                        if let Some(new_started) = started.checked_sub_signed(duration) {
+                            *started = new_started;
+                            tx_categories
+                                .insert(&category_uuid_s, serialize(&category).unwrap())?;
+                        } else {
+                            return Ok(Err(StatusJson::new(
+                                Status::BadRequest,
+                                "Duration subtract resulted in overflow",
+                            )));
+                        }
+
+                        Ok(Ok(Redirect::to(uri!(index))))
+                    }
+                    None => Ok(Err(StatusJson::new(
+                        Status::BadRequest,
+                        "No active session",
+                    ))),
+                }
+            }
+        }
+    })?
+}
+
 #[get("/session/<session_uuid>/edit")]
 pub fn session_edit(
     _auth: Authorized,