conduit/src/pdu.rs

use crate::{Error, Result};
use js_int::UInt;
use ruma::{
    events::{
        pdu::EventHash, room::member::MemberEventContent, AnyRoomEvent, AnyStateEvent,
        AnyStrippedStateEvent, AnySyncRoomEvent, AnySyncStateEvent, EventJson, EventType,
        StateEvent,
    },
    identifiers::{EventId, RoomId, ServerName, 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: Box<ServerName>,
    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 {
    pub fn redact(&mut self) -> Result<()> {
        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()
            .ok_or_else(|| Error::bad_database("PDU in db has invalid content."))?;

        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();

        Ok(())
    }

    pub fn to_sync_room_event(&self) -> EventJson<AnySyncRoomEvent> {
        let json = serde_json::to_string(&self).expect("PDUs are always valid");
        serde_json::from_str::<AnySyncRoomEvent>(&json)
            .map(EventJson::from)
            .expect("AnySyncRoomEvent can always be built from a full PDU event")
    }
    pub fn to_room_event(&self) -> EventJson<AnyRoomEvent> {
        let json = serde_json::to_string(&self).expect("PDUs are always valid");
        serde_json::from_str::<AnyRoomEvent>(&json)
            .map(EventJson::from)
            .expect("AnyRoomEvent can always be built from a full PDU event")
    }
    pub fn to_state_event(&self) -> EventJson<AnyStateEvent> {
        let json = serde_json::to_string(&self).expect("PDUs are always valid");
        serde_json::from_str::<AnyStateEvent>(&json)
            .map(EventJson::from)
            .expect("AnyStateEvent can always be built from a full PDU event")
    }
    pub fn to_sync_state_event(&self) -> EventJson<AnySyncStateEvent> {
        let json = serde_json::to_string(&self).expect("PDUs are always valid");
        serde_json::from_str::<AnySyncStateEvent>(&json)
            .map(EventJson::from)
            .expect("AnySyncStateEvent can always be built from a full PDU event")
    }
    pub fn to_stripped_state_event(&self) -> EventJson<AnyStrippedStateEvent> {
        let json = serde_json::to_string(&self).expect("PDUs are always valid");
        serde_json::from_str::<AnyStrippedStateEvent>(&json)
            .map(EventJson::from)
            .expect("AnyStrippedStateEvent can always be built from a full PDU event")
    }
    pub fn to_member_event(&self) -> EventJson<StateEvent<MemberEventContent>> {
        let json = serde_json::to_string(&self).expect("PDUs are always valid");
        serde_json::from_str::<StateEvent<MemberEventContent>>(&json)
            .map(EventJson::from)
            .expect("StateEvent<MemberEventContent> can always be built from a full PDU event")
    }
}