ferret/pkg/stdlib/objects/keep.go

package objects

import (
	"context"

	"github.com/MontFerret/ferret/pkg/runtime/core"
	"github.com/MontFerret/ferret/pkg/runtime/values"
)

/*
 * Returns a new object with only given keys.
 * @params src (Object) - source object.
 * @params keys (Array Of String OR Strings) - keys that need to be keeped.
 * @returns (Object) - New Object with only given keys.
 */
func Keep(_ context.Context, args ...core.Value) (core.Value, error) {
	err := core.ValidateArgs(args, 2, core.MaxArgs)

	if err != nil {
		return values.None, err
	}

	err = core.ValidateType(args[0], core.ObjectType)

	if err != nil {
		return values.None, err
	}

	keys := values.NewArrayWith(args[1:]...)

	if len(args) == 2 && args[1].Type() == core.ArrayType {
		keys = args[1].(*values.Array)
	}

	err = validateArrayOfStrings(keys)

	if err != nil {
		return values.None, err
	}

	obj := args[0].(*values.Object)
	resultObj := values.NewObject()

	var key values.String
	var val core.Value
	var exists values.Boolean

	for idx := values.NewInt(0); idx < keys.Length(); idx++ {
		key = keys.Get(idx).(values.String)
		if val, exists = obj.Get(key); exists {
			resultObj.Set(key, val)
		}
	}

	return resultObj, nil
}

func validateArrayOfStrings(arr *values.Array) (err error) {
	for idx := values.NewInt(0); idx < arr.Length(); idx++ {
		err = core.ValidateType(arr.Get(idx), core.StringType)
		if err != nil {
			break
		}
	}
	return
}