73 lines
1.3 KiB
Go
73 lines
1.3 KiB
Go
package fzf
|
|
|
|
import (
|
|
"bufio"
|
|
"io"
|
|
"os"
|
|
|
|
"github.com/junegunn/fzf/src/util"
|
|
)
|
|
|
|
// Reader reads from command or standard input
|
|
type Reader struct {
|
|
pusher func([]byte) bool
|
|
eventBox *util.EventBox
|
|
delimNil bool
|
|
}
|
|
|
|
// ReadSource reads data from the default command or from standard input
|
|
func (r *Reader) ReadSource() {
|
|
if util.IsTty() {
|
|
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) {
|
|
delim := byte('\n')
|
|
if r.delimNil {
|
|
delim = '\000'
|
|
}
|
|
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) {
|
|
r.eventBox.Set(EvtReadNew, nil)
|
|
}
|
|
}
|
|
if err != nil {
|
|
break
|
|
}
|
|
}
|
|
}
|
|
|
|
func (r *Reader) readFromStdin() {
|
|
r.feed(os.Stdin)
|
|
}
|
|
|
|
func (r *Reader) readFromCommand(cmd string) {
|
|
listCommand := util.ExecCommand(cmd)
|
|
out, err := listCommand.StdoutPipe()
|
|
if err != nil {
|
|
return
|
|
}
|
|
err = listCommand.Start()
|
|
if err != nil {
|
|
return
|
|
}
|
|
defer listCommand.Wait()
|
|
r.feed(out)
|
|
}
|