conduit/src/pdu.rs

92 lines
3.1 KiB
Rust
Raw Normal View History

2020-04-04 09:53:37 +00:00
use js_int::UInt;
2020-06-05 16:19:26 +00:00
use ruma::{
2020-06-06 20:34:08 +00:00
api::federation::pdu::EventHash,
2020-06-05 16:19:26 +00:00
events::{
collections::all::{RoomEvent, StateEvent},
stripped::AnyStrippedStateEvent,
EventJson, EventType,
},
identifiers::{EventId, RoomId, UserId},
};
2020-04-04 09:53:37 +00:00
use serde::{Deserialize, Serialize};
2020-06-01 18:58:49 +00:00
use serde_json::json;
2020-04-04 09:53:37 +00:00
use std::collections::HashMap;
#[derive(Deserialize, Serialize)]
pub struct PduEvent {
pub event_id: EventId,
pub room_id: RoomId,
pub sender: UserId,
pub origin: String,
pub origin_server_ts: UInt,
#[serde(rename = "type")]
pub kind: EventType,
pub content: serde_json::Value,
#[serde(skip_serializing_if = "Option::is_none")]
pub state_key: Option<String>,
pub prev_events: Vec<EventId>,
pub depth: UInt,
pub auth_events: Vec<EventId>,
#[serde(skip_serializing_if = "Option::is_none")]
pub redacts: Option<EventId>,
#[serde(default, skip_serializing_if = "serde_json::Map::is_empty")]
pub unsigned: serde_json::Map<String, serde_json::Value>,
pub hashes: EventHash,
pub signatures: HashMap<String, HashMap<String, String>>,
}
impl PduEvent {
2020-05-26 08:27:51 +00:00
pub fn redact(&mut self) {
self.unsigned.clear();
let allowed = match self.kind {
EventType::RoomMember => vec!["membership"],
EventType::RoomCreate => vec!["creator"],
EventType::RoomJoinRules => vec!["join_rule"],
EventType::RoomPowerLevels => vec![
"ban",
"events",
"events_default",
"kick",
"redact",
"state_default",
"users",
"users_default",
],
EventType::RoomHistoryVisibility => vec!["history_visibility"],
_ => vec![],
};
let old_content = self.content.as_object_mut().unwrap(); // TODO error
let mut new_content = serde_json::Map::new();
for key in allowed {
if let Some(value) = old_content.remove(key) {
new_content.insert(key.to_owned(), value);
}
}
2020-06-01 18:58:49 +00:00
self.unsigned.insert(
"redacted_because".to_owned(),
json!({"content": {}, "type": "m.room.redaction"}),
);
2020-05-26 08:27:51 +00:00
self.content = new_content.into();
}
2020-04-23 12:27:50 +00:00
pub fn to_room_event(&self) -> EventJson<RoomEvent> {
2020-04-04 09:53:37 +00:00
// Can only fail in rare circumstances that won't ever happen here, see
// https://docs.rs/serde_json/1.0.50/serde_json/fn.to_string.html
let json = serde_json::to_string(&self).unwrap();
2020-04-23 12:27:50 +00:00
// EventJson's deserialize implementation always returns `Ok(...)`
serde_json::from_str::<EventJson<RoomEvent>>(&json).unwrap()
2020-04-04 09:53:37 +00:00
}
pub fn to_state_event(&self) -> EventJson<StateEvent> {
let json = serde_json::to_string(&self).unwrap();
serde_json::from_str::<EventJson<StateEvent>>(&json).unwrap()
}
2020-04-23 12:27:50 +00:00
pub fn to_stripped_state_event(&self) -> EventJson<AnyStrippedStateEvent> {
let json = serde_json::to_string(&self).unwrap();
2020-04-23 12:27:50 +00:00
serde_json::from_str::<EventJson<AnyStrippedStateEvent>>(&json).unwrap()
}
2020-04-04 09:53:37 +00:00
}