chore: Delete sled-state-inspector
The sled store is on its way out and nobody is using this.pull/1553/head
Jonas Platte
4 months ago
committed by
Jonas Platte
parent
b1062a67e0
commit
95f1867816
@ -1,22 +0,0 @@
|
||||
[package]
|
||||
name = "sled-state-inspector"
|
||||
version = "0.1.0"
|
||||
edition = "2021"
|
||||
publish = false
|
||||
|
||||
[[bin]]
|
||||
name = "sled-state-inspector"
|
||||
test = false
|
||||
|
||||
[dependencies]
|
||||
atty = "0.2.14"
|
||||
clap = "3.2.4"
|
||||
futures = { version = "0.3.21", default-features = false, features = ["executor"] }
|
||||
matrix-sdk-base = { path = "../../crates/matrix-sdk-base", version = "0.6.0"}
|
||||
matrix-sdk-sled = { path = "../../crates/matrix-sdk-sled", version = "0.2.0"}
|
||||
ruma = { workspace = true }
|
||||
rustyline = "10.0.0"
|
||||
rustyline-derive = "0.7.0"
|
||||
serde = { workspace = true }
|
||||
serde_json = { workspace = true }
|
||||
syntect = { version = "5.0.0", default-features = false, features = ["dump-load", "parsing", "regex-fancy"] }
|
||||
@ -1,367 +0,0 @@
|
||||
use std::{fmt::Debug, sync::Arc};
|
||||
|
||||
use atty::Stream;
|
||||
use clap::{Arg, ArgMatches, Command as Argparse};
|
||||
use futures::executor::block_on;
|
||||
use matrix_sdk_base::{RoomInfo, StateStore};
|
||||
use matrix_sdk_sled::SledStateStore;
|
||||
use ruma::{events::StateEventType, OwnedRoomId, OwnedUserId, RoomId};
|
||||
use rustyline::{
|
||||
completion::{Completer, Pair},
|
||||
error::ReadlineError,
|
||||
highlight::{Highlighter, MatchingBracketHighlighter},
|
||||
hint::{Hinter, HistoryHinter},
|
||||
validate::{MatchingBracketValidator, Validator},
|
||||
CompletionType, Config, Context, EditMode, Editor,
|
||||
};
|
||||
use rustyline_derive::Helper;
|
||||
use serde::Serialize;
|
||||
use syntect::{
|
||||
dumps::from_binary,
|
||||
easy::HighlightLines,
|
||||
highlighting::{Style, ThemeSet},
|
||||
parsing::SyntaxSet,
|
||||
util::{as_24_bit_terminal_escaped, LinesWithEndings},
|
||||
};
|
||||
|
||||
#[derive(Clone)]
|
||||
struct Inspector {
|
||||
store: Arc<SledStateStore>,
|
||||
printer: Printer,
|
||||
}
|
||||
|
||||
#[derive(Helper)]
|
||||
struct InspectorHelper {
|
||||
store: Arc<SledStateStore>,
|
||||
_highlighter: MatchingBracketHighlighter,
|
||||
_validator: MatchingBracketValidator,
|
||||
_hinter: HistoryHinter,
|
||||
}
|
||||
|
||||
impl InspectorHelper {
|
||||
const EVENT_TYPES: &'static [&'static str] = &[
|
||||
"m.room.aliases",
|
||||
"m.room.avatar",
|
||||
"m.room.canonical_alias",
|
||||
"m.room.create",
|
||||
"m.room.encryption",
|
||||
"m.room.guest_access",
|
||||
"m.room.history_visibility",
|
||||
"m.room.join_rules",
|
||||
"m.room.name",
|
||||
"m.room.power_levels",
|
||||
"m.room.tombstone",
|
||||
"m.room.topic",
|
||||
];
|
||||
|
||||
fn new(store: Arc<SledStateStore>) -> Self {
|
||||
Self {
|
||||
store,
|
||||
_highlighter: MatchingBracketHighlighter::new(),
|
||||
_validator: MatchingBracketValidator::new(),
|
||||
_hinter: HistoryHinter {},
|
||||
}
|
||||
}
|
||||
|
||||
fn complete_event_types(&self, arg: Option<&&str>) -> Vec<Pair> {
|
||||
Self::EVENT_TYPES
|
||||
.iter()
|
||||
.map(|&t| Pair { display: t.to_owned(), replacement: format!("{t} ") })
|
||||
.filter(|r| if let Some(arg) = arg { r.replacement.starts_with(arg) } else { true })
|
||||
.collect()
|
||||
}
|
||||
|
||||
fn complete_rooms(&self, arg: Option<&&str>) -> Vec<Pair> {
|
||||
let rooms: Vec<RoomInfo> =
|
||||
block_on(async { StateStore::get_room_infos(&*self.store).await.unwrap() });
|
||||
|
||||
rooms
|
||||
.into_iter()
|
||||
.map(|r| Pair {
|
||||
display: r.room_id().to_string(),
|
||||
replacement: format!("{} ", r.room_id()),
|
||||
})
|
||||
.filter(|r| if let Some(arg) = arg { r.replacement.starts_with(arg) } else { true })
|
||||
.collect()
|
||||
}
|
||||
}
|
||||
|
||||
impl Completer for InspectorHelper {
|
||||
type Candidate = Pair;
|
||||
|
||||
fn complete(
|
||||
&self,
|
||||
line: &str,
|
||||
pos: usize,
|
||||
_: &Context<'_>,
|
||||
) -> Result<(usize, Vec<Pair>), ReadlineError> {
|
||||
let args: Vec<&str> = line.split_ascii_whitespace().collect();
|
||||
|
||||
let commands = vec![
|
||||
("get-state", "get a state event in the given room"),
|
||||
("get-profiles", "get all the stored profiles in the given room"),
|
||||
("list-rooms", "list all rooms"),
|
||||
("get-members", "get all the membership events in the given room"),
|
||||
]
|
||||
.iter()
|
||||
.map(|(r, d)| Pair { display: format!("{r} ({d})"), replacement: format!("{r} ") })
|
||||
.collect();
|
||||
|
||||
if args.is_empty() {
|
||||
Ok((pos, commands))
|
||||
} else if args.len() == 1 {
|
||||
if (args[0] == "get-state" || args[0] == "get-members" || args[0] == "get-profiles")
|
||||
&& line.ends_with(' ')
|
||||
{
|
||||
Ok((args[0].len() + 1, self.complete_rooms(args.get(1))))
|
||||
} else {
|
||||
Ok((
|
||||
0,
|
||||
commands.into_iter().filter(|c| c.replacement.starts_with(args[0])).collect(),
|
||||
))
|
||||
}
|
||||
} else if args.len() == 2 {
|
||||
if args[0] == "get-state" {
|
||||
if line.ends_with(' ') {
|
||||
Ok((args[0].len() + args[1].len() + 2, self.complete_event_types(args.get(2))))
|
||||
} else {
|
||||
Ok((args[0].len() + 1, self.complete_rooms(args.get(1))))
|
||||
}
|
||||
} else if args[0] == "get-members" || args[0] == "get-profiles" {
|
||||
Ok((args[0].len() + 1, self.complete_rooms(args.get(1))))
|
||||
} else {
|
||||
Ok((pos, vec![]))
|
||||
}
|
||||
} else if args.len() == 3 {
|
||||
if args[0] == "get-state" {
|
||||
Ok((args[0].len() + args[1].len() + 2, self.complete_event_types(args.get(2))))
|
||||
} else {
|
||||
Ok((pos, vec![]))
|
||||
}
|
||||
} else {
|
||||
Ok((pos, vec![]))
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl Hinter for InspectorHelper {
|
||||
type Hint = String;
|
||||
}
|
||||
|
||||
impl Highlighter for InspectorHelper {}
|
||||
|
||||
impl Validator for InspectorHelper {}
|
||||
|
||||
#[derive(Clone, Debug)]
|
||||
struct Printer {
|
||||
ps: Arc<SyntaxSet>,
|
||||
ts: Arc<ThemeSet>,
|
||||
json: bool,
|
||||
color: bool,
|
||||
}
|
||||
|
||||
impl Printer {
|
||||
fn new(json: bool, color: bool) -> Self {
|
||||
let syntax_set: SyntaxSet = from_binary(include_bytes!("../syntaxes.bin"));
|
||||
let themes: ThemeSet = from_binary(include_bytes!("../themes.bin"));
|
||||
|
||||
Self { ps: syntax_set.into(), ts: themes.into(), json, color }
|
||||
}
|
||||
|
||||
fn pretty_print_struct<T: Debug + Serialize>(&self, data: &T) {
|
||||
let data = if self.json {
|
||||
serde_json::to_string_pretty(data).expect("Can't serialize struct")
|
||||
} else {
|
||||
format!("{data:#?}")
|
||||
};
|
||||
|
||||
let syntax = if self.json {
|
||||
self.ps.find_syntax_by_extension("rs").expect("Can't find rust syntax extension")
|
||||
} else {
|
||||
self.ps.find_syntax_by_extension("json").expect("Can't find json syntax extension")
|
||||
};
|
||||
|
||||
if self.color {
|
||||
let mut h = HighlightLines::new(syntax, &self.ts.themes["Forest Night"]);
|
||||
|
||||
for line in LinesWithEndings::from(&data) {
|
||||
let ranges: Vec<(Style, &str)> =
|
||||
h.highlight_line(line, &self.ps).expect("Failed to highlight line");
|
||||
let escaped = as_24_bit_terminal_escaped(&ranges[..], false);
|
||||
print!("{escaped}");
|
||||
}
|
||||
|
||||
// Clear the formatting
|
||||
println!("\x1b[0m");
|
||||
} else {
|
||||
println!("{data}");
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl Inspector {
|
||||
fn new(database_path: &str, json: bool, color: bool) -> Self {
|
||||
let printer = Printer::new(json, color);
|
||||
let store = Arc::new(
|
||||
SledStateStore::builder()
|
||||
.path(database_path.into())
|
||||
.build()
|
||||
.expect("Can't open sled database"),
|
||||
);
|
||||
|
||||
Self { store, printer }
|
||||
}
|
||||
|
||||
async fn run(&self, matches: ArgMatches) {
|
||||
match matches.subcommand() {
|
||||
Some(("get-profiles", args)) => {
|
||||
let room_id = RoomId::parse(args.value_of("room-id").unwrap()).unwrap();
|
||||
|
||||
self.get_profiles(room_id).await;
|
||||
}
|
||||
|
||||
Some(("get-members", args)) => {
|
||||
let room_id = RoomId::parse(args.value_of("room-id").unwrap()).unwrap();
|
||||
|
||||
self.get_members(room_id).await;
|
||||
}
|
||||
Some(("list-rooms", _)) => self.list_rooms().await,
|
||||
Some(("get-display-names", args)) => {
|
||||
let room_id = RoomId::parse(args.value_of("room-id").unwrap()).unwrap();
|
||||
let display_name = args.value_of("display-name").unwrap().to_owned();
|
||||
self.get_display_name_owners(room_id, display_name).await;
|
||||
}
|
||||
Some(("get-state", args)) => {
|
||||
let room_id = RoomId::parse(args.value_of("room-id").unwrap()).unwrap();
|
||||
let event_type =
|
||||
StateEventType::try_from(args.value_of("event-type").unwrap()).unwrap();
|
||||
self.get_state(room_id, event_type).await;
|
||||
}
|
||||
_ => unreachable!(),
|
||||
}
|
||||
}
|
||||
|
||||
async fn list_rooms(&self) {
|
||||
let rooms: Vec<RoomInfo> = StateStore::get_room_infos(&*self.store).await.unwrap();
|
||||
self.printer.pretty_print_struct(&rooms);
|
||||
}
|
||||
|
||||
async fn get_display_name_owners(&self, room_id: OwnedRoomId, display_name: String) {
|
||||
let users = self.store.get_users_with_display_name(&room_id, &display_name).await.unwrap();
|
||||
self.printer.pretty_print_struct(&users);
|
||||
}
|
||||
|
||||
async fn get_profiles(&self, room_id: OwnedRoomId) {
|
||||
let joined: Vec<OwnedUserId> =
|
||||
StateStore::get_joined_user_ids(&*self.store, &room_id).await.unwrap();
|
||||
|
||||
for member in joined {
|
||||
let event = self.store.get_profile(&room_id, &member).await.unwrap();
|
||||
self.printer.pretty_print_struct(&event);
|
||||
}
|
||||
}
|
||||
|
||||
async fn get_members(&self, room_id: OwnedRoomId) {
|
||||
let joined: Vec<OwnedUserId> =
|
||||
StateStore::get_joined_user_ids(&*self.store, &room_id).await.unwrap();
|
||||
|
||||
for member in joined {
|
||||
let event = self.store.get_member_event(&room_id, &member).await.unwrap();
|
||||
self.printer.pretty_print_struct(&event);
|
||||
}
|
||||
}
|
||||
|
||||
async fn get_state(&self, room_id: OwnedRoomId, event_type: StateEventType) {
|
||||
self.printer.pretty_print_struct(
|
||||
&self.store.get_state_event(&room_id, event_type, "").await.unwrap(),
|
||||
);
|
||||
}
|
||||
|
||||
fn subcommands() -> Vec<Argparse<'static>> {
|
||||
vec![
|
||||
Argparse::new("list-rooms"),
|
||||
Argparse::new("get-members").arg(Arg::new("room-id").required(true).validator(|r| {
|
||||
RoomId::parse(r).map(|_| ()).map_err(|_| "Invalid room id given".to_owned())
|
||||
})),
|
||||
Argparse::new("get-profiles").arg(Arg::new("room-id").required(true).validator(|r| {
|
||||
RoomId::parse(r).map(|_| ()).map_err(|_| "Invalid room id given".to_owned())
|
||||
})),
|
||||
Argparse::new("get-display-names")
|
||||
.arg(Arg::new("room-id").required(true).validator(|r| {
|
||||
RoomId::parse(r).map(|_| ()).map_err(|_| "Invalid room id given".to_owned())
|
||||
}))
|
||||
.arg(Arg::new("display-name").required(true)),
|
||||
Argparse::new("get-state")
|
||||
.arg(Arg::new("room-id").required(true).validator(|r| {
|
||||
RoomId::parse(r).map(|_| ()).map_err(|_| "Invalid room id given".to_owned())
|
||||
}))
|
||||
.arg(Arg::new("event-type").required(true).validator(|e| {
|
||||
StateEventType::try_from(e)
|
||||
.map(|_| ())
|
||||
.map_err(|_| "Invalid event type".to_owned())
|
||||
})),
|
||||
]
|
||||
}
|
||||
|
||||
async fn parse_and_run(&self, input: &str) {
|
||||
let argparse = Argparse::new("state-inspector")
|
||||
.disable_version_flag(true)
|
||||
.disable_help_flag(true)
|
||||
.no_binary_name(true)
|
||||
.subcommand_required(true)
|
||||
.arg_required_else_help(true)
|
||||
.subcommands(Inspector::subcommands());
|
||||
|
||||
match argparse.try_get_matches_from(input.split_ascii_whitespace()) {
|
||||
Ok(m) => {
|
||||
self.run(m).await;
|
||||
}
|
||||
Err(e) => {
|
||||
println!("{e}");
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
fn main() {
|
||||
let argparse = Argparse::new("state-inspector")
|
||||
.disable_version_flag(true)
|
||||
.arg(Arg::new("database").required(true))
|
||||
.arg(
|
||||
Arg::new("json")
|
||||
.long("json")
|
||||
.help("set the output to raw json instead of Rust structs")
|
||||
.global(true)
|
||||
.takes_value(false),
|
||||
)
|
||||
.subcommands(Inspector::subcommands());
|
||||
|
||||
let matches = argparse.get_matches();
|
||||
|
||||
let database_path = matches.value_of("database").expect("No database path");
|
||||
let json = matches.is_present("json");
|
||||
let color = atty::is(Stream::Stdout);
|
||||
|
||||
let inspector = Inspector::new(database_path, json, color);
|
||||
|
||||
if matches.subcommand().is_none() {
|
||||
let config = Config::builder()
|
||||
.history_ignore_space(true)
|
||||
.completion_type(CompletionType::List)
|
||||
.edit_mode(EditMode::Emacs)
|
||||
.build();
|
||||
|
||||
let helper = InspectorHelper::new(inspector.store.clone());
|
||||
|
||||
let mut rl =
|
||||
Editor::<InspectorHelper>::with_config(config).expect("Failed to create Editor");
|
||||
rl.set_helper(Some(helper));
|
||||
|
||||
while let Ok(input) = rl.readline(">> ") {
|
||||
rl.add_history_entry(input.as_str());
|
||||
block_on(inspector.parse_and_run(input.as_str()));
|
||||
}
|
||||
} else {
|
||||
block_on(inspector.run(matches));
|
||||
}
|
||||
}
|
||||
Binary file not shown.
Binary file not shown.
Loading…
Reference in New Issue