AST.go

type IfStatNode struct {
	Condition  Node
	ThenBranch Node
	ElseBranch Node
}

func (n *IfStatNode) Interpret(i *Interpreter) (Node, error) {
	conditionNode, err := n.Condition.Interpret(i)
	if err != nil {
		return nil, err
	}

	conditional, ok := conditionNode.(*BoolLiteral)

	if !ok {
		return nil, fmt.Errorf("expected conditional to be boolean literal, got %T", conditionNode)
	}

	var node Node
	var e error
	if conditional.Value {
		node, e = n.ThenBranch.Interpret(i)
	} else if n.ElseBranch != nil {
		node, e = n.ElseBranch.Interpret(i)
	}

	return node, e
}