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

Add internal quoting style switch to @sh

Browse Source
This commit is contained in:
Vit Zikmund 2023-02-06 14:31:52 +00:00
parent c2f7e0d82c
commit 8ebc0e5801
1 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
pkg/yqlib/encoder_sh.go
View File

@ -12,10 +12,11 @@ import (
var unsafeChars = regexp.MustCompile(`[^\w@%+=:,./-]`)
type shEncoder struct {
quoteAll bool
}
func NewShEncoder() Encoder {
return &shEncoder{}
return &shEncoder{false}
}
func (e *shEncoder) CanHandleAliases() bool {
@ -39,7 +40,7 @@ func (e *shEncoder) Encode(writer io.Writer, originalNode *yaml.Node) error {
return writeString(writer, e.encode(originalNode.Value))
}
// put any shell-unsafe characters into a single-quoted block, close the block lazily
// put any (shell-unsafe) characters into a single-quoted block, close the block lazily
func (e *shEncoder) encode(input string) string {
const quote = '\''
var inQuoteBlock = false
@ -57,8 +58,8 @@ func (e *shEncoder) encode(input string) string {
// escape the quote with a backslash
encoded.WriteRune('\\')
} else {
if unsafeChars.MatchString(string(ir)) && !inQuoteBlock {
// start a quote block on an unsafe character
if e.shouldQuote(ir) && !inQuoteBlock {
// start a quote block for any (unsafe) characters
encoded.WriteRune(quote)
inQuoteBlock = !inQuoteBlock
}
@ -72,3 +73,7 @@ func (e *shEncoder) encode(input string) string {
}
return encoded.String()
}
func (e *shEncoder) shouldQuote(ir rune) bool {
return e.quoteAll || unsafeChars.MatchString(string(ir))
}