package proto

Pipello · Oct 10, 2024 · 308a102 · 308a102
1 parent fed350f
commit 308a102
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 8 deletions.
diff --git a/definition/model_definition.go b/definition/model_definition.go
@@ -226,10 +226,7 @@ type CompleteGenerator struct {
 }
 
 func (m *CompleteGenerator) getProtoServiceFilePath() string {
-	if m.Config.ProtoBufFileName != "" {
-		return "./api/" + m.Config.ProtoBufFileName
-	}
-	return fmt.Sprintf("./api/%s.proto", strcase.ToSnake(m.Config.GRPCServiceName))
+	return fmt.Sprintf("./api/%s.proto", strcase.ToSnake(m.Config.ServiceName))
 }
 
 func (m *CompleteGenerator) readCustomBlocks() {
@@ -301,9 +298,9 @@ func getFilePath(name string) string {
 }
 
 type Config struct {
-	ProtoBufFileName string
-	GRPCServiceName  string
-	RepositoryName   string
+	ServiceName     string
+	GRPCServiceName string
+	RepositoryName  string
 }
 
 type Generator interface {

diff --git a/definition/service.proto.tpl b/definition/service.proto.tpl
@@ -2,7 +2,7 @@
 
 syntax = "proto3";
 
-package main;
+package {{.Config.ServiceName}};
 
 option go_package = "{{.Config.RepositoryName}}/api";