211c61d cli: Add module for common arguments — heartwood

Radicle Heartwood Protocol & Stack

cli: Add module for common arguments

Erik Kundt committed 6 months ago

commit 211c61d0253186f57eea96d59d632d0e51080758
parent efe10f95be9fde1cda84f0631ec8f6546ac7228d

4 files changed +71 -41

added crates/radicle-cli/src/args.rs

@@ -0,0 +1,65 @@

 use radicle::prelude::{NodeId, RepoId};
 use thiserror::Error;
 #[derive(Clone, Debug)]
 pub(super) enum BlockTarget {
     Node(NodeId),
     Repo(RepoId),
 }
 #[derive(Debug, Error)]
 #[error("invalid repository or node specified (RID parsing failed with: '{repo}', NID parsing failed with: '{node}'))")]
 pub(super) struct ParseTargetError {
     repo: radicle::identity::IdError,
     node: radicle::crypto::PublicKeyError,
 }
 impl std::str::FromStr for BlockTarget {
     type Err = ParseTargetError;
     fn from_str(val: &str) -> Result<Self, Self::Err> {
         val.parse::<RepoId>()
             .map(BlockTarget::Repo)
             .or_else(|repo| {
                 val.parse::<NodeId>()
                     .map(BlockTarget::Node)
                     .map_err(|node| ParseTargetError { repo, node })
             })
     }
 }
 impl std::fmt::Display for BlockTarget {
     fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
         match self {
             Self::Node(nid) => nid.fmt(f),
             Self::Repo(rid) => rid.fmt(f),
         }
     }
 }
 #[cfg(test)]
 mod test {
     use crate::args::ParseTargetError;
     use std::str::FromStr;
     use super::BlockTarget;
     #[test]
     fn should_parse_nid() {
         let target = BlockTarget::from_str("z6MkiswaKJ85vafhffCGBu2gdBsYoDAyHVBWRxL3j297fwS9");
         assert!(target.is_ok())
     }
     #[test]
     fn should_parse_rid() {
         let target = BlockTarget::from_str("rad:z3Tr6bC7ctEg2EHmLvknUr29mEDLH");
         assert!(target.is_ok())
     }
     #[test]
     fn should_not_parse() {
         let err = BlockTarget::from_str("bee").unwrap_err();
         assert!(matches!(err, ParseTargetError { .. }));
     }
 }

modified crates/radicle-cli/src/commands/block.rs

@@ -2,10 +2,9 @@ mod args;

 use radicle::node::policy::Policy;
 use crate::args::BlockTarget;
 use crate::terminal as term;
 use args::Target;
 pub use args::Args;
 pub(crate) use args::ABOUT;

@@ -14,8 +13,8 @@ pub fn run(args: Args, ctx: impl term::Context) -> anyhow::Result<()> {

     let mut policies = profile.policies_mut()?;
     let updated = match args.target {
         Target::Node(nid) => policies.set_follow_policy(&nid, Policy::Block)?,
         Target::Repo(rid) => policies.set_seed_policy(&rid, Policy::Block)?,
         BlockTarget::Node(nid) => policies.set_follow_policy(&nid, Policy::Block)?,
         BlockTarget::Repo(rid) => policies.set_seed_policy(&rid, Policy::Block)?,
     };
     if updated {
         term::success!("Policy for {} set to 'block'", args.target);

modified crates/radicle-cli/src/commands/block/args.rs

@@ -1,44 +1,9 @@

 use clap::Parser;
 use thiserror::Error;
 use radicle::prelude::{NodeId, RepoId};
 use crate::args::BlockTarget;
 pub(crate) const ABOUT: &str = "Block repositories or nodes from being seeded or followed";
 #[derive(Clone, Debug)]
 pub(super) enum Target {
     Node(NodeId),
     Repo(RepoId),
 }
 #[derive(Debug, Error)]
 #[error("invalid repository or node specified (RID parsing failed with: '{repo}', NID parsing failed with: '{node}'))")]
 pub(super) struct ParseTargetError {
     repo: radicle::identity::IdError,
     node: radicle::crypto::PublicKeyError,
 }
 impl std::str::FromStr for Target {
     type Err = ParseTargetError;
     fn from_str(val: &str) -> Result<Self, Self::Err> {
         val.parse::<RepoId>().map(Target::Repo).or_else(|repo| {
             val.parse::<NodeId>()
                 .map(Target::Node)
                 .map_err(|node| ParseTargetError { repo, node })
         })
     }
 }
 impl std::fmt::Display for Target {
     fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
         match self {
             Self::Node(nid) => nid.fmt(f),
             Self::Repo(rid) => rid.fmt(f),
         }
     }
 }
 #[derive(Parser, Debug)]
 #[command(about = ABOUT, disable_version_flag = true)]
 pub struct Args {

@@ -46,7 +11,7 @@ pub struct Args {

     ///
     /// [example values: rad:z3Tr6bC7ctEg2EHmLvknUr29mEDLH, z6MkiswaKJ85vafhffCGBu2gdBsYoDAyHVBWRxL3j297fwS9]
     #[arg(value_name = "RID|NID")]
     pub(super) target: Target,
     pub(super) target: BlockTarget,
 }
 #[cfg(test)]

modified crates/radicle-cli/src/lib.rs

@@ -1,6 +1,7 @@

 #![allow(clippy::collapsible_if)]
 #![allow(clippy::or_fun_call)]
 #![allow(clippy::too_many_arguments)]
 pub mod args;
 pub mod commands;
 pub mod git;
 pub mod node;