generator

type Config struct {
	Verbose         bool
	GenerateMain    bool // whether stuff in the main package should be generated
	GenerateInvoker bool // generate main.go with invoker when main package generate
	Version         string
	NoFastAPI       bool
	ModuleName      string
	ServiceName     string
	Use             string
	IDLType         string
	Includes        util.StringSlice
	ThriftOptions   util.StringSlice
	ProtobufOptions util.StringSlice
	IDL             string // the IDL file passed on the command line
	OutputPath      string // the output path for main pkg and kitex_gen
	PackagePrefix   string
	CombineService  bool // combine services to one service
	CopyIDL         bool
	Features        []feature
}

type File struct {
	Name    string
	Content string
}

type Generator interface {
	GenerateService(pkg *PackageInfo) ([]*File, error)
	GenerateMainPackage(pkg *PackageInfo) ([]*File, error)
}

type HandleFunc func(*Task, *PackageInfo) (*File, error)

type MethodInfo struct {
	PkgInfo
	ServiceName            string
	Name                   string
	RawName                string
	Oneway                 bool
	Void                   bool
	Args                   []*Parameter
	Resp                   *Parameter
	Exceptions             []*Parameter
	ArgStructName          string
	ResStructName          string
	IsResponseNeedRedirect bool // int -> int*
	GenArgResultStruct     bool
	ClientStreaming        bool
	ServerStreaming        bool
}

type Middleware func(HandleFunc) HandleFunc

type PackageInfo struct {
	Namespace    string            // a dot-separated string for generating service package under kitex_gen
	Dependencies map[string]string // package name => import path, used for searching imports
	*ServiceInfo                   // the target service

	// the belowing fields will be filled and used by the gnerator
	Codec            string
	Version          string
	RealServiceName  string
	Imports          map[string]string // import path => alias
	ExternalKitexGen string
	Features         []feature
}

type Parameter struct {
	Deps    []PkgInfo
	Name    string
	RawName string
	Type    string // *PkgA.StructB
}

type PkgInfo struct {
	PkgRefName string
	ImportPath string
}

type ServiceInfo struct {
	PkgInfo
	ServiceName     string
	RawServiceName  string
	ServiceTypeName func() string
	Base            *ServiceInfo
	Methods         []*MethodInfo
	CombineServices []*ServiceInfo
	HasStreaming    bool
}

type Task struct {
	Name string
	Path string
	Text string
	*template.Template
}