| // Copyright 2018 The Fuchsia Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| package codegen |
| |
| import ( |
| "embed" |
| "go/format" |
| "text/template" |
| |
| "go.fuchsia.dev/fuchsia/tools/fidl/lib/fidlgen" |
| ) |
| |
| //go:embed *.tmpl |
| var templates embed.FS |
| |
| type formatter struct{} |
| |
| func (f formatter) Format(source []byte) ([]byte, error) { |
| return format.Source(source) |
| } |
| |
| var gofmt fidlgen.Formatter = formatter{} |
| |
| type Generator struct { |
| *fidlgen.Generator |
| } |
| |
| func NewGenerator() Generator { |
| return Generator{fidlgen.NewGenerator("GoTemplates", templates, gofmt, |
| template.FuncMap{})} |
| } |
| |
| func (gen Generator) GenerateImplFile(tree Root, filename string) error { |
| return gen.GenerateFile(filename, "GenerateLibraryFile", tree) |
| } |
| |
| func (gen *Generator) GeneratePkgNameFile(tree Root, filename string) error { |
| return fidlgen.WriteFileIfChanged(filename, []byte(tree.PackageName)) |
| } |