malvin 0.2.4

Non-interactive research and coding agent
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
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112

use std::path::{Path, PathBuf};

use chrono::{DateTime, Duration, Utc};

use super::{dir_size, run_dir_timestamp};
use crate::log_gc_config::LogsGcConfig;
use crate::output::print_log_warning;

pub(crate) fn prune_run_dirs(
    run_dirs: &mut Vec<PathBuf>,
    config: &LogsGcConfig,
    protect: Option<&Path>,
) -> (usize, u64) {
    if run_dirs.is_empty() {
        return (0, 0);
    }
    let count_or_age = over_count_cap(run_dirs.len(), config.max_count)
        || over_age_limit(run_dirs.last(), config.max_age_days);
    if !count_or_age && config.max_bytes.is_none() {
        return (0, 0);
    }
    let mut sizes: Vec<u64> = run_dirs.iter().map(|p| dir_size(p)).collect();
    let mut total_bytes: u64 = sizes.iter().copied().sum();
    if !count_or_age && !over_byte_cap(total_bytes, config.max_bytes) {
        return (0, 0);
    }
    let mut removed = 0usize;
    let mut freed = 0u64;
    while needs_prune(run_dirs, total_bytes, config) {
        match remove_oldest_run(run_dirs, &mut sizes, &mut total_bytes, protect) {
            Some(size) => {
                removed += 1;
                freed = freed.saturating_add(size);
            }
            None => break,
        }
    }
    (removed, freed)
}

pub(crate) fn needs_prune(run_dirs: &[PathBuf], total_bytes: u64, config: &LogsGcConfig) -> bool {
    if run_dirs.is_empty() {
        return false;
    }
    if over_byte_cap(total_bytes, config.max_bytes) {
        return true;
    }
    if over_count_cap(run_dirs.len(), config.max_count) {
        return true;
    }
    over_age_limit(run_dirs.last(), config.max_age_days)
}

pub(crate) const fn over_byte_cap(total_bytes: u64, max_bytes: Option<u64>) -> bool {
    let Some(cap) = max_bytes else {
        return false;
    };
    total_bytes > cap
}

pub(crate) fn over_count_cap(run_count: usize, max_count: u64) -> bool {
    max_count > 0 && u64::try_from(run_count).unwrap_or(u64::MAX) > max_count
}

pub(crate) fn over_age_limit(oldest: Option<&PathBuf>, max_age_days: u64) -> bool {
    if max_age_days == 0 {
        return false;
    }
    let Some(path) = oldest else {
        return false;
    };
    let Some(name) = path.file_name().and_then(|n| n.to_str()) else {
        return false;
    };
    let ts = run_dir_timestamp(name).or_else(|| mtime_as_utc(path));
    let cutoff = Utc::now() - Duration::days(i64::try_from(max_age_days).unwrap_or(i64::MAX));
    ts.is_some_and(|t| t < cutoff)
}

pub(crate) fn mtime_as_utc(path: &Path) -> Option<DateTime<Utc>> {
    let modified = path.metadata().ok()?.modified().ok()?;
    Some(DateTime::<Utc>::from(modified))
}

fn remove_oldest_run(
    run_dirs: &mut Vec<PathBuf>,
    sizes: &mut Vec<u64>,
    total_bytes: &mut u64,
    protect: Option<&Path>,
) -> Option<u64> {
    let mut idx = run_dirs.len();
    while idx > 0 {
        idx -= 1;
        let path = run_dirs[idx].clone();
        if protect.is_some_and(|p| p == path.as_path()) {
            continue;
        }
        let size = sizes[idx];
        match std::fs::remove_dir_all(&path) {
            Ok(()) => {
                run_dirs.remove(idx);
                sizes.remove(idx);
                *total_bytes = total_bytes.saturating_sub(size);
                return Some(size);
            }
            Err(e) => {
                print_log_warning(&format!("could not prune log run {}: {e}", path.display()));
            }
        }
    }
    None
}