yunzerwebsiteallinone/go/controllers/backend_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 BackendNotebookController struct {
	beego.Controller
}

func (c *BackendNotebookController) backendNotebookClaims() (*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 != "backend" {
		return nil, orm.ErrNoRows
	}
	return claims, nil
}

func (c *BackendNotebookController) nbJsonErr(httpStatus, bizCode int, msg string) {
	c.Ctx.Output.SetStatus(httpStatus)
	c.Data["json"] = map[string]interface{}{"code": bizCode, "msg": msg}
	_ = c.ServeJSON()
}

func (c *BackendNotebookController) nbOk(data interface{}) {
	c.Data["json"] = map[string]interface{}{"code": 200, "msg": "success", "data": data}
	_ = c.ServeJSON()
}

// List GET /backend/notebook/list
func (c *BackendNotebookController) List() {
	claims, err := c.backendNotebookClaims()
	if err != nil {
		c.nbJsonErr(401, 401, "未登录或无权限")
		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.BackendNotebook)).
		Filter("is_deleted", 0).
		Filter("tid", claims.TenantId).
		Filter("user_id", claims.UserID)

	if keyword != "" {
		qs = qs.Filter("title__icontains", keyword)
	}

	total, _ := qs.Count()

	var list []models.BackendNotebook
	_, err = qs.OrderBy("-update_time", "-create_time").
		Limit(pageSize).Offset((page - 1) * pageSize).
		All(&list)
	if err != nil && err != orm.ErrNoRows {
		c.nbJsonErr(500, 500, "查询失败")
		return
	}
	if list == nil {
		list = []models.BackendNotebook{}
	}

	c.nbOk(map[string]interface{}{"list": list, "total": total})
}

// Detail GET /backend/notebook/detail/:id
func (c *BackendNotebookController) Detail() {
	claims, err := c.backendNotebookClaims()
	if err != nil {
		c.nbJsonErr(401, 401, "未登录或无权限")
		return
	}

	id, err := strconv.ParseUint(c.Ctx.Input.Param(":id"), 10, 64)
	if err != nil || id == 0 {
		c.nbJsonErr(400, 400, "无效ID")
		return
	}

	var note models.BackendNotebook
	err = models.Orm.QueryTable(new(models.BackendNotebook)).
		Filter("id", id).
		Filter("is_deleted", 0).
		Filter("tid", claims.TenantId).
		Filter("user_id", claims.UserID).
		One(&note)
	if err != nil {
		c.nbJsonErr(404, 404, "笔记不存在")
		return
	}

	c.nbOk(note)
}

// Create POST /backend/notebook/create
func (c *BackendNotebookController) Create() {
	claims, err := c.backendNotebookClaims()
	if err != nil {
		c.nbJsonErr(401, 401, "未登录或无权限")
		return
	}

	raw, err := io.ReadAll(c.Ctx.Request.Body)
	if err != nil {
		c.nbJsonErr(400, 400, "参数错误")
		return
	}
	var payload struct {
		Title   string `json:"title"`
		Content string `json:"content"`
	}
	if err := json.Unmarshal(raw, &payload); err != nil {
		c.nbJsonErr(400, 400, "参数错误")
		return
	}

	payload.Title = strings.TrimSpace(payload.Title)
	if payload.Title == "" {
		payload.Title = "无标题"
	}

	userID := uint64(claims.UserID)
	note := &models.BackendNotebook{
		Tid:       claims.TenantId,
		Title:     payload.Title,
		Content:   payload.Content,
		UserID:    &userID,
		UserName:  &claims.Username,
		IsDeleted: 0,
	}

	id, err := models.Orm.Insert(note)
	if err != nil {
		c.nbJsonErr(500, 500, "创建失败")
		return
	}

	note.ID = uint64(id)
	c.nbOk(note)
}

// Update POST /backend/notebook/update/:id
func (c *BackendNotebookController) Update() {
	claims, err := c.backendNotebookClaims()
	if err != nil {
		c.nbJsonErr(401, 401, "未登录或无权限")
		return
	}

	id, err := strconv.ParseUint(c.Ctx.Input.Param(":id"), 10, 64)
	if err != nil || id == 0 {
		c.nbJsonErr(400, 400, "无效ID")
		return
	}

	raw, err := io.ReadAll(c.Ctx.Request.Body)
	if err != nil {
		c.nbJsonErr(400, 400, "参数错误")
		return
	}
	var payload struct {
		Title   string `json:"title"`
		Content string `json:"content"`
	}
	if err := json.Unmarshal(raw, &payload); err != nil {
		c.nbJsonErr(400, 400, "参数错误")
		return
	}

	payload.Title = strings.TrimSpace(payload.Title)
	if payload.Title == "" {
		payload.Title = "无标题"
	}

	var note models.BackendNotebook
	err = models.Orm.QueryTable(new(models.BackendNotebook)).
		Filter("id", id).
		Filter("is_deleted", 0).
		Filter("tid", claims.TenantId).
		Filter("user_id", claims.UserID).
		One(&note)
	if err != nil {
		c.nbJsonErr(404, 404, "笔记不存在")
		return
	}

	now := time.Now()
	_, err = models.Orm.QueryTable(new(models.BackendNotebook)).
		Filter("id", id).
		Update(map[string]interface{}{
			"title":       payload.Title,
			"content":     payload.Content,
			"update_time": now,
		})
	if err != nil {
		c.nbJsonErr(500, 500, "更新失败")
		return
	}

	note.Title = payload.Title
	note.Content = payload.Content
	note.UpdateTime = &now
	c.nbOk(note)
}

// Delete DELETE /backend/notebook/delete/:id
func (c *BackendNotebookController) Delete() {
	claims, err := c.backendNotebookClaims()
	if err != nil {
		c.nbJsonErr(401, 401, "未登录或无权限")
		return
	}

	id, err := strconv.ParseUint(c.Ctx.Input.Param(":id"), 10, 64)
	if err != nil || id == 0 {
		c.nbJsonErr(400, 400, "无效ID")
		return
	}

	var note models.BackendNotebook
	err = models.Orm.QueryTable(new(models.BackendNotebook)).
		Filter("id", id).
		Filter("is_deleted", 0).
		Filter("tid", claims.TenantId).
		Filter("user_id", claims.UserID).
		One(&note)
	if err != nil {
		c.nbJsonErr(404, 404, "笔记不存在")
		return
	}

	now := time.Now()
	_, err = models.Orm.QueryTable(new(models.BackendNotebook)).
		Filter("id", id).
		Update(map[string]interface{}{
			"is_deleted":  1,
			"delete_time": now,
		})
	if err != nil {
		c.nbJsonErr(500, 500, "删除失败")
		return
	}

	c.nbOk(nil)
}