main-spex 0.26.0

AI-first development framework that turns specifications into verified code
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
113
114
115
116
117
118

use std::collections::HashMap;

use super::spex_config::PreexistingError;

#[cfg(test)]
#[path = "spex_yaml_edit_test.rs"]
mod spex_yaml_edit_test;

pub fn set_preexisting_errors(
    raw: &str,
    by_package: &HashMap<String, Vec<PreexistingError>>,
) -> String {
    let lines: Vec<&str> = raw.lines().collect();
    let mut out = String::with_capacity(raw.len());
    let mut pending: Option<String> = None;
    let mut in_scope = false;

    let mut i = 0;
    while i < lines.len() {
        let line = lines[i];
        let indent = indent_of(line);
        let trimmed = line.trim_start();

        let package_start = indent == 2 && trimmed.starts_with("- package_name:");
        let closes_previous_section = package_start || (indent == 0 && !trimmed.is_empty());

        if closes_previous_section {
            if let Some(block) = pending.take() {
                out.push_str(&block);
            }
            in_scope = false;
        }

        if package_start {
            let name = trimmed["- package_name:".len()..].trim();
            if let Some(entries) = by_package.get(name) {
                in_scope = true;
                if !entries.is_empty() {
                    pending = Some(render_check_ignore(entries, 4));
                }
            }
            out.push_str(line);
            out.push('\n');
            i += 1;
            continue;
        }

        if in_scope && indent == 4 && trimmed.starts_with("check_ignore:") {
            i += 1;
            while i < lines.len() && indent_of(lines[i]) > 4 {
                i += 1;
            }
            continue;
        }

        out.push_str(line);
        out.push('\n');
        i += 1;
    }

    if let Some(block) = pending.take() {
        out.push_str(&block);
    }
    out
}

pub(super) fn render_check_ignore(entries: &[PreexistingError], indent: usize) -> String {
    let pad = " ".repeat(indent);
    let mut block = format!("{pad}check_ignore:\n");
    for entry in entries {
        block.push_str(&format!("{pad}  - rule_id: {}\n", entry.rule_id));
        block.push_str(&format!("{pad}    check: {}\n", entry.check));
        if !entry.files.is_empty() {
            block.push_str(&format!("{pad}    files:\n"));
            for file in &entry.files {
                block.push_str(&format!("{pad}      - {file}\n"));
            }
        }
    }
    block
}

pub fn set_cli_version(raw: &str, version: &str) -> String {
    let new_line = format!("cli_version: {version}");

    if raw.lines().any(|line| line.starts_with("cli_version:")) {
        let mut out = String::with_capacity(raw.len());
        for line in raw.lines() {
            if line.starts_with("cli_version:") {
                out.push_str(&new_line);
            } else {
                out.push_str(line);
            }
            out.push('\n');
        }
        return out;
    }

    let mut out = String::with_capacity(raw.len() + new_line.len() + 1);
    let mut inserted = false;
    for line in raw.lines() {
        out.push_str(line);
        out.push('\n');
        if !inserted && line.starts_with("project_name:") {
            out.push_str(&new_line);
            out.push('\n');
            inserted = true;
        }
    }
    if !inserted {
        out.insert_str(0, &format!("{new_line}\n"));
    }
    out
}

pub(super) fn indent_of(line: &str) -> usize {
    line.chars().take_while(|ch| *ch == ' ').count()
}