cargo-mend 0.21.0

Opinionated visibility auditing for Rust crates and workspaces
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

use std::cmp::Reverse;
use std::collections::BTreeMap;
use std::fs;
use std::path::Path;

use anyhow::Context;
use anyhow::Result;

use super::TaggedFix;
use super::ValidatedFixSet;
use crate::reporting::AppliedFixCounts;

/// Writes every fix in `fixes` and returns the edits that actually landed, per
/// notice kind. A fix whose range no longer fits its file is skipped and
/// counted nowhere — the run must not claim an edit it did not write.
pub(in crate::fixes) fn apply_fixes(fixes: &ValidatedFixSet) -> Result<AppliedFixCounts> {
    let mut by_file: BTreeMap<&Path, Vec<&TaggedFix>> = BTreeMap::new();
    for tagged in fixes.tagged() {
        by_file
            .entry(tagged.fix.path.as_path())
            .or_default()
            .push(tagged);
    }
    let mut applied = AppliedFixCounts::default();
    for (path, mut file_fixes) in by_file {
        let mut text = fs::read_to_string(path)
            .with_context(|| format!("failed to read {}", path.display()))?;
        // Apply later edits first so earlier offsets remain valid. When two
        // fixes share a start offset (a replacement at [N..M] and an insertion
        // at [N..N]), apply the wider replacement first so [N..M] still
        // targets the original bytes.
        file_fixes.sort_by_key(|tagged| Reverse((tagged.fix.start, tagged.fix.end)));
        for tagged in file_fixes {
            let fix = &tagged.fix;
            if fix.end <= text.len() && fix.start <= fix.end {
                text.replace_range(fix.start..fix.end, &fix.replacement);
                if let Some(fix_kind) = tagged.fix_kind {
                    applied.record(fix_kind);
                }
            }
        }
        fs::write(path, text).with_context(|| format!("failed to write {}", path.display()))?;
    }

    Ok(applied)
}