git-revise 0.1.8

// use std::cell::RefCell;

// pub trait GitDiff {
//     fn git_diff(
//         repo: &git2::Repository,
//         exclude_files: &[String],
//     ) -> crate::error::ReviseResult<String> {
//         let mut opts = git2::DiffOptions::new();

//         // 设置 diff 选项以只包含暂存区的更改
//         opts.include_untracked(false)
//             .show_untracked_content(false)
//             .include_ignored(false);

//         // 获取 HEAD 指向的 tree
//         let head = repo.head()?.peel_to_tree()?;

//         // 获取暂存区的 index
//         let index = repo.index()?;

//         // 比较 HEAD 和暂存区的 index
//         let diff = repo.diff_tree_to_index(
//             Some(&head),
//             Some(&index),
//             Some(&mut opts),
//         )?;

//         let content = RefCell::new(String::new());
//         let current_file_excluded = RefCell::new(false);

//         diff.foreach(
//             &mut |delta, _| {
//                 if let Some(path) = delta.new_file().path() {
//                     if let Some(file_name) = path.file_name() {
//                         if let Some(file_name_str) = file_name.to_str() {
//                             *current_file_excluded.borrow_mut() =
// exclude_files
// .contains(&file_name_str.to_string());                         }
//                     }
//                 }
//                 true // 总是返回 true 以继续处理下一个文件
//             },
//             None,
//             None,
//             Some(&mut |_, _, line| {
//                 if !*current_file_excluded.borrow() {
//                     let prefix = match line.origin() {
//                         '+' => "+",
//                         '-' => "-",
//                         _ => " ",
//                     };
//                     content.borrow_mut().push_str(prefix);
//                     content
//                         .borrow_mut()
//                         .push_str(&String::from_utf8_lossy(line.content()));
//                 }
//                 true // 总是返回 true 以继续处理下一行
//             }),
//         )?;

//         Ok(content.into_inner())
//     }
// }

// // pub trait GitDiff {
// //     fn git_diff(repo: &git2::Repository) ->
// // crate::error::ReviseResult<String> {         let mut opts =
// // git2::DiffOptions::new(); // 创建一个新的DiffOptions实例

// //         let head = repo.head()?.peel_to_tree()?; // 获取HEAD指向的tree
// //         let diff = repo.diff_tree_to_workdir(Some(&head), Some(&mut
// opts))?; // // 获取当前工作目录与HEAD的差异

// //         let mut content = String::new(); // 预分配内存

// //         diff.foreach(
// //             &mut |_, _| true,
// //             None,
// //             None,
// //             Some(&mut |_, _, line| {
// //                 let prefix = match line.origin() {
// //                     '+' => "+",
// //                     '-' => "-",
// //                     _ => " ",
// //                 };
// //                 content.push_str(prefix);
// //
// content.push_str(&String::from_utf8_lossy(line.content())); //
// content.push('\n'); //                 true
// //             }),
// //         )?;
// //         Ok(content)
// //     }
// // }

// #[cfg(test)]
// mod tests {
//     use super::*;

//     #[test]
//     fn test_git_diff() {
//         struct GItDiffImpl;
//         impl GitDiff for GItDiffImpl {}

//         let _ =
//             GItDiffImpl::git_diff(&git2::Repository::open(".").unwrap(), &[
//                 "README.md".to_string(),
//             ])
//             .unwrap();
//     }
// }

use std::process::Command;

use crate::error::ReviseResult;

pub trait GitDiff {
    fn git_diff(exclude_files: &[String]) -> ReviseResult<String> {
        let mut args = vec!["diff", "--staged"];

        let exclude_args: Vec<String> = exclude_files
            .iter()
            .map(|file| format!(":!{file}"))
            .collect();
        args.extend(exclude_args.iter().map(std::string::String::as_str));

        let mut binding = Command::new("git");
        let command = binding.args(&args);
        let output = command.output()?;

        if !output.status.success() {
            return Err(anyhow::anyhow!(
                "Git diff failed: {}",
                String::from_utf8_lossy(&output.stderr)
            ));
        }
        let diff = String::from_utf8(output.stdout)?;
        Ok(diff)
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    #[ignore]
    fn test_git_diff() {
        struct GitDiffImpl;
        impl GitDiff for GitDiffImpl {}

        let diff =
            GitDiffImpl::git_diff(&["CHANGELOG.md".to_string()]).unwrap();
        println!("Git diff: {diff}");
    }
}