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

support windows colors

Browse Source
This commit is contained in:
Yasuhiro Matsumoto
2017-09-19 00:27:47 +09:00
parent 2dc153de8a
commit 686fad4ef4
1 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
cmd/chroma/main.go
View File

@@ -7,6 +7,7 @@ import (
"io/ioutil" "io/ioutil"
"os" "os"
"os/signal" "os/signal"
"runtime"
"runtime/pprof" "runtime/pprof"
"strings" "strings"
@@ -17,6 +18,8 @@ import (
"github.com/alecthomas/chroma/formatters/html" "github.com/alecthomas/chroma/formatters/html"
"github.com/alecthomas/chroma/lexers" "github.com/alecthomas/chroma/lexers"
"github.com/alecthomas/chroma/styles" "github.com/alecthomas/chroma/styles"
"github.com/mattn/go-colorable"
"github.com/mattn/go-isatty"
) )
var ( var (
@@ -59,7 +62,11 @@ command, for Go.
}() }()
defer pprof.StopCPUProfile() defer pprof.StopCPUProfile()
} }
w := bufio.NewWriterSize(os.Stdout, 16384) var out io.Writer = os.Stdout
if runtime.GOOS == "windows" && isatty.IsTerminal(os.Stdout.Fd()) {
out = colorable.NewColorableStdout()
}
w := bufio.NewWriterSize(out, 16384)
defer w.Flush() defer w.Flush()
if *htmlFlag { if *htmlFlag {
*formatterFlag = "html" *formatterFlag = "html"