git_commit_helper_cli/

review.rs

use anyhow::Result;
use std::process::Command;
use crate::config::Config;
use crate::ai_service;
use crate::github;
use crate::gerrit;
use log::{debug, info};
use crate::terminal_format::Style;

pub async fn review_remote_changes(config: &Config, url: &str) -> Result<String> {
    debug!("开始审查远程代码改动: {}", url);

    // 获取改动信息和 diff 内容
    let (change_message, diff) = if url.contains("github.com") {
        if url.contains("/pull/") {
            let pr_info = github::get_pr_info(url).await?;
            let diff = github::get_pr_diff(url).await?;
            (pr_info, diff)
        } else if url.contains("/commit/") {
            let msg = github::get_commit_info(url).await?;
            let diff = github::get_commit_diff(url).await?;
            (msg, diff)
        } else {
            return Err(anyhow::anyhow!("无效的GitHub URL，必须是PR或commit链接"));
        }
    } else if url.contains("/+/") {
        // Gerrit 链接
        let msg = gerrit::get_change_info(url).await?;
        let diff = gerrit::get_change_diff(url).await?;
        (msg, diff)
    } else {
        return Err(anyhow::anyhow!("无效的URL，必须是GitHub或Gerrit链接"));
    };

    if diff.trim().is_empty() {
        return Err(anyhow::anyhow!("未发现任何代码改动"));
    }

    // 翻译改动信息（如果存在且包含英文）
    let mut review = String::new();
    if !change_message.is_empty() {
        // 分离标题和描述
        let mut parts = change_message.splitn(2, "\n描述：\n");
        let title = parts.next().unwrap().trim_start_matches("标题：").trim();
        let description = parts.next();

        let mut info = String::new();

        // 处理标题
        let title_info = if title.chars().any(|c| c.is_ascii_alphabetic()) {
            // 如果标题包含英文字符
            let translator = ai_service::create_translator(config).await?;
            let prompt = format!("请将以下 PR 标题翻译成中文：\n\n{}", title);
            let chinese = translator.chat("你是一个代码提交信息翻译助手。", &prompt).await?;
            format!("标题：{}\n中文翻译：{}\n", title, chinese)
        } else {
            format!("标题：{}\n", title)
        };
        info.push_str(&title_info);

        // 处理描述（如果存在）
        if let Some(desc) = description {
            if !desc.trim().is_empty() {
                if desc.chars().any(|c| c.is_ascii_alphabetic()) {
                    // 如果描述包含英文字符
                    let translator = ai_service::create_translator(config).await?;
                    let prompt = format!("请将以下 PR 描述翻译成中文：\n\n{}", desc);
                    let chinese = translator.chat("你是一个代码提交信息翻译助手。", &prompt).await?;
                    info.push_str(&format!("\n描述：\n{}\n中文翻译：\n{}\n", desc, chinese));
                } else {
                    info.push_str(&format!("\n描述：\n{}\n", desc));
                }
            }
        }

        review.push_str(&info);
        review.push('\n');
    }

    // 代码审查
    let translator = ai_service::create_translator(config).await?;
    info!("正在使用 {:?} 服务进行代码审查...", config.default_service);

    let system_prompt = get_review_prompt();
    let review_result = translator.chat(&system_prompt, &diff).await?;
    review.push_str(&review_result);

    // 终端格式化输出
    let review = format_review_for_terminal(&review);
    Ok(review)
}

// 终端格式化review内容
fn format_review_for_terminal(input: &str) -> String {
    let mut out = String::new();
    let mut in_report = false;
    for line in input.lines() {
        if line.trim().is_empty() {
            out.push('\n');
            continue;
        }
        if line.starts_with("标题：") {
            out.push_str(&Style::separator());
            out.push_str(&Style::title(line));
        } else if line.starts_with("中文翻译：") {
            out.push_str(&Style::green(line));
        } else if line.starts_with("描述：") {
            out.push_str(&Style::blue(line));
        } else if line.starts_with("代码审查报告：") {
            out.push_str(&Style::separator());
            out.push_str(&Style::yellow(line));
            in_report = true;
        } else if line.starts_with("警告") {
            out.push_str(&Style::yellow(line));
        } else if line.starts_with("错误") {
            out.push_str(&Style::red(line));
        } else if in_report {
            out.push_str(&Style::plain(line));
        } else {
            out.push_str(&Style::plain(line));
        }
    }
    out.push_str(&Style::separator());
    out
}

