cabret/operation/categorize.go

package operation

import (
	"fmt"
	"log"

	"github.com/aziis98/cabret"
	"github.com/iancoleman/strcase"
)

func init() {
	registerType("categorize", &Categorize{})
}

type Categorize struct {
	Key              string
	CategoryVariable string
}

func getKey[T any](m map[string]any, key string, defaultValue ...T) (T, error) {
	v, ok := m[key]
	if !ok {
		if len(defaultValue) > 0 {
			return defaultValue[0], nil
		}

		var zero T
		return zero, fmt.Errorf(`missing "%s" field`, key)
	}

	value, ok := v.(T)
	if !ok {
		var zero T
		return zero, fmt.Errorf(`expected %T but got "%v" of type %T`, zero, v, v)
	}

	return value, nil
}

func (op *Categorize) Load(config map[string]any) error {
	var err error

	op.Key, err = getKey[string](config, "key")
	if err != nil {
		return err
	}

	op.CategoryVariable, err = getKey(config, "bind", "Category")
	if err != nil {
		return err
	}

	return nil
}

func (op *Categorize) ProcessList(contents []cabret.Content) ([]cabret.Content, error) {
	key := strcase.ToCamel(op.Key)

	categories := map[string][]cabret.Content{}

	for _, content := range contents {
		v, ok := content.Metadata[key]
		if !ok {
			log.Printf(`[operation.Categorize] item has no categories`)
			continue
		}

		cats, ok := v.([]any)
		if !ok {
			return nil, fmt.Errorf(`expected a list but got "%v" of type %T`, v, v)
		}

		for _, v := range cats {
			cat, ok := v.(string)
			if !ok {
				return nil, fmt.Errorf(`expected a string but got "%v" of type %T`, v, v)
			}

			log.Printf(`[operation.Categorize] found item with category "%s"`, cat)

			categories[cat] = append(categories[cat], content)
		}
	}

	result := []cabret.Content{}
	for name, contents := range categories {
		log.Printf(`[operation.Categorize] generating category with %v item(s)`, len(contents))

		result = append(result, cabret.Content{
			Type: "metadata-only",
			Metadata: cabret.Map{
				op.CategoryVariable: name,
				"Items":             contents,
			},
		})
	}

	return result, nil
}