Resolves #404 - Add template support for attributes (#469)

steve-r-west · web-flow · commit 923ace5bf93d · 2024-08-20T15:24:47.000-07:00
diff --git a/cmd/create.go b/cmd/create.go
@@ -185,11 +185,12 @@ func NewCreateCommand(parentCmd *cobra.Command) func() {
 								})
 							} else { // This is an attribute value
 								return completion.Complete(completion.Request{
-									Type:       completion.CompleteAttributeValue,
-									Resource:   resource,
-									Verb:       completion.Create,
-									Attribute:  args[len(args)-1],
-									ToComplete: toComplete,
+									Type:           completion.CompleteAttributeValue,
+									Resource:       resource,
+									Verb:           completion.Create,
+									Attribute:      args[len(args)-1],
+									ToComplete:     toComplete,
+									AllowTemplates: true,
 								})
 							}
 						} else {
diff --git a/cmd/delete.go b/cmd/delete.go
@@ -161,11 +161,12 @@ func NewDeleteCommand(parentCmd *cobra.Command) func() {
 						})
 					} else { // This is an attribute value
 						return completion.Complete(completion.Request{
-							Type:       completion.CompleteAttributeValue,
-							Resource:   resource,
-							Verb:       completion.Delete,
-							Attribute:  args[len(args)-1],
-							ToComplete: toComplete,
+							Type:           completion.CompleteAttributeValue,
+							Resource:       resource,
+							Verb:           completion.Delete,
+							Attribute:      args[len(args)-1],
+							ToComplete:     toComplete,
+							AllowTemplates: true,
 						})
 					}
 				}
diff --git a/cmd/helper.go b/cmd/helper.go
@@ -299,7 +299,9 @@ epcc %s %s%s key [] => %s
 
 # To send a nested object use the . character to nest values deeper.
 epcc %s %s%s key.some.child hello key.some.other goodbye => %s
-`,
+
+# Attributes can also be generated using Go templates and Sprig (https://masterminds.github.io/sprig/) functions.
+epcc %s %s%s key 'Test {{ randAlphaNum 6 | upper }} Value' => %s`,
 		verb, resource.SingularName, id, toJsonExample([]string{"key", "b"}, resource),
 		verb, resource.SingularName, id, toJsonExample([]string{"key", "1"}, resource),
 		verb, resource.SingularName, id, toJsonExample([]string{"key", "\"1\""}, resource),
@@ -310,6 +312,7 @@ epcc %s %s%s key.some.child hello key.some.other goodbye => %s
 		verb, resource.SingularName, id, toJsonExample([]string{"key[0]", "a", "key[1]", "true"}, resource),
 		verb, resource.SingularName, id, toJsonExample([]string{"key", "[]"}, resource),
 		verb, resource.SingularName, id, toJsonExample([]string{"key.some.child", "hello", "key.some.other", "goodbye"}, resource),
+		verb, resource.SingularName, id, toJsonExample([]string{"key", "Test {{ randAlphaNum 6 | upper }} Value"}, resource),
 	)
 }
 
diff --git a/cmd/login.go b/cmd/login.go
@@ -428,11 +428,12 @@ var loginAccountManagement = &cobra.Command{
 				})
 			} else {
 				return completion.Complete(completion.Request{
-					Type:       completion.CompleteAttributeValue,
-					Verb:       completion.Create,
-					Resource:   res,
-					Attributes: usedAttributes,
-					ToComplete: toComplete,
+					Type:           completion.CompleteAttributeValue,
+					Verb:           completion.Create,
+					Resource:       res,
+					Attributes:     usedAttributes,
+					ToComplete:     toComplete,
+					AllowTemplates: true,
 				})
 			}
 
diff --git a/cmd/root.go b/cmd/root.go
@@ -290,7 +290,6 @@ func Execute() {
 
 		os.Exit(1)
 	} else {
-
 		os.Exit(0)
 	}
 }
diff --git a/cmd/runbooks.go b/cmd/runbooks.go
@@ -421,11 +421,9 @@ func processRunbookVariablesOnCommand(runbookActionRunActionCommand *cobra.Comma
 			}
 		} else {
 			description := ""
-
 			if variable.Description != nil {
 				description = variable.Description.Short
 			}
-
 			runbookActionRunActionCommand.Flags().StringVar(runbookStringArguments[key], key, variable.Default, description)
 		}
 
