Assure root size is properly computed

1 files changed, 7 insertions, 0 deletions

diff --git a/src/common.rs b/src/common.rs
index 86e5ac4..11f7d7f 100644
--- a/src/common.rs
+++ b/src/common.rs
@@ -1,7 +1,14 @@
+use crate::interactive::{Tree, TreeIndex};
 use jwalk::WalkDir;
 use std::fmt;
 use std::path::Path;
 
+pub(crate) fn get_size_or_panic(tree: &Tree, node_idx: TreeIndex) -> u64 {
+    tree.node_weight(node_idx)
+        .expect("node should always be retrievable with valid index")
+        .size
+}
+
 /// Specifies a way to format bytes
 #[derive(Clone, Copy)]
 pub enum ByteFormat {