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

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" "io/ioutil"
"os" "os"
"runtime/pprof" "runtime/pprof"
"strings"
"gopkg.in/alecthomas/kingpin.v3-unstable" "gopkg.in/alecthomas/kingpin.v3-unstable"
@ -42,21 +43,30 @@ func main() {
defer w.Flush() defer w.Flush()
writer := getWriter(w) writer := getWriter(w)
if len(*filesArgs) == 0 { if len(*filesArgs) == 0 {
lex("", os.Stdin, writer) contents, err := ioutil.ReadAll(os.Stdin)
kingpin.FatalIfError(err, "")
lex("", string(contents), writer)
} else { } else {
for _, filename := range *filesArgs { for _, filename := range *filesArgs {
r, err := os.Open(filename) contents, err := ioutil.ReadFile(filename)
kingpin.FatalIfError(err, "") kingpin.FatalIfError(err, "")
lex(filename, r, writer) lex(filename, string(contents), writer)
r.Close()
} }
} }
} }
func listAll() { func listAll() {
fmt.Printf("lexers:") fmt.Println("lexers:")
for _, l := range lexers.Registry.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.Println()
fmt.Printf("styles:") fmt.Printf("styles:")
@ -71,12 +81,9 @@ func listAll() {
fmt.Println() fmt.Println()
} }
func lex(path string, r io.Reader, writer func(*chroma.Token)) { func lex(path string, contents string, writer func(*chroma.Token)) {
contents, err := ioutil.ReadAll(r)
kingpin.FatalIfError(err, "")
kingpin.FatalIfError(err, "")
lexer := chroma.Coalesce(selexer(path)) lexer := chroma.Coalesce(selexer(path))
err = lexer.Tokenise(nil, string(contents), writer) err := lexer.Tokenise(nil, string(contents), writer)
kingpin.FatalIfError(err, "") kingpin.FatalIfError(err, "")
} }