Start9/rpc-toolkit
2
0
Fork 0
mirror of https://github.com/Start9Labs/rpc-toolkit.git synced 2026-03-26 02:11:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

allow mutating command in CliApp before running

Browse Source
This commit is contained in:
Aiden McClelland
2026-01-16 14:01:33 -07:00
parent 406ee9e88b
commit 39e547ff99
1 changed files with 12 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
src/cli.rs
View File

@@ -27,6 +27,7 @@ pub struct CliApp<Context: crate::Context + Clone, Config: CommandFactory + From
_phantom: PhantomData<(Context, Config)>, _phantom: PhantomData<(Context, Config)>,
make_ctx: Box<dyn FnOnce(Config) -> Result<Context, RpcError> + Send + Sync>, make_ctx: Box<dyn FnOnce(Config) -> Result<Context, RpcError> + Send + Sync>,
root_handler: ParentHandler<Context>, root_handler: ParentHandler<Context>,
mut_cmd: Option<Box<dyn FnOnce(clap::Command) -> clap::Command + Send + Sync>>,
} }
impl<Context: crate::Context + Clone, Config: CommandFactory + FromArgMatches> impl<Context: crate::Context + Clone, Config: CommandFactory + FromArgMatches>
CliApp<Context, Config> CliApp<Context, Config>
@@ -39,8 +40,16 @@ impl<Context: crate::Context + Clone, Config: CommandFactory + FromArgMatches>
_phantom: PhantomData::new(), _phantom: PhantomData::new(),
make_ctx: Box::new(make_ctx), make_ctx: Box::new(make_ctx),
root_handler, root_handler,
mut_cmd: None,
} }
} }
pub fn mutate_command(
mut self,
f: impl FnOnce(clap::Command) -> clap::Command + Send + Sync + 'static,
) -> Self {
self.mut_cmd = Some(Box::new(f));
self
}
pub fn run(self, args: impl IntoIterator<Item = OsString>) -> Result<(), RpcError> { pub fn run(self, args: impl IntoIterator<Item = OsString>) -> Result<(), RpcError> {
let mut cmd = Config::command(); let mut cmd = Config::command();
for (name, handler) in &self.root_handler.subcommands.1 { for (name, handler) in &self.root_handler.subcommands.1 {
@@ -48,6 +57,9 @@ impl<Context: crate::Context + Clone, Config: CommandFactory + FromArgMatches>
cmd = cmd.subcommand(cli.cli_command().name(name)); cmd = cmd.subcommand(cli.cli_command().name(name));
} }
} }
if let Some(f) = self.mut_cmd {
cmd = f(cmd);
}
let matches = cmd.get_matches_from(args); let matches = cmd.get_matches_from(args);
let config = Config::from_arg_matches(&matches)?; let config = Config::from_arg_matches(&matches)?;
let ctx = (self.make_ctx)(config)?; let ctx = (self.make_ctx)(config)?;