gen

type CodeGenerator struct {
	ServiceGenerator api.ServiceGenerator
}

type Generator interface {
	// MangleType generates a unique name for the given type that may be used
	// in private helper types and functions. This name is unique across all
	// types in the current context including imported types.
	MangleType(spec compile.TypeSpec) string

	// TextTemplate renders the given template with the given template
	// context.
	TextTemplate(s string, data interface{}, opts ...TemplateOption) (string, error)

	// DeclareFromTemplate renders the given template, and includes the
	// declarations from the generated Go code in the package being generated.
	// The next call to Write will write these declarations and any other
	// declarations that have not been written so far.
	DeclareFromTemplate(s string, data interface{}, opts ...TemplateOption) error

	// EnsureDeclared is similar to DeclareFromTemplate except that it simply
	// ignores conflicting definitions.
	EnsureDeclared(s string, data interface{}, opts ...TemplateOption) error

	// LookupTypeName returns the fully qualified name that should be used for
	// the given Thrift type. It imports the corresponding Go package if
	// necessary.
	//
	// This must be called with user-defined types only.
	LookupTypeName(compile.TypeSpec) (string, error)

	// LookupConstantName returns the fully qualified name that should be used
	// for the given Thrift constant. It imports the corresponding Go package if
	// necessary.
	LookupConstantName(*compile.Constant) (string, error)

	// Import ensures that the given package has been imported in the generated
	// code. Returns the name that should be used to reference the imported
	// module.
	Import(path string) string

	// Write the generated code to the given Writer and start a new file for
	// this package. All consecutive calls to DeclareFromTemplate will be
	// accumulated until the next Write call.
	//
	// A single Write must corresponds to a single file in the generated
	// package.
	//
	// The FileSet argument is deprecated and will be ignored.
	Write(w io.Writer, _ *token.FileSet) error
}

type GeneratorOptions struct {
	Importer    ThriftPackageImporter
	ImportPath  string
	PackageName string

	NoZap                 bool
	EnumTextMarshalStrict bool
}

type Namespace interface {
	// Generates a new name based on the given name. Prefers the provided name
	// but the name may be altered if it conflicts with another name in the
	// current scope.
	//
	// The returned name is also reserved so that future names cannot conflict
	// with it.
	NewName(name string) string

	// Reserve the given name in this namespace. Fails with an error if the name
	// is already taken.
	Reserve(name string) error

	// Forget the given name. Nothing happens if this name wasn't already
	// taken in this namespace. The name will NOT be removed from parent
	// namespaces.
	Forget(name string)

	// Create a new Child namespace. The child namespace cannot use any names
	// defined in this namespace or any of its parent namespaces.
	Child() Namespace

	// Rotate returns the oldest known name that was used for the given
	// requested name, and moves it to the end of the list of names used for
	// that requested name.
	//
	// This may be used to access the names used in a template dynamically in a
	// deterministic order if recording them in local variables is not possible.
	Rotate(name string) (string, error)
}

type Options struct {
	// OutputDir is the directory into which all generated code is written.
	//
	// This must be an absolute path.
	OutputDir string

	// PackagePrefix controls the import path prefix for all generated
	// packages.
	PackagePrefix string

	// ThriftRoot is the directory within whose tree all Thrift files consumed
	// are contained. The locations of the Thrift files relative to the
	// ThriftFile determines the module structure in OutputDir.
	//
	// This must be an absolute path.
	ThriftRoot string

	// NoRecurse determines whether code should be generated for included Thrift
	// files as well. If true, code gets generated only for the first module.
	NoRecurse bool

	// If true, we will not generate versioncheck.go files.
	NoVersionCheck bool

	// Code generation plugin
	Plugin CodeGenerator

	// Do not generate types.go
	NoTypes bool

	// Do not generate constants.go
	NoConstants bool

	// Do not generate service helpers
	NoServiceHelpers bool

	// Do not embed IDLs in generated code
	NoEmbedIDL bool

	// Do not generate Zap logging code
	NoZap bool

	// Name of the file to be generated by ThriftRW.
	OutputFile string

	// Generates an error on MarshalText and MarshalJSON if the enum value is
	// unrecognized.
	EnumTextMarshalStrict bool
}

type StreamGenerator struct {
	// contains filtered or unexported fields
}

type TemplateOption func(Generator, *template.Template) *template.Template

type ThriftPackageImporter interface {
	// RelativePackage returns the import path for the top-level package of the
	// given Thrift file relative to the ImportPrefix.
	RelativePackage(file string) (importPath string, err error)

	// RelativePackage returns the relative path of the given Thrift file
	// relative to the ThriftRoot.
	RelativeThriftFilePath(file string) (relPath string, err error)

	// Package returns the import path for the top-level package of the given Thrift
	// file.
	Package(file string) (importPath string, err error)
}

type WireGenerator struct {
	// contains filtered or unexported fields
}