From 1c80cbf28bbcfe0b33c1abc999719dbf9b6329c1 Mon Sep 17 00:00:00 2001
From: Joe Ardent <code@ardent.nebcorp.com>
Date: Tue, 20 Mar 2018 19:23:21 -0700
Subject: [PATCH] Make printing slightly prettier

---
 src/main.rs | 24 ++++++++++++++++++++----
 1 file changed, 20 insertions(+), 4 deletions(-)

diff --git a/src/main.rs b/src/main.rs
index 5d4492a..72d3d97 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -271,8 +271,10 @@ fn display_node<S: Into<String>>(
     let mut is = indentation_str.into();
     print_this_node(node_to_print, is_first, depth, is.as_ref());
 
-    is = is.replace("└──", "   ");
-    is = is.replace("├──", "│  ");
+    is = is.replace("└─┬", "  ");
+    is = is.replace("└──", "  ");
+    is = is.replace("├──", "│ ");
+    is = is.replace("├─┬", "│ ");
 
     let printable_node_slashes = node_to_print.dir.name.matches('/').count();
 
@@ -287,15 +289,29 @@ fn display_node<S: Into<String>>(
     });
 
     let mut is_biggest = true;
+    let mut has_display_children = false;
     for node in to_display {
         if node_to_print.children.contains(node) {
+            let has_children = node.children.len() > 0;
             if node.dir.name.matches("/").count() == printable_node_slashes + 1 {
                 num_sibblings -= 1;
+                for ref n in node.children.iter() {
+                    has_display_children = has_display_children || to_display.contains(n);
+                }
+                let has_children = has_children && has_display_children;
                 let tree_chars = {
                     if num_sibblings == 0 {
-                        "└──"
+                        if has_children {
+                            "└─┬"
+                        } else {
+                            "└──"
+                        }
                     } else {
-                        "├──"
+                        if has_children {
+                            "├─┬"
+                        } else {
+                            "├──"
+                        }
                     }
                 };
                 display_node(