59 lines
1.8 KiB
Rust
59 lines
1.8 KiB
Rust
|
use std::fs;
|
||
|
|
use ratatui::buffer::Buffer;
|
||
|
|
use ratatui::layout::Rect;
|
||
|
|
use ratatui::prelude::Style;
|
||
|
|
use ratatui::widgets::{Block, Borders, Widget};
|
||
|
|
use tui_tree_widget::{Tree, TreeItem};
|
||
|
|
use uuid::Uuid;
|
||
|
|
|
||
|
|
#[derive(Clone, Copy, Debug)]
|
||
|
|
pub struct Explorer<'a> {
|
||
|
|
root_path: &'a std::path::Path,
|
||
|
|
}
|
||
|
|
|
||
|
|
impl<'a> Explorer<'a> {
|
||
|
|
pub fn new(path: &'a std::path::Path) -> Self {
|
||
|
|
Explorer { root_path: path }
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn draw(self, area: Rect, buf: &mut Buffer) {
|
||
|
|
let tree_item = Self::build_tree_from_path(self.root_path.to_path_buf());
|
||
|
|
Tree::new(&tree_item.children())
|
||
|
|
.expect("Failed to build tree.")
|
||
|
|
.style(Style::default())
|
||
|
|
.block(Block::default().borders(Borders::ALL))
|
||
|
|
.render(area, buf);
|
||
|
|
}
|
||
|
|
|
||
|
|
fn build_tree_from_path(path: std::path::PathBuf) -> TreeItem<'static, String> {
|
||
|
|
let mut children = vec![];
|
||
|
|
if let Ok(entries) = fs::read_dir(&path) {
|
||
|
|
let mut paths = entries
|
||
|
|
.map(|res| res.map(|e| e.path()))
|
||
|
|
.collect::<Result<Vec<_>, std::io::Error>>()
|
||
|
|
.expect("");
|
||
|
|
paths.sort();
|
||
|
|
for path in paths {
|
||
|
|
if path.is_dir() {
|
||
|
|
children.push(Self::build_tree_from_path(path));
|
||
|
|
} else {
|
||
|
|
children.push(TreeItem::new_leaf(
|
||
|
|
Uuid::new_v4().to_string(),
|
||
|
|
path.file_name().unwrap().to_string_lossy().to_string(),
|
||
|
|
));
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
TreeItem::new(
|
||
|
|
Uuid::new_v4().to_string(),
|
||
|
|
path.file_name()
|
||
|
|
.unwrap_or_default()
|
||
|
|
.to_string_lossy()
|
||
|
|
.to_string(),
|
||
|
|
children,
|
||
|
|
)
|
||
|
|
.expect("Failed to build tree from path.")
|
||
|
|
}
|
||
|
|
}
|