Add basic lua script command (#641)

This will allow for locally running lua code against our engine for testing w/o a full deployment.  Also supports a separate context file to test sideloaded information that can be added to context.

Author: michaeljguarino

cmd/command/cd/cd_services.go

@@ -2,15 +2,18 @@ package cd
 
 import (
 	"fmt"
+	"path/filepath"
 	"strings"
 
 	"github.com/pluralsh/plural-cli/pkg/common"
+	lua "github.com/yuin/gopher-lua"
 
 	gqlclient "github.com/pluralsh/console/go/client"
 	"github.com/pluralsh/plural-cli/pkg/cd/template"
 	"github.com/pluralsh/plural-cli/pkg/console"
 	"github.com/pluralsh/plural-cli/pkg/utils"
 	"github.com/pluralsh/polly/containers"
+	"github.com/pluralsh/polly/luautils"
 	"github.com/samber/lo"
 	"github.com/urfave/cli"
 	"k8s.io/apimachinery/pkg/util/yaml"
@@ -115,6 +118,25 @@ func (p *Plural) cdServiceCommands() []cli.Command {
 				},
 			},
 		},
+		{
+			Name:   "lua",
+			Action: p.handleLuaTemplate,
+			Usage:  "Templates a .lua file using the Plural defined lua engine and returns the result",
+			Flags: []cli.Flag{
+				cli.StringFlag{
+					Name:  "lua-file",
+					Usage: "The .lua file you want to attempt to template.",
+				},
+				cli.StringFlag{
+					Name:  "context",
+					Usage: "A yaml context file to imitate the internal service template context",
+				},
+				cli.StringFlag{
+					Name:  "dir",
+					Usage: "The directory to run the lua script from, defaults to the current working directory",
+				},
+			},
+		},
 		{
 			Name:      "delete",
 			ArgsUsage: "@{cluster-handle}/{serviceName}",
@@ -277,6 +299,77 @@ func (p *Plural) handleTemplateService(c *cli.Context) error {
 	return printResult(res)
 }
 
+func (p *Plural) handleLuaTemplate(c *cli.Context) error {
+	if err := p.InitConsoleClient(consoleToken, consoleURL); err != nil {
+		return err
+	}
+
+	luaFile := c.String("lua-file")
+	context := c.String("context")
+	dir := c.String("dir")
+	if dir == "" {
+		dir = "."
+	}
+
+	if luaFile == "" {
+		return fmt.Errorf("expected --lua-file flag")
+	}
+
+	luaStr, err := utils.ReadFile(luaFile)
+	if err != nil {
+		return err
+	}
+
+	ctx := map[string]interface{}{}
+	if context != "" {
+		if err := utils.YamlFile(context, &ctx); err != nil {
+			return err
+		}
+	}
+
+	values := map[interface{}]interface{}{}
+	valuesFiles := []string{}
+
+	dir, err = filepath.Abs(dir)
+	if err != nil {
+		return err
+	}
+	proc := luautils.NewProcessor(dir)
+	defer proc.L.Close()
+
+	// Register global values and valuesFiles in Lua
+	valuesTable := proc.L.NewTable()
+	proc.L.SetGlobal("values", valuesTable)
+
+	valuesFilesTable := proc.L.NewTable()
+	proc.L.SetGlobal("valuesFiles", valuesFilesTable)
+	proc.L.SetGlobal("cluster", luautils.GoValueToLuaValue(proc.L, ctx["cluster"]))
+	proc.L.SetGlobal("configuration", luautils.GoValueToLuaValue(proc.L, ctx["configuration"]))
+	proc.L.SetGlobal("contexts", luautils.GoValueToLuaValue(proc.L, ctx["contexts"]))
+	proc.L.SetGlobal("imports", luautils.GoValueToLuaValue(proc.L, ctx["imports"]))
+
+	if err := proc.L.DoString(luaStr); err != nil {
+		return err
+	}
+
+	if err := luautils.MapLua(proc.L.GetGlobal("values").(*lua.LTable), &values); err != nil {
+		return err
+	}
+
+	if err := luautils.MapLua(proc.L.GetGlobal("valuesFiles").(*lua.LTable), &valuesFiles); err != nil {
+		return err
+	}
+
+	result := map[string]interface{}{
+		"values":      luautils.SanitizeValue(values),
+		"valuesFiles": valuesFiles,
+	}
+
+	utils.Highlight("Final lua output:\n\n")
+	utils.NewYAMLPrinter(result).PrettyPrint()
+	return nil
+}
+
 func (p *Plural) handleCloneClusterService(c *cli.Context) error {
 	if err := p.InitConsoleClient(consoleToken, consoleURL); err != nil {
 		return err

go.mod

@@ -41,7 +41,7 @@ require (
 	github.com/pluralsh/console/go/controller v0.0.0-20250430123026-a75a764f035b
 	github.com/pluralsh/gqlclient v1.12.2
 	github.com/pluralsh/plural-operator v0.5.5
-	github.com/pluralsh/polly v0.2.0
+	github.com/pluralsh/polly v0.2.7
 	github.com/posthog/posthog-go v1.4.10
 	github.com/samber/lo v1.50.0
 	github.com/urfave/cli v1.22.16

go.sum

@@ -724,6 +724,14 @@ github.com/pluralsh/plural-operator v0.5.5 h1:57GxniNjUa3hpHgvFr9oDonFgvDUC8XDD5
 github.com/pluralsh/plural-operator v0.5.5/go.mod h1:WIXiz26/WDcUn0FA7Q1jPxmfsm98U1/JL8YpIdKVLX0=
 github.com/pluralsh/polly v0.2.0 h1:i0Yu13Lwef9u7hyiqHLyDZhkT1G1uXcaK6Ph7EeXjxo=
 github.com/pluralsh/polly v0.2.0/go.mod h1:o+oi+FXNThipKNyZsqCemqwY0JdkvhEjQqi8aTw8l4M=
+github.com/pluralsh/polly v0.2.4 h1:OyxUq324sTE/ZM5AVUZAgeZCLxM77gfqlB6uIOGihFo=
+github.com/pluralsh/polly v0.2.4/go.mod h1:GpeJrbPW0XedsoTXYRPhS5mc6GqIYLocOaRRSsZGisQ=
+github.com/pluralsh/polly v0.2.5 h1:JO6rqZChuV+SQ4xry07XctZDW3DqYaWqEWsrio8k88A=
+github.com/pluralsh/polly v0.2.5/go.mod h1:GpeJrbPW0XedsoTXYRPhS5mc6GqIYLocOaRRSsZGisQ=
+github.com/pluralsh/polly v0.2.6 h1:BS2ubNJoXM2HwVEJxcgpew/j9ErYEzk91ArkqUavr5w=
+github.com/pluralsh/polly v0.2.6/go.mod h1:GpeJrbPW0XedsoTXYRPhS5mc6GqIYLocOaRRSsZGisQ=
+github.com/pluralsh/polly v0.2.7 h1:Sk0rW9jGH82jdCFGpz2B8X4qz+LcBxadl2ToUFoS9Q0=
+github.com/pluralsh/polly v0.2.7/go.mod h1:GpeJrbPW0XedsoTXYRPhS5mc6GqIYLocOaRRSsZGisQ=
 github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
 github.com/pmezard/go-difflib v1.0.1-0.20181226105442-5d4384ee4fb2 h1:Jamvg5psRIccs7FGNTlIRMkT8wgtp5eCXdBlqhYGL6U=
 github.com/pmezard/go-difflib v1.0.1-0.20181226105442-5d4384ee4fb2/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=

test/lua/basic.lua

@@ -0,0 +1,5 @@
+values = {}
+valuesFiles = {}
+local parts = {"a", "b", "c"}
+local joined = utils.pathJoin(parts)
+values["joined"] = joined

test/lua/fleeta/cluster.yaml

@@ -0,0 +1,9 @@
+spec:
+  region: us-east-1
+  nodes:
+  - name: node1
+    type: t3.small
+    count: 1
+  - name: node2
+    type: t3.small
+    count: 1

test/lua/fleeta/dev/cluster-a/cluster.yaml

@@ -0,0 +1,2 @@
+cluster: dev-cluster-a
+spec: {}

test/lua/fleeta/dev/cluster.yaml

@@ -0,0 +1,6 @@
+spec:
+  region: us-east-2
+  nodes:
+  - name: node1
+    type: t3.small
+    count: 1

test/lua/fleeta/prod/cluster-a/cluster.yaml

@@ -0,0 +1,2 @@
+cluster: prod-cluster-a
+spec: {}

test/lua/fleeta/prod/cluster-b/cluster.yaml

@@ -0,0 +1,2 @@
+cluster: prod-cluster-b
+spec: {}

test/lua/fsmerge.lua

@@ -0,0 +1,36 @@
+values = {}
+valuesFiles = {}
+
+files, err = fs.walk("test/lua/fleeta")
+if err then
+    error(err)
+    return
+end
+
+table.sort(files, function(a, b)
+    local aParts = utils.splitString(a, "/")
+    local bParts = utils.splitString(b, "/")
+    return #aParts < #bParts
+end)
+
+local workset = {}
+
+for _, file in ipairs(files) do
+    local f = fs.read(file)
+    local asYaml = encoding.yamlDecode(f)
+    local parent = {}
+    local parts = utils.splitString(file, "/")
+    local i = #parts - 1
+    while i > 0 do
+        local parentFile = table.concat(parts, "/", 1, i - 1) .. "/cluster.yaml"
+        if  workset[parentFile] then
+            parent = workset[parentFile]
+            break
+        end
+        i = i - 1
+    end
+    workset[file] = utils.merge(parent, asYaml)
+    if workset[file].cluster then
+        values[workset[file].cluster] = workset[file]
+    end
+end