fzf/src/reader.go

package fzf

// #include <unistd.h>
import "C"

import (
	"bufio"
	"io"
	"os"
	"os/exec"
)

const defaultCommand = `find * -path '*/\.*' -prune -o -type f -print -o -type l -print 2> /dev/null`

// Reader reads from command or standard input
type Reader struct {
	pusher   func(string)
	eventBox *EventBox
}

// ReadSource reads data from the default command or from standard input
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 = defaultCommand
		}
		r.readFromCommand(cmd)
	} else {
		r.readFromStdin()
	}
	r.eventBox.Set(EvtReadFin, 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(EvtReadNew, nil)
		}
	}
}

func (r *Reader) readFromStdin() {
	r.feed(os.Stdin)
}

func (r *Reader) readFromCommand(cmd string) {
	listCommand := exec.Command("sh", "-c", cmd)
	out, err := listCommand.StdoutPipe()
	if err != nil {
		return
	}
	err = listCommand.Start()
	if err != nil {
		return
	}
	defer listCommand.Wait()
	r.feed(out)
}