diff options
-rw-r--r-- | src/output/details.rs | 10 |
1 files changed, 9 insertions, 1 deletions
diff --git a/src/output/details.rs b/src/output/details.rs index 49c3fd1..c2bbfc0 100644 --- a/src/output/details.rs +++ b/src/output/details.rs @@ -116,6 +116,7 @@ use std::io; use std::path::PathBuf; use std::string::ToString; use std::ops::Add; +use std::iter::repeat; use colours::Colours; use column::{Alignment, Column, Cell}; @@ -683,7 +684,14 @@ impl<U> Table<U> where U: Users { // necessary to maintain information about the previously-printed // lines, as the output will change based on whether the // *previous* entry was the last in its directory. - stack.reserve(row.depth + 1); + // TODO: Replace this by Vec::resize() when it becomes stable (1.5.0) + let stack_len = stack.len(); + if row.depth + 1 > stack_len { + stack.extend(repeat(TreePart::Edge).take(row.depth + 1 - stack_len)); + } else { + stack = stack[..(row.depth + 1)].into(); + } + stack[row.depth] = if row.last { TreePart::Corner } else { TreePart::Edge }; for i in 1 .. row.depth + 1 { |