blob: 0907df0248d29d35dabc122abaae124c66e81fbb [file] [log] [blame]
// Copyright 2023 The Go 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 main
import (
// config holds the configuration for a package.
type config struct {
Package string
ProtoImportPath string `yaml:"protoImportPath"`
// Import path for the support package needed by the generated code.
SupportImportPath string `yaml:"supportImportPath"`
// The types to process. Only these types and the types they depend
// on will be output.
// The key is the name of the proto type.
Types map[string]*typeConfig
// Omit the types in this list, even if they would normally be output.
// Elements can be globs.
OmitTypes []string `yaml:"omitTypes"`
// Converter functions for types not in the proto package.
// Each value should be "tofunc, fromfunc"
Converters map[string]string
type typeConfig struct {
// The name for the veneer type, if different.
Name string
// The prefix of the proto enum values. It will be removed.
ProtoPrefix string `yaml:"protoPrefix"`
// The prefix for the veneer enum values, if different from the type name.
VeneerPrefix string `yaml:"veneerPrefix"`
// Overrides for enum values.
ValueNames map[string]string `yaml:"valueNames"`
// Overrides for field types. Map key is proto field name.
Fields map[string]fieldConfig
// Custom conversion functions: "tofunc, fromfunc"
ConvertToFrom string `yaml:"convertToFrom"`
// Doc string for the type, omitting the initial type name.
Doc string
// Verb to place after type name in doc. Default: "is".
// Ignored if Doc is non-empty.
DocVerb string `yaml:"docVerb"`
type fieldConfig struct {
Name string // veneer name
Type string // veneer type
// Omit from output.
Omit bool
func (c *config) init() {
for protoName, tc := range c.Types {
if tc == nil {
tc = &typeConfig{Name: protoName}
c.Types[protoName] = tc
if tc.Name == "" {
tc.Name = protoName
func (tc *typeConfig) init() {
if tc.VeneerPrefix == "" {
tc.VeneerPrefix = tc.Name
func readConfigFile(filename string) (*config, error) {
if filename == "" {
return nil, errors.New("missing config file")
f, err := os.Open(filename)
if err != nil {
return nil, err
defer f.Close()
dec := yaml.NewDecoder(f)
var c config
if err := dec.Decode(&c); err != nil {
return nil, fmt.Errorf("reading %s: %w", filename, err)
return &c, nil