mirror of
https://forgejo.ellis.link/continuwuation/continuwuity.git
synced 2025-07-28 02:38:30 +00:00
d98ce2c7b9
The first part of getting admin command docs on the website. There's also the beginnings of manpage generation here, although it's kinda sus and I'm not sure how it's supposed to work. I'll leave that to anyone who wants to package it. We introduce the beginings of the xtask pattern here - we do a lot of file generation, I thought it would be best to avoid doing that on every compilation. It also helps avoid lots of runtime deps. We'll need to document generating this stuff & probably add pre-commit hooks for it, though.
70 lines
1.7 KiB
Rust
70 lines
1.7 KiB
Rust
use clap::Subcommand;
|
|
use conduwuit::Result;
|
|
use futures::StreamExt;
|
|
use ruma::{OwnedRoomId, OwnedUserId};
|
|
|
|
use crate::{admin_command, admin_command_dispatch};
|
|
|
|
#[admin_command_dispatch]
|
|
#[derive(Debug, Subcommand)]
|
|
/// All the getters and iterators from src/database/key_value/account_data.rs
|
|
pub enum AccountDataCommand {
|
|
/// - Returns all changes to the account data that happened after `since`.
|
|
ChangesSince {
|
|
/// Full user ID
|
|
user_id: OwnedUserId,
|
|
/// UNIX timestamp since (u64)
|
|
since: u64,
|
|
/// Optional room ID of the account data
|
|
room_id: Option<OwnedRoomId>,
|
|
},
|
|
|
|
/// - Searches the account data for a specific kind.
|
|
AccountDataGet {
|
|
/// Full user ID
|
|
user_id: OwnedUserId,
|
|
/// Account data event type
|
|
kind: String,
|
|
/// Optional room ID of the account data
|
|
room_id: Option<OwnedRoomId>,
|
|
},
|
|
}
|
|
|
|
#[admin_command]
|
|
async fn changes_since(
|
|
&self,
|
|
user_id: OwnedUserId,
|
|
since: u64,
|
|
room_id: Option<OwnedRoomId>,
|
|
) -> Result {
|
|
let timer = tokio::time::Instant::now();
|
|
let results: Vec<_> = self
|
|
.services
|
|
.account_data
|
|
.changes_since(room_id.as_deref(), &user_id, since, None)
|
|
.collect()
|
|
.await;
|
|
let query_time = timer.elapsed();
|
|
|
|
self.write_str(&format!("Query completed in {query_time:?}:\n\n```rs\n{results:#?}\n```"))
|
|
.await
|
|
}
|
|
|
|
#[admin_command]
|
|
async fn account_data_get(
|
|
&self,
|
|
user_id: OwnedUserId,
|
|
kind: String,
|
|
room_id: Option<OwnedRoomId>,
|
|
) -> Result {
|
|
let timer = tokio::time::Instant::now();
|
|
let results = self
|
|
.services
|
|
.account_data
|
|
.get_raw(room_id.as_deref(), &user_id, &kind)
|
|
.await;
|
|
let query_time = timer.elapsed();
|
|
|
|
self.write_str(&format!("Query completed in {query_time:?}:\n\n```rs\n{results:#?}\n```"))
|
|
.await
|
|
}
|