-
Notifications
You must be signed in to change notification settings - Fork 0
/
parser.go
106 lines (88 loc) · 1.77 KB
/
parser.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
package pipeparser
import (
"bytes"
"os/exec"
"strings"
pipe "github.com/b4b4r07/go-pipe"
)
const (
stateStart = "start"
stateArgs = "args"
stateQuotes = "quotes"
singleQuoteChar = '\''
doubleQuoteChar = '"'
backslashChar = '\\'
tabChar = '\t'
spaceChar = ' '
)
func parse(input string) []string {
var args []string
state := stateStart
current := ""
quote := string(doubleQuoteChar)
isEscapeNext := true
for i := 0; i < len(input); i++ {
c := input[i]
if state == stateQuotes {
if string(c) != quote {
current += string(c)
} else {
args = append(args, current)
current = ""
state = stateStart
}
continue
}
if isEscapeNext {
current += string(c)
isEscapeNext = false
continue
}
if c == backslashChar {
isEscapeNext = true
continue
}
if c == doubleQuoteChar || c == singleQuoteChar {
state = stateQuotes
quote = string(c)
continue
}
if state == stateArgs {
if c == spaceChar || c == tabChar {
args = append(args, current)
current = ""
state = stateStart
} else {
current += string(c)
}
continue
}
if c != spaceChar && c != tabChar {
state = stateArgs
current += string(c)
}
}
if current != "" {
args = append(args, current)
}
return args
}
func buildCommands(input string) []*exec.Cmd {
commands := []*exec.Cmd{}
commandStrings := strings.Split(input, " | ")
for _, s := range commandStrings {
parsed := parse(s)
command := exec.Command(parsed[0], parsed[1:]...)
commands = append(commands, command)
}
return commands
}
// Run executes the pipe command
func Run(input string) (bytes.Buffer, error) {
var b bytes.Buffer
pipeErr := pipe.Command(&b, buildCommands(input)...)
if pipeErr != nil {
return b, pipeErr
}
return b, nil
}