wonder
8e8ea6a07d
Deploy PR-Helper / deploy (push) Successful in 2m9s
- 新增 PR 描述 tab：流式生成 + 双栏实时预览（渲染效果 + Markdown 源码） - 新增 AI 代码审查 tab：Top-N/并发设置、历史记录、进度条、总结、备注、PDF 导出、内联 Diff - 保留 Diff 视图中的 AI 评审快捷内联按钮 - Repo handler 传递 TopN/Concurrency 设置 - 删除独立的 generate.html 和 review.html 页面及路由
101 lines
2.4 KiB
Go
101 lines
2.4 KiB
Go
package handlers
|
|
|
|
import (
|
|
"database/sql"
|
|
"net/http"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
|
|
"github.com/HoHD/PR-Helper/models"
|
|
)
|
|
|
|
type PageHandler struct {
|
|
db *sql.DB
|
|
}
|
|
|
|
func NewPageHandler(db *sql.DB) *PageHandler {
|
|
return &PageHandler{db: db}
|
|
}
|
|
|
|
// userData returns a gin.H with the current user info for template rendering.
|
|
func userData(c *gin.Context) gin.H {
|
|
user := GetCurrentUser(c)
|
|
if user == nil {
|
|
return gin.H{}
|
|
}
|
|
return gin.H{"User": user}
|
|
}
|
|
|
|
func (h *PageHandler) Index(c *gin.Context) {
|
|
user := GetCurrentUser(c)
|
|
if user == nil {
|
|
c.Redirect(http.StatusFound, "/login")
|
|
return
|
|
}
|
|
repos, _ := h.listRepos(user.ID)
|
|
data := userData(c)
|
|
data["Repos"] = repos
|
|
c.HTML(http.StatusOK, "pages/index.html", data)
|
|
}
|
|
|
|
func (h *PageHandler) Repo(c *gin.Context) {
|
|
user := GetCurrentUser(c)
|
|
topN := "20"
|
|
concurrency := "5"
|
|
if user != nil {
|
|
h.db.QueryRow("SELECT value FROM user_settings WHERE user_id = ? AND `key` = 'review.top_n'", user.ID).Scan(&topN)
|
|
h.db.QueryRow("SELECT value FROM user_settings WHERE user_id = ? AND `key` = 'review.concurrency'", user.ID).Scan(&concurrency)
|
|
}
|
|
if topN == "" {
|
|
topN = "20"
|
|
}
|
|
if concurrency == "" {
|
|
concurrency = "5"
|
|
}
|
|
data := userData(c)
|
|
data["ID"] = c.Param("id")
|
|
data["TopN"] = topN
|
|
data["Concurrency"] = concurrency
|
|
c.HTML(http.StatusOK, "pages/repo.html", data)
|
|
}
|
|
|
|
func (h *PageHandler) Settings(c *gin.Context) {
|
|
user := GetCurrentUser(c)
|
|
settings := make(map[string]string)
|
|
for key := range models.DefaultSettings {
|
|
settings[key] = ""
|
|
}
|
|
if user != nil {
|
|
rows, err := h.db.Query("SELECT `key`, value FROM user_settings WHERE user_id = ?", user.ID)
|
|
if err == nil {
|
|
defer rows.Close()
|
|
for rows.Next() {
|
|
var k, v string
|
|
if rows.Scan(&k, &v) == nil {
|
|
settings[k] = v
|
|
}
|
|
}
|
|
}
|
|
}
|
|
data := userData(c)
|
|
data["Settings"] = settings
|
|
c.HTML(http.StatusOK, "pages/settings.html", data)
|
|
}
|
|
|
|
func (h *PageHandler) listRepos(userID int64) ([]models.Repository, error) {
|
|
rows, err := h.db.Query(`SELECT id, url, local_path, size_bytes, cloned_at, last_used FROM repositories WHERE user_id = ? ORDER BY last_used DESC`, userID)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer rows.Close()
|
|
var repos []models.Repository
|
|
for rows.Next() {
|
|
var r models.Repository
|
|
if err := rows.Scan(&r.ID, &r.URL, &r.LocalPath, &r.SizeBytes, &r.ClonedAt, &r.LastUsed); err != nil {
|
|
continue
|
|
}
|
|
repos = append(repos, r)
|
|
}
|
|
return repos, nil
|
|
}
|