vibe-action 0.1.4

Command router — execute shell commands and LLM prompts via simple YAML actions.
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

//! YAML formatting utilities with comment injection support.
//! Provides types and functions for adding documentation comments to YAML configuration files.

/// Comment annotation for a YAML field with multiple comment lines.
pub enum YamlComment {
    /// Comment before a field: ("field_name", vec!["line1", "line2"])
    Field(&'static str, Vec<&'static str>),
}

/// Adds comments before specified top-level fields in a YAML string.
pub fn add_comments(yaml: String, comments: Vec<YamlComment>) -> String {
    let mut result = yaml.trim().to_string();
    let mut is_first = true;

    for comment in comments {
        let YamlComment::Field(field, lines) = comment;

        let comment_block = lines
            .iter()
            .map(|l| format!("# {}", l.trim()).trim().to_string())
            .collect::<Vec<_>>()
            .join("\n");

        if is_first {
            result = result.replace(
                &format!("{}:", field),
                &format!("{}\n{}:", comment_block, field),
            );
            is_first = false;
        } else {
            let pattern = format!("\n{}:", field);
            let replacement = format!("\n\n{}\n{}:", comment_block, field);
            if result.contains(&pattern) {
                result = result.replace(&pattern, &replacement);
            }
        }
    }
    result
}