tenebrous-sheets/src/routes/characters.rs

use crate::db::{Dao, TenebrousDbConn};
use crate::errors::Error;
use crate::models::characters::Visibility;
use crate::models::users::User;
use rocket::request::Form;
use rocket::response::Redirect;
use rocket_contrib::templates::Template;
use std::collections::HashMap;

pub(crate) fn routes() -> Vec<rocket::Route> {
    routes![
        view_character,
        new_character,
        create_new_character,
        new_character_not_logged_in,
        edit_character
    ]
}

#[derive(FromForm)]
struct NewCharacterForm {
    name: String, //TODO add game system
}

#[get("/<username>/<character_id>")]
fn view_character(
    character_id: i32,
    username: String,
    conn: TenebrousDbConn,
    logged_in_user: Option<&User>,
) -> Result<Template, Error> {
    let user = conn.load_user(&username)?.ok_or(Error::NotFound)?;

    let character = conn
        .load_character(character_id)?
        .and_then(|c| c.as_visible_for(logged_in_user))
        .ok_or(Error::NotFound)?;

    let mut context = HashMap::new();
    context.insert("name", character.character_name);
    context.insert("username", user.username);
    Ok(Template::render("view_character", context))
}

#[get("/new")]
fn new_character(logged_in_user: &User, conn: TenebrousDbConn) -> Result<Template, Error> {
    let context = HashMap::<String, String>::new();
    Ok(Template::render("new_character", context))
}

#[post("/new", data = "<form>")]
fn create_new_character(
    form: Form<NewCharacterForm>,
    logged_in_user: &User,
    conn: TenebrousDbConn,
) -> Result<Redirect, Error> {
    //TODO redirect to character edit page
    //TODO redirect back to new character page with an error and filled-out form if validation errors.
    //TODO add game system.
    use crate::models::characters::NewCharacter;
    use crate::models::proto::cofd::CofdSheet;
    use prost::bytes::BytesMut;
    use prost::Message;

    let new_character = CofdSheet::default();
    let mut buf = BytesMut::with_capacity(std::mem::size_of_val(&new_character));
    new_character.encode(&mut buf)?;

    let insert = NewCharacter {
        user_id: logged_in_user.id,
        viewable: true,
        character_name: &form.name,
        data_type: std::any::type_name::<CofdSheet>(),
        data_version: 1,
        data: &buf,
    };

    conn.insert_character(insert)?;
    Ok(super::common::redirect_to_index())
}

#[get("/new", rank = 2)]
fn new_character_not_logged_in() -> Redirect {
    super::common::redirect_to_login()
}

#[get("/<owner>/<character_id>/edit")]
fn edit_character(
    character_id: i32,
    owner: String,
    logged_in_user: Option<&User>,
    conn: TenebrousDbConn,
) -> Result<Template, Error> {
    let owner = conn.load_user(&owner)?.ok_or(Error::NotFound)?;
    let logged_in_user = logged_in_user.ok_or(Error::NotLoggedIn)?;
    let character = conn.load_character(character_id)?.ok_or(Error::NotFound)?;

    if logged_in_user != &owner {
        return Err(Error::NoPermission);
    }

    let mut context = HashMap::new();
    context.insert("name", character.character_name);
    context.insert("username", owner.username);
    Ok(Template::render("view_character", context))
}