ucl/ucl/objs.go

package ucl

import (
	"context"
	"errors"
	"fmt"
	"reflect"
	"strconv"
	"strings"
	"time"

	"github.com/lmika/gopkgs/fp/slices"
)

type Object interface {
	String() string
	Truthy() bool
}

type Listable interface {
	Len() int
	Index(i int) Object
}

type Iterable interface {
	HasNext() bool

	// Next returns the next object from the iterable if one exists, otherwise
	// returns nil, false.
	Next(ctx context.Context) (Object, error)
}

type ModListable interface {
	Listable

	// Insert adds a new item to the list. idx can be a positive
	// number from 0 to len(), in which case the object will be inserted
	// at that position.  If idx is negative, then the item will be inserted
	// at that position from the right.
	Insert(idx int, obj Object) error
}

type Hashable interface {
	Len() int
	Value(k string) Object
	Each(func(k string, v Object) error) error
}

type ListObject []Object

func NewListObject() *ListObject {
	return &ListObject{}
}

func (lo *ListObject) Append(o Object) {
	*lo = append(*lo, o)
}

func (lo *ListObject) Insert(idx int, obj Object) error {
	if idx != -1 {
		return errors.New("not supported")
	}
	*lo = append(*lo, obj)
	return nil
}

func (s *ListObject) String() string {
	return fmt.Sprintf("%v", []Object(*s))
}

func (s *ListObject) Truthy() bool {
	return len(*s) > 0
}

func (s *ListObject) Len() int {
	return len(*s)
}

func (s *ListObject) Index(i int) Object {
	return (*s)[i]
}

type iteratorObject struct {
	Iterable
}

func (i iteratorObject) String() string {
	return "iterator{}"
}

func (i iteratorObject) Truthy() bool {
	return i.Iterable.HasNext()
}

type hashObject map[string]Object

func (s hashObject) String() string {
	if len(s) == 0 {
		return "[:]"
	}

	sb := strings.Builder{}
	sb.WriteString("[")
	for k, v := range s {
		if sb.Len() != 1 {
			sb.WriteString(" ")
		}
		sb.WriteString(k)
		sb.WriteString(":")
		sb.WriteString(v.String())
	}
	sb.WriteString("]")
	return sb.String()
}

func (s hashObject) Truthy() bool {
	return len(s) > 0
}

func (s hashObject) Len() int {
	return len(s)
}

func (s hashObject) Value(k string) Object {
	return s[k]
}

func (s hashObject) Each(fn func(k string, v Object) error) error {
	for k, v := range s {
		if err := fn(k, v); err != nil {
			return err
		}
	}
	return nil
}

type StringObject string

func (s StringObject) String() string {
	return string(s)
}

func (s StringObject) Truthy() bool {
	return string(s) != ""
}

type IntObject int

func (i IntObject) String() string {
	return strconv.Itoa(int(i))
}

func (i IntObject) Truthy() bool {
	return i != 0
}

type boolObject bool

func (b boolObject) String() string {
	if b {
		return "true"
	}
	return "false"
}

func (b boolObject) Truthy() bool {
	return bool(b)
}

type timeObject time.Time

func (t timeObject) String() string {
	return time.Time(t).Format(time.RFC3339)
}

func (t timeObject) Truthy() bool {
	return !time.Time(t).IsZero()
}

func toGoValue(obj Object) (interface{}, bool) {
	switch v := obj.(type) {
	case nil:
		return nil, true
	case StringObject:
		return string(v), true
	case IntObject:
		return int(v), true
	case boolObject:
		return bool(v), true
	case timeObject:
		return time.Time(v), true
	case *ListObject:
		xs := make([]interface{}, 0, len(*v))
		for _, va := range *v {
			x, ok := toGoValue(va)
			if !ok {
				continue
			}
			xs = append(xs, x)
		}
		return xs, true
	case hashObject:
		xs := make(map[string]interface{})
		for k, va := range v {
			x, ok := toGoValue(va)
			if !ok {
				continue
			}
			xs[k] = x
		}
		return xs, true
	case iteratorObject:
		return v.Iterable, true
	case Iterable:
		return v, true
	case proxyObject:
		return v.p, true
	case listableProxyObject:
		return v.orig.Interface(), true
	case structProxyObject:
		return v.orig.Interface(), true
	}

	return nil, false
}

func fromGoValue(v any) (Object, error) {
	switch t := v.(type) {
	case Object:
		return t, nil
	case Iterable:
		return iteratorObject{t}, nil
	case nil:
		return nil, nil
	case string:
		return StringObject(t), nil
	case int:
		return IntObject(t), nil
	case bool:
		return boolObject(t), nil
	case time.Time:
		return timeObject(t), nil
	}

	return fromGoReflectValue(reflect.ValueOf(v))
}

