Paperclip/pkg/internal/server/api/attachment_dir_api.go

103 lines
2.3 KiB
Go
Raw Normal View History

2024-07-25 14:02:26 +00:00
package api
2024-07-25 14:12:00 +00:00
import (
2024-08-06 15:45:58 +00:00
"strconv"
2024-07-25 14:12:00 +00:00
"strings"
"git.solsynth.dev/hydrogen/paperclip/pkg/internal/database"
"git.solsynth.dev/hydrogen/paperclip/pkg/internal/models"
2024-08-06 15:45:58 +00:00
"git.solsynth.dev/hydrogen/paperclip/pkg/internal/services"
2024-07-25 14:12:00 +00:00
"github.com/gofiber/fiber/v2"
)
func listAttachment(c *fiber.Ctx) error {
take := c.QueryInt("take", 0)
offset := c.QueryInt("offset", 0)
if take > 100 {
take = 100
}
tx := database.C
2024-07-25 14:12:00 +00:00
2024-08-06 17:02:03 +00:00
needQuery := true
2024-08-06 15:45:58 +00:00
var result = make([]models.Attachment, take)
var idxList []uint
2024-08-02 13:52:32 +00:00
if len(c.Query("id")) > 0 {
2024-08-06 15:45:58 +00:00
var pendingQueryId []uint
idx := strings.Split(c.Query("id"), ",")
for p, raw := range idx {
id, err := strconv.Atoi(raw)
if err != nil {
continue
} else {
idxList = append(idxList, uint(id))
}
if val, ok := services.GetAttachmentCache(uint(id)); ok {
result[p] = val
} else {
pendingQueryId = append(pendingQueryId, uint(id))
}
}
tx = tx.Where("id IN ?", pendingQueryId)
2024-08-06 17:02:03 +00:00
needQuery = len(pendingQueryId) > 0
} else {
// Do sort this when doesn't filter by the id
// Because the sort will mess up the result
tx = tx.Order("created_at DESC")
2024-08-02 13:52:32 +00:00
}
if len(c.Query("author")) > 0 {
var author models.Account
if err := database.C.Where("name = ?", c.Query("author")).First(&author).Error; err != nil {
return fiber.NewError(fiber.StatusBadRequest, err.Error())
} else {
tx = tx.Where("account_id = ?", author.ID)
}
2024-07-25 14:12:00 +00:00
}
if usage := c.Query("usage"); len(usage) > 0 {
tx = tx.Where("usage IN ?", strings.Split(usage, " "))
2024-07-25 14:12:00 +00:00
}
if original := c.QueryBool("original", false); original {
tx = tx.Where("ref_id IS NULL")
}
2024-07-25 14:12:00 +00:00
var count int64
countTx := tx
if err := countTx.Model(&models.Attachment{}).Count(&count).Error; err != nil {
return fiber.NewError(fiber.StatusBadRequest, err.Error())
}
2024-08-06 17:02:03 +00:00
if needQuery {
var out []models.Attachment
if err := tx.Offset(offset).Limit(take).Preload("Account").Find(&out).Error; err != nil {
return fiber.NewError(fiber.StatusBadRequest, err.Error())
}
if len(idxList) == 0 {
result = out
} else {
for _, item := range out {
for p, id := range idxList {
if item.ID == id {
result[p] = item
}
2024-08-06 15:45:58 +00:00
}
}
}
}
2024-08-06 17:02:03 +00:00
for _, item := range result {
services.CacheAttachment(item.ID, item)
}
2024-07-25 14:12:00 +00:00
return c.JSON(fiber.Map{
"count": count,
2024-08-06 15:45:58 +00:00
"data": result,
2024-07-25 14:12:00 +00:00
})
}