91 lines
2.9 KiB
Rust
91 lines
2.9 KiB
Rust
use std::sync::Arc;
|
|
|
|
use anyhow::Context;
|
|
use uuid::Uuid;
|
|
use warp::{Filter, Rejection, Reply};
|
|
use warp::filters::BoxedFilter;
|
|
|
|
use crate::message::Message;
|
|
use crate::State;
|
|
use crate::util::HOUSING_ZONES;
|
|
use crate::web::{AnyhowRejection, WebError};
|
|
|
|
pub fn write(state: Arc<State>) -> BoxedFilter<(impl Reply, )> {
|
|
warp::post()
|
|
.and(warp::path("messages"))
|
|
.and(warp::path::end())
|
|
.and(super::get_id(Arc::clone(&state)))
|
|
.and(warp::body::content_length_limit(1024))
|
|
.and(warp::body::json())
|
|
.and_then(move |(id, extra), message: Message| logic(Arc::clone(&state), id, extra, message))
|
|
.boxed()
|
|
}
|
|
|
|
async fn logic(state: Arc<State>, id: i64, extra: i64, message: Message) -> Result<impl Reply, Rejection> {
|
|
let housing = HOUSING_ZONES.contains(&message.territory);
|
|
if housing && (message.world.is_none() || message.ward.is_none()) {
|
|
return Err(warp::reject::custom(WebError::MissingHousingInfo));
|
|
}
|
|
|
|
if !housing && (message.ward.is_some() || message.plot.is_some()) {
|
|
return Err(warp::reject::custom(WebError::UnnecessaryHousingInfo));
|
|
}
|
|
|
|
let text = {
|
|
let packs = state.packs.read().await;
|
|
let pack = packs.get(&message.pack_id)
|
|
.ok_or(WebError::InvalidPackId)
|
|
.map_err(warp::reject::custom)?;
|
|
pack.format(
|
|
message.template_1,
|
|
message.word_1_list.and_then(|list| message.word_1_word.map(|word| (list, word))),
|
|
message.conjunction,
|
|
message.template_2,
|
|
message.word_2_list.and_then(|list| message.word_2_word.map(|word| (list, word))),
|
|
)
|
|
.ok_or(WebError::InvalidIndex)
|
|
.map_err(warp::reject::custom)?
|
|
};
|
|
|
|
let existing = sqlx::query!(
|
|
// language=sqlite
|
|
"select count(*) as count from messages where user = ?",
|
|
id
|
|
)
|
|
.fetch_one(&state.db)
|
|
.await
|
|
.context("could not get count of messages")
|
|
.map_err(AnyhowRejection)
|
|
.map_err(warp::reject::custom)?;
|
|
|
|
if existing.count >= 10 + extra as i32 {
|
|
return Err(warp::reject::custom(WebError::TooManyMessages));
|
|
}
|
|
|
|
let message_id = Uuid::new_v4().simple().to_string();
|
|
let territory = message.territory as i64;
|
|
|
|
sqlx::query!(
|
|
// language=sqlite
|
|
"insert into messages (id, user, territory, world, ward, plot, x, y, z, yaw, message, glyph) values (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)",
|
|
message_id,
|
|
id,
|
|
territory,
|
|
message.world,
|
|
message.ward,
|
|
message.plot,
|
|
message.x,
|
|
message.y,
|
|
message.z,
|
|
message.yaw,
|
|
text,
|
|
message.glyph,
|
|
)
|
|
.execute(&state.db)
|
|
.await
|
|
.context("could not insert message into database")
|
|
.map_err(AnyhowRejection)
|
|
.map_err(warp::reject::custom)?;
|
|
Ok(message_id)
|
|
}
|