func fromGoReflectValue(resVal reflect.Value) (Object, error) {
	if !resVal.IsValid() {
		return nil, nil
	}

	switch resVal.Kind() {
	case reflect.Slice:
		return listableProxyObject{v: resVal, orig: resVal}, nil
	case reflect.Struct:
		return newStructProxyObject(resVal, resVal), nil
	case reflect.Pointer:
		switch resVal.Elem().Kind() {
		case reflect.Slice:
			return listableProxyObject{v: resVal.Elem(), orig: resVal}, nil
		case reflect.Struct:
			return newStructProxyObject(resVal.Elem(), resVal), nil
		}

		return fromGoReflectValue(resVal.Elem())
	}

	return proxyObject{resVal.Interface()}, nil
}

type macroArgs struct {
	eval     evaluator
	ec       *evalCtx
	hasPipe  bool
	pipeArg  Object
	ast      *astCmd
	argShift int
}

func (ma macroArgs) nargs() int {
	return len(ma.ast.InvokeArgs[ma.argShift:])
}

func (ma *macroArgs) shift(n int) {
	ma.argShift += n
}

func (ma macroArgs) identIs(ctx context.Context, n int, expectedIdent string) bool {
	if n >= len(ma.ast.InvokeArgs[ma.argShift:]) {
		return false
	}

	if len(ma.ast.InvokeArgs[ma.argShift+n].DotSuffix) != 0 {
		return false
	}

	lit := ma.ast.InvokeArgs[ma.argShift+n].Arg.Ident
	if lit == nil {
		return false
	}

	return lit.String() == expectedIdent
}

func (ma *macroArgs) shiftIdent(ctx context.Context) (string, bool) {
	if ma.argShift >= len(ma.ast.InvokeArgs) {
		return "", false
	}

	if len(ma.ast.InvokeArgs[ma.argShift].DotSuffix) != 0 {
		return "", false
	}

	lit := ma.ast.InvokeArgs[ma.argShift].Arg.Ident
	if lit != nil {
		ma.argShift += 1
		return lit.String(), true
	}
	return "", false
}

func (ma macroArgs) evalArg(ctx context.Context, n int) (Object, error) {
	if n >= len(ma.ast.InvokeArgs[ma.argShift:]) {
		return nil, errors.New("not enough arguments") // FIX
	}

	return ma.eval.evalDot(ctx, ma.ec, ma.ast.InvokeArgs[ma.argShift+n])
}

func (ma macroArgs) evalBlock(ctx context.Context, n int, args []Object, pushScope bool) (Object, error) {
	obj, err := ma.evalArg(ctx, n)
	if err != nil {
		return nil, err
	}

	switch v := obj.(type) {
	case blockObject:
		ec := ma.ec
		if pushScope {
			ec = ec.fork()
		}
		for i, n := range v.block.Names {
			if i < len(args) {
				ec.setOrDefineVar(n, args[i])
			}
		}

		return ma.eval.evalBlock(ctx, ec, v.block)
	case StringObject:
		iv := ma.ec.lookupInvokable(string(v))
		if iv == nil {
			return nil, errors.New("'" + string(v) + "' is not invokable")
		}

		return iv.invoke(ctx, invocationArgs{
			eval: ma.eval,
			inst: ma.eval.inst,
			ec:   ma.ec,
			args: args,
		})
	}

	return nil, errors.New("expected an invokable arg")
}

type errObject struct{ err error }

func (eo errObject) String() string {
	return "error:" + eo.err.Error()
}

func (eo errObject) Truthy() bool {
	return true
}

type invocationArgs struct {
	eval   evaluator
	inst   *Inst
	ec     *evalCtx
	args   []Object
	kwargs map[string]*ListObject
}

func (ia invocationArgs) expectArgn(x int) error {
	if len(ia.args) < x {
		return errors.New("expected at least " + strconv.Itoa(x) + " args")
	}
	return nil
}

func (ia invocationArgs) stringArg(i int) (string, error) {
	if len(ia.args) < i {
		return "", errors.New("expected at least " + strconv.Itoa(i) + " args")
	}
	s, ok := ia.args[i].(fmt.Stringer)
	if !ok {
		return "", errors.New("expected a string arg")
	}
	return s.String(), nil
}

func (ia invocationArgs) intArg(i int) (int, error) {
	if len(ia.args) < i {
		return 0, errors.New("expected at least " + strconv.Itoa(i) + " args")
	}

	switch v := ia.args[i].(type) {
	case IntObject:
		return int(v), nil
	default:
		return 0, errors.New("expected an int arg")
	}
}

func (ia invocationArgs) invokableArg(i int) (invokable, error) {
	if len(ia.args) < i {
		return nil, errors.New("expected at least " + strconv.Itoa(i) + " args")
	}

	switch v := ia.args[i].(type) {
	case invokable:
		return v, nil
	case StringObject:
		iv := ia.ec.lookupInvokable(string(v))
		if iv == nil {
			return nil, errors.New("'" + string(v) + "' is not invokable")
		}
		return iv, nil
	}
	return nil, errors.New("expected an invokable arg")
}

