Add platform-specific terminal init & restore

tea.go

@@ -6,8 +6,6 @@ import (
 	"log"
 	"log/syslog"
 	"strings"
-
-	"github.com/pkg/term"
 )
 
 // Msg represents an action. It's used by Update to update the UI.
@@ -39,7 +37,6 @@ type Program struct {
 	update        Update
 	view          View
 	subscriptions []Sub
-	rw            io.ReadWriter
 }
 
 // ErrMsg is just a regular message containing an error. We handle it in Update
@@ -93,18 +90,11 @@ func (p *Program) Start() error {
 		linesRendered int
 	)
 
-	tty, err := term.Open("/dev/tty")
+	err := initTerminal()
 	if err != nil {
 		return err
 	}
-
+	defer restoreTerminal()
-	p.rw = tty
-	tty.SetRaw()
-	hideCursor()
-	defer func() {
-		showCursor()
-		tty.Restore()
-	}()
 
 	// Initialize program
 	model, cmd = p.init()
@@ -122,7 +112,7 @@ func (p *Program) Start() error {
 	// need that we're enabling it by default.
 	go func() {
 		for {
-			msg, _ := ReadKey(p.rw)
+			msg, _ := ReadKey(os.Stdin)
 			msgs <- KeyMsg(msg)
 		}
 	}()
@@ -183,7 +173,7 @@ func (p *Program) render(model Model, linesRendered int) int {
 	if linesRendered > 0 {
 		clearLines(linesRendered)
 	}
-	io.WriteString(p.rw, view)
+	io.WriteString(os.Stdout, view)
 	return strings.Count(view, "\r\n")
 }
 

tty_unix.go

@@ -0,0 +1,28 @@
+// +build darwin dragonfly freebsd linux netbsd openbsd solaris
+
+package tea
+
+import (
+	"github.com/pkg/term"
+)
+
+var (
+	tty *term.Term
+)
+
+func initTerminal() error {
+	var err error
+	tty, err = term.Open("/dev/tty")
+	if err != nil {
+		return err
+	}
+
+	tty.SetRaw()
+	hideCursor()
+	return nil
+}
+
+func restoreTerminal() {
+	showCursor()
+	tty.Restore()
+}

tty_windows.go

@@ -0,0 +1,12 @@
+// +build windows
+
+package tea
+
+func initTerminal() error {
+	hideCursor()
+	return nil
+}
+
+func restoreTerminal() {
+	showCursor()
+}