path: root/src/controllers/film.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 rocket::{get, post};
use rocket::form::{Form, FromForm};
use rocket::http::Status;
use rocket::response::Redirect;
use rocket::serde::json::Json;
use rocket_dyn_templates::Template;
use serde::Serialize;
use std::collections::HashMap;
use std::result::Result;

#[get("/", format = "application/json", rank = 1)]
pub async fn get_films_json(db: db::Connection) -> Json<Vec<models::Film>> {
    Json(db.get_films().await.unwrap())
}

#[get("/", rank = 2)]
pub async fn list_films(db: db::Connection) -> Result<Template, Status> {
    #[derive(Serialize)]
    struct Context {
        films: Vec<models::Film>,
    }

    let ctx = Context { films: db.get_films().await.map_err(|_| Status::InternalServerError)? };
    Ok(Template::render("film/list", &ctx))
}

#[get("/new")]
pub fn new_film() -> Template {
    let ctx: HashMap<String, String> = HashMap::new();
    Template::render("film/new", &ctx)
}

#[derive(FromForm)]
pub struct NewFilmForm {
    title: String,
    url: Option<String>,
}

#[post("/", format = "application/x-www-form-urlencoded", data = "<film>")]
pub async fn create_film(db: db::Connection, film: Form<NewFilmForm>) -> Result<Redirect, Status> {
    db.create_film(film.title.to_owned(), film.url.to_owned()).await.map_err(|_| Status::InternalServerError)?;
    Ok(Redirect::to("films"))
}