AI 代码自动评审组件开发

public class OpenAiCodeReview {
    public static void main(String[] args) throws Exception{
        System.out.println("开始CI/CD");
        // 1.检出代码diff,
        ProcessBuilder processBuilder = new ProcessBuilder("git", "diff", "HEAD~1", "HEAD");
        processBuilder.directory(new File("."));
        Process process = processBuilder.start();
        
        BufferedReader reader = new BufferedReader(new InputStreamReader((process.getInputStream())));
        String line;
        StringBuilder diffCode = new StringBuilder();
        while((line = reader.readLine()) != null) {
            diffCode.append(line).append('\n');
        }
        
        int exitCode = process.waitFor();
        System.out.println("Exited with code:" + exitCode);
        System.out.println("评审代码: " + diffCode);
    }

}

public class ChatCompletionSyncResponse {
    private List<Choice> choices;
    public static class Choice{
        private Message message;
        public Message getMessage() {
            return message;
        }
        public void setMessage(Message message) {
            this.message = message;
        }

    }

    public static class Message{
        private String role;
        private String content;

        public String getRole() {
            return role;
        }
        public String getContent() {
            return content;
        }
        public void setRole(String role) {
            this.role = role;
        }
        public void setContent(String content) {
            this.content = content;
        }

    }
    public List<Choice> getChoices() {
        return choices;
    }
    public void setChoices(List<Choice> choices) {
        this.choices = choices;
    }

}

private static String codeReview(String diffCode)throws Exception{
        String apiKey = System.getenv("GLM_KEY");
        //String apiKey = "a96e4f3b1ac040859ab86fdfc7bcdede.QpOgtBdp83eRWQMd";
        String token = BearerTokenUtils.getToken(apiKey);

        URL url = new URL("https://open.bigmodel.cn/api/paas/v4/chat/completions");
        HttpURLConnection connection = (HttpURLConnection) url.openConnection();
        connection.setRequestMethod("POST");
        connection.setRequestProperty("Authorization", "Bearer " + token);
        connection.setRequestProperty("Content-Type","application/json");
        connection.setRequestProperty("User-Agent","Mozilla/4.0 (compatible; MSIE 5.0; Windows NT; DigExt)");
        connection.setDoOutput(true);

        ChatCompletionRequest chatCompletionRequest = new ChatCompletionRequest();
        chatCompletionRequest.setModel(Model.GLM_4_5_Air.getCode());
        chatCompletionRequest.setMessages(new ArrayList<ChatCompletionRequest.Prompt>(){
            private static final long serialVersionUID = -7988151926241837899L;
            {
                add(new ChatCompletionRequest.Prompt("user","你是一个高级编程架构师，精通各类场景方案、架构设计和编程语言请，请您根据git diff记录，对代码做出评审。代码如下: " ));
                add(new ChatCompletionRequest.Prompt("user",diffCode));
            }
        });


        try(OutputStream os = connection.getOutputStream()){
            byte[] input = JSON.toJSONString(chatCompletionRequest).getBytes(StandardCharsets.UTF_8);
            os.write(input);
        }

        int responseCode = connection.getResponseCode();
        System.out.println("Response Code: " + responseCode);

        BufferedReader in = new BufferedReader(new InputStreamReader(connection.getInputStream()));
        String inputLine;

        StringBuilder context = new StringBuilder();
        while ((inputLine=in.readLine())!=null){
            context.append(inputLine);
        }
        in.close();;
        connection.disconnect();
        System.out.println("评审结果："+context.toString());

        ChatCompletionSyncResponse response = JSON.parseObject(context.toString(), ChatCompletionSyncResponse.class);
        return response.getChoices().get(0).getMessage().getContent();
    }

/**
* 日志文件存储
* @param token
* @param log
* @return
* @throws Exception
*/
private static String writeLog(String token, String log) throws Exception{
    Git git = Git.cloneRepository()
            .setURI("https://github.com/orbisz/openai-code-review-log.git")
            .setDirectory(new File("repo"))
            .setCredentialsProvider(new UsernamePasswordCredentialsProvider(token, ""))
            .call();

    String dateFolderName = new SimpleDateFormat("yyyy-MM-dd").format(new Date());
    File dateFolder = new File("repo/" + dateFolderName);
    if (!dateFolder.exists()){
        dateFolder.mkdirs();
    }
    String fileName = generateRandomString(12) + ".md";
    File newFile = new File(dateFolder, fileName);
    try(FileWriter writer = new FileWriter(newFile)){
        writer.write(log);
    }
    git.add().addFilepattern(dateFolderName+"/"+fileName).call();
    git.commit().setMessage("Add new file via GitHub Actions").call();
    git.push().setCredentialsProvider(new UsernamePasswordCredentialsProvider(token, "")).call();

    System.out.println("Changes have been pushed to the repository.");
    return "https://github.com/orbisz/openai-code-review-log/blob/master/"+dateFolderName+"/"+fileName;
}

