file_tag_prefixer.go

package config

import (
	"fmt"
	"reflect"

	"github.com/fatih/structtag"
)

type fileTagPrefixer struct {
	prefix string
}

var _ TagModifier = &fileTagPrefixer{}

// NewFileTagPrefixer creates a new TagModifier which adds a prefix to the
// values of `file` tags. This can be used to register one config multiple
// times and have their initialization be read from different keys in a
// config file.
func NewFileTagPrefixer(prefix string) TagModifier {
	return &fileTagPrefixer{
		prefix: prefix,
	}
}

func (p *fileTagPrefixer) AlterFieldTag(fieldType reflect.StructField, tags *structtag.Tags) error {
	tag, err := tags.Get(FileTag)
	if err != nil {
		return nil
	}

	return tags.Set(&structtag.Tag{
		Key:  FileTag,
		Name: fmt.Sprintf("%s_%s", p.prefix, tag.Name),
	})
}