func (ia invocationArgs) fork(args []Object) invocationArgs {
	return invocationArgs{
		eval:   ia.eval,
		inst:   ia.inst,
		ec:     ia.ec,
		args:   args,
		kwargs: nil,
	}
}

func (ia invocationArgs) shift(i int) invocationArgs {
	if len(ia.args) < i {
		return ia
	}
	return invocationArgs{
		eval:   ia.eval,
		inst:   ia.inst,
		ec:     ia.ec,
		args:   ia.args[i:],
		kwargs: ia.kwargs,
	}
}

// invokable is an Object that can be executed as a command
type invokable interface {
	invoke(ctx context.Context, args invocationArgs) (Object, error)
}

type macroable interface {
	invokeMacro(ctx context.Context, args macroArgs) (Object, error)
}

type pipeInvokable interface {
	invokable
}

type invokableFunc func(ctx context.Context, args invocationArgs) (Object, error)

func (i invokableFunc) invoke(ctx context.Context, args invocationArgs) (Object, error) {
	return i(ctx, args)
}

type blockObject struct {
	block    *astBlock
	closedEC *evalCtx
}

func (bo blockObject) String() string {
	return "block"
}

func (bo blockObject) Truthy() bool {
	return len(bo.block.Statements) > 0
}

func (bo blockObject) invoke(ctx context.Context, args invocationArgs) (Object, error) {
	ec := bo.closedEC.fork()
	for i, n := range bo.block.Names {
		if i < len(args.args) {
			ec.setOrDefineVar(n, args.args[i])
		}
	}

	return args.eval.evalBlock(ctx, ec, bo.block)
}

type macroFunc func(ctx context.Context, args macroArgs) (Object, error)

func (i macroFunc) invokeMacro(ctx context.Context, args macroArgs) (Object, error) {
	return i(ctx, args)
}

func isTruthy(obj Object) bool {
	if obj == nil {
		return false
	}
	return obj.Truthy()
}

type proxyObject struct {
	p interface{}
}

func (p proxyObject) String() string {
	return fmt.Sprintf("proxyObject{%T}", p.p)
}

func (p proxyObject) Truthy() bool {
	return p.p != nil
}

type listableProxyObject struct {
	v    reflect.Value
	orig reflect.Value
}

func (p listableProxyObject) String() string {
	return fmt.Sprintf("listableProxyObject{%v}", p.v.Type())
}

func (p listableProxyObject) Truthy() bool {
	return p.v.Len() > 0
}

func (p listableProxyObject) Len() int {
	return p.v.Len()
}

func (p listableProxyObject) Index(i int) Object {
	e, err := fromGoValue(p.v.Index(i).Interface())
	if err != nil {
		return nil
	}
	return e
}

type structProxyObject struct {
	v    reflect.Value
	orig reflect.Value
	vf   []reflect.StructField
}

func newStructProxyObject(v reflect.Value, orig reflect.Value) structProxyObject {
	return structProxyObject{
		v:    v,
		orig: orig,
		vf:   slices.Filter(reflect.VisibleFields(v.Type()), func(t reflect.StructField) bool { return t.IsExported() }),
	}
}

func (s structProxyObject) String() string {
	return fmt.Sprintf("structProxyObject{%v}", s.v.Type())
}

func (s structProxyObject) Truthy() bool {
	return true
}

func (s structProxyObject) Len() int {
	return len(s.vf)
}

func (s structProxyObject) Value(k string) Object {
	f := s.v.FieldByName(k)
	if !f.IsValid() {
		return nil
	}

	if f.Kind() == reflect.Ptr {
		if f.IsNil() {
			return nil
		}
		f = f.Elem()
	}

	e, err := fromGoValue(f.Interface())
	if err != nil {
		return nil
	}
	return e
}

func (s structProxyObject) Each(fn func(k string, v Object) error) error {
	for _, f := range s.vf {
		v, err := fromGoValue(s.v.FieldByName(f.Name).Interface())
		if err != nil {
			v = nil
		}

		if err := fn(f.Name, v); err != nil {
			return err
		}
	}
	return nil
}

type OpaqueObject struct {
	v any
}

func Opaque(v any) OpaqueObject {
	return OpaqueObject{v: v}
}

func (p OpaqueObject) String() string {
	return fmt.Sprintf("opaque{%T}", p.v)
}

func (p OpaqueObject) Truthy() bool {
	return p.v != nil
}

type errBreak struct {
	isCont bool
	ret    Object
}

func (e errBreak) Error() string {
	if e.isCont {
		return "continue"
	}
	return "break"
}

type errReturn struct {
	ret Object
}

func (e errReturn) Error() string {
	return "return"
}

var ErrHalt = errors.New("halt")