Add templateDir variable

encoder.go

@@ -18,6 +18,7 @@ type GenericTemplateBasedEncoder struct {
 	service        *descriptor.ServiceDescriptorProto
 	file           *descriptor.FileDescriptorProto
 	debug          bool
+	destinationDir string
 }
 
 type Ast struct {
@@ -27,6 +28,7 @@ type Ast struct {
 	GoPWD          string                             `json:"go-pwd,omitempty"`
 	PWD            string                             `json:"pwd"`
 	Debug          bool                               `json:"debug"`
+	DestinationDir string                             `json:"destination-dir"`
 	File           *descriptor.FileDescriptorProto    `json:"file"`
 	RawFilename    string                             `json:"raw-filename"`
 	Filename       string                             `json:"filename"`
@@ -35,12 +37,13 @@ type Ast struct {
 	Environment    []string                           `json:"environment"`
 }
 
-func NewGenericTemplateBasedEncoder(templateDir string, service *descriptor.ServiceDescriptorProto, file *descriptor.FileDescriptorProto, debug bool) (e *GenericTemplateBasedEncoder) {
+func NewGenericTemplateBasedEncoder(templateDir string, service *descriptor.ServiceDescriptorProto, file *descriptor.FileDescriptorProto, debug bool, destinationDir string) (e *GenericTemplateBasedEncoder) {
 	e = &GenericTemplateBasedEncoder{
 		service:        service,
 		file:           file,
 		templateDir:    templateDir,
 		debug:          debug,
+		destinationDir: destinationDir,
 	}
 
 	if debug {
@@ -95,6 +98,7 @@ func (e *GenericTemplateBasedEncoder) genAst(templateFilename string) (*Ast, err
 		GoPWD:          goPwd,
 		File:           e.file,
 		TemplateDir:    e.templateDir,
+		DestinationDir: e.destinationDir,
 		RawFilename:    templateFilename,
 		Filename:       "",
 		Environment:    os.Environ(),

main.go

@@ -30,6 +30,7 @@ func main() {
 
 	// Parse parameters
 	templateDir := "./templates"
+	destinationDir := "."
 	debug := false
 	if parameter := g.Request.GetParameter(); parameter != "" {
 		for _, param := range strings.Split(parameter, ",") {
@@ -42,6 +43,9 @@ func main() {
 			case "template_dir":
 				templateDir = parts[1]
 				break
+			case "destination_dir":
+				destinationDir = parts[1]
+				break
 			case "debug":
 				switch strings.ToLower(parts[1]) {
 				case "true", "t":
@@ -60,7 +64,7 @@ func main() {
 	// Generate the encoders
 	for _, file := range g.Request.GetProtoFile() {
 		for _, service := range file.GetService() {
-			encoder := NewGenericTemplateBasedEncoder(templateDir, service, file, debug)
+			encoder := NewGenericTemplateBasedEncoder(templateDir, service, file, debug, destinationDir)
 			g.Response.File = append(g.Response.File, encoder.Files()...)
 		}
 	}