✨Feat: 支持 AI 总结代码修改内容

This commit is contained in:
2025-10-13 19:20:48 +08:00
parent c4acef5887
commit 023f0b9590
2 changed files with 38 additions and 2 deletions
@@ -14,4 +14,6 @@ public class AIConstant {
"[Modified Files]: ...\n" +
"【输出格式】" +
"[总结]: ...";
public static final String DEFAULT_ERROR_RESPONSE = "*变化文件数量过多,请在仓库中查看";
}
@@ -2,7 +2,9 @@ package cn.hezhaohui.webhook.service;
import cn.hezhaohui.lark.entity.CardData;
import cn.hezhaohui.lark.util.Lark;
import cn.hezhaohui.webhook.constant.AIConstant;
import cn.hezhaohui.webhook.entity.GiteaPushEvent;
import jakarta.annotation.Resource;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;
@@ -13,6 +15,12 @@ import java.util.stream.Stream;
@Slf4j
public class HookService {
@Resource
private AIService aiService;
@Resource
private FileService fileService;
public void handlePush(GiteaPushEvent event) {
List<String> commitMsgList = event.getCommits().stream()
.map(GiteaPushEvent.Commit::getMessage)
@@ -20,22 +28,48 @@ public class HookService {
List<String> addedFiles = event.getCommits().stream()
.flatMap(commit -> commit.getAdded().stream())
.distinct()
.toList();
List<String> removedFiles = event.getCommits().stream()
.flatMap(commit -> commit.getRemoved().stream())
.distinct()
.toList();
List<String> modifiedFiles = event.getCommits().stream()
.flatMap(commit -> commit.getModified().stream())
.distinct()
.toList();
if (addedFiles.isEmpty() && removedFiles.isEmpty() && modifiedFiles.isEmpty()) {
return;
}
String ai = AIConstant.DEFAULT_ERROR_RESPONSE;
if (addedFiles.size() + removedFiles.size() + modifiedFiles.size() <= 20) {
String addedFilesContent = fileService.getAllFiles(event.getRepository().getHtml_url(), addedFiles);
String removedFilesContent = fileService.getAllFiles(event.getRepository().getHtml_url(), removedFiles);
String modifiedFilesContent = fileService.getAllFiles(event.getRepository().getHtml_url(), modifiedFiles);
String allFilesContentString = "[Added Files]: \n" +
addedFilesContent +
"\n[Removed Files]: \n" +
removedFilesContent +
"\n[Modified Files]: \n" +
modifiedFilesContent;
String input = "[Commit Messages]: \n" +
String.join("\n", commitMsgList) +
"\n" + allFilesContentString;
ai = aiService.getAIResponse(input);
}
log.info("commitMsgList: {}\n", commitMsgList);
log.info("addedFiles: {}\n", addedFiles);
log.info("removedFiles: {}\n", removedFiles);
log.info("modifiedFiles: {}\n", modifiedFiles);
// TODO: 调用 AI 模块
CardData cardData = CardData.builder()
.title("[Gitea] 仓库状态变化")
@@ -44,7 +78,7 @@ public class HookService {
.msg(String.join("\n", commitMsgList))
.tag("PUSH")
.subtitle("Ciallo~(∠・ω< )⌒★")
.ai("待开发")
.ai(ai)
.build();
try {