injm 0.5.0

A CLI tool that injects content into marked regions in source files.
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

use std::fs;
use std::io::{self, Read};

use crate::core::checker::{check_duplicated_ids, check_missing_ids};
use crate::core::inject::inject;
use crate::core::parser::parse_patterns;
use crate::core::types::{MarkerBlock, OutputID, Result};

pub fn run(
    input: Vec<String>,
    output: Vec<String>,
    dry_run: bool,
    ids: Vec<OutputID>,
) -> Result<()> {
    let output_files = parse_patterns(&output)?;

    let input_blocks: Vec<MarkerBlock> = if input.is_empty() {
        stdin_blocks(ids)?
    } else {
        let input_files = parse_patterns(&input)?;
        check_missing_ids(&output_files, &input_files)?;
        input_files
            .into_iter()
            .flat_map(|file| file.blocks)
            .collect()
    };

    check_duplicated_ids(&input_blocks)?;

    for output_file in output_files {
        let replaced = inject(&output_file.content, &output_file.blocks, &input_blocks)?;
        if dry_run {
            println!("{replaced}");
        } else {
            fs::write(output_file.path, replaced)?;
        }
    }

    Ok(())
}

fn read_stdin() -> Result<String> {
    let mut input = String::new();
    io::stdin().read_to_string(&mut input)?;
    Ok(input)
}
fn stdin_blocks(ids: Vec<OutputID>) -> Result<Vec<MarkerBlock>> {
    let stdin = read_stdin()?;
    let input_ids = ids.into_iter().flatten().collect();
    Ok(vec![MarkerBlock {
        input_content: Some(stdin),
        begin_line: 0,
        end_line: 0,
        input_ids,
        output_id: None,
    }])
}