wonder
9c843b79ba
- 新增 users 和 user_settings 表，repositories/analyses/review_notes 添加 user_id 列 - 实现基于邮箱+密码的注册登录，密码使用 bcrypt 哈希 - 使用 gin-contrib/sessions cookie-based session 管理 - 所有仓库、分析记录、review notes 按用户隔离 - 用户设置（LLM 配置、review 参数、缓存配置）独立存储 - 新增登录/注册页面，导航栏显示用户邮箱和退出按钮 - 前端 fetch 请求统一添加 credentials: 'same-origin' - 支持 SESSION_SECRET 环境变量配置会话密钥
113 lines
2.7 KiB
Go
113 lines
2.7 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) {
|
|
data := userData(c)
|
|
data["ID"] = c.Param("id")
|
|
c.HTML(http.StatusOK, "pages/repo.html", data)
|
|
}
|
|
|
|
func (h *PageHandler) Generate(c *gin.Context) {
|
|
data := userData(c)
|
|
data["ID"] = c.Param("id")
|
|
c.HTML(http.StatusOK, "pages/generate.html", data)
|
|
}
|
|
|
|
func (h *PageHandler) Review(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/review.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
|
|
}
|