mirror of
https://github.com/MontFerret/ferret.git
synced 2024-12-23 01:24:30 +02:00
65 lines
1.3 KiB
Go
65 lines
1.3 KiB
Go
package objects
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/MontFerret/ferret/pkg/runtime/core"
|
|
"github.com/MontFerret/ferret/pkg/runtime/values"
|
|
"github.com/MontFerret/ferret/pkg/runtime/values/types"
|
|
)
|
|
|
|
// KeepKeys 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 KeepKeys(_ 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], types.Object)
|
|
|
|
if err != nil {
|
|
return values.None, err
|
|
}
|
|
|
|
keys := values.NewArrayWith(args[1:]...)
|
|
|
|
if len(args) == 2 && args[1].Type().Equals(types.Array) {
|
|
keys = args[1].(*values.Array)
|
|
}
|
|
|
|
err = validateArrayOf(types.String, 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
|
|
|
|
keys.ForEach(func(keyVal core.Value, idx int) bool {
|
|
key = keyVal.(values.String)
|
|
|
|
if val, exists = obj.Get(key); exists {
|
|
cloneable, ok := val.(core.Cloneable)
|
|
|
|
if ok {
|
|
val = cloneable.Clone()
|
|
}
|
|
|
|
resultObj.Set(key, val)
|
|
}
|
|
|
|
return true
|
|
})
|
|
|
|
return resultObj, nil
|
|
}
|