dolt/codegen/nomwriter.go

package gen

import (
	"io"

	. "github.com/attic-labs/noms/dbg"
	"github.com/clipperhouse/typewriter"
)

var (
	templates = typewriter.TemplateSlice{}
)

func init() {
	Chk.NoError(typewriter.Register(&nomWriter{}))
}

type nomWriter struct{}

func (nw *nomWriter) Name() string {
	return "noms"
}

func (nw *nomWriter) Imports(t typewriter.Type) []typewriter.ImportSpec {
	return []typewriter.ImportSpec{}
}

func (nw *nomWriter) Write(w io.Writer, typ typewriter.Type) error {
	tag, found := typ.FindTag(nw)

	if !found {
		return nil
	}

	tmpl, err := templates.ByTag(typ, tag)
	if err != nil {
		return err
	}

	w.Write([]byte(`
// DO NOT EDIT
//
// This file was generated by a tool.
// See http://clipperhouse.github.io/gen for details.
`))

	if err := tmpl.Execute(w, typ); err != nil {
		return err
	}
	return nil
}