From 2fa7524def10da8f87f3ada2ddd35bfc011f721d Mon Sep 17 00:00:00 2001
From: Jake McGinty <me@jake.su>
Date: Mon, 14 Jun 2021 18:50:21 +0900
Subject: [PATCH] client: add list-cidrs function

Closes #99
---
 client/src/main.rs | 25 +++++++++++++++++++++++++
 1 file changed, 25 insertions(+)

diff --git a/client/src/main.rs b/client/src/main.rs
index af23bf9..410c6d4 100644
--- a/client/src/main.rs
+++ b/client/src/main.rs
@@ -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)?,