backend增加租户简称登录
This commit is contained in:
+317
-312
@@ -1,312 +1,317 @@
|
||||
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(¬e)
|
||||
|
||||
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(¬e)
|
||||
|
||||
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(¬e)
|
||||
|
||||
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)
|
||||
}
|
||||
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("-pinned", "-id").
|
||||
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(¬e)
|
||||
|
||||
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"`
|
||||
Pinned int8 `json:"pinned"`
|
||||
}
|
||||
|
||||
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,
|
||||
Pinned: payload.Pinned,
|
||||
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"`
|
||||
Pinned int8 `json:"pinned"`
|
||||
}
|
||||
|
||||
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(¬e)
|
||||
|
||||
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,
|
||||
"pinned": payload.Pinned,
|
||||
"update_time": now,
|
||||
})
|
||||
|
||||
if err != nil {
|
||||
jsonResponse(&c.Controller, 500, 500, "更新失败", nil)
|
||||
return
|
||||
}
|
||||
|
||||
note.Title = payload.Title
|
||||
note.Content = payload.Content
|
||||
note.Pinned = payload.Pinned
|
||||
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(¬e)
|
||||
|
||||
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)
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user