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

Make --list more detailed.

Browse Source
This commit is contained in:
Alec Thomas 2017-06-06 17:12:22 +10:00
parent 5f7faa38d5
commit ca512c335b
1 changed files with 18 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
cmd/chroma/main.go
View File

@ -7,6 +7,7 @@ import (
"io/ioutil"
"os"
"runtime/pprof"
"strings"
"gopkg.in/alecthomas/kingpin.v3-unstable"
@ -42,21 +43,30 @@ func main() {
defer w.Flush()
writer := getWriter(w)
if len(*filesArgs) == 0 {
lex("", os.Stdin, writer)
contents, err := ioutil.ReadAll(os.Stdin)
kingpin.FatalIfError(err, "")
lex("", string(contents), writer)
} else {
for _, filename := range *filesArgs {
r, err := os.Open(filename)
contents, err := ioutil.ReadFile(filename)
kingpin.FatalIfError(err, "")
lex(filename, r, writer)
r.Close()
lex(filename, string(contents), writer)
}
}
}
func listAll() {
fmt.Printf("lexers:")
fmt.Println("lexers:")
for _, l := range lexers.Registry.Lexers {
fmt.Printf(" %s", l.Config().Name)
config := l.Config()
fmt.Printf(" %s\n", config.Name)
filenames := []string{}
filenames = append(filenames, config.Filenames...)
filenames = append(filenames, config.AliasFilenames...)
fmt.Printf(" aliases: %s\n", strings.Join(config.Aliases, " "))
fmt.Printf(" filenames: %s\n", strings.Join(filenames, " "))
fmt.Printf(" mimetypes: %s\n", strings.Join(config.MimeTypes, " "))
fmt.Printf(" priority: %d\n", config.Priority)
}
fmt.Println()
fmt.Printf("styles:")
@ -71,12 +81,9 @@ func listAll() {
fmt.Println()
}
func lex(path string, r io.Reader, writer func(*chroma.Token)) {
contents, err := ioutil.ReadAll(r)
kingpin.FatalIfError(err, "")
kingpin.FatalIfError(err, "")
func lex(path string, contents string, writer func(*chroma.Token)) {
lexer := chroma.Coalesce(selexer(path))
err = lexer.Tokenise(nil, string(contents), writer)
err := lexer.Tokenise(nil, string(contents), writer)
kingpin.FatalIfError(err, "")
}