path: root/src/controllers/room.rs

/*
    Social program for Ramaskrik.
    Copyright (C) 2019  Harald Eilertsen <haraldei@anduin.net>

    This program is free software: you can redistribute it and/or modify
    it under the terms of the GNU Affero General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU Affero General Public License for more details.

    You should have received a copy of the GNU Affero General Public License
    along with this program.  If not, see <https://www.gnu.org/licenses/>.
*/

use crate::{
    db,
    models,
};

use std::result::Result;
use rocket::{get, post};
use rocket::request::{Form, FromForm};
use rocket::response::Redirect;
use rocket_contrib::{
    json::Json,
    templates::Template,
};
use serde::Serialize;
use std::collections::HashMap;
use std::error::Error;

#[derive(Serialize)]
struct Context {
    rooms : Vec<models::Room>,
}

#[get("/", format = "application/json", rank = 1)]
pub fn get_rooms_json(db: db::Connection) -> Json<Vec<models::Room>> {
    Json(db.get_rooms().unwrap())
}

#[get("/", rank = 2)]
pub fn list_rooms(db: db::Connection) -> Result<Template, Box<dyn Error>> {
    let ctx = Context { rooms: db.get_rooms()? };
    Ok(Template::render("room/list", &ctx))
}

#[get("/new")]
pub fn new_room() -> Result<Template, Box<dyn Error>> {
    let ctx: HashMap<String, String> = HashMap::new();
    Ok(Template::render("room/new", &ctx))
}

#[derive(FromForm)]
pub struct NewRoomForm {
    name: String,
}

#[post("/", format = "application/x-www-form-urlencoded", data = "<room>")]
pub fn create_room(db: db::Connection, room: Form<NewRoomForm>) -> Result<Redirect, Box<dyn Error>> {
    db.create_room(&room.name)?;
    Ok(Redirect::to("/rooms"))
}