terraform-provider-stackitprivatepreview/stackit/internal/services/postgresflexalpha/flavor/functions.go

package postgresFlexAlphaFlavor

import (
	"context"
	"fmt"

	"github.com/stackitcloud/stackit-sdk-go/services/postgresflex/v3alpha1api"
)

type flavorsClientReader interface {
	GetFlavorsRequest(
		ctx context.Context,
		projectId, region string,
	) v3alpha1api.ApiGetFlavorsRequestRequest
}

func getAllFlavors(ctx context.Context, client flavorsClientReader, projectId, region string) (
	[]v3alpha1api.ListFlavors,
	error,
) {
	getAllFilter := func(_ v3alpha1api.ListFlavors) bool { return true }
	flavorList, err := getFlavorsByFilter(ctx, client, projectId, region, getAllFilter)
	if err != nil {
		return nil, err
	}
	return flavorList, nil
}

// getFlavorsByFilter is a helper function to retrieve flavors using a filtern function.
// Hint: The API does not have a GetFlavors endpoint, only ListFlavors
func getFlavorsByFilter(
	ctx context.Context,
	client flavorsClientReader,
	projectId, region string,
	filter func(db v3alpha1api.ListFlavors) bool,
) ([]v3alpha1api.ListFlavors, error) {
	if projectId == "" || region == "" {
		return nil, fmt.Errorf("listing postgresflex flavors: projectId and region are required")
	}

	const pageSize = 25

	var result = make([]v3alpha1api.ListFlavors, 0)

	for page := int32(1); ; page++ {
		res, err := client.GetFlavorsRequest(ctx, projectId, region).
			Page(page).Size(pageSize).Sort(v3alpha1api.FLAVORSORT_ID_ASC).Execute()
		if err != nil {
			return nil, fmt.Errorf("requesting flavors list (page %d): %w", page, err)
		}

		// If the API returns no flavors, we have reached the end of the list.
		if len(res.Flavors) == 0 {
			break
		}

		for _, flavor := range res.Flavors {
			if filter(flavor) {
				result = append(result, flavor)
			}
		}
	}

	return result, nil
}