use js_int::UInt; use ruma_events::{ collections::all::{RoomEvent, StateEvent}, stripped::AnyStrippedStateEvent, EventJson, EventType, }; use ruma_federation_api::EventHash; use ruma_identifiers::{EventId, RoomId, UserId}; use serde::{Deserialize, Serialize}; use serde_json::json; 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, pub prev_events: Vec, pub depth: UInt, pub auth_events: Vec, #[serde(skip_serializing_if = "Option::is_none")] pub redacts: Option, #[serde(default, skip_serializing_if = "serde_json::Map::is_empty")] pub unsigned: serde_json::Map, pub hashes: EventHash, pub signatures: HashMap>, } impl PduEvent { 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); } } self.unsigned.insert( "redacted_because".to_owned(), json!({"content": {}, "type": "m.room.redaction"}), ); self.content = new_content.into(); } pub fn to_room_event(&self) -> EventJson { // 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(); // EventJson's deserialize implementation always returns `Ok(...)` serde_json::from_str::>(&json).unwrap() } pub fn to_state_event(&self) -> EventJson { let json = serde_json::to_string(&self).unwrap(); serde_json::from_str::>(&json).unwrap() } pub fn to_stripped_state_event(&self) -> EventJson { let json = serde_json::to_string(&self).unwrap(); serde_json::from_str::>(&json).unwrap() } }