pub async fn review_changes(config: &Config, no_review: bool) -> Result<Option<String>> {
    // 如果命令行指定了 --no-review 或配置文件中禁用了 ai_review，则跳过审查
    if no_review {
        info!("已通过 --no-review 参数禁用代码审查");
        return Ok(None);
    }

    if !config.ai_review {
        info!("AI代码审查功能已在配置中禁用，可以使用 git-commit-helper ai-review --enable 启用");
        return Ok(None);
    }

    // 获取当前改动的差异
    let diff = get_staged_changes()?;
    if diff.trim().is_empty() {
        info!("没有检测到暂存的代码改动");
        return Ok(None);
    }

    // 使用配置的 AI 服务进行代码审查
    let translator = ai_service::create_translator(config).await?;
    info!("正在使用 {:?} 服务进行代码审查...", config.default_service);

    let system_prompt = get_review_prompt();
    let review = translator.chat(&system_prompt, &diff).await?;

    Ok(Some(review))
}

// 构建代码审查提示语
fn get_review_prompt() -> String {
    // 获取配置文件路径
    let prompt_path = crate::config::Config::config_path()
        .expect("无法获取配置目录")
        .parent()
        .expect("无法获取父目录")
        .join("review_prompt.txt");

    // 如果文件存在则读取，否则使用默认提示语
    if prompt_path.exists() {
        info!("正在使用 {:?} 提示词文件, 进行代码审查...", prompt_path.display());
        std::fs::read_to_string(&prompt_path).unwrap_or_else(|err| {
            log::error!("Failed to read review prompt file {:?}: {}", prompt_path.display(), err);
            DEFAULT_REVIEW_PROMPT.to_string()
        })
    } else {
        DEFAULT_REVIEW_PROMPT.to_string()
    }
}

const DEFAULT_REVIEW_PROMPT:&str = r#"您是一位专业的代码审查者，请对以下代码变更进行审查并给出中文评价。请着重关注：

1. 代码质量：
   - 代码是否清晰易懂
   - 变量和函数命名是否恰当
   - 代码结构是否合理

2. 潜在问题：
   - 可能的bug
   - 边界条件处理
   - 异常情况的处理
   - 资源使用和释放

3. 最佳实践：
   - 是否符合编程规范
   - 是否遵循设计模式
   - 代码重用性
   - 模块化和解耦

4. 性能考虑：
   - 算法效率
   - 资源使用效率
   - 可能的性能瓶颈

5. 安全性：
   - 输入验证
   - 数据安全
   - 权限检查

请以"代码审查报告："开头，使用简洁的语言描述发现的问题和改进建议。如果代码符合最佳实践，也请给出正面的评价。
"#;

fn get_staged_changes() -> Result<String> {
    let output = Command::new("git")
        .args(&["diff", "--cached"])
        .output()?;

    if !output.status.success() {
        return Err(anyhow::anyhow!("获取暂存区改动失败"));
    }

    Ok(String::from_utf8(output.stdout)?)
}

pub fn should_skip_review(message: &str) -> bool {
    message.starts_with("Merge") ||
    message.starts_with("Cherry-pick") ||
    message.starts_with("Revert")
}

pub async fn review_local_commit(config: &Config, commit_id: &str) -> Result<String> {
    debug!("开始审查本地commit: {}", commit_id);

    // 获取diff内容
    let diff = get_commit_diff(commit_id)?;

    if diff.trim().is_empty() {
        return Err(anyhow::anyhow!("未发现任何代码改动"));
    }

    // 代码审查
    info!("正在使用 {:?} 服务进行代码审查...", config.default_service);
    let mut review = String::new();
    let translator = ai_service::create_translator(config).await?;
    let system_prompt = get_review_prompt();
    let review_result = translator.chat(&system_prompt, &diff).await?;
    review.push_str(&review_result);

    Ok(review)
}

fn get_commit_diff(commit_id: &str) -> Result<String> {
    let output = Command::new("git")
        .args(&["show", "--pretty=format:", commit_id])
        .output()?;

    if !output.status.success() {
        return Err(anyhow::anyhow!("获取commit差异失败"));
    }

    Ok(String::from_utf8(output.stdout)?)
}