feat: Phase 4 — LLM 集成,PR 描述生成与 AI 代码审查

This commit is contained in:
2026-06-19 19:43:18 +08:00
parent 7ac3a3726b
commit afb270721c
8 changed files with 783 additions and 29 deletions
+61 -2
View File
@@ -2,8 +2,11 @@ package handlers
import (
"database/sql"
"encoding/json"
"fmt"
"net/http"
"github.com/HoHD/PR-Helper/services"
"github.com/gin-gonic/gin"
)
@@ -15,7 +18,63 @@ func NewGenerateHandler(db *sql.DB) *GenerateHandler {
return &GenerateHandler{db: db}
}
// Generate handles POST /api/repos/:id/generate — stub for Phase 4
// Generate handles POST /api/repos/:id/generate — SSE streaming PR description generation.
func (h *GenerateHandler) Generate(c *gin.Context) {
c.JSON(http.StatusNotImplemented, gin.H{"error": "generate not implemented yet — coming in Phase 4"})
id := c.Param("id")
// Get repo info
var localPath string
err := h.db.QueryRow(`SELECT local_path FROM repositories WHERE id = ?`, id).Scan(&localPath)
if err == sql.ErrNoRows {
c.JSON(http.StatusNotFound, gin.H{"error": "repository not found"})
return
}
if err != nil {
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
return
}
// Parse request
var req struct {
Base string `json:"base" binding:"required"`
Head string `json:"head" binding:"required"`
}
if err := c.ShouldBindJSON(&req); err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": "base and head are required"})
return
}
// Set SSE headers
c.Header("Content-Type", "text/event-stream")
c.Header("Cache-Control", "no-cache")
c.Header("Connection", "keep-alive")
c.Header("X-Accel-Buffering", "no")
c.Status(http.StatusOK)
flusher, ok := c.Writer.(http.Flusher)
if !ok {
c.JSON(http.StatusInternalServerError, gin.H{"error": "streaming not supported"})
return
}
sendEvent := func(event string, data interface{}) {
jsonData, _ := json.Marshal(data)
fmt.Fprintf(c.Writer, "event: %s\ndata: %s\n\n", event, jsonData)
flusher.Flush()
}
// Update last_used
h.db.Exec(`UPDATE repositories SET last_used = datetime('now') WHERE id = ?`, id)
// Generate PR description
pr, err := services.GeneratePR(h.db, localPath, req.Base, req.Head, sendEvent)
if err != nil {
sendEvent("error", map[string]interface{}{"message": err.Error()})
return
}
// Save analysis to DB
resultJSON, _ := json.Marshal(pr)
h.db.Exec(`INSERT INTO analyses (repo_id, type, base_ref, head_ref, result) VALUES (?, 'pr_description', ?, ?, ?)`,
id, req.Base, req.Head, string(resultJSON))
}
+82 -2
View File
@@ -2,8 +2,12 @@ package handlers
import (
"database/sql"
"encoding/json"
"fmt"
"net/http"
"strconv"
"github.com/HoHD/PR-Helper/services"
"github.com/gin-gonic/gin"
)
@@ -15,9 +19,85 @@ func NewReviewHandler(db *sql.DB) *ReviewHandler {
return &ReviewHandler{db: db}
}
// Review handles POST /api/repos/:id/review — stub for Phase 4
// Review handles POST /api/repos/:id/review — SSE streaming AI code review.
func (h *ReviewHandler) Review(c *gin.Context) {
c.JSON(http.StatusNotImplemented, gin.H{"error": "review not implemented yet — coming in Phase 4"})
id := c.Param("id")
// Get repo info
var localPath string
err := h.db.QueryRow(`SELECT local_path FROM repositories WHERE id = ?`, id).Scan(&localPath)
if err == sql.ErrNoRows {
c.JSON(http.StatusNotFound, gin.H{"error": "repository not found"})
return
}
if err != nil {
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
return
}
// Parse request
var req struct {
Base string `json:"base" binding:"required"`
Head string `json:"head" binding:"required"`
TopN *int `json:"top_n"`
}
if err := c.ShouldBindJSON(&req); err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": "base and head are required"})
return
}
// Determine Top-N: request value > settings default (20)
topN := 20
if req.TopN != nil {
topN = *req.TopN
} else {
var topNStr string
h.db.QueryRow(`SELECT value FROM settings WHERE key = 'review.top_n'`).Scan(&topNStr)
if topNStr != "" {
if n, err := strconv.Atoi(topNStr); err == nil && n > 0 {
topN = n
}
}
}
// Set SSE headers
c.Header("Content-Type", "text/event-stream")
c.Header("Cache-Control", "no-cache")
c.Header("Connection", "keep-alive")
c.Header("X-Accel-Buffering", "no")
c.Status(http.StatusOK)
flusher, ok := c.Writer.(http.Flusher)
if !ok {
c.JSON(http.StatusInternalServerError, gin.H{"error": "streaming not supported"})
return
}
sendEvent := func(event string, data interface{}) {
jsonData, _ := json.Marshal(data)
fmt.Fprintf(c.Writer, "event: %s\ndata: %s\n\n", event, jsonData)
flusher.Flush()
}
// Update last_used
h.db.Exec(`UPDATE repositories SET last_used = datetime('now') WHERE id = ?`, id)
// Run AI review
err = services.GenerateReview(h.db, localPath, req.Base, req.Head, topN, sendEvent)
if err != nil {
sendEvent("error", map[string]interface{}{"message": err.Error()})
return
}
// Save analysis to DB
resultData := map[string]interface{}{
"base": req.Base,
"head": req.Head,
"top_n": topN,
}
resultJSON, _ := json.Marshal(resultData)
h.db.Exec(`INSERT INTO analyses (repo_id, type, base_ref, head_ref, result) VALUES (?, 'code_review', ?, ?, ?)`,
id, req.Base, req.Head, string(resultJSON))
}
// SaveNotes handles POST /api/repos/:id/review/notes — stub for Phase 5