push

src/rest/mod.rs

@@ -1,9 +1,10 @@
 mod restday;
 mod restreg;
+mod restuser;
 
 use std::ops::Deref;
 
-use chrono::{Duration, Local, NaiveDate};
+use chrono::{Datelike, Duration, Local, NaiveDate};
 use rocket::{
     form::{self, Form, ValueField},
     fs::FileServer,
@@ -37,7 +38,17 @@ impl Deref for NaiveDateForm {
 #[get("/")]
 async fn index(db: &State<DatabaseConnection>, user: user::Model) -> Template {
     let mut data = Vec::new();
-    for i in 0..6 {
+
+    let mut show_next_n_days = 6;
+    if user.is_cox {
+        let end_of_year = NaiveDate::from_ymd_opt(Local::now().year(), 5, 31).unwrap();
+        show_next_n_days = end_of_year
+            .signed_duration_since(Local::now().date_naive())
+            .num_days()
+            + 1;
+    }
+
+    for i in 0..show_next_n_days {
         let date = (Local::now() + Duration::days(i)).date_naive();
         let day = day::Model::find_or_create_day(date, db.inner()).await;
         data.push(DayWithTrips::new(day, db.inner()).await);
@@ -62,6 +73,12 @@ fn savename(name: Form<NameForm>, cookies: &CookieJar) -> Redirect {
     Redirect::to("/")
 }
 
+#[get("/logout")]
+fn logout(cookies: &CookieJar) -> Redirect {
+    cookies.remove(Cookie::new("name", ""));
+    Redirect::to("/")
+}
+
 #[catch(401)] //unauthorized
 fn unauthorized_error() -> Redirect {
     Redirect::to("/name")
@@ -72,8 +89,9 @@ pub async fn start() -> Rocket<Build> {
         .attach(Template::fairing())
         .manage(Database::connect("sqlite://db.sqlite").await.unwrap())
         .mount("/public", FileServer::from("static/"))
-        .mount("/", routes![index, name, savename])
+        .mount("/", routes![index, name, savename, logout])
         .mount("/day", restday::routes())
         .mount("/register", restreg::routes())
+        .mount("/user", restuser::routes())
         .register("/", catchers![unauthorized_error])
 }

src/rest/restreg.rs

@@ -20,6 +20,10 @@ async fn register(db: &State<DatabaseConnection>, register: Form<RegisterForm>)
         .unwrap()
         .expect("There's no trip on this date (yet)");
 
+    if !day.open_registration {
+        return Redirect::to("/");
+    }
+
     let user = user::Model::find_or_create_user(&register.name, db.inner()).await;
 
     let day = format!("{}", day.day.format("%Y-%m-%d"));

src/rest/restuser.rs

@@ -0,0 +1,42 @@
+use rocket::{form::Form, response::Redirect, Route, State};
+use rocket_dyn_templates::{context, Template};
+use sea_orm::{ActiveModelTrait, DatabaseConnection, EntityTrait, Set};
+
+use crate::models::{day, user};
+
+use super::NaiveDateForm;
+
+#[get("/")]
+async fn index(db: &State<DatabaseConnection>, user: user::AdminUser) -> Template {
+    let users = user::Entity::find().all(db.inner()).await.unwrap();
+
+    Template::render("user/index", context! {user, users})
+}
+
+#[derive(FromForm)]
+struct UserEditForm {
+    is_cox: bool,
+    is_admin: bool,
+}
+
+#[put("/<id>", data = "<data>")]
+async fn update(
+    db: &State<DatabaseConnection>,
+    id: i32,
+    data: Form<UserEditForm>,
+    _user: user::AdminUser,
+) -> Redirect {
+    let new_user = user::ActiveModel {
+        id: Set(id),
+        is_cox: Set(data.is_cox),
+        is_admin: Set(data.is_admin),
+        ..Default::default()
+    };
+    new_user.update(db.inner()).await.unwrap();
+
+    Redirect::to("/user")
+}
+
+pub fn routes() -> Vec<Route> {
+    routes![index, update]
+}