Add concise Display implementation for DirectoryTree

ProjectLighthouseCAU · Mar 18, 2024 · f1d0337 · f1d0337
1 parent 9324629
commit f1d0337
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 2 deletions.
diff --git a/lighthouse-client/examples/list_root.rs b/lighthouse-client/examples/list_root.rs
@@ -7,7 +7,7 @@ async fn run(url: &str, auth: Authentication) -> Result<()> {
     info!("Connected to the Lighthouse server");
 
     let tree = lh.list(&[], ()).await?.payload;
-    info!("Got {:#?}", tree);
+    info!("Got {}", tree);
 
     Ok(())
 }

diff --git a/lighthouse-protocol/src/payload.rs b/lighthouse-protocol/src/payload.rs
@@ -1,4 +1,4 @@
-use std::collections::HashMap;
+use std::{collections::HashMap, fmt};
 
 use serde::{Serialize, Deserialize};
 
@@ -18,3 +18,23 @@ pub enum Model {
 pub struct DirectoryTree {
     pub entries: HashMap<String, Option<DirectoryTree>>,
 }
+
+impl fmt::Display for DirectoryTree {
+    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
+        write!(f, "{{")?;
+        let count = self.entries.len();
+        for (i, (key, value)) in self.entries.iter().enumerate() {
+            write!(f, "\"{}\": ", key)?;
+            if let Some(value) = value {
+                write!(f, "{}", value)?;
+            } else {
+                write!(f, "None")?;
+            }
+            if i < count - 1 {
+                write!(f, ", ")?;
+            }
+        }
+        write!(f, "}}")?;
+        Ok(())
+    }
+}