expr/operator-builtin.go

// Copyright (c) 2024 Celestino Amoroso (celestino.amoroso@gmail.com).
// All rights reserved.

// operator-builtin.go
package expr

import "io"

//-------- builtin term

func newBuiltinTerm(tk *Token) (inst *term) {
	return &term{
		tk:       *tk,
		children: make([]*term, 0, 1),
		position: posPrefix,
		priority: priSign,
		evalFunc: evalBuiltin,
	}
}

func evalBuiltin(ctx ExprContext, self *term) (v any, err error) {
	var childValue any

	if childValue, err = self.evalPrefix(ctx); err != nil {
		return
	}

	count := 0
	if isString(childValue) {
		module, _ := childValue.(string)
		count, err = ImportInContextByGlobPattern(ctx, module)
	} else {
		var moduleSpec any
		it := NewAnyIterator(childValue)
		for moduleSpec, err = it.Next(); err == nil; moduleSpec, err = it.Next() {
			if module, ok := moduleSpec.(string); ok {
				if ImportInContext(ctx, module) {
					count++
				} else {
					err = self.Errorf("unknown module %q", module)
					break
				}
			} else {
				err = self.Errorf("expected string at item nr %d, got %T", it.Index()+1, moduleSpec)
				break
			}
		}
		if err == io.EOF {
			err = nil
		}
	}
	if err == nil {
		v = int64(count)
	}
	return
}

// init
func init() {
	registerTermConstructor(SymKwBuiltin, newBuiltinTerm)
}