diff --git a/routers/controllers/admin.go b/routers/controllers/admin.go index 3a72dff..9da1d3a 100644 --- a/routers/controllers/admin.go +++ b/routers/controllers/admin.go @@ -378,3 +378,14 @@ func AdminListShare(c *gin.Context) { c.JSON(200, ErrorResponse(err)) } } + +// AdminDeleteShare 批量删除分享 +func AdminDeleteShare(c *gin.Context) { + var service admin.ShareBatchService + if err := c.ShouldBindJSON(&service); err == nil { + res := service.Delete(c) + c.JSON(200, res) + } else { + c.JSON(200, ErrorResponse(err)) + } +} diff --git a/routers/router.go b/routers/router.go index c237bfd..90da1ee 100644 --- a/routers/router.go +++ b/routers/router.go @@ -393,6 +393,8 @@ func InitMasterRouter() *gin.Engine { { // 列出分享 share.POST("list", controllers.AdminListShare) + // 删除 + share.POST("delete", controllers.AdminDeleteShare) } } diff --git a/service/admin/share.go b/service/admin/share.go new file mode 100644 index 0000000..12d3d4f --- /dev/null +++ b/service/admin/share.go @@ -0,0 +1,79 @@ +package admin + +import ( + model "github.com/HFO4/cloudreve/models" + "github.com/HFO4/cloudreve/pkg/hashid" + "github.com/HFO4/cloudreve/pkg/serializer" + "github.com/gin-gonic/gin" + "strings" +) + +// ShareBatchService 分享批量操作服务 +type ShareBatchService struct { + ID []uint `json:"id" binding:"min=1"` +} + +// Delete 删除文件 +func (service *ShareBatchService) Delete(c *gin.Context) serializer.Response { + if err := model.DB.Where("id in (?)", service.ID).Delete(&model.Share{}).Error; err != nil { + return serializer.DBErr("无法删除分享", err) + } + return serializer.Response{} +} + +// Shares 列出分享 +func (service *AdminListService) Shares() serializer.Response { + var res []model.Share + total := 0 + + tx := model.DB.Model(&model.Share{}) + if service.OrderBy != "" { + tx = tx.Order(service.OrderBy) + } + + for k, v := range service.Conditions { + tx = tx.Where(k+" = ?", v) + } + + if len(service.Searches) > 0 { + search := "" + for k, v := range service.Searches { + search += k + " like '%" + v + "%' OR " + } + search = strings.TrimSuffix(search, " OR ") + tx = tx.Where(search) + } + + // 计算总数用于分页 + tx.Count(&total) + + // 查询记录 + tx.Limit(service.PageSize).Offset((service.Page - 1) * service.PageSize).Find(&res) + + // 查询对应用户,同时计算HashID + users := make(map[uint]model.User) + hashIDs := make(map[uint]string, len(res)) + for _, file := range res { + users[file.UserID] = model.User{} + hashIDs[file.ID] = hashid.HashID(file.ID, hashid.ShareID) + } + + userIDs := make([]uint, 0, len(users)) + for k := range users { + userIDs = append(userIDs, k) + } + + var userList []model.User + model.DB.Where("id in (?)", userIDs).Find(&userList) + + for _, v := range userList { + users[v.ID] = v + } + + return serializer.Response{Data: map[string]interface{}{ + "total": total, + "items": res, + "users": users, + "ids": hashIDs, + }} +}