tree-tui 0.1.3

An interactive terminal UI for visualizing directories: code, size, and git stats.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45

//! Walk the directory off the async runtime and deliver a built skeleton [`Tree`].
//!
//! The walk is cheap (structure + sizes only, no file contents), so it runs on a
//! blocking thread and the UI shows a brief spinner. Expensive per-lens metrics
//! are computed later, lazily, when a lens is first opened (see `collect` and the
//! event loop).

use std::path::PathBuf;
use std::time::{Duration, Instant};

use tokio::sync::oneshot;

use crate::collect;
use crate::model::{Tree, build_skeleton};

/// The result of the initial directory walk.
pub struct ScanOutcome {
    pub tree: Tree,
    pub duration: Duration,
    /// Whether the root is inside a git repository (gates the git lenses).
    pub repo: bool,
    /// Short hash of the repository's current `HEAD`, if the root is a git repo.
    pub head: Option<String>,
}

/// Walk `root` on a blocking thread, returning a receiver for the skeleton.
pub fn spawn(root: PathBuf, root_label: String) -> oneshot::Receiver<ScanOutcome> {
    let (tx, rx) = oneshot::channel();
    tokio::task::spawn_blocking(move || {
        let started = Instant::now();
        let result = collect::walk(&root);
        let tree = build_skeleton(&result.files, &result.dirs, root_label);
        let repo = collect::is_repo(&root);
        let head = collect::head_short_hash(&root);

        // The receiver is dropped if the user quits mid-scan; ignore that.
        let _ = tx.send(ScanOutcome {
            tree,
            duration: started.elapsed(),
            repo,
            head,
        });
    });
    rx
}