249 lines
7.1 KiB
Go
249 lines
7.1 KiB
Go
package controller
|
||
|
||
import (
|
||
"github.com/gofiber/fiber/v2"
|
||
_ "leafdev.top/Leaf/leaf-library-3/internal/entity"
|
||
"leafdev.top/Leaf/leaf-library-3/internal/pkg/validator"
|
||
"leafdev.top/Leaf/leaf-library-3/internal/services/auth"
|
||
"leafdev.top/Leaf/leaf-library-3/internal/services/collection"
|
||
"leafdev.top/Leaf/leaf-library-3/internal/services/workspace"
|
||
"leafdev.top/Leaf/leaf-library-3/internal/types/dto"
|
||
"leafdev.top/Leaf/leaf-library-3/internal/types/errs"
|
||
)
|
||
|
||
type CollectionController struct {
|
||
collectionService *collection.Service
|
||
workspaceService *workspace.Service
|
||
authService *auth.Service
|
||
}
|
||
|
||
func NewCollectionController(
|
||
collectionService *collection.Service,
|
||
workspaceService *workspace.Service,
|
||
authService *auth.Service,
|
||
) *CollectionController {
|
||
return &CollectionController{
|
||
collectionService: collectionService,
|
||
workspaceService: workspaceService,
|
||
authService: authService,
|
||
}
|
||
}
|
||
|
||
// CreateCollection 创建集合
|
||
// @Summary 创建集合
|
||
// @Description 在工作空间下创建新的集合
|
||
// @Tags Collection
|
||
// @Accept json
|
||
// @Produce json
|
||
// @Param request body dto.CreateCollectionRequest true "创建集合请求"
|
||
// @Success 200 {object} dto.Response{data=entity.Collection}
|
||
// @Failure 400 {object} dto.Response{data=[]dto.ValidateError} "参数验证失败"
|
||
// @Failure 404 {object} dto.Response "工作空间不存在"
|
||
// @Router /collections [post]
|
||
func (c *CollectionController) CreateCollection(ctx *fiber.Ctx) error {
|
||
var req dto.CreateCollectionRequest
|
||
if err := ctx.BodyParser(&req); err != nil {
|
||
return err
|
||
}
|
||
|
||
if validationErrors, ok, err := validator.Struct(req); !ok {
|
||
if err != nil {
|
||
return err
|
||
}
|
||
return dto.Ctx(ctx).Message((*validationErrors)[0].Message).Status(fiber.StatusBadRequest).Send()
|
||
}
|
||
|
||
// 检查工作空间是否存在
|
||
exists, err := c.workspaceService.Exists(ctx.Context(), req.WorkspaceID)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
if !exists {
|
||
return dto.Ctx(ctx).Error(errs.ErrWorkspaceNotExists).Status(fiber.StatusNotFound).Send()
|
||
}
|
||
|
||
collection, err := c.collectionService.Create(ctx.Context(), req.WorkspaceID, req.Name)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
return dto.Ctx(ctx).Success(collection).Send()
|
||
}
|
||
|
||
// GetCollection 获取集合
|
||
// @Summary 获取集合
|
||
// @Description 根据集合ID获取集合详情
|
||
// @Tags Collection
|
||
// @Accept json
|
||
// @Produce json
|
||
// @Param id path int true "集<><E99B86>ID"
|
||
// @Success 200 {object} dto.Response{data=entity.Collection}
|
||
// @Failure 404 {object} dto.Response "集合不存在"
|
||
// @Router /collections/{id} [get]
|
||
func (c *CollectionController) GetCollection(ctx *fiber.Ctx) error {
|
||
var req dto.GetCollectionRequest
|
||
if err := ctx.ParamsParser(&req); err != nil {
|
||
return err
|
||
}
|
||
|
||
exists, err := c.collectionService.Exists(ctx.Context(), req.ID)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
if !exists {
|
||
return dto.Ctx(ctx).Error(errs.ErrCollectionNotExists).Status(fiber.StatusNotFound).Send()
|
||
}
|
||
|
||
collection, err := c.collectionService.Get(ctx.Context(), req.ID)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
return dto.Ctx(ctx).Success(collection).Send()
|
||
}
|
||
|
||
// ListCollections 列出工作空间下的所有集合
|
||
// @Summary 列出集合列表
|
||
// @Description 获取指定工作空间下的所有集合
|
||
// @Tags Collection
|
||
// @Accept json
|
||
// @Produce json
|
||
// @Param workspace_id path int true "工作空间ID"
|
||
// @Success 200 {object} dto.Response{data=[]entity.Collection}
|
||
// @Failure 404 {object} dto.Response "工作空间不存在"
|
||
// @Router /workspaces/{workspace_id}/collections [get]
|
||
func (c *CollectionController) ListCollections(ctx *fiber.Ctx) error {
|
||
var req dto.ListCollectionsRequest
|
||
if err := ctx.ParamsParser(&req); err != nil {
|
||
return err
|
||
}
|
||
|
||
// 检<><E6A380><EFBFBD>工作空间是否存在
|
||
exists, err := c.workspaceService.Exists(ctx.Context(), req.WorkspaceID)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
if !exists {
|
||
return dto.Ctx(ctx).Error(errs.ErrWorkspaceNotExists).Status(fiber.StatusNotFound).Send()
|
||
}
|
||
|
||
collections, err := c.collectionService.List(ctx.Context(), req.WorkspaceID)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
return dto.Ctx(ctx).Success(collections).Send()
|
||
}
|
||
|
||
// UpdateCollection 更新集合
|
||
// @Summary 更新集合
|
||
// @Description 更新集合的名称等信息
|
||
// @Tags Collection
|
||
// @Accept json
|
||
// @Produce json
|
||
// @Param id path int true "集合ID"
|
||
// @Param request body dto.UpdateCollectionRequest true "更新集合请求"
|
||
// @Success 200 {object} dto.Response{data=entity.Collection}
|
||
// @Failure 400 {object} dto.Response{data=[]dto.ValidateError} "参数验证失败"
|
||
// @Failure 404 {object} dto.Response "集合不存在"
|
||
// @Router /collections/{id} [put]
|
||
func (c *CollectionController) UpdateCollection(ctx *fiber.Ctx) error {
|
||
var params dto.CollectionIDParam
|
||
if err := ctx.ParamsParser(¶ms); err != nil {
|
||
return err
|
||
}
|
||
|
||
var req dto.UpdateCollectionRequest
|
||
if err := ctx.BodyParser(&req); err != nil {
|
||
return err
|
||
}
|
||
|
||
if validationErrors, ok, err := validator.Struct(req); !ok {
|
||
if err != nil {
|
||
return err
|
||
}
|
||
return dto.Ctx(ctx).Data(validationErrors).Status(fiber.StatusBadRequest).Send()
|
||
}
|
||
|
||
// 检查集合是否存在
|
||
collection, err := c.collectionService.Get(ctx.Context(), params.ID)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
if collection == nil {
|
||
return dto.Ctx(ctx).Error(errs.ErrCollectionNotExists).Status(fiber.StatusNotFound).Send()
|
||
}
|
||
|
||
// 检查用户是否有权限访问该集合
|
||
user := c.authService.GetUser(ctx)
|
||
workspace, err := c.workspaceService.Get(ctx.Context(), collection.WorkspaceId)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
isMember, err := c.workspaceService.IsMember(ctx.Context(), user.ID, workspace)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
if !isMember {
|
||
return dto.Ctx(ctx).Error(errs.ErrNoPermission).Status(fiber.StatusForbidden).Send()
|
||
}
|
||
|
||
// 更新集合
|
||
collection, err = c.collectionService.Update(ctx.Context(), params.ID, req.Name)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
return dto.Ctx(ctx).Success(collection).Send()
|
||
}
|
||
|
||
// DeleteCollection 删除集合
|
||
// @Summary 删除集合
|
||
// @Description 删除指定的集合
|
||
// @Tags Collection
|
||
// @Accept json
|
||
// @Produce json
|
||
// @Param id path int true "集合ID"
|
||
// @Success 200 {object} dto.Response
|
||
// @Failure 404 {object} dto.Response "集合不存在"
|
||
// @Router /collections/{id} [delete]
|
||
func (c *CollectionController) DeleteCollection(ctx *fiber.Ctx) error {
|
||
var params dto.CollectionIDParam
|
||
if err := ctx.ParamsParser(¶ms); err != nil {
|
||
return err
|
||
}
|
||
|
||
// 检查集合是否存在
|
||
collection, err := c.collectionService.Get(ctx.Context(), params.ID)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
if collection == nil {
|
||
return dto.Ctx(ctx).Error(errs.ErrCollectionNotExists).Status(fiber.StatusNotFound).Send()
|
||
}
|
||
|
||
// 检查用户是否有权限访问该集合
|
||
user := c.authService.GetUser(ctx)
|
||
workspace, err := c.workspaceService.Get(ctx.Context(), collection.WorkspaceId)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
isMember, err := c.workspaceService.IsMember(ctx.Context(), user.ID, workspace)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
if !isMember {
|
||
return dto.Ctx(ctx).Error(errs.ErrNoPermission).Status(fiber.StatusForbidden).Send()
|
||
}
|
||
|
||
// 删除集合
|
||
err = c.collectionService.Delete(ctx.Context(), params.ID)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
return dto.Ctx(ctx).Success(nil).Send()
|
||
}
|