rusdu 0.1.2

Rust rewrite of ncdu — NCurses Disk Usage analyzer
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
46
47
48
49
50
51
52
53
54
55
56
57
58

use anyhow::{anyhow, Result};
use std::path::Path;
use std::process::Command;

pub fn spawn_shell(dir_path: &Path, read_only: bool) -> Result<()> {
    if read_only {
        return Err(anyhow!("Cannot spawn shell in read-only mode"));
    }

    // Determine shell executable
    let shell_exe = if cfg!(windows) {
        std::env::var("COMSPEC").unwrap_or_else(|_| "powershell.exe".to_string())
    } else {
        std::env::var("NCDU_SHELL")
            .or_else(|_| std::env::var("SHELL"))
            .unwrap_or_else(|_| "/bin/sh".to_string())
    };

    // Increment NCDU_LEVEL env variable
    let ncdu_level = std::env::var("NCDU_LEVEL")
        .ok()
        .and_then(|val| val.parse::<i32>().ok())
        .unwrap_or(0)
        + 1;

    // Suspend crossterm TUI raw mode before launching shell
    crossterm::terminal::disable_raw_mode()?;
    crossterm::execute!(
        std::io::stdout(),
        crossterm::terminal::LeaveAlternateScreen,
        crossterm::cursor::Show
    )?;

    let mut cmd = Command::new(&shell_exe);
    cmd.current_dir(dir_path)
        .env("NCDU_LEVEL", ncdu_level.to_string());

    let status = cmd.status();

    // Re-enable TUI raw mode
    crossterm::terminal::enable_raw_mode()?;
    crossterm::execute!(
        std::io::stdout(),
        crossterm::terminal::EnterAlternateScreen,
        crossterm::cursor::Hide
    )?;

    match status {
        Ok(s) => {
            if s.success() {
                Ok(())
            } else {
                Err(anyhow!("Shell exited with error status"))
            }
        }
        Err(e) => Err(anyhow!("Failed to spawn shell {:?}: {}", shell_exe, e)),
    }
}