expr_logical.go

package rel

import (
	"context"
	"fmt"

	"github.com/arr-ai/wbnf/parser"
)

type AndExpr struct {
	ExprScanner
	a, b Expr
}

func NewAndExpr(scanner parser.Scanner, a, b Expr) Expr {
	return AndExpr{ExprScanner: ExprScanner{Src: scanner}, a: a, b: b}
}

func (e AndExpr) String() string {
	return fmt.Sprintf("(%v) && (%v)", e.a, e.b)
}

func (e AndExpr) Eval(ctx context.Context, local Scope) (Value, error) {
	a, err := e.a.Eval(ctx, local)
	if err != nil {
		return nil, WrapContextErr(err, e, local)
	}
	if !a.IsTrue() {
		return a, nil
	}

	b, err := e.b.Eval(ctx, local)
	if err != nil {
		return nil, WrapContextErr(err, e, local)
	}

	return b, nil
}

type OrExpr struct {
	ExprScanner
	a, b Expr
}

func NewOrExpr(scanner parser.Scanner, a, b Expr) Expr {
	return OrExpr{ExprScanner: ExprScanner{Src: scanner}, a: a, b: b}
}

func (e OrExpr) String() string {
	return fmt.Sprintf("(%v) || (%v)", e.a, e.b)
}

func (e OrExpr) Eval(ctx context.Context, local Scope) (Value, error) {
	a, err := e.a.Eval(ctx, local)
	if err != nil {
		return nil, WrapContextErr(err, e, local)
	}
	if a.IsTrue() {
		return a, nil
	}

	b, err := e.b.Eval(ctx, local)
	if err != nil {
		return nil, WrapContextErr(err, e, local)
	}

	return b, nil
}