You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
100 lines
1.9 KiB
Go
100 lines
1.9 KiB
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
|
|
}
|