Merge pull request #2 from mattn/windows

Windows Support

README.md

@@ -41,7 +41,26 @@ func main() {
   isColorEnabled := true
   banner.Init(os.Stdout, isEnabled, isColorEnabled, bytes.NewBufferString("My Custom Banner"))
 }
+```
+
+If using windows, use go-colorable. This works in all-platforms.
+
+```go
+package main
+
+import (
+	"bytes"
+	"os"
+
+	"github.com/dimiro1/banner"
+	"github.com/mattn/go-colorable"
+)
 
+func main() {
+  isEnabled := true
+  isColorEnabled := true
+  banner.Init(colorable.NewColorableStdout(), isEnabled, isColorEnabled, bytes.NewBufferString("My Custom Banner"))
+}
 ```
 
 # API

autoload/autoload.go

@@ -11,6 +11,7 @@ import (
 	"os"
 
 	"github.com/dimiro1/banner"
+	"github.com/mattn/go-colorable"
 )
 
 func init() {
@@ -36,5 +37,5 @@ func init() {
 		return
 	}
 
-	banner.Init(os.Stdout, isEnabled, isColorEnabled, in)
+	banner.Init(colorable.NewColorableStdout(), isEnabled, isColorEnabled, in)
 }

color.go

@@ -4,7 +4,11 @@
 
 package banner
 
-import "runtime"
+import (
+	"os"
+
+	"github.com/mattn/go-isatty"
+)
 
 const (
 	encodeStart = "\033["
@@ -156,15 +160,19 @@ func (a ansiColor) BrightWhite() string {
 	return outputANSI(a.isColorEnabled, "97")
 }
 
+func (a ansiBackground) Reset() string {
+	return outputANSI(a.isColorEnabled, encodeReset)
+}
+
 func outputANSI(isColorEnabled bool, code string) string {
 	if isColorEnabled && isANSIEnabled() {
 		return encodeStart + code + encodeEnd
 	}
 
 	return ""
+
 }
 
 func isANSIEnabled() bool {
-	// Windows is always disabled.
-	return isTerminal() && (runtime.GOOS != "windows")
+	return isatty.IsTerminal(os.Stdout.Fd())
 }

example/banner.txt

@@ -15,4 +15,4 @@ Compiler: {{ .Compiler }}
 ENV: {{ .Env "GOPATH" }}
 Now: {{ .Now "Monday, 2 Jan 2006" }}
 {{ .AnsiColor.BrightGreen }}This text will appear in Green
-{{ .AnsiColor.BrightRed }}This text will appear in Red
+{{ .AnsiColor.BrightRed }}This text will appear in Red{{ .AnsiColor.Default }}