rustdu 0.1.1

Терминальный анализатор дискового пространства (аналог ncdu на Rust)
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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88

use std::collections::HashMap;
use std::fs;
use std::path::PathBuf;
use std::sync::Mutex;
use once_cell::sync::Lazy;

// Кэш размеров папок (чтобы не пересчитывать повторно)
static SIZE_CACHE: Lazy<Mutex<HashMap<PathBuf, u64>>> = Lazy::new(|| Mutex::new(HashMap::new()));

#[derive(Debug, Clone)]
pub struct FileEntry {
    pub name: String,
    pub path: PathBuf,
    pub size: u64,
    pub is_dir: bool,
}

/// Рекурсивно вычисляет размер папки с использованием кэша
fn dir_size(path: &PathBuf) -> u64 {
    // Проверяем кэш
    if let Some(cached) = SIZE_CACHE.lock().unwrap().get(path) {
        return *cached;
    }

    let mut total = 0;
    if let Ok(entries) = fs::read_dir(path) {
        for entry in entries.flatten() {
            let meta = match entry.metadata() {
                Ok(m) => m,
                Err(_) => continue,
            };
            if meta.is_file() {
                total += meta.len();
            } else if meta.is_dir() {
                let sub_size = dir_size(&entry.path());
                total += sub_size;
            }
        }
    }
    // Сохраняем в кэш
    SIZE_CACHE.lock().unwrap().insert(path.clone(), total);
    total
}

/// Сканирует директорию, возвращает список и общий размер всех элементов в ней
pub fn scan_directory(path: &PathBuf) -> anyhow::Result<(Vec<FileEntry>, u64)> {
    let mut entries = Vec::new();
    let read_dir = fs::read_dir(path)?;
    let mut total_size = 0;

    for entry in read_dir {
        let entry = entry?;
        let metadata = entry.metadata()?;
        let name = entry.file_name().to_string_lossy().to_string();
        let path = entry.path();
        let is_dir = metadata.is_dir();
        let size = if is_dir {
            dir_size(&path)
        } else {
            metadata.len()
        };
        total_size += size;
        entries.push(FileEntry { name, path, size, is_dir });
    }

    // Сортируем по размеру (убывание) — как в ncdu
    entries.sort_by(|a, b| b.size.cmp(&a.size));
    Ok((entries, total_size))
}

/// Удаляет файл или папку (рекурсивно)
pub fn delete_entry(path: &PathBuf) -> anyhow::Result<()> {
    if path.is_dir() {
        fs::remove_dir_all(path)?;
    } else {
        fs::remove_file(path)?;
    }
    // Инвалидируем кэш для родительской папки
    if let Some(parent) = path.parent() {
        SIZE_CACHE.lock().unwrap().remove(parent);
    }
    Ok(())
}

/// Очистка кэша (например, при выходе)
pub fn clear_cache() {
    SIZE_CACHE.lock().unwrap().clear();
}