ferret/pkg/runtime/expressions/param.go

package expressions

import (
	"context"
	"github.com/MontFerret/ferret/pkg/runtime/collections"
	"github.com/MontFerret/ferret/pkg/runtime/core"
	"github.com/MontFerret/ferret/pkg/runtime/values"
)

type ParameterExpression struct {
	src  core.SourceMap
	name string
}

func NewParameterExpression(src core.SourceMap, name string) (*ParameterExpression, error) {
	if name == "" {
		return nil, core.Error(core.ErrMissedArgument, "name")
	}

	return &ParameterExpression{src, name}, nil
}

func (e *ParameterExpression) Iterate(ctx context.Context, scope *core.Scope) (collections.Iterator, error) {
	value, err := e.Exec(ctx, scope)

	if err != nil {
		return nil, core.SourceError(e.src, err)
	}

	iter, err := collections.ToIterator(value)

	if err != nil {
		return nil, core.SourceError(e.src, err)
	}

	return iter, nil
}

func (e *ParameterExpression) Exec(ctx context.Context, scope *core.Scope) (core.Value, error) {
	param, err := core.ParamFrom(ctx, e.name)

	if err != nil {
		return values.None, core.SourceError(e.src, err)
	}

	return param, nil
}