116 lines
3 KiB
Rust
116 lines
3 KiB
Rust
use std::collections::BTreeMap;
|
|
|
|
use ruma::MilliSecondsSinceUnixEpoch;
|
|
use serde::Deserialize;
|
|
use serde_json::value::{RawValue as RawJsonValue, Value as JsonValue, to_raw_value};
|
|
|
|
use super::Pdu;
|
|
use crate::{Result, err, implement, is_true};
|
|
|
|
#[implement(Pdu)]
|
|
pub fn remove_transaction_id(&mut self) -> Result {
|
|
use BTreeMap as Map;
|
|
|
|
let Some(unsigned) = &self.unsigned else {
|
|
return Ok(());
|
|
};
|
|
|
|
let mut unsigned: Map<&str, Box<RawJsonValue>> = serde_json::from_str(unsigned.get())
|
|
.map_err(|e| err!(Database("Invalid unsigned in pdu event: {e}")))?;
|
|
|
|
unsigned.remove("transaction_id");
|
|
self.unsigned = to_raw_value(&unsigned)
|
|
.map(Some)
|
|
.expect("unsigned is valid");
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[implement(Pdu)]
|
|
pub fn add_age(&mut self) -> Result {
|
|
use BTreeMap as Map;
|
|
|
|
let mut unsigned: Map<&str, Box<RawJsonValue>> = self
|
|
.unsigned
|
|
.as_deref()
|
|
.map(RawJsonValue::get)
|
|
.map_or_else(|| Ok(Map::new()), serde_json::from_str)
|
|
.map_err(|e| err!(Database("Invalid unsigned in pdu event: {e}")))?;
|
|
|
|
// deliberately allowing for the possibility of negative age
|
|
let now: i128 = MilliSecondsSinceUnixEpoch::now().get().into();
|
|
let then: i128 = self.origin_server_ts.into();
|
|
let this_age = now.saturating_sub(then);
|
|
|
|
unsigned.insert("age", to_raw_value(&this_age)?);
|
|
self.unsigned = Some(to_raw_value(&unsigned)?);
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[implement(Pdu)]
|
|
pub fn add_relation(&mut self, name: &str, pdu: Option<&Pdu>) -> Result {
|
|
use serde_json::Map;
|
|
|
|
let mut unsigned: Map<String, JsonValue> = self
|
|
.unsigned
|
|
.as_deref()
|
|
.map(RawJsonValue::get)
|
|
.map_or_else(|| Ok(Map::new()), serde_json::from_str)
|
|
.map_err(|e| err!(Database("Invalid unsigned in pdu event: {e}")))?;
|
|
|
|
let pdu = pdu
|
|
.map(serde_json::to_value)
|
|
.transpose()?
|
|
.unwrap_or_else(|| JsonValue::Object(Map::new()));
|
|
|
|
unsigned
|
|
.entry("m.relations")
|
|
.or_insert(JsonValue::Object(Map::new()))
|
|
.as_object_mut()
|
|
.map(|object| object.insert(name.to_owned(), pdu));
|
|
|
|
self.unsigned = Some(to_raw_value(&unsigned)?);
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[implement(Pdu)]
|
|
pub fn contains_unsigned_property<F>(&self, property: &str, is_type: F) -> bool
|
|
where
|
|
F: FnOnce(&JsonValue) -> bool,
|
|
{
|
|
self.get_unsigned_as_value()
|
|
.get(property)
|
|
.map(is_type)
|
|
.is_some_and(is_true!())
|
|
}
|
|
|
|
#[implement(Pdu)]
|
|
pub fn get_unsigned_property<T>(&self, property: &str) -> Result<T>
|
|
where
|
|
T: for<'de> Deserialize<'de>,
|
|
{
|
|
self.get_unsigned_as_value()
|
|
.get_mut(property)
|
|
.map(JsonValue::take)
|
|
.map(serde_json::from_value)
|
|
.ok_or(err!(Request(NotFound("property not found in unsigned object"))))?
|
|
.map_err(|e| err!(Database("Failed to deserialize unsigned.{property} into type: {e}")))
|
|
}
|
|
|
|
#[implement(Pdu)]
|
|
#[must_use]
|
|
pub fn get_unsigned_as_value(&self) -> JsonValue {
|
|
self.get_unsigned::<JsonValue>().unwrap_or_default()
|
|
}
|
|
|
|
#[implement(Pdu)]
|
|
pub fn get_unsigned<T>(&self) -> Result<JsonValue> {
|
|
self.unsigned
|
|
.as_ref()
|
|
.map(|raw| raw.get())
|
|
.map(serde_json::from_str)
|
|
.ok_or(err!(Request(NotFound("\"unsigned\" property not found in pdu"))))?
|
|
.map_err(|e| err!(Database("Failed to deserialize \"unsigned\" into value: {e}")))
|
|
}
|