-
Notifications
You must be signed in to change notification settings - Fork 1
/
shlex.ts
94 lines (85 loc) · 1.81 KB
/
shlex.ts
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
export interface ISplitOptions {
brackets: [string, string][],
split: string
escape: string
keepBrace?: boolean
}
export const defaultSplitOptions: ISplitOptions = {
brackets: [['"', '"'], ["'", "'"]],
split: ' ',
escape: '\\'
}
/**
*
* @param ss
* @param options
*
* ```js
* > split('')
* []
* > split('a:b "c:d e:f"')
* ['a:b', 'c:d e:f']
* > split('a "b c" "d e"')
* ['a', 'b c', 'd e']
* ```
*/
export function split (ss: string, options: ISplitOptions = defaultSplitOptions) {
const bracketStack = {
data: [] as string[],
push (c: string) {
this.data.push(c)
},
pop () {
return this.data.pop()
},
peek () {
return this.data.length > 0 ? this.data[this.data.length - 1] : undefined
}
}
const tokenStack = {
data: [] as string[],
currentChars: [] as string[],
addChar (c: string) {
this.currentChars.push(c)
},
flush () {
const d = this.currentChars.join('')
if (d) {
this.data.push(d)
}
this.currentChars = []
}
}
let prev = ''
ss.split('').map((c) => {
if (prev === options.escape) {
tokenStack.addChar(c)
} else {
let canAddChar = true
for (const [op, cl] of options.brackets) {
if (c === cl) {
if (bracketStack.peek() === op) {
bracketStack.pop()
canAddChar = false
break
}
}
if (c === op) {
bracketStack.push(c)
canAddChar = false
break
}
}
if (c === options.split && !bracketStack.peek()) {
tokenStack.flush()
} else {
if (options.keepBrace || canAddChar) {
tokenStack.addChar(c)
}
}
}
prev = c
})
tokenStack.flush()
return tokenStack.data.map(s => s.trim()).filter(s => s)
}