2020-12-27 21:03:10 +00:00
|
|
|
use crate::db::{Dao, TenebrousDbConn};
|
|
|
|
use crate::errors::Error;
|
2021-01-03 22:05:28 +00:00
|
|
|
use crate::models::characters::Character;
|
2021-01-01 00:40:48 +00:00
|
|
|
use crate::models::proto::cofd::*;
|
2020-12-27 21:03:10 +00:00
|
|
|
use crate::models::users::User;
|
2021-01-01 00:40:48 +00:00
|
|
|
use std::collections::btree_map::{Entry, OccupiedEntry};
|
2020-12-27 21:03:10 +00:00
|
|
|
|
|
|
|
pub(crate) fn routes() -> Vec<rocket::Route> {
|
|
|
|
routes![
|
|
|
|
cofd::update_basic_info,
|
|
|
|
cofd::update_attributes,
|
|
|
|
cofd::update_attribute,
|
|
|
|
cofd::update_skills,
|
2021-01-02 14:51:24 +00:00
|
|
|
cofd::update_skill_value,
|
2020-12-27 21:03:10 +00:00
|
|
|
cofd::add_condition,
|
|
|
|
cofd::remove_condition
|
|
|
|
]
|
|
|
|
}
|
|
|
|
|
2020-12-29 14:01:07 +00:00
|
|
|
/// Load the character belonging to the given user, as long as they're
|
|
|
|
/// the owner of that character. Returns an error if user is not
|
|
|
|
/// logged in, the owner of the character is not found, or the logged
|
|
|
|
/// in user does not have the permission to access this character.
|
|
|
|
async fn load_character(
|
2020-12-29 23:06:41 +00:00
|
|
|
conn: &TenebrousDbConn<'_>,
|
2020-12-29 14:01:07 +00:00
|
|
|
logged_in_user: Option<&User>,
|
2021-01-02 22:03:10 +00:00
|
|
|
owner: &str,
|
2020-12-29 14:01:07 +00:00
|
|
|
character_id: i32,
|
|
|
|
) -> Result<Character, Error> {
|
|
|
|
let logged_in_user = logged_in_user.ok_or(Error::NotLoggedIn)?;
|
2020-12-31 20:19:45 +00:00
|
|
|
|
2020-12-29 14:01:07 +00:00
|
|
|
let character: Character = conn
|
|
|
|
.load_character(character_id)
|
|
|
|
.await?
|
|
|
|
.ok_or(Error::NotFound)?;
|
|
|
|
|
2021-01-02 22:03:10 +00:00
|
|
|
if &logged_in_user.username != owner {
|
2020-12-29 14:01:07 +00:00
|
|
|
return Err(Error::NoPermission);
|
|
|
|
}
|
|
|
|
|
|
|
|
Ok(character)
|
|
|
|
}
|
|
|
|
|
2020-12-27 21:03:10 +00:00
|
|
|
/// Protobuf-based REST endpoints for editing a character.
|
|
|
|
mod cofd {
|
|
|
|
use super::*;
|
2021-01-01 00:40:48 +00:00
|
|
|
use crate::models::proto::cofd::cofd_sheet::Skill;
|
2020-12-27 21:03:10 +00:00
|
|
|
use crate::models::proto::{cofd::api::*, cofd::*, Proto};
|
|
|
|
|
2021-01-02 14:51:24 +00:00
|
|
|
fn find_skill_entry<'a>(
|
|
|
|
sheet: &'a mut CofdSheet,
|
|
|
|
skill_name: &'a str,
|
|
|
|
) -> Option<OccupiedEntry<'a, String, Skill>> {
|
|
|
|
let all_skills = vec![
|
|
|
|
&mut sheet.mental_skills,
|
|
|
|
&mut sheet.physical_skills,
|
|
|
|
&mut sheet.social_skills,
|
|
|
|
];
|
|
|
|
|
|
|
|
// Search all skill lists for this value using "workaround" to
|
|
|
|
// break value from for loops.
|
|
|
|
let skill: Option<OccupiedEntry<_, _>> = 'l: loop {
|
|
|
|
for skill_map in all_skills {
|
|
|
|
if let Entry::Occupied(entry) = skill_map.entry(skill_name.to_owned()) {
|
|
|
|
break 'l Some(entry);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
break None;
|
|
|
|
};
|
|
|
|
|
|
|
|
skill
|
|
|
|
}
|
|
|
|
|
|
|
|
fn find_skill<'a>(sheet: &'a mut CofdSheet, skill_name: &'a str) -> Option<&'a mut Skill> {
|
|
|
|
find_skill_entry(sheet, skill_name).map(|entry| entry.into_mut())
|
|
|
|
}
|
|
|
|
|
2020-12-27 21:03:10 +00:00
|
|
|
#[post("/cofd/<owner>/<character_id>/basic-info", data = "<info>")]
|
|
|
|
pub(super) fn update_basic_info<'a>(
|
|
|
|
owner: String,
|
|
|
|
character_id: i32,
|
|
|
|
info: Proto<BasicInfo>,
|
|
|
|
) -> &'a str {
|
|
|
|
"lol"
|
|
|
|
}
|
|
|
|
|
|
|
|
#[post("/cofd/<owner>/<character_id>/attributes", data = "<info>")]
|
|
|
|
pub(super) fn update_attributes<'a>(
|
|
|
|
owner: String,
|
|
|
|
character_id: i32,
|
|
|
|
info: Proto<Attributes>,
|
|
|
|
) -> &'a str {
|
|
|
|
"lol"
|
|
|
|
}
|
|
|
|
|
2021-01-02 22:11:30 +00:00
|
|
|
#[post("/rpc/cofd/update_attribute", data = "<req>")]
|
2020-12-27 21:49:08 +00:00
|
|
|
pub(super) async fn update_attribute<'a>(
|
2021-01-02 22:11:30 +00:00
|
|
|
req: Proto<UpdateAttributeRequest>,
|
2020-12-29 23:06:41 +00:00
|
|
|
conn: TenebrousDbConn<'_>,
|
2020-12-27 21:49:08 +00:00
|
|
|
logged_in_user: Option<&User>,
|
|
|
|
) -> Result<&'a str, Error> {
|
2021-01-02 22:11:30 +00:00
|
|
|
let mut character = load_character(
|
|
|
|
&conn,
|
|
|
|
logged_in_user,
|
|
|
|
&req.character_username,
|
|
|
|
req.character_id,
|
|
|
|
)
|
|
|
|
.await?;
|
2020-12-27 21:49:08 +00:00
|
|
|
|
2021-01-02 22:11:30 +00:00
|
|
|
let mut sheet: CofdSheet = character.try_deserialize()?;
|
|
|
|
let value = req.attribute_value;
|
|
|
|
match req.attribute_name.to_lowercase().as_ref() {
|
|
|
|
"strength" => Ok(sheet.strength = value),
|
|
|
|
"dexterity" => Ok(sheet.dexterity = value),
|
|
|
|
"stamina" => Ok(sheet.stamina = value),
|
|
|
|
"intelligence" => Ok(sheet.intelligence = value),
|
|
|
|
"wits" => Ok(sheet.wits = value),
|
|
|
|
"resolve" => Ok(sheet.resolve = value),
|
|
|
|
"presence" => Ok(sheet.presence = value),
|
|
|
|
"manipulation" => Ok(sheet.manipulation = value),
|
|
|
|
"composure" => Ok(sheet.composure = value),
|
2020-12-27 21:49:08 +00:00
|
|
|
_ => Err(Error::InvalidInput),
|
|
|
|
}?;
|
|
|
|
|
|
|
|
character.update_data(sheet)?;
|
2020-12-29 23:06:41 +00:00
|
|
|
conn.update_character_sheet(&character).await?;
|
2020-12-27 21:49:08 +00:00
|
|
|
Ok("lol")
|
2020-12-27 21:03:10 +00:00
|
|
|
}
|
|
|
|
|
2021-01-02 14:51:24 +00:00
|
|
|
#[patch(
|
|
|
|
"/cofd/<owner>/<character_id>/skills",
|
|
|
|
data = "<skill_update>",
|
|
|
|
rank = 1
|
|
|
|
)]
|
2021-01-01 00:40:48 +00:00
|
|
|
pub(super) async fn update_skills<'a>(
|
2020-12-27 21:03:10 +00:00
|
|
|
owner: String,
|
|
|
|
character_id: i32,
|
2021-01-01 00:40:48 +00:00
|
|
|
skill_update: Proto<SkillUpdate>,
|
2020-12-29 23:06:41 +00:00
|
|
|
conn: TenebrousDbConn<'_>,
|
2021-01-01 00:40:48 +00:00
|
|
|
logged_in_user: Option<&User>,
|
|
|
|
) -> Result<&'a str, Error> {
|
2021-01-02 22:03:10 +00:00
|
|
|
let mut character = load_character(&conn, logged_in_user, &owner, character_id).await?;
|
2021-01-01 00:40:48 +00:00
|
|
|
let mut sheet: CofdSheet = character.try_deserialize()?;
|
2021-01-02 14:51:24 +00:00
|
|
|
let updated_skill: &Skill = skill_update.skill.as_ref().ok_or(Error::InvalidInput)?;
|
|
|
|
let skill_entry = find_skill_entry(&mut sheet, &skill_update.name);
|
2021-01-01 00:40:48 +00:00
|
|
|
|
|
|
|
skill_entry
|
2021-01-02 14:51:24 +00:00
|
|
|
.map(|mut entry| entry.insert(updated_skill.clone()))
|
2021-01-01 00:40:48 +00:00
|
|
|
.ok_or(Error::InvalidInput)?;
|
|
|
|
|
|
|
|
println!(
|
|
|
|
"updated skill {} with {:?}",
|
|
|
|
skill_update.name, skill_update.skill
|
|
|
|
);
|
|
|
|
|
|
|
|
character.update_data(sheet)?;
|
|
|
|
conn.update_character_sheet(&character).await?;
|
|
|
|
Ok("lol")
|
2020-12-27 21:03:10 +00:00
|
|
|
}
|
|
|
|
|
2021-01-02 22:03:10 +00:00
|
|
|
#[post("/rpc/cofd/update_skill_value", data = "<request>")]
|
2021-01-02 14:51:24 +00:00
|
|
|
pub(super) async fn update_skill_value<'a>(
|
2021-01-02 22:03:10 +00:00
|
|
|
request: Proto<UpdateSkillValueRequest>,
|
2021-01-02 14:51:24 +00:00
|
|
|
conn: TenebrousDbConn<'_>,
|
|
|
|
logged_in_user: Option<&User>,
|
|
|
|
) -> Result<&'a str, Error> {
|
2021-01-02 22:03:10 +00:00
|
|
|
println!("{:#?}", request);
|
|
|
|
let mut character = load_character(
|
|
|
|
&conn,
|
|
|
|
logged_in_user,
|
|
|
|
&request.character_username,
|
|
|
|
request.character_id,
|
|
|
|
)
|
|
|
|
.await?;
|
|
|
|
|
2021-01-02 14:51:24 +00:00
|
|
|
let mut sheet: CofdSheet = character.try_deserialize()?;
|
2021-01-02 22:03:10 +00:00
|
|
|
let skill: Option<&mut Skill> = find_skill(&mut sheet, &request.skill_name);
|
2021-01-02 14:51:24 +00:00
|
|
|
|
|
|
|
skill
|
2021-01-02 22:03:10 +00:00
|
|
|
.map(|s| s.dots = request.skill_value)
|
2021-01-02 14:51:24 +00:00
|
|
|
.ok_or(Error::InvalidInput)?;
|
|
|
|
|
|
|
|
println!("updated skill value",);
|
|
|
|
|
|
|
|
character.update_data(sheet)?;
|
|
|
|
conn.update_character_sheet(&character).await?;
|
|
|
|
Ok("lol")
|
|
|
|
}
|
|
|
|
|
2020-12-27 21:03:10 +00:00
|
|
|
#[put("/cofd/<owner>/<character_id>/conditions", data = "<info>")]
|
|
|
|
pub(super) fn add_condition<'a>(
|
|
|
|
owner: String,
|
|
|
|
character_id: i32,
|
|
|
|
info: Proto<Condition>,
|
|
|
|
) -> &'a str {
|
|
|
|
"lol"
|
|
|
|
}
|
|
|
|
|
|
|
|
#[delete("/cofd/<owner>/<character_id>/conditions", data = "<info>")]
|
|
|
|
pub(super) fn remove_condition<'a>(
|
|
|
|
owner: String,
|
|
|
|
character_id: i32,
|
|
|
|
info: Proto<Condition>,
|
|
|
|
) -> &'a str {
|
|
|
|
"lol"
|
|
|
|
}
|
|
|
|
}
|