vise/engine/loop.go

package engine

import (
	"bufio"
	"context"
	"fmt"
	"io"
	"log"
	"strings"
)

// Loop starts an engine execution loop with the given symbol as the starting node.
//
// The root reads inputs from the provided reader, one line at a time.
//
// It will execute until running out of bytecode in the buffer.
//
// Any error not handled by the engine will terminate the oop and return an error.
//
// Rendered output is written to the provided writer.
func Loop(en *Engine, startSym string, ctx context.Context, reader io.Reader, writer io.Writer) error {
	err := en.Init(startSym, ctx)
	if err != nil {
		return fmt.Errorf("cannot init: %v\n", err)
	}

	err = en.WriteResult(writer, ctx)
	if err != nil {
		return err
	}
	writer.Write([]byte{0x0a})

	running := true
	bufReader := bufio.NewReader(reader)
	for running {
		in, err := bufReader.ReadString('\n')
		if err == io.EOF {
			log.Printf("EOF found, that's all folks")
			return nil
		}
		if err != nil {
			return fmt.Errorf("cannot read input: %v\n", err)
		}
		in = strings.TrimSpace(in)
		running, err = en.Exec([]byte(in), ctx)
		if err != nil {
			return fmt.Errorf("unexpected termination: %v\n", err)
		}
		err = en.WriteResult(writer, ctx)
		if err != nil {
			return err
		}
		writer.Write([]byte{0x0a})

	}
	return nil
}