fzf/src/reader.go

73 lines
1.3 KiB
Go
Raw Normal View History

2015-01-01 19:49:30 +00:00
package fzf
import (
"bufio"
"io"
"os"
2015-01-12 03:56:17 +00:00
"github.com/junegunn/fzf/src/util"
2015-01-01 19:49:30 +00:00
)
2015-01-11 18:01:24 +00:00
// Reader reads from command or standard input
2015-01-01 19:49:30 +00:00
type Reader struct {
pusher func([]byte) bool
2015-01-12 03:56:17 +00:00
eventBox *util.EventBox
delimNil bool
2015-01-01 19:49:30 +00:00
}
2015-01-11 18:01:24 +00:00
// ReadSource reads data from the default command or from standard input
2015-01-01 19:49:30 +00:00
func (r *Reader) ReadSource() {
2015-01-12 03:56:17 +00:00
if util.IsTty() {
2015-01-01 19:49:30 +00:00
cmd := os.Getenv("FZF_DEFAULT_COMMAND")
if len(cmd) == 0 {
2015-01-11 18:01:24 +00:00
cmd = defaultCommand
2015-01-01 19:49:30 +00:00
}
r.readFromCommand(cmd)
} else {
r.readFromStdin()
}
2015-01-11 18:01:24 +00:00
r.eventBox.Set(EvtReadFin, nil)
2015-01-01 19:49:30 +00:00
}
func (r *Reader) feed(src io.Reader) {
delim := byte('\n')
if r.delimNil {
delim = '\000'
}
2016-08-15 16:52:24 +00:00
reader := bufio.NewReaderSize(src, readerBufferSize)
for {
// ReadBytes returns err != nil if and only if the returned data does not
// end in delim.
bytea, err := reader.ReadBytes(delim)
if len(bytea) > 0 {
if err == nil {
bytea = bytea[:len(bytea)-1]
}
if r.pusher(bytea) {
2015-07-21 18:21:20 +00:00
r.eventBox.Set(EvtReadNew, nil)
}
2015-01-01 19:49:30 +00:00
}
if err != nil {
break
}
2015-01-01 19:49:30 +00:00
}
}
func (r *Reader) readFromStdin() {
r.feed(os.Stdin)
}
func (r *Reader) readFromCommand(cmd string) {
listCommand := util.ExecCommand(cmd)
2015-01-01 19:49:30 +00:00
out, err := listCommand.StdoutPipe()
if err != nil {
return
}
err = listCommand.Start()
if err != nil {
return
}
defer listCommand.Wait()
r.feed(out)
}