diff --git a/cmd/update.go b/cmd/update.go
@@ -161,11 +161,12 @@ func NewUpdateCommand(parentCmd *cobra.Command) func() {
 						})
 					} else { // This is an attribute value
 						return completion.Complete(completion.Request{
-							Type:       completion.CompleteAttributeValue,
-							Resource:   resource,
-							Verb:       completion.Update,
-							Attribute:  args[len(args)-1],
-							ToComplete: toComplete,
+							Type:           completion.CompleteAttributeValue,
+							Resource:       resource,
+							Verb:           completion.Update,
+							Attribute:      args[len(args)-1],
+							ToComplete:     toComplete,
+							AllowTemplates: true,
 						})
 					}
 				} else {
diff --git a/docs/runbook-development.md b/docs/runbook-development.md
@@ -171,17 +171,17 @@ description:
   short: "A hello world runbook"
 actions:
   sequential-sleeps:
-  variables:
-    count:
-      type: INT
-      default: 2
-      description:
-        short: "The number of sleeps"
-  commands:
-    - |2
-      {{- range untilStep 0 .count 1}}
-        - sleep 1
-      {{- end -}}
+    variables:
+      count:
+        type: INT
+        default: 2
+        description:
+          short: "The number of sleeps"
+    commands:
+      - |2
+        {{- range untilStep 0 .count 1}}
+          - sleep 1
+        {{- end -}}
   concurrent-sleeps:
     variables:
       count:
diff --git a/external/completion/completion.go b/external/completion/completion.go
@@ -1,9 +1,11 @@
 package completion
 
 import (
+	"fmt"
 	"github.com/elasticpath/epcc-cli/external/aliases"
 	"github.com/elasticpath/epcc-cli/external/resources"
 	"github.com/spf13/cobra"
+	"os"
 	"regexp"
 	"strconv"
 	"strings"
@@ -48,8 +50,9 @@ type Request struct {
 	QueryParam string
 	Header     string
 	// The current string argument being completed
-	ToComplete string
-	NoAliases  bool
+	ToComplete     string
+	NoAliases      bool
+	AllowTemplates bool
 }
 
 func Complete(c Request) ([]string, cobra.ShellCompDirective) {
@@ -295,6 +298,79 @@ func Complete(c Request) ([]string, cobra.ShellCompDirective) {
 					results = append(results, supportedFileTypes...)
 				}
 			}
+
+			if c.AllowTemplates {
+				lastPipe := strings.LastIndex(c.ToComplete, "|")
+				prefix := ""
+				if lastPipe == -1 {
+					prefix = "{{ "
+				} else {
+					prefix = c.ToComplete[0:lastPipe+1] + " "
+				}
+
+				myResults := []string{}
+				myResults = append(myResults,
+					prefix+"date",
+					prefix+"now",
+					prefix+"randAlphaNum",
+					prefix+"randAlpha",
+					prefix+"randAscii",
+					prefix+"randNumeric",
+					prefix+"randAlphaNum",
+					prefix+"randAlpha",
+					prefix+"randAscii",
+					prefix+"randNumeric",
+					prefix+"pseudoRandAlphaNum",
+					prefix+"pseudoRandAlpha",
+					prefix+"pseudoRandNumeric",
+					prefix+"pseudoRandString",
+					prefix+"pseudoRandInt",
+					prefix+"uuidv4",
+					prefix+"duration",
+				)
+
+				if prefix != "{{ " {
+					// Functions that make sense as continuations
+					myResults = append(myResults,
+						prefix+"trim",
+						prefix+"trimAll",
+						prefix+"trimSuffix",
+						prefix+"trimPrefix",
+						prefix+"upper",
+						prefix+"lower",
+						prefix+"title",
+						prefix+"repeat",
+						prefix+"substr",
+						prefix+"nospace",
+						prefix+"trunc",
+						prefix+"abbrev",
+						prefix+"initials",
+						prefix+"wrap",
+						prefix+"cat",
+						prefix+"replace",
+						prefix+"snakecase",
+						prefix+"camelcase",
+						prefix+"kebabcase",
+						prefix+"swapcase",
+						prefix+"shufflecase",
+					)
+				}
+
+				re := regexp.MustCompile(`env\s+[A-Za-z]*\s*$`)
+				if re.MatchString(c.ToComplete) {
+					for _, v := range os.Environ() {
+						myResults = append(myResults,
+							fmt.Sprintf("%venv \"%v\"", prefix, strings.Split(v, "=")[0]),
+						)
+					}
+				} else {
+					myResults = append(myResults, prefix+"env")
+				}
+				//myResults = append(myResults, strings.TrimSuffix(c.ToComplete, " ")+" }}", strings.TrimSuffix(c.ToComplete, " ")+" |")
+				for _, r := range myResults {
+					results = append(results, r+" |", r+" }}")
+				}
+			}
 		}
 	}
 
