124 lines
1.8 KiB
Go
124 lines
1.8 KiB
Go
package shellwords
|
|
|
|
import (
|
|
"errors"
|
|
)
|
|
|
|
type ErrParse struct {
|
|
Line string
|
|
Position string
|
|
}
|
|
|
|
func Parse(line string) ([]string, error) {
|
|
var args []string
|
|
buf := ""
|
|
var escaped, doubleQuoted, singleQuoted bool
|
|
backtick := ""
|
|
|
|
got := false
|
|
cursor := 0
|
|
|
|
runes := []rune(line)
|
|
|
|
for _, r := range runes {
|
|
if escaped {
|
|
buf += string(r)
|
|
escaped = false
|
|
continue
|
|
}
|
|
|
|
if r == '\\' {
|
|
if singleQuoted {
|
|
buf += string(r)
|
|
} else {
|
|
escaped = true
|
|
}
|
|
continue
|
|
}
|
|
|
|
if isSpace(r) {
|
|
switch {
|
|
case singleQuoted, doubleQuoted:
|
|
buf += string(r)
|
|
backtick += string(r)
|
|
case got:
|
|
cursor += len(buf)
|
|
args = append(args, buf)
|
|
buf = ""
|
|
got = false
|
|
}
|
|
continue
|
|
}
|
|
|
|
switch r {
|
|
case '"':
|
|
if !singleQuoted {
|
|
if doubleQuoted {
|
|
got = true
|
|
}
|
|
doubleQuoted = !doubleQuoted
|
|
continue
|
|
}
|
|
case '\'':
|
|
if !doubleQuoted {
|
|
if singleQuoted {
|
|
got = true
|
|
}
|
|
singleQuoted = !singleQuoted
|
|
continue
|
|
}
|
|
}
|
|
|
|
got = true
|
|
buf += string(r)
|
|
}
|
|
|
|
if got {
|
|
args = append(args, buf)
|
|
}
|
|
|
|
if escaped || singleQuoted || doubleQuoted {
|
|
// the number of characters to highlight
|
|
var (
|
|
pos = cursor + 5
|
|
start = string(runes[max(cursor-100, 0) : pos-1])
|
|
end = string(runes[pos+1 : min(cursor+100, len(runes))])
|
|
part = ""
|
|
)
|
|
|
|
for i := pos - 1; i >= 0 && i < len(runes) && i < pos+2; i++ {
|
|
if runes[i] == '\\' {
|
|
part += "\\"
|
|
}
|
|
part += string(runes[i])
|
|
}
|
|
|
|
return nil, errors.New(
|
|
"Unexpected quote or escape: " + start + "__" + part + "__" + end)
|
|
}
|
|
|
|
return args, nil
|
|
}
|
|
|
|
func isSpace(r rune) bool {
|
|
switch r {
|
|
case ' ', '\t', '\r', '\n':
|
|
return true
|
|
}
|
|
return false
|
|
}
|
|
|
|
func min(i, j int) int {
|
|
if i < j {
|
|
return i
|
|
}
|
|
return j
|
|
}
|
|
|
|
func max(i, j int) int {
|
|
if i < j {
|
|
return j
|
|
}
|
|
return i
|
|
}
|