/**
 * 生成指定长度的随机字符串
 * @param length
 * @return
 */
private static String generateRandomString(int length) {
    String characters = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789";
    StringBuilder sb = new StringBuilder(length);
    for (int i = 0; i < length; i++) {
        int index = (int) (Math.random() * characters.length());
        sb.append(characters.charAt(index));
    }
    return sb.toString();
}

String commitAuthor = System.getenv("COMMIT_AUTHOR");
        String commitBranch = System.getenv("COMMIT_BRANCH");
        String repo = System.getenv("REPOSITORY");
        int idx = repo.indexOf('/');
        String repository = (idx >= 0 && idx < repo.length() - 1)
                ? repo.substring(idx + 1)
                : repo;
        String fileName = commitAuthor + "-" + repository + "-" + commitBranch + "-" + generateRandomString(4) + ".md";

public class WXAccessTokenUtils {
    private static final String APPID = "";
    private static final String SECRET = "";
    private static final String GRANT_TYPE = "client_credential";
    private static final String URL_TEMPLATE = "https://api.weixin.qq.com/cgi-bin/token?grant_type=%s&appid=%s&secret=%s";

    public static String getAccessToken(){
        try{
            String urlString = String.format(URL_TEMPLATE, GRANT_TYPE, APPID, SECRET);
            URL url = new URL(urlString);
            HttpURLConnection connection = (HttpURLConnection) url.openConnection();
            connection.setRequestMethod("GET");

            int responseCode = connection.getResponseCode();
            System.out.println("Response Code:"+responseCode);

            if(responseCode == HttpURLConnection.HTTP_OK){
                BufferedReader in = new BufferedReader((new InputStreamReader(connection.getInputStream())));
                String inputLine;
                StringBuilder response = new StringBuilder();

                while ((inputLine = in.readLine())!=null){
                    response.append(inputLine);
                }
                in.close();

                System.out.println("Response:"+response.toString());

                Token token = JSON.parseObject(response.toString(),Token.class);

                return token.getAccess_token();
            }else{
                System.out.println("GET request failed");
                return null;
            }
        }catch (Exception e){
            e.printStackTrace();
            return null;
        }
    }
    public static class Token{
        private String access_token;
        private Integer expires_in;

        public String getAccess_token() {
            return access_token;
        }
        public void setAccess_token(String access_token) {
            this.access_token = access_token;
        }
        public Integer getExpires_in() {
            return expires_in;
        }
        public void setExpires_in(Integer expires_in) {
            this.expires_in = expires_in;
        }
    }
}

public class Message {

    private String touser = "";
    private String template_id = "";
    private String url = "";
    private Map<String, Map<String, String>> data = new HashMap<>();

    public void put(String key, String value) {
        data.put(key, new HashMap<>() {
            private static final long serialVersionUID = 7092338402387318563L;
            {
                put("value", value);
            }
        });
    }
    ... setter and getter
}

/**
 * 微信公众号消息通知
 * @param logUrl
 */
private static void pushMessage(String logUrl){
    String accessToken = WXAccessTokenUtils.getAccessToken();
    System.out.println(accessToken);

    String repo = System.getenv("REPOSITORY");
    int idx = repo.indexOf('/');
    String repository = (idx >= 0 && idx < repo.length() - 1)
            ? repo.substring(idx + 1)
            : repo;
    Message message = new Message();
    message.put("project",repository);
    message.put("author",System.getenv("COMMIT_AUTHOR"));
    message.put("branch",System.getenv("COMMIT_BRANCH"));
    message.put("review",logUrl);
    message.setUrl(logUrl);

    String url = String.format("https://api.weixin.qq.com/cgi-bin/message/template/send?access_token=%s",accessToken);

    sendPostRequest(url,JSON.toJSONString(message));
}

/**
 * 发送POST请求
 * @param urlString
 * @param jsonBody
 */
