eightball/PluralKit
1
0
Fork 0
mirror of https://github.com/PluralKit/PluralKit.git synced 2026-02-04 13:06:50 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

refactor(commands): make tree type properly hygienic

Browse source
This commit is contained in:
dusk 2025-01-11 20:25:41 +09:00
parent 3120e62dda
commit ee45fca6ab
No known key found for this signature in database
2 changed files with 21 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

15
crates/commands/src/lib.rs
View file

@ -63,12 +63,9 @@ pub fn parse_command(input: String) -> CommandResult {
let mut flags: HashMap<String, Option<String>> = HashMap::new();
loop {
println!("possible: {:?}", local_tree.branches.keys());
let next = next_token(
local_tree.branches.keys().cloned().collect(),
input.clone(),
current_pos,
);
let possible_tokens = local_tree.possible_tokens().cloned().collect::<Vec<_>>();
println!("possible: {:?}", possible_tokens);
let next = next_token(possible_tokens.clone(), input.clone(), current_pos);
println!("next: {:?}", next);
match next {
Ok((found_token, arg, new_pos)) => {
@ -87,14 +84,14 @@ pub fn parse_command(input: String) -> CommandResult {
args.push(arg.raw.to_string());
}
if let Some(next_tree) = local_tree.branches.get(&found_token) {
if let Some(next_tree) = local_tree.get_branch(&found_token) {
local_tree = next_tree.clone();
} else {
panic!("found token could not match tree, at {input}");
}
}
Err(None) => {
if let Some(command_ref) = local_tree.current_command_key {
if let Some(command_ref) = local_tree.callback() {
println!("{command_ref} {params:?}");
return CommandResult::Ok {
command: ParsedCommand {
@ -106,6 +103,8 @@ pub fn parse_command(input: String) -> CommandResult {
};
}
println!("{possible_tokens:?}");
// todo: check if last token is a common incorrect unquote (multi-member names etc)
// todo: check if this is a system name in pk;s command
return CommandResult::Err {

16
crates/commands/src/tree.rs
View file

@ -5,8 +5,8 @@ use crate::{commands::Command, Token};
#[derive(Debug, Clone)]
pub struct TreeBranch {
pub current_command_key: Option<SmolStr>,
pub branches: OrderMap<Token, TreeBranch>,
current_command_key: Option<SmolStr>,
branches: OrderMap<Token, TreeBranch>,
}
impl TreeBranch {
@ -36,4 +36,16 @@ impl TreeBranch {
},
);
}
pub fn callback(&self) -> Option<SmolStr> {
self.current_command_key.clone()
}
pub fn possible_tokens(&self) -> impl Iterator<Item = &Token> {
self.branches.keys()
}
pub fn get_branch(&self, token: &Token) -> Option<&TreeBranch> {
self.branches.get(token)
}
}