feat: use less memory for builtin arguments (#103)

* feat: try to use less memory for builtin arguments

* fix(lint): hoist nil check to for loop

Signed-off-by: Chris Gianelloni <[email protected]>

---------

Signed-off-by: Chris Gianelloni <[email protected]>
Co-authored-by: Chris Gianelloni <[email protected]>

Author: rvcas

builtin/default_function.go

@@ -253,7 +253,7 @@ var Builtins map[string]DefaultFunction = map[string]DefaultFunction{
 	"indexArray":    IndexArray,
 }
 
-var defaultFunctionForceCount = [TotalBuiltinCount]int{
+var defaultFunctionForceCount = [TotalBuiltinCount]uint{
 	// Integer functions
 	AddInteger:            0,
 	SubtractInteger:       0,
@@ -369,11 +369,11 @@ var defaultFunctionForceCount = [TotalBuiltinCount]int{
 	IndexArray:    1,
 }
 
-func (f DefaultFunction) ForceCount() int {
+func (f DefaultFunction) ForceCount() uint {
 	return defaultFunctionForceCount[f]
 }
 
-var defaultFunctionArity = [TotalBuiltinCount]int{
+var defaultFunctionArity = [TotalBuiltinCount]uint{
 	// Integer functions
 	AddInteger:            2,
 	SubtractInteger:       2,
@@ -489,7 +489,7 @@ var defaultFunctionArity = [TotalBuiltinCount]int{
 	IndexArray:    2,
 }
 
-func (f DefaultFunction) Arity() int {
+func (f DefaultFunction) Arity() uint {
 	return defaultFunctionArity[f]
 }
 

cek/builtin_args.go

@@ -0,0 +1,57 @@
+package cek
+
+import (
+	"iter"
+
+	"github.com/blinklabs-io/plutigo/syn"
+)
+
+type argHolder[T syn.Eval] [6]Value[T]
+
+func newArgHolder[T syn.Eval]() argHolder[T] {
+	return argHolder[T]([6]Value[T]{})
+}
+
+type BuiltinArgs[T syn.Eval] struct {
+	data Value[T]
+	next *BuiltinArgs[T]
+}
+
+func (b *BuiltinArgs[T]) Iter() iter.Seq[Value[T]] {
+	return func(yield func(Value[T]) bool) {
+		temp := b
+		acc := []Value[T]{}
+
+		for temp != nil {
+
+			acc = append(acc, temp.data)
+
+			temp = temp.next
+		}
+
+		for i := len(acc) - 1; i >= 0; i-- {
+			if !yield(acc[i]) {
+				return
+			}
+		}
+	}
+}
+
+func (b *BuiltinArgs[T]) Extend(data Value[T]) *BuiltinArgs[T] {
+	return &BuiltinArgs[T]{
+		data: data,
+		next: b,
+	}
+}
+
+func (b *BuiltinArgs[T]) Extract(holder *argHolder[T], count uint) {
+	temp := b
+
+	for temp != nil {
+
+		holder[count-1] = temp.data
+
+		temp = temp.next
+		count--
+	}
+}