brunel/handlers/packages/packages.go

package handlers_packages

import (
	"brunel/domain"
	"brunel/fastmap"
	"brunel/helpers"
	"brunel/packages"
	"fmt"
	"sort"
	"strings"

	"github.com/gofiber/fiber/v2"
)

type PackagesResponse struct {
	Total    int                                            `json:"total"`
	Packages *fastmap.Fastmap[string, domain.SourcePackage] `json:"packages"`
}

func Packages(c *fiber.Ctx) error {
	pageNum := c.QueryInt("page", 1)
	pageSize := c.QueryInt("pageSize", 250)
	search := strings.ToLower(c.Query("search"))
	filter := c.Query("filter")

	cacheKey := fmt.Sprintf("packages:%d:%d:%s:%s", pageNum, pageSize, search, filter)
	if cachedResponse, ok := helpers.Cache.Get(cacheKey); ok {
		return c.Status(fiber.StatusOK).JSON(cachedResponse)
	}

	// Adjust pageNum to be 0-based for GetPage
	adjustedPageNum := pageNum - 1
	if adjustedPageNum < 0 {
		adjustedPageNum = 0
	}

	packs := packages.GetPackages()

	// Convert map to slice for sorting
	packSlice := make([]struct {
		Key   string
		Value domain.SourcePackage
	}, 0, packs.Len())

	packs.Iter(func(k string, v domain.SourcePackage) bool {
		packSlice = append(packSlice, struct {
			Key   string
			Value domain.SourcePackage
		}{k, v})
		return true
	})

	// Stable sort the slice
	sort.SliceStable(packSlice, func(i, j int) bool {
		return packSlice[i].Key < packSlice[j].Key
	})

	finalReturn := fastmap.New[string, domain.SourcePackage]()

	for _, item := range packSlice {
		k, source := item.Key, item.Value

		matchesFilter := filter == ""
		matchesSearch := search == "" || strings.Contains(strings.ToLower(k), search)

		source.Packages.Iter(func(key string, value domain.PackageInfo) bool {
			if !matchesFilter && value.Status == domain.PackageStatus(filter) {
				matchesFilter = true
			}
			if !matchesSearch && strings.Contains(strings.ToLower(key), search) {
				matchesSearch = true
			}
			return !(matchesFilter && matchesSearch) // stop iterating if we've found matches for both
		})

		if matchesFilter && matchesSearch {
			finalReturn.Set(k, source)
		}
	}

	result := finalReturn.GetPage(adjustedPageNum, pageSize)

	response := PackagesResponse{
		Total:    finalReturn.Len(),
		Packages: result,
	}

	helpers.Cache.Set(cacheKey, response)
	return c.Status(fiber.StatusOK).JSON(response)
}