From 404b73e058019d33f7a6c5ac02e6277659f7764f Mon Sep 17 00:00:00 2001
From: Konrad Niemiec <konradjniemiec@gmail.com>
Date: Mon, 15 Jan 2024 18:11:33 -0800
Subject: [PATCH] works for some proto multining

---
 cmd/lekko/gen.go | 26 ++++++++++++++++----------
 1 file changed, 16 insertions(+), 10 deletions(-)

diff --git a/cmd/lekko/gen.go b/cmd/lekko/gen.go
index 3eb0d5a8..e60cc446 100644
--- a/cmd/lekko/gen.go
+++ b/cmd/lekko/gen.go
@@ -35,6 +35,7 @@ import (
 	"golang.org/x/mod/modfile"
 	"google.golang.org/protobuf/encoding/protojson"
 	"google.golang.org/protobuf/proto"
+	"google.golang.org/protobuf/reflect/protoreflect"
 	"google.golang.org/protobuf/reflect/protoregistry"
 	"google.golang.org/protobuf/types/known/anypb"
 	"google.golang.org/protobuf/types/known/structpb"
@@ -239,18 +240,16 @@ func (c *SafeLekkoClient) {{$.FuncName}}(ctx *{{$.StaticType}}) {{$.RetType}} {
 
 	const protoTemplateBody = `// {{$.Description}}
 func (c *LekkoClient) {{$.FuncName}}(ctx context.Context) (*{{$.RetType}}, error) {
-	result := &{{$.RetType}}{}
-	err := c.{{$.GetFunction}}(ctx, "{{$.Namespace}}", "{{$.Key}}", result)
-	return result, err
+       result := &{{$.RetType}}{}
+       err := c.{{$.GetFunction}}(ctx, "{{$.Namespace}}", "{{$.Key}}", result)
+       return result, err
 }
 
 // {{$.Description}}
-func (c *SafeLekkoClient) {{$.FuncName}}(ctx context.Context) *{{$.RetType}} {
-	result := &{{$.RetType}}{}
-	c.{{$.GetFunction}}(ctx, "{{$.Namespace}}", "{{$.Key}}", result)
-	return result
-}
-`
+func (c *SafeLekkoClient) {{$.FuncName}}(ctx *{{$.StaticType}}) *{{$.RetType}} {
+{{range  $.NaturalLanguage}}{{ . }}
+{{end}}}`
+
 	const jsonTemplateBody = `// {{$.Description}}
 func (c *LekkoClient) {{$.FuncName}}(ctx context.Context, result interface{}) error {
 	return c.{{$.GetFunction}}(ctx, "{{$.Namespace}}", "{{$.Key}}", result)
@@ -295,6 +294,7 @@ func (c *SafeLekkoClient) {{$.FuncName}}(ctx context.Context, result interface{}
 		getFunction = "GetJSON"
 		templateBody = jsonTemplateBody
 	case featurev1beta1.FeatureType_FEATURE_TYPE_PROTO:
+		templateBody = protoTemplateBody
 		getFunction = "GetProto"
 		//templateBody = protoTemplateBody
 		// we don't need the import path so sending in empty string
@@ -459,5 +459,11 @@ func translateRetValue(val *anypb.Any, protoType *protoImport) string {
 	if protoType == nil {
 		return string(try.To1(protojson.MarshalOptions{Resolver: typeRegistry}.Marshal(msg)))
 	}
-	return fmt.Sprintf("&%s.%s{}", protoType.PackageAlias, protoType.Type)
+	// todo multiline formatting
+	var lines []string
+	msg.ProtoReflect().Range(func(f protoreflect.FieldDescriptor, val protoreflect.Value) bool {
+		lines = append(lines, fmt.Sprintf("%s: %s", strcase.UpperCamelCase(f.TextName()), val.String()))
+		return true
+	})
+	return fmt.Sprintf("&%s.%s{%s}", protoType.PackageAlias, protoType.Type, strings.Join(lines, ", "))
 }