yunzerwebsite/yunzerwebsiteallinone
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

实现记事本功能

Browse Source
This commit is contained in:
hero920103
2026-06-17 17:49:59 +08:00
parent 673b84a0b3
commit 4e1b30b3cc
14 changed files with 1383 additions and 214 deletions
Download Patch File Download Diff File Expand all files Collapse all files
go/controllers/platform_notebook.go
+312
View File
@@ -0,0 +1,312 @@
package controllers
import (
"encoding/json"
"io"
"server/models"
"server/pkg/jwtutil"
"strconv"
"strings"
"time"
"github.com/beego/beego/v2/client/orm"
beego "github.com/beego/beego/v2/server/web"
)
type PlatformNotebookController struct {
beego.Controller
}
// requireAuth 验证平台用户权限
func requireNotebookAuth(c *beego.Controller) (*jwtutil.Claims, error) {
auth := c.Ctx.Request.Header.Get("Authorization")
if auth == "" {
return nil, orm.ErrNoRows
}
parts := strings.SplitN(auth, " ", 2)
if len(parts) != 2 || parts[0] != "Bearer" {
return nil, orm.ErrNoRows
}
claims, err := jwtutil.ParseToken(parts[1])
if err != nil {
return nil, err
}
if claims.UserType != "platform" {
return nil, orm.ErrNoRows
}
return claims, nil
}
// jsonResponse 统一JSON响应
func jsonResponse(c *beego.Controller, httpStatus, code int, msg string, data interface{}) {
c.Ctx.Output.SetStatus(httpStatus)
resp := map[string]interface{}{
"code": code,
"msg": msg,
}
if data != nil {
resp["data"] = data
}
c.Data["json"] = resp
_ = c.ServeJSON()
}
// List 获取笔记列表
// GET /platform/notebook/list
func (c *PlatformNotebookController) List() {
claims, err := requireNotebookAuth(&c.Controller)
if err != nil {
jsonResponse(&c.Controller, 401, 401, "未登录或无权限", nil)
return
}
page, _ := c.GetInt("page", 1)
pageSize, _ := c.GetInt("pageSize", 20)
keyword := strings.TrimSpace(c.GetString("keyword"))
if page < 1 {
page = 1
}
if pageSize < 1 || pageSize > 100 {
pageSize = 20
}
qs := models.Orm.QueryTable(new(models.PlatformNotebook)).
Filter("is_deleted", 0).
Filter("user_id", claims.UserID)
if keyword != "" {
qs = qs.Filter("title__icontains", keyword)
}
total, err := qs.Count()
if err != nil {
jsonResponse(&c.Controller, 500, 500, "查询失败", nil)
return
}
var list []models.PlatformNotebook
_, err = qs.OrderBy("-update_time", "-create_time").
Limit(pageSize).
Offset((page - 1) * pageSize).
All(&list)
if err != nil && err != orm.ErrNoRows {
jsonResponse(&c.Controller, 500, 500, "查询失败", nil)
return
}
if list == nil {
list = []models.PlatformNotebook{}
}
jsonResponse(&c.Controller, 200, 200, "success", map[string]interface{}{
"list": list,
"total": total,
})
}
// Detail 获取笔记详情
// GET /platform/notebook/detail/:id
func (c *PlatformNotebookController) Detail() {
claims, err := requireNotebookAuth(&c.Controller)
if err != nil {
jsonResponse(&c.Controller, 401, 401, "未登录或无权限", nil)
return
}
id, err := strconv.ParseUint(c.Ctx.Input.Param(":id"), 10, 64)
if err != nil || id == 0 {
jsonResponse(&c.Controller, 400, 400, "无效ID", nil)
return
}
var note models.PlatformNotebook
err = models.Orm.QueryTable(new(models.PlatformNotebook)).
Filter("id", id).
Filter("is_deleted", 0).
Filter("user_id", claims.UserID).
One(&note)
if err != nil {
if err == orm.ErrNoRows {
jsonResponse(&c.Controller, 404, 404, "笔记不存在", nil)
} else {
jsonResponse(&c.Controller, 500, 500, "查询失败", nil)
}
return
}
jsonResponse(&c.Controller, 200, 200, "success", note)
}
// Create 创建笔记
// POST /platform/notebook/create
func (c *PlatformNotebookController) Create() {
claims, err := requireNotebookAuth(&c.Controller)
if err != nil {
jsonResponse(&c.Controller, 401, 401, "未登录或无权限", nil)
return
}
raw, err := io.ReadAll(c.Ctx.Request.Body)
if err != nil {
jsonResponse(&c.Controller, 400, 400, "参数错误", nil)
return
}
var payload struct {
Title string `json:"title"`
Content string `json:"content"`
}
if err := json.Unmarshal(raw, &payload); err != nil {
jsonResponse(&c.Controller, 400, 400, "参数错误", nil)
return
}
payload.Title = strings.TrimSpace(payload.Title)
if payload.Title == "" {
payload.Title = "无标题"
}
userID := uint64(claims.UserID)
note := &models.PlatformNotebook{
Title: payload.Title,
Content: payload.Content,
UserID: &userID,
UserName: &claims.Username,
IsDeleted: 0,
}
id, err := models.Orm.Insert(note)
if err != nil {
jsonResponse(&c.Controller, 500, 500, "创建失败", nil)
return
}
note.ID = uint64(id)
jsonResponse(&c.Controller, 200, 200, "创建成功", note)
}
// Update 更新笔记
// POST /platform/notebook/update/:id
func (c *PlatformNotebookController) Update() {
claims, err := requireNotebookAuth(&c.Controller)
if err != nil {
jsonResponse(&c.Controller, 401, 401, "未登录或无权限", nil)
return
}
id, err := strconv.ParseUint(c.Ctx.Input.Param(":id"), 10, 64)
if err != nil || id == 0 {
jsonResponse(&c.Controller, 400, 400, "无效ID", nil)
return
}
raw, err := io.ReadAll(c.Ctx.Request.Body)
if err != nil {
jsonResponse(&c.Controller, 400, 400, "参数错误", nil)
return
}
var payload struct {
Title string `json:"title"`
Content string `json:"content"`
}
if err := json.Unmarshal(raw, &payload); err != nil {
jsonResponse(&c.Controller, 400, 400, "参数错误", nil)
return
}
payload.Title = strings.TrimSpace(payload.Title)
if payload.Title == "" {
payload.Title = "无标题"
}
// 验证笔记是否存在且属于当前用户
var note models.PlatformNotebook
err = models.Orm.QueryTable(new(models.PlatformNotebook)).
Filter("id", id).
Filter("is_deleted", 0).
Filter("user_id", claims.UserID).
One(&note)
if err != nil {
if err == orm.ErrNoRows {
jsonResponse(&c.Controller, 404, 404, "笔记不存在", nil)
} else {
jsonResponse(&c.Controller, 500, 500, "查询失败", nil)
}
return
}
now := time.Now()
_, err = models.Orm.QueryTable(new(models.PlatformNotebook)).
Filter("id", id).
Update(map[string]interface{}{
"title": payload.Title,
"content": payload.Content,
"update_time": now,
})
if err != nil {
jsonResponse(&c.Controller, 500, 500, "更新失败", nil)
return
}
note.Title = payload.Title
note.Content = payload.Content
note.UpdateTime = &now
jsonResponse(&c.Controller, 200, 200, "更新成功", note)
}
// Delete 删除笔记（软删除）
// DELETE /platform/notebook/delete/:id
func (c *PlatformNotebookController) Delete() {
claims, err := requireNotebookAuth(&c.Controller)
if err != nil {
jsonResponse(&c.Controller, 401, 401, "未登录或无权限", nil)
return
}
id, err := strconv.ParseUint(c.Ctx.Input.Param(":id"), 10, 64)
if err != nil || id == 0 {
jsonResponse(&c.Controller, 400, 400, "无效ID", nil)
return
}
// 验证笔记是否存在且属于当前用户
var note models.PlatformNotebook
err = models.Orm.QueryTable(new(models.PlatformNotebook)).
Filter("id", id).
Filter("is_deleted", 0).
Filter("user_id", claims.UserID).
One(&note)
if err != nil {
if err == orm.ErrNoRows {
jsonResponse(&c.Controller, 404, 404, "笔记不存在", nil)
} else {
jsonResponse(&c.Controller, 500, 500, "查询失败", nil)
}
return
}
now := time.Now()
_, err = models.Orm.QueryTable(new(models.PlatformNotebook)).
Filter("id", id).
Update(map[string]interface{}{
"is_deleted": 1,
"delete_time": now,
})
if err != nil {
jsonResponse(&c.Controller, 500, 500, "删除失败", nil)
return
}
jsonResponse(&c.Controller, 200, 200, "删除成功", nil)
}
go/models/init.go
+1
View File
@@ -62,6 +62,7 @@ func Init(_ string) {
new(PlatformAccountPoolWindsurf),
new(PlatformAccountPoolCursor),
new(PlatformAccountPoolCodex),
new(PlatformNotebook),
new(CmsArticleCategory),
new(CmsArticle),
go/models/platform_notebook.go
+20
View File
@@ -0,0 +1,20 @@
package models
import "time"
// PlatformNotebook 平台记事本表: yz_platform_notebook
type PlatformNotebook struct {
ID uint64 `orm:"column(id);pk;auto" json:"id"`
Title string `orm:"column(title);size(255)" json:"title"`
Content string `orm:"column(content);type(longtext);null" json:"content"`
UserID *uint64 `orm:"column(user_id);null" json:"user_id"`
UserName *string `orm:"column(user_name);size(100);null" json:"user_name"`
IsDeleted int8 `orm:"column(is_deleted);default(0)" json:"is_deleted"`
CreateTime time.Time `orm:"column(create_time);auto_now_add;type(datetime)" json:"create_time"`
UpdateTime *time.Time `orm:"column(update_time);type(datetime);null" json:"update_time"`
DeleteTime *time.Time `orm:"column(delete_time);type(datetime);null" json:"delete_time"`
}
func (m *PlatformNotebook) TableName() string {
return "yz_platform_notebook"
}
go/routers/platform/platform.go
+7
View File
@@ -232,4 +232,11 @@ func Register() {
beego.Router("/platform/accountPool/codex/unextract", &controllers.PlatformAccountPoolCodexController{}, "post:Unextract")
beego.Router("/platform/accountPool/codex/replenish", &controllers.PlatformAccountPoolCodexController{}, "post:Replenish")
beego.Router("/platform/accountPool/codex/probeToken", &controllers.PlatformAccountPoolCodexController{}, "post:ProbeToken")
// 记事本管理
beego.Router("/platform/notebook/list", &controllers.PlatformNotebookController{}, "get:List")
beego.Router("/platform/notebook/detail/:id", &controllers.PlatformNotebookController{}, "get:Detail")
beego.Router("/platform/notebook/create", &controllers.PlatformNotebookController{}, "post:Create")
beego.Router("/platform/notebook/update/:id", &controllers.PlatformNotebookController{}, "post:Update")
beego.Router("/platform/notebook/delete/:id", &controllers.PlatformNotebookController{}, "delete:Delete")
}