summaryrefslogtreecommitdiff
path: root/src/reader.go
blob: 0e1f0a9c0026f280cb712585a15d187dc1419e9a (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
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)
}