Reorganize files

1 files changed, 36 insertions, 0 deletions

diff --git a/input/stdin.go b/input/stdin.go
new file mode 100644
index 0000000..5acfa5e
--- /dev/null
+++ b/input/stdin.go
@@ -0,0 +1,36 @@
+package input
+
+import (
+	"bufio"
+	"fmt"
+	"io"
+	"os"
+
+	"github.com/Shugyousha/stasher/work"
+)
+
+type StdinInput struct {
+	retchan chan *work.Work
+}
+
+func (i *StdinInput) Start() chan *work.Work {
+	i.retchan = make(chan *work.Work, 100)
+	r := bufio.NewReader(os.Stdin)
+
+	go func() {
+		for {
+			bs, err := r.ReadBytes(byte('\n'))
+			if err == io.EOF {
+				break
+			}
+			if err != nil {
+				fmt.Printf("Error when reading input from Stdin: %q", err)
+				os.Exit(1)
+			}
+			i.retchan <- &work.Work{Data: bs}
+		}
+		close(i.retchan)
+	}()
+
+	return i.retchan
+}