private static void sendPostRequest(String urlString, String jsonBody) {
    try {
        URL url = new URL(urlString);
        HttpURLConnection conn = (HttpURLConnection) url.openConnection();
        conn.setRequestMethod("POST");
        conn.setRequestProperty("Content-Type", "application/json; utf-8");
        conn.setRequestProperty("Accept", "application/json");
        conn.setDoOutput(true);

        try (OutputStream os = conn.getOutputStream()) {
            byte[] input = jsonBody.getBytes(StandardCharsets.UTF_8);
            os.write(input, 0, input.length);
        }

        try (Scanner scanner = new Scanner(conn.getInputStream(), StandardCharsets.UTF_8.name())) {
            String response = scanner.useDelimiter("\\A").next();
            System.out.println(response);
        }
    } catch (Exception e) {
        e.printStackTrace();
    }
}

/**
* 获取diffCode
*
* @return diffCode 代码差异
* @throws Exception
*/
public String diff() throws IOException, InterruptedException {
// 1.git log: 得到git上一次提交的hash, 使用git diff HEAD HEAD~1 HEAD的方式有时不可靠, %H指的是提交完整Hash值
ProcessBuilder logProcessBuilder = new ProcessBuilder("git", "log", "-1", "--pretty=format:%H");
logProcessBuilder.directory(new File("."));
Process logProcess = logProcessBuilder.start();

BufferedReader logReader = new BufferedReader(new InputStreamReader(logProcess.getInputStream()));
String latestCommitLogHash = logReader.readLine();  // 每次提交都会生成一个hash值, 用来标识这次提交
logReader.close();
logProcess.waitFor();

// 2.git diff: 获取代码diff
ProcessBuilder diffProcessBuilder = new ProcessBuilder("git", "diff", latestCommitLogHash + "^", latestCommitLogHash);
diffProcessBuilder.directory(new File("."));
Process diffProcess = diffProcessBuilder.start();

BufferedReader diffReader = new BufferedReader(new InputStreamReader(diffProcess.getInputStream()));
StringBuilder diffCode = new StringBuilder();
String line;
while((line = diffReader.readLine()) != null) {
diffCode.append(line).append("\n");
}
diffReader.close();

// 3.等待git diff执行完成
int waitCode = diffProcess.waitFor();
if (waitCode != 0) {
logger.info("failed to execute git diff and get code diff, waitCode:{}", waitCode);
}

// 4.返回diffCode
return diffCode.toString();
}

/**
 * 提交日志并推送到远程仓库
 *
 * @param reviewLog 评审日志
 * @return logURL 评审结果路径
 * @Expection GitAPIException
 */
public String commitAndPush(String reviewLog) throws Exception {
    // 1.克隆日志仓库
    Git git = Git.cloneRepository()
            .setURI(githubCommitLogUri)
            .setDirectory(new File("repo"))
            .setCredentialsProvider(new UsernamePasswordCredentialsProvider(githubToken, ""))
            .call();

    // 2.创建日志文件夹
    String dateFolderName = new SimpleDateFormat("yyyy-mm-dd").format(new Date());
    File dateFolder = new File("repo/" + dateFolderName);
    if (!dateFolder.exists()) { // 如果不存在则创建
        dateFolder.mkdirs();
    }

    // 3.创建日志文件&写入日志
    String fileName = project + "-" + branch + "-" + author + System.currentTimeMillis() + "-" + RandomUtil.randomString(4) + ".md";
    File file = new File(dateFolder, fileName);
    try (FileWriter writer = new FileWriter(file)) {
        writer.write(reviewLog);
    }

    // 4.将文件添加到暂存区并提交
    git.add().addFilepattern(dateFolderName + "/" + fileName).call();
    git.commit().setMessage("Add new file via Github Actions").call();
    git.push().setCredentialsProvider(new UsernamePasswordCredentialsProvider(githubToken, "")).call();

    logger.info("openai-code-review push and commit done!, fileName:{}", fileName);

    // 5.返回日志路径方便后续模板消息使用
    return githubCommitLogUri + "/blob/main/" + dateFolderName + "/" + fileName;
}

/**
 * @Author: Lz
 * Date: 2025/3/9 22:50
 * Module function: openai接口, 定义规范方法, 可以接入不同的大模型
 */
public interface IOpenAi {

    ChatCompletionSyncResponseDTO completion(ChatCompletionRequestDTO requestDTO) throws Exception;
}

/**
 * openai实例-ChatGlm
 *
 * @param requestDTO
 * @return
 * @throws Exception
 */
