snippy 0.1.3

A command-line tool for that makes using LLMs for code generation a breeze
Documentation 
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

use crate::applier::utils::print_diff;
use crate::applier::Applier;
use crate::errors::ClipboardError;
use crate::extractor::ParsedBlock;
use crate::utils::{read_file_async, remove_file_async, write_file_async};
use async_trait::async_trait;
use regex::Regex;
use std::path::PathBuf;
use tracing::{debug, error, info};

#[derive(Clone)]
pub struct SearchReplaceApplier {
    base_path: PathBuf,
}

impl SearchReplaceApplier {
    pub fn new(base_path: &PathBuf) -> Self {
        SearchReplaceApplier {
            base_path: base_path.clone(),
        }
    }
}

#[async_trait]
impl Applier for SearchReplaceApplier {
    async fn apply(&self, block: &ParsedBlock) -> Result<(), ClipboardError> {
        let path = self.base_path.join(&block.filename);
        debug!("Applying search-replace to file: {}", path.display());

        let search_replace_re =
            Regex::new(r"(?s)<<<+\s*SEARCH\r?\n(.*?)===+[^\r\n]*\r?\n(.*?)>>>+\s*REPLACE")
                .expect("Invalid regex");

        let original_content = read_file_async(&path).await.unwrap_or_default();
        let mut current_content = original_content.replace("\r\n", "\n");
        let block_content = block.content.replace("\r\n", "\n");

        let mut successful_replacements = 0;
        for cap in search_replace_re.captures_iter(&block_content) {
            let search_content = cap.get(1).map_or("", |m| m.as_str());
            let replace_content = cap.get(2).map_or("", |m| m.as_str());
            debug!(
                "Search: '{}'\nReplace: '{}'",
                search_content, replace_content
            );

            if search_content.trim().is_empty() {
                debug!("Empty search content, replacing with replace content");
                current_content = replace_content.to_string();
                successful_replacements += 1;
            } else if current_content.contains(search_content) {
                debug!("Found search content in file: '{}'", search_content);
                current_content = current_content.replace(search_content, replace_content);
                successful_replacements += 1;
            } else {
                let search_content_without_newline = search_content.trim_end();
                let replace_content_without_newline = replace_content.trim_end();
                if current_content.contains(search_content_without_newline) {
                    debug!(
                        "Found search content without newline in file: '{}'",
                        search_content_without_newline
                    );
                    current_content = current_content.replace(
                        search_content_without_newline,
                        replace_content_without_newline,
                    );
                    successful_replacements += 1;
                } else {
                    error!(
                        "Failed to find content to replace in file {}: '{}'",
                        path.display(),
                        search_content
                    );
                }
            }
        }

        if successful_replacements == 0 {
            return Err(ClipboardError::ContentApplicationError(format!(
                "Failed to find any content to replace in file {}",
                path.display()
            )));
        }

        if current_content.trim().is_empty() {
            remove_file_async(&path)
                .await
                .map_err(|e| ClipboardError::IoError(e.to_string()))?;
            info!("Deleted file {}", path.display());
        } else {
            write_file_async(&path, &current_content)
                .await
                .map_err(|e| ClipboardError::IoError(e.to_string()))?;
            info!("Applied search-replace to {}", path.display());
        }

        print_diff(
            &path.display().to_string(),
            &original_content,
            &current_content,
        );
        Ok(())
    }
}