move EntryMarkMap into Mark widget

2 files changed, 11 insertions, 4 deletions

diff --git a/src/interactive/widgets/entries.rs b/src/interactive/widgets/entries.rs
index bef91cd..db510de 100644
--- a/src/interactive/widgets/entries.rs
+++ b/src/interactive/widgets/entries.rs
@@ -1,6 +1,6 @@
 use crate::interactive::{
-    widgets::{COLOR_MARKED, COLOR_MARKED_DARK, COLOR_MARKED_DARKER},
-    DisplayOptions, EntryDataBundle, EntryMarkMap,
+    widgets::{EntryMarkMap, COLOR_MARKED, COLOR_MARKED_DARK, COLOR_MARKED_DARKER},
+    DisplayOptions, EntryDataBundle,
 };
 use dua::traverse::{Tree, TreeIndex};
 use itertools::Itertools;
diff --git a/src/interactive/widgets/mark.rs b/src/interactive/widgets/mark.rs
index 9f94592..be18d0e 100644
--- a/src/interactive/widgets/mark.rs
+++ b/src/interactive/widgets/mark.rs
@@ -1,8 +1,8 @@
-use crate::interactive::{widgets::COLOR_MARKED_LIGHT, CursorDirection, EntryMark, EntryMarkMap};
+use crate::interactive::{widgets::COLOR_MARKED_LIGHT, CursorDirection};
 use dua::traverse::{Tree, TreeIndex};
 use dua::{path_of, ByteFormat};
 use itertools::Itertools;
-use std::borrow::Borrow;
+use std::{borrow::Borrow, collections::BTreeMap, path::PathBuf};
 use termion::{event::Key, event::Key::*};
 use tui::{
     buffer::Buffer,
@@ -14,6 +14,13 @@ use tui::{
 };
 use tui_react::{List, ListProps};
 
+pub type EntryMarkMap = BTreeMap<TreeIndex, EntryMark>;
+pub struct EntryMark {
+    pub size: u64,
+    pub path: PathBuf,
+    pub index: usize,
+}
+
 #[derive(Default)]
 pub struct MarkPane {
     selected: Option<usize>,