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

package api

import (
	"fmt"
	"github.com/spf13/viper"
	"gorm.io/datatypes"
	"strings"

	"git.solsynth.dev/hydrogen/paperclip/pkg/internal/database"
	"git.solsynth.dev/hydrogen/paperclip/pkg/internal/models"
	"git.solsynth.dev/hydrogen/paperclip/pkg/internal/services"
	"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

	needQuery := true

	var result = make([]models.Attachment, take)
	var idxList []string

	if len(c.Query("id")) > 0 {
		var pendingQueryId []string
		idx := strings.Split(c.Query("id"), ",")
		for p, raw := range idx {
			idxList = append(idxList, raw)
			if val, ok := services.GetAttachmentCache(raw); ok {
				result[p] = val
			} else {
				pendingQueryId = append(pendingQueryId, raw)
			}
		}
		tx = tx.Where("rid IN ?", pendingQueryId)
		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")

		// Do not expose un-public indexable attachments
		prefix := viper.GetString("database.prefix")
		tx = tx.
			Joins(fmt.Sprintf("JOIN %sattachment_pools ON %sattachment_pools.id = %sattachments.pool_id", prefix, prefix, prefix)).
			Where(datatypes.JSONQuery(fmt.Sprintf("%sattachment_pools.config", prefix)).Equals(true, "public_indexable"))
	}

	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)
		}
	}

	if pools := c.Query("pools"); len(pools) > 0 {
		prefix := viper.GetString("database.prefix")
		poolAliases := strings.Split(pools, ",")
		tx = tx.
			Joins(fmt.Sprintf("JOIN %sattachment_pools ON %sattachment_pools.id = %sattachments.pool_id", prefix, prefix, prefix)).
			Where(fmt.Sprintf("%sattachment_pools.alias IN ?", prefix), poolAliases)
	}

	if original := c.QueryBool("original", false); original {
		tx = tx.Where("ref_id IS NULL")
	}

	var count int64
	countTx := tx
	if err := countTx.Model(&models.Attachment{}).Count(&count).Error; err != nil {
		return fiber.NewError(fiber.StatusBadRequest, err.Error())
	}

	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.Rid == id {
						result[p] = item
					}
				}
			}
		}
	}

	for _, item := range result {
		services.CacheAttachment(item)
	}

	return c.JSON(fiber.Map{
		"count": count,
		"data":  result,
	})
}