mirror of
https://github.com/MontFerret/ferret.git
synced 2025-07-07 00:56:53 +02:00
Feature/#9 array functions (#57)
* #9 Added 'APPEND' function * #9 Added 'FIRST' function * #9 Added 'FLATTEN' function * #9 Added 'INTERSECTION' function * #9 Added 'LAST' function * #9 Added 'MINUS' function * #9 Added 'NTH' function * #9 Added 'OUTERSECTION' function * #9 Added 'POP' function * #9 Added 'POSITION' function * #9 Added 'PUSH' function * Fixed nil pointer exception in value parser * #9 Added 'REMOVE_NTH' function * #9 Added 'REMOVE_VALUE' function * #9 Added 'REMOVE_VALUES' function * #9 Added 'REVERSE' function * #9 Added 'SHIFT' function * #9 Added 'SLICE' function * Removed meme * #9 Added 'SORTED' function * #9 Added SORTED_UNIQUE function * #9 Added 'UNION' function * #9 Added 'UNION_DISTINCT' function * #9 Added 'UNIQUE' function * #9 Added 'UNSHIFT' function * #9 Made more strict optional arg validation * #9 Fixed linting errors
This commit is contained in:
74
pkg/stdlib/arrays/append.go
Normal file
74
pkg/stdlib/arrays/append.go
Normal file
@ -0,0 +1,74 @@
|
||||
package arrays
|
||||
|
||||
import (
|
||||
"context"
|
||||
"github.com/MontFerret/ferret/pkg/runtime/core"
|
||||
"github.com/MontFerret/ferret/pkg/runtime/values"
|
||||
)
|
||||
|
||||
/*
|
||||
* Appends a new item to an array and returns a new array with a given element.
|
||||
* If ``uniqueOnly`` is set to true, then will add the item only if it's unique.
|
||||
* @param arr (Array) - Target array.
|
||||
* @param item (Value) - Target value to add.
|
||||
* @returns arr (Array) - New array.
|
||||
*/
|
||||
func Append(_ context.Context, args ...core.Value) (core.Value, error) {
|
||||
err := core.ValidateArgs(args, 2, 3)
|
||||
|
||||
if err != nil {
|
||||
return values.None, err
|
||||
}
|
||||
|
||||
err = core.ValidateType(args[0], core.ArrayType)
|
||||
|
||||
if err != nil {
|
||||
return values.None, err
|
||||
}
|
||||
|
||||
arr := args[0].(*values.Array)
|
||||
arg := args[1]
|
||||
unique := values.False
|
||||
|
||||
if len(args) > 2 {
|
||||
err = core.ValidateType(args[2], core.BooleanType)
|
||||
|
||||
if err != nil {
|
||||
return values.None, err
|
||||
}
|
||||
|
||||
unique = args[2].(values.Boolean)
|
||||
}
|
||||
|
||||
next := values.NewArray(int(arr.Length()) + 1)
|
||||
|
||||
if !unique {
|
||||
arr.ForEach(func(item core.Value, idx int) bool {
|
||||
next.Push(item)
|
||||
|
||||
return true
|
||||
})
|
||||
|
||||
next.Push(arg)
|
||||
|
||||
return next, nil
|
||||
}
|
||||
|
||||
hasDuplicate := false
|
||||
|
||||
arr.ForEach(func(item core.Value, idx int) bool {
|
||||
next.Push(item)
|
||||
|
||||
if !hasDuplicate {
|
||||
hasDuplicate = item.Compare(arg) == 0
|
||||
}
|
||||
|
||||
return true
|
||||
})
|
||||
|
||||
if !hasDuplicate {
|
||||
next.Push(arg)
|
||||
}
|
||||
|
||||
return next, nil
|
||||
}
|
Reference in New Issue
Block a user