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::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;
use std::result::Result;

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

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

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

#[get("/new")]
pub fn new_film() -> Result<Template, Box<dyn Error>> {
    let ctx: HashMap<String, String> = HashMap::new();
    Ok(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 fn create_film(db: db::Connection, film: Form<NewFilmForm>) -> Result<Redirect, Box<dyn Error>> {
    db.create_film(&film.title, &film.url)?;
    Ok(Redirect::to("films"))
}