-
Notifications
You must be signed in to change notification settings - Fork 22
/
ts.nu
84 lines (75 loc) · 1.92 KB
/
ts.nu
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
# | A bunch of commands for developing `tree-sitter-nu`
#
# # requirements
# - tree-sitter cli
def list-tests-in-file [file: path] {
open --raw $file
| to text
| lines
| do {|lines|
($lines | skip 1)
| zip $lines
| zip ($lines | skip 2)
} $in
| each {|groups| $groups | flatten}
| filter {|group| $group.1 =~ "=+$" and $group.2 =~ "=+$"}
| each {|group| $group.0}
}
def list-tests-in-dir [dir: path] {
ls $dir
| each {|it|
if $it.type == "dir" {
list-tests-in-dir $it.name | flatten
} else if $it.type == "file" {
list-tests-in-file $it.name
}
}
| flatten
}
def "completion ts test" [] {
list-tests-in-dir ./corpus
}
# Generates a parser from the grammar
export def "ts gen" [] { # -> nothing
clear
tree-sitter generate
}
# Performs highlighting on the specified file(s) and
# prints the highlighted result(s) to stdout
export def "ts hl" [ # -> string
glob: string # the file(s) to highlight
] {
clear
tree-sitter highlight $glob
}
# Runs all tests in the repo
export def "ts test" [ # -> string
test_name?: string@"completion ts test" # specific test name to run
] {
clear
if ($test_name == null) {
tree-sitter test
} else {
tree-sitter test -f $test_name
}
}
# Parses the specified file(s) and prints their ASTs to
# std-out
export def "ts parse" [ # -> string
glob: string, # the file(s) to parse
--debug(-d), # print debug info along with the AST
--stat(-s), # only print success stats instead of the AST
] {
clear
if $debug {
tree-sitter parse $glob -d
} else if $stat {
tree-sitter parse $glob -sq
} else {
tree-sitter parse $glob
}
}
# Generates a default config file and prints out its path
export def "ts config" [] {
tree-sitter init-config
}