fix: parse cmd error with single quotes

This commit is contained in:
Lykin 2024-09-02 18:50:51 +08:00
parent a9c7cb1cd2
commit 2388f309d8
1 changed files with 23 additions and 14 deletions

View File

@ -135,29 +135,38 @@ func SplitCmd(cmd string) []string {
var result []string
var curStr strings.Builder
var preChar int32
inQuotes := false
var quotesChar int32
for _, char := range []rune(cmd) {
if char == '"' && preChar != '\\' {
inQuotes = !inQuotes
} else if char == ' ' && !inQuotes {
if curStr.Len() > 0 {
if part, e := strconv.Unquote(`"` + curStr.String() + `"`); e == nil {
result = append(result, part)
cmdRune := []rune(cmd)
for _, char := range cmdRune {
if (char == '"' || char == '\'') && preChar != '\\' && (quotesChar == 0 || quotesChar == char) {
if quotesChar != 0 {
quotesChar = 0
} else {
quotesChar = char
}
} else if char == ' ' && quotesChar == 0 {
result = append(result, curStr.String())
curStr.Reset()
}
} else {
curStr.WriteRune(char)
}
preChar = char
}
result = append(result, curStr.String())
if curStr.Len() > 0 {
if part, e := strconv.Unquote(`"` + curStr.String() + `"`); e == nil {
result = append(result, part)
result = sliceutil.FilterMap(result, func(i int) (string, bool) {
var part = strings.TrimSpace(result[i])
if len(part) <= 0 {
return "", false
}
if strings.Contains(part, "\\") {
if unquotePart, e := strconv.Unquote(`"` + part + `"`); e == nil {
return unquotePart, true
}
}
return part, true
})
return result
}