go/chroma
1
0
Fork 0
mirror of https://github.com/alecthomas/chroma.git synced 2025-03-31 22:05:17 +02:00
Code Issues Releases Activity

Add "tokens" formatter.

Browse Source
This commit is contained in:
Alec Thomas 2017-06-06 16:04:57 +10:00
parent 5749aebe42
commit 5f7faa38d5
2 changed files with 17 additions and 7 deletions
cmd/chroma
main.go
formatters
tokens.go

8
cmd/chroma/main.go

@ -17,10 +17,9 @@ import (
)
var (
profileFlag = kingpin.Flag("profile", "Enable profiling to file.").PlaceHolder("FILE").String()
profileFlag = kingpin.Flag("profile", "Enable profiling to file.").PlaceHolder("FILE").Hidden().String()
listFlag = kingpin.Flag("list", "List lexers, styles and formatters.").Bool()
tokensFlag = kingpin.Flag("tokens", "Dump raw tokens.").Bool()
lexerFlag = kingpin.Flag("lexer", "Lexer to use when formatting.").Default("autodetect").Short('l').String()
styleFlag = kingpin.Flag("style", "Style to use for formatting.").Short('s').Default("swapoff").String()
formatterFlag = kingpin.Flag("formatter", "Formatter to use.").Default("terminal").Short('f').String()
@ -89,11 +88,6 @@ func selexer(path string) chroma.Lexer {
}
func getWriter(w io.Writer) func(*chroma.Token) {
if *tokensFlag {
return func(token *chroma.Token) {
fmt.Printf("%#v\n", token)
}
}
style := styles.Get(*styleFlag)
formatter := formatters.Get(*formatterFlag)
// formatter := formatters.TTY8

16
formatters/tokens.go Normal file

@ -0,0 +1,16 @@
package formatters
import (
"fmt"
"io"
"github.com/alecthomas/chroma"
"github.com/alecthomas/chroma/styles"
)
// Tokens formatter outputs the raw token structures.
var Tokens = Register("tokens", FormatterFunc(func(w io.Writer, s *styles.Style) (func(*chroma.Token), error) {
return func(token *chroma.Token) {
fmt.Fprintln(w, token.GoString())
}, nil
}))