avo/internal/gen/gen.go

package gen

import (
	"go/format"

	"github.com/mmcloughlin/avo/internal/inst"
	"github.com/mmcloughlin/avo/printer"
)

// Interface of an instruction code generator.
type Interface interface {
	Generate([]inst.Instruction) ([]byte, error)
}

// Func adapts a function to Interface.
type Func func([]inst.Instruction) ([]byte, error)

// Generate calls f.
func (f Func) Generate(is []inst.Instruction) ([]byte, error) {
	return f(is)
}

// Builder constructs a code generator.
type Builder func(printer.Config) Interface

// GoFmt formats Go code produced from the given generator.
func GoFmt(i Interface) Interface {
	return Func(func(is []inst.Instruction) ([]byte, error) {
		b, err := i.Generate(is)
		if err != nil {
			return nil, err
		}
		formatted, err := format.Source(b)
		if err != nil {
			return b, err
		}
		return formatted, nil
	})
}