go/chroma
1
0
Fork 0
mirror of https://github.com/alecthomas/chroma.git synced 2025-02-21 19:06:18 +02:00
Code Issues Releases Activity

Add --check flag to aid in lexer development

Browse Source
This commit is contained in:
Orivej Desh 2017-10-03 12:02:34 +00:00
parent 6d281667bb
commit feb2cdccf3
1 changed files with 21 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
cmd/chroma/main.go
View File

@ -34,6 +34,7 @@ var (
listFlag = kingpin.Flag("list", "List lexers, styles and formatters.").Bool() listFlag = kingpin.Flag("list", "List lexers, styles and formatters.").Bool()
unbufferedFlag = kingpin.Flag("unbuffered", "Do not buffer output.").Bool() unbufferedFlag = kingpin.Flag("unbuffered", "Do not buffer output.").Bool()
traceFlag = kingpin.Flag("trace", "Trace lexer states as they are traversed.").Bool() traceFlag = kingpin.Flag("trace", "Trace lexer states as they are traversed.").Bool()
checkFlag = kingpin.Flag("check", "Do not format, check for tokenization errors instead.").Bool()
lexerFlag = kingpin.Flag("lexer", "Lexer to use when formatting.").PlaceHolder("autodetect").Short('l').Enum(lexers.Names(true)...) lexerFlag = kingpin.Flag("lexer", "Lexer to use when formatting.").PlaceHolder("autodetect").Short('l').Enum(lexers.Names(true)...)
styleFlag = kingpin.Flag("style", "Style to use for formatting.").Short('s').Default("swapoff").Enum(styles.Names()...) styleFlag = kingpin.Flag("style", "Style to use for formatting.").Short('s').Default("swapoff").Enum(styles.Names()...)
@ -167,10 +168,14 @@ command, for Go.
for _, filename := range *filesArgs { for _, filename := range *filesArgs {
contents, err := ioutil.ReadFile(filename) contents, err := ioutil.ReadFile(filename)
kingpin.FatalIfError(err, "") kingpin.FatalIfError(err, "")
if *checkFlag {
check(filename, lex(filename, string(contents)))
} else {
format(w, style, lex(filename, string(contents))) format(w, style, lex(filename, string(contents)))
} }
} }
} }
}
func listAll() { func listAll() {
fmt.Println("lexers:") fmt.Println("lexers:")
@ -236,3 +241,18 @@ func format(w io.Writer, style *chroma.Style, it chroma.Iterator) {
err := formatter.Format(w, style, it) err := formatter.Format(w, style, it)
kingpin.FatalIfError(err, "") kingpin.FatalIfError(err, "")
} }
func check(filename string, it chroma.Iterator) {
line, col := 1, 0
for token := it(); token != nil; token = it() {
if token.Type == chroma.Error {
fmt.Printf("%s:%d:%d %q\n", filename, line, col, token.String())
}
for _, c := range token.String() {
col++
if c == '\n' {
line, col = line+1, 0
}
}
}
}