From 11605f6b630391bc90982314d333a947bb69a954 Mon Sep 17 00:00:00 2001
From: Kim Alvefur <zash@zash.se>
Date: Fri, 4 Aug 2023 12:55:07 +0200
Subject: [PATCH] Respect string Style in Lua encoder

Lua has 'single', "double" and [[ long ]] strings.
---
 pkg/yqlib/encoder_lua.go | 23 ++++++++++++++++++++++-
 pkg/yqlib/lua_test.go    |  4 ++--
 2 files changed, 24 insertions(+), 3 deletions(-)

diff --git a/pkg/yqlib/encoder_lua.go b/pkg/yqlib/encoder_lua.go
index b5816b57..8506348d 100644
--- a/pkg/yqlib/encoder_lua.go
+++ b/pkg/yqlib/encoder_lua.go
@@ -80,7 +80,28 @@ func (le *luaEncoder) PrintLeadingContent(writer io.Writer, content string) erro
 }
 
 func (le *luaEncoder) encodeString(writer io.Writer, node *yaml.Node) error {
-	return writeString(writer, "\""+le.escape.Replace(node.Value)+"\"")
+	quote := "\""
+	switch node.Style {
+	case yaml.LiteralStyle, yaml.FoldedStyle, yaml.FlowStyle:
+		for i := 0; i < 10; i++ {
+			if !strings.Contains(node.Value, "]"+strings.Repeat("=", i)+"]") {
+				err := writeString(writer, "["+strings.Repeat("=", i)+"[\n")
+				if err != nil {
+					return err
+				}
+				err = writeString(writer, node.Value)
+				if err != nil {
+					return err
+				}
+				return writeString(writer, "]"+strings.Repeat("=", i)+"]")
+			}
+		}
+	case yaml.SingleQuotedStyle:
+		quote = "'"
+
+		// falltrough to regular ol' string
+	}
+	return writeString(writer, quote+le.escape.Replace(node.Value)+quote)
 }
 
 func (le *luaEncoder) writeIndent(writer io.Writer) error {
diff --git a/pkg/yqlib/lua_test.go b/pkg/yqlib/lua_test.go
index df5b98ee..6c8b9916 100644
--- a/pkg/yqlib/lua_test.go
+++ b/pkg/yqlib/lua_test.go
@@ -47,8 +47,8 @@ numbers: [123,456]
 	{
 		skipDoc:      true,
 		description:  "Scalar str",
-		input:        "str: |\n  foo\n  bar\n",
-		expected:     "return {\n\t[\"str\"] = \"foo\\nbar\\n\";\n};\n",
+		input:        "str: |\n  foo\n  bar\nanother: 'single'\nand: \"double\"",
+		expected:     "return {\n\t[\"str\"] = [[\nfoo\nbar\n]];\n\t[\"another\"] = 'single';\n\t[\"and\"] = \"double\";\n};\n",
 		scenarioType: "encode",
 	},
 	{