From f3177305d5572b26f135fc045481358b4eb1bf69 Mon Sep 17 00:00:00 2001
From: Junegunn Choi <junegunn.c@gmail.com>
Date: Fri, 2 Jan 2015 04:49:30 +0900
Subject: Rewrite fzf in Go

---
 src/reader.go | 60 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 60 insertions(+)
 create mode 100644 src/reader.go

(limited to 'src/reader.go')

diff --git a/src/reader.go b/src/reader.go
new file mode 100644
index 00000000..0e1f0a9c
--- /dev/null
+++ b/src/reader.go
@@ -0,0 +1,60 @@
+package fzf
+
+// #include <unistd.h>
+import "C"
+
+import (
+	"bufio"
+	"fmt"
+	"io"
+	"os"
+	"os/exec"
+)
+
+const DEFAULT_COMMAND = "find * -path '*/\\.*' -prune -o -type f -print -o -type l -print 2> /dev/null"
+
+type Reader struct {
+	pusher   func(string)
+	eventBox *EventBox
+}
+
+func (r *Reader) ReadSource() {
+	if int(C.isatty(C.int(os.Stdin.Fd()))) != 0 {
+		cmd := os.Getenv("FZF_DEFAULT_COMMAND")
+		if len(cmd) == 0 {
+			cmd = DEFAULT_COMMAND
+		}
+		r.readFromCommand(cmd)
+	} else {
+		r.readFromStdin()
+	}
+	r.eventBox.Set(EVT_READ_FIN, nil)
+}
+
+func (r *Reader) feed(src io.Reader) {
+	if scanner := bufio.NewScanner(src); scanner != nil {
+		for scanner.Scan() {
+			r.pusher(scanner.Text())
+			r.eventBox.Set(EVT_READ_NEW, nil)
+		}
+	}
+}
+
+func (r *Reader) readFromStdin() {
+	r.feed(os.Stdin)
+}
+
+func (r *Reader) readFromCommand(cmd string) {
+	arg := fmt.Sprintf("%q", cmd)
+	listCommand := exec.Command("sh", "-c", arg[1:len(arg)-1])
+	out, err := listCommand.StdoutPipe()
+	if err != nil {
+		return
+	}
+	err = listCommand.Start()
+	if err != nil {
+		return
+	}
+	defer listCommand.Wait()
+	r.feed(out)
+}
-- 
cgit v1.2.3