yunzerwebsiteallinone/go/controllers/platform_notebook.go

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)
}