suyiiyii/yq
1
0
Fork 0
mirror of https://github.com/mikefarah/yq.git synced 2026-07-10 08:35:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

Print simple keys unquoted in Lua output

Browse Source 
String keys that satisfy the requirements for variable names can be used
as keys without quotes in tables.
This commit is contained in:
Kim Alvefur 2023-07-30 20:54:59 +02:00
parent 21c64fa945
commit 21fda2eb3d
1 changed files with 21 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
pkg/yqlib/encoder_lua.go
View File

@ -89,6 +89,22 @@ func (le *luaEncoder) encodeArray(writer io.Writer, node *yaml.Node) error {
return writeString(writer, "}")
}
func needsQuoting(s string) bool {
// [%a_][%w_]*
for i, c := range s {
if i == 0 {
if !((c >= 'A' && c <= 'Z') || (c >= 'a' && c <= 'z') || c == '_') {
return true
}
} else {
if !((c >= '0' && c <= '9') || (c >= 'A' && c <= 'Z') || (c >= 'a' && c <= 'z') || c == '_') {
return true
}
}
}
return false
}
func (le *luaEncoder) encodeMap(writer io.Writer, node *yaml.Node) error {
err := writeString(writer, "{")
if err != nil {
@ -105,6 +121,11 @@ func (le *luaEncoder) encodeMap(writer io.Writer, node *yaml.Node) error {
if err != nil {
return err
}
} else if child.Tag == "!!str" && !needsQuoting(child.Value) {
err = writeString(writer, child.Value+"=")
if err != nil {
return err
}
} else {
// key
err := writeString(writer, "[")