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

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

pub struct DiffApplier {
    base_path: PathBuf,
}

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

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

        let original_content = read_file_async(&file_path).await.unwrap_or_default();
        let new_content = apply_diff(&file_path, &original_content, &block.content).await?;

        write_file_async(&file_path, &new_content).await?;

        info!("Applied diff to {:?}", file_path);
        print_diff(&block.filename, &original_content, &block.content);
        Ok(())
    }
}

pub async fn apply_diff(
    path: &PathBuf,
    current_content: &str,
    diff_content: &str,
) -> Result<String, ClipboardError> {
    let patch_result = Patch::from_str(diff_content);

    match patch_result {
        Ok(patch) => match diffy::apply(current_content, &patch) {
            Ok(new_content) => Ok(new_content),
            Err(e) => Err(ClipboardError::DiffError(format!(
                "Failed to apply diff for file {}: {}",
                path.display(),
                e.to_string()
            ))),
        },
        Err(e) => Err(ClipboardError::DiffError(format!(
            "Failed to parse diff for file {}: {}",
            path.display(),
            e.to_string()
        ))),
    }
}