diff --git a/external/completion/completion_test.go b/external/completion/completion_test.go
@@ -1,6 +1,7 @@
 package completion
 
 import (
+	"github.com/elasticpath/epcc-cli/external/resources"
 	"github.com/spf13/cobra"
 	"github.com/stretchr/testify/require"
 	"testing"
@@ -72,3 +73,73 @@ func TestHeaderValueWithNonNilValueCompletes(t *testing.T) {
 	require.Equal(t, compDir, cobra.ShellCompDirectiveNoFileComp)
 	require.Contains(t, completions, "USD")
 }
+
+func TestAttributeValueWithNoTemplating(t *testing.T) {
+	// Fixture Setup
+	toComplete := ""
+	acct := resources.MustGetResourceByName("password-profiles")
+	request := Request{
+		Type:       CompleteAttributeValue,
+		Verb:       Create,
+		ToComplete: toComplete,
+		Attribute:  "username_format",
+		Resource:   acct,
+	}
+
+	// Exercise SUT
+	completions, compDir := Complete(request)
+
+	// Verify Results
+	require.Equal(t, compDir, cobra.ShellCompDirectiveNoFileComp)
+	require.Contains(t, completions, "any")
+	require.Contains(t, completions, "email")
+	require.Equal(t, 2, len(completions))
+}
+
+func TestAttributeValueWithTemplating(t *testing.T) {
+	// Fixture Setup
+	toComplete := ""
+	acct := resources.MustGetResourceByName("password-profiles")
+	request := Request{
+		Type:           CompleteAttributeValue,
+		Verb:           Create,
+		ToComplete:     toComplete,
+		Attribute:      "username_format",
+		Resource:       acct,
+		AllowTemplates: true,
+	}
+
+	// Exercise SUT
+	completions, compDir := Complete(request)
+
+	// Verify Results
+	require.Equal(t, compDir, cobra.ShellCompDirectiveNoFileComp)
+	require.Contains(t, completions, "any")
+	require.Contains(t, completions, "email")
+	require.Contains(t, completions, `{{\ randAlphaNum\ |`)
+	require.Contains(t, completions, `{{\ randAlphaNum\ }}`)
+}
+
+func TestAttributeValueWithTemplatingAndPipe(t *testing.T) {
+	// Fixture Setup
+	toComplete := "{{ randAlphaNum 3 | "
+	acct := resources.MustGetResourceByName("password-profiles")
+	request := Request{
+		Type:           CompleteAttributeValue,
+		Verb:           Create,
+		ToComplete:     toComplete,
+		Attribute:      "username_format",
+		Resource:       acct,
+		AllowTemplates: true,
+	}
+
+	// Exercise SUT
+	completions, compDir := Complete(request)
+
+	// Verify Results
+	require.Equal(t, compDir, cobra.ShellCompDirectiveNoFileComp)
+	require.Contains(t, completions, "any")
+	require.Contains(t, completions, "email")
+	require.Contains(t, completions, `{{\ randAlphaNum\ 3\ |\ upper\ |`)
+	require.Contains(t, completions, `{{\ randAlphaNum\ 3\ |\ lower\ }}`)
+}
diff --git a/external/json/to_json.go b/external/json/to_json.go
@@ -5,6 +5,7 @@ import (
 	"fmt"
 	"github.com/elasticpath/epcc-cli/external/aliases"
 	"github.com/elasticpath/epcc-cli/external/resources"
+	"github.com/elasticpath/epcc-cli/external/templates"
 	"github.com/itchyny/gojq"
 	log "github.com/sirupsen/logrus"
 	"regexp"
@@ -53,6 +54,9 @@ func toJsonObject(args []string, noWrapping bool, compliant bool, attributes map
 		key := args[i]
 		val := args[i+1]
 
+		// Try and process the argument as a helm template
+		val = templates.Render(val)
+
 		jsonKey := key
 		switch {
 		case key == "type" || key == "id":
@@ -265,6 +269,9 @@ func formatValue(v string) string {
 	} else if match, _ := regexp.MatchString("^\\[\\]$", v); match {
 		return v
 	} else {
+		v = strings.ReplaceAll(v, "\\", "\\\\")
+		v = strings.ReplaceAll(v, `"`, `\"`)
+
 		return fmt.Sprintf("\"%s\"", v)
 	}
 }
diff --git a/external/json/to_json_test.go b/external/json/to_json_test.go
diff --git a/external/runbooks/runbook_rendering.go b/external/runbooks/runbook_rendering.go
diff --git a/external/templates/funcs.go b/external/templates/funcs.go
diff --git a/external/templates/template.go b/external/templates/template.go

Original file line number	Diff line number	Diff line change
`@@ -290,7 +290,6 @@ func Execute() {`
`290`	`290`
`291`	`291`	`os.Exit(1)`
`292`	`292`	`} else {`
`293`		`-`
`294`	`293`	`os.Exit(0)`
`295`	`294`	`}`
`296`	`295`	`}`
Original file line number	Diff line number	Diff line change
`@@ -421,11 +421,9 @@ func processRunbookVariablesOnCommand(runbookActionRunActionCommand *cobra.Comma`
`421`	`421`	`}`
`422`	`422`	`} else {`
`423`	`423`	`description := ""`
`424`		`-`
`425`	`424`	`if variable.Description != nil {`
`426`	`425`	`description = variable.Description.Short`
`427`	`426`	`}`
`428`		`-`
`429`	`427`	`runbookActionRunActionCommand.Flags().StringVar(runbookStringArguments[key], key, variable.Default, description)`
`430`	`428`	`}`
`431`	`429`