vpn/innernet
1
0
Fork 0
mirror of https://github.com/tonarino/innernet.git synced 2024-11-28 08:58:37 +02:00
Code Issues Releases Activity

client: add list-cidrs function

Browse Source 
Closes #99
This commit is contained in:
Jake McGinty 2021-06-14 18:50:21 +09:00
parent 647ec7ca3e
commit 2fa7524def
1 changed files with 25 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
client/src/main.rs
View File

@ -171,6 +171,15 @@ enum Command {
opts: DeleteCidrOpts,
},
/// List CIDRs.
ListCidrs {
interface: Interface,
/// Display CIDRs in tree format
#[structopt(short, long)]
tree: bool,
},
/// Disable an enabled peer.
DisablePeer { interface: Interface },
@ -604,6 +613,21 @@ fn delete_cidr(interface: &InterfaceName, opts: DeleteCidrOpts) -> Result<(), Er
Ok(())
}
fn list_cidrs(interface: &InterfaceName, tree: bool) -> Result<(), Error> {
let data_store = DataStore::open(interface)?;
if tree {
let cidr_tree = CidrTree::new(data_store.cidrs());
colored::control::set_override(false);
print_tree(&cidr_tree, &[], 0);
colored::control::unset_override();
} else {
for cidr in data_store.cidrs() {
println!("{} {}", cidr.cidr, cidr.name);
}
}
Ok(())
}
fn add_peer(interface: &InterfaceName, opts: AddPeerOpts) -> Result<(), Error> {
let InterfaceConfig { server, .. } = InterfaceConfig::from_interface(interface)?;
let api = Api::new(&server);
@ -1036,6 +1060,7 @@ fn run(opt: Opts) -> Result<(), Error> {
Command::RenamePeer { interface, opts } => rename_peer(&interface, opts)?,
Command::AddCidr { interface, opts } => add_cidr(&interface, opts)?,
Command::DeleteCidr { interface, opts } => delete_cidr(&interface, opts)?,
Command::ListCidrs { interface, tree } => list_cidrs(&interface, tree)?,
Command::DisablePeer { interface } => enable_or_disable_peer(&interface, false)?,
Command::EnablePeer { interface } => enable_or_disable_peer(&interface, true)?,
Command::AddAssociation { interface, opts } => add_association(&interface, opts)?,