Fix potential newlines in version file breaking code generation

loopholelabs · Apr 4, 2024 · 43d41f9 · 43d41f9
1 parent 49e5f5f
commit 43d41f9
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/pkg/generator/generator.go b/pkg/generator/generator.go
@@ -18,6 +18,7 @@ package generator
 
 import (
 	"bytes"
+	"strings"
 	"text/template"
 
 	generator "github.com/loopholelabs/polyglot/generator/golang"
@@ -121,7 +122,7 @@ func (g *Generator) Generate(req *pluginpb.CodeGeneratorRequest) (res *pluginpb.
 		numStreamMethods, numMethods := extractNumberOfMethods(f)
 
 		err = templ.ExecuteTemplate(genFile, "prebase.templ", map[string]interface{}{
-			"pluginVersion":       version.Version,
+			"pluginVersion":       strings.TrimSpace(version.Version),
 			"sourcePath":          f.Desc.Path(),
 			"package":             packageName,
 			"requiredImports":     requiredImports,

diff --git a/pkg/generator/test/test.frpc.go b/pkg/generator/test/test.frpc.go