add command to get the state of all rooms

src/cli/get_room_states.rs

@@ -0,0 +1,71 @@
+#![warn(clippy::missing_docs_in_private_items)]
+
+//! Implementation of the `get-room-states` command
+
+use std::{cmp::Ordering, sync::Arc};
+
+use ruma::events::StateEventType;
+use serde::Serialize;
+
+use super::GetRoomStatesArgs;
+use crate::{
+    config, database::KeyValueDatabase, error, services, PduEvent, Services,
+};
+
+mod cache;
+
+/// Serializable information about a state event
+#[derive(Serialize, PartialEq, Eq)]
+struct StateEvent {
+    /// The kind of state event
+    kind: StateEventType,
+
+    /// The `state_key` of the event
+    key: String,
+
+    /// The event itself
+    event: Arc<PduEvent>,
+}
+
+impl Ord for StateEvent {
+    fn cmp(&self, other: &Self) -> Ordering {
+        Ordering::Equal
+            .then_with(|| self.event.room_id.cmp(&other.event.room_id))
+            .then_with(|| self.kind.cmp(&other.kind))
+            .then_with(|| self.key.cmp(&other.key))
+            .then_with(|| self.event.event_id.cmp(&other.event.event_id))
+    }
+}
+
+impl PartialOrd for StateEvent {
+    fn partial_cmp(&self, other: &Self) -> Option<Ordering> {
+        Some(self.cmp(other))
+    }
+}
+
+/// Subcommand entrypoint
+pub(crate) async fn run(
+    args: GetRoomStatesArgs,
+) -> Result<(), error::DumpStateCommand> {
+    use error::DumpStateCommand as Error;
+
+    let config = config::load(args.config.config.as_ref()).await?;
+
+    let db = Box::leak(Box::new(
+        KeyValueDatabase::load_or_create(&config).map_err(Error::Database)?,
+    ));
+
+    Services::new(db, config, None)
+        .map_err(Error::InitializeServices)?
+        .install();
+
+    services().globals.err_if_server_name_changed()?;
+
+    db.apply_migrations().await.map_err(Error::Database)?;
+
+    let room_states = cache::get_room_states().await;
+
+    serde_json::to_writer(std::io::stdout(), &room_states)?;
+
+    Ok(())
+}