@Override
public ChatCompletionSyncResponseDTO completion(ChatCompletionRequestDTO requestDTO) throws Exception {
    String token = BearerTokenUtils.getToken(apiSecretKey);
    // 1.创建HTTP连接对象
    URL url = new URL(apiHost);
    HttpURLConnection connection = (HttpURLConnection) url.openConnection();
    connection.setRequestMethod("POST");
    connection.setRequestProperty("Authorization", token);
    connection.setRequestProperty("Content-type", "application/json; utf-8");
    connection.setRequestProperty("Accept", "application/json");
    connection.setRequestProperty("User-Agent", "Mozilla/4.0 (compatible; MSIE 5.0; Windows NT; DigExt)");
    connection.setDoOutput(true);

    // 2.读取响应数据
    try (OutputStream os = connection.getOutputStream()) {
        byte[] input = JSON.toJSONString(requestDTO).getBytes(StandardCharsets.UTF_8);
        os.write(input, 0, input.length);
    }

    // 3.处理响应结果
    BufferedReader reader = new BufferedReader(new InputStreamReader(connection.getInputStream()));
    StringBuilder content = new StringBuilder();
    String line;
    while ((line = reader.readLine()) != null) {
        content.append(line).append("\n");
    }
    reader.close();

    // 4.返回响应结果
    return JSON.parseObject(content.toString(), ChatCompletionSyncResponseDTO.class);
}

/**
 * 发送微信模板消息
 *
 * @param logUrl 日志评审日子, 不是发送微信模板消息的URL！！！
 * @param data 模板消息数据
 * @throws Exception
 */
public void sendTemplateMessage(String logUrl, Map<String, Map<String, String>> data) throws Exception {
    // 1.获取access_token
    String accessToken = WXAccessTokenUtils.getAccessToken(appId, appSecret);
    URL url = new URL("https://api.weixin.qq.com/cgi-bin/message/template/send?access_token=" + accessToken);
    HttpURLConnection connection = (HttpURLConnection) url.openConnection();
    connection.setRequestMethod("POST");;
    connection.setRequestProperty("Content-Type", "application/json; utf-8");
    connection.setRequestProperty("Accept", "application/json");
    connection.setDoOutput(true);

    // 2.创建模板消息对象
    TemplateMessageDTO templateMessageDTO = new TemplateMessageDTO();
    templateMessageDTO.setUrl(logUrl);
    templateMessageDTO.setData(data);

    // 3.发送模板消息, 即将消息以字节形式写入输入流, 然后发送到服务器(微信)
    // 这里使用try-with-resources语法, 确保流关闭
    try (OutputStream os = connection.getOutputStream()) {
        byte[] input = JSON.toJSONString(templateMessageDTO).getBytes(StandardCharsets.UTF_8);
        os.write(input, 0, input.length);
    }

    // 4.响应结果解析, 即通过Scanner从输入流中读取数据
    try (Scanner scanner = new Scanner(connection.getInputStream(), StandardCharsets.UTF_8.name())) {
        String response = scanner.useDelimiter("\\A").next(); // 确保读出所有响应数据
        logger.info("openai-code-review weixin TemplateMessage response:{}", response);
    }
}

/**
 * @Author: Lz
 * Date: 2025/3/9 16:36
 * Module function: 评审服务接口
 */
public interface IOpenAiCodeReviewService {

    void exec();
} 

/**
 * @Author: Lz
 * Date: 2025/3/9 16:36
 * Module function: 代码评审服务抽象类-用于定义模板, 就是流程, 具体实现由子类去做
 */
public abstract class AbstractOpenAiCodeReviewService implements IOpenAiCodeReviewService{
    private final Logger logger = (Logger)LoggerFactory.getLogger(AbstractOpenAiCodeReviewService.class);

    protected final GitCommand gitCommand;

    protected final IOpenAi openAi;

    protected final WeiXin weiXin;

    public AbstractOpenAiCodeReviewService(GitCommand gitCommand, IOpenAi openAi, WeiXin weiXin) {
        this.gitCommand = gitCommand;
        this.openAi = openAi;
        this.weiXin = weiXin;
    }

    /**
     * 定义执行流程
     */
    @Override
    public void exec() {
        try {
            // 1.获得提交代码: 返回diff_code
            String diffCode = getDiffCode();
            // 2.代码评审记录: 返回评审日志
            String log = codeReview(diffCode);
            // 3.评审记录保存: 返回评审日志URL
            String logUrl = recordCodeReview(log);
            // 4.发送消息通知: 模板消息, 评审日志
            pushMessage(logUrl);
        } catch(Exception e) {
            logger.info("openai-code-review error: " + e);
        }
    }

    protected abstract String getDiffCode() throws Exception;

    protected abstract String codeReview(String diffCode) throws Exception;

    protected abstract String recordCodeReview(String log) throws Exception;

    protected abstract void pushMessage(String logUrl) throws Exception;

public class OpenAiCodeReviewService extends AbstractOpenAiCodeReviewService {

    public OpenAiCodeReviewService(GitCommand gitCommand, IOpenAi openAi, WeiXin weiXin) {
        super(gitCommand, openAi, weiXin);
    }

    /**
     * 获取diffCode
     *
     * @return diffCode
     * @throws Exception
     */
    @Override
    protected String getDiffCode() throws Exception {
        return gitCommand.diff();
    }

    /**
     * 代码评审
     *
     * @param diffCode
     * @return 代码评审结果 log
     * @throws Exception
     */
    @Override
    protected String codeReview(String diffCode) throws Exception {
        // 构造Request请求对象
        ChatCompletionRequestDTO requestDTO = new ChatCompletionRequestDTO();
        requestDTO.setModel(Model.GLM_4_FLASH.getCode());
        requestDTO.setMessages(new ArrayList<>() {
            private static final long serialVersionUID = -7988151926241837899L;
            {
                add(new ChatCompletionRequestDTO.Prompt("user", "你是一个高级编程架构师，精通各类场景方案、架构设计和编程语言请，请您根据git diff记录，对代码做出评审。代码如下:"));
                add(new ChatCompletionRequestDTO.Prompt("user", diffCode));
            }
        });
        ChatCompletionSyncResponseDTO completion = openAi.completion(requestDTO);
        ChatCompletionSyncResponseDTO.Message message = completion.getChoices().get(0).getMessage();
        return message.getContent();
    }

    /**
     * 代码评审日志写入
     *
     * @param log
     * @return logUrl
     * @throws Exception
     */
    @Override
    protected String recordCodeReview(String log) throws Exception {
        return gitCommand.commitAndPush(log);
    }

    /**
     * 发送微信模板消息
     *
     * @param logUrl
     */
    @Override
    protected void pushMessage(String logUrl) throws Exception {
        // 1.构造请求数据Data
        HashMap<String, Map<String, String>> data = new HashMap<>();
        TemplateMessageDTO.put(data, TemplateMessageDTO.TemplateKey.PROJECT_NAME, gitCommand.getProject());
        TemplateMessageDTO.put(data, TemplateMessageDTO.TemplateKey.BRANCH_NAME, gitCommand.getBranch());
        TemplateMessageDTO.put(data, TemplateMessageDTO.TemplateKey.COMMIT_AUTHOR, gitCommand.getAuthor());
        TemplateMessageDTO.put(data, TemplateMessageDTO.TemplateKey.COMMIT_MESSAGE, gitCommand.getMessage());

        // 2.调用服务
        weiXin.sendTemplateMessage(logUrl, data);
    }


}

public class OpenAiCodeReview {

    private static Logger logger = LoggerFactory.getLogger(OpenAiCodeReview.class);

    /**
     * 程序入口
     * 
     * @param args
     */
    public static void main(String[] args) {
        // 1.GitCommand
        GitCommand gitCommand = new GitCommand(
                getEnv("GITHUB_REVIEW_LOG_URI"),
                getEnv("GITHUB_TOKEN"),
                getEnv("COMMIT_AUTHOR"),
                getEnv("COMMIT_PROJECT"),
                getEnv("COMMIT_BRANCH"),
                getEnv("COMMIT_MESSAGE")
        );

        // 2.ChatGlm
        ChatGlm chatGlm = new ChatGlm(
                getEnv("CHATGLM_API_HOST"),
                getEnv("CHATGLM_API_SECRET_KEY")
        );

        // 3.WeiXin
        WeiXin weiXin = new WeiXin(
                getEnv("WEIXIN_APPID"),
                getEnv("WEIXIN_APP_SECRET"),
                getEnv("WEIXIN_TOUSER"),
                getEnv("WEIXIN_TEMPLATE_ID")
        );

        OpenAiCodeReviewService service = new OpenAiCodeReviewService(gitCommand, chatGlm, weiXin);
        service.exec();

        logger.info("openai-code-review done!");
    }

    /**
     * 读取环境变量
     *
     * @param key
     * @return
     */
    private static String getEnv(String key) {
        String env = System.getenv(key);
        if (env == null || env.isEmpty()) {
            throw new RuntimeException("key is null");
        }
        return env;
    }

}