main: attach os.Stdin to command with -run (#66)

myitcv · Feb 27, 2019 · 331f452 · 331f452
1 parent 392b803
commit 331f452
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 2 deletions.
diff --git a/main.go b/main.go
@@ -347,6 +347,7 @@ func mainerr() error {
 			case *fRun:
 				run := exec.Command(target, runArgs...)
 				run.Args[0] = path.Base(mp.ImportPath)
+				run.Stdin = os.Stdin
 				run.Stdout = os.Stdout
 				run.Stderr = os.Stderr
 				if err := run.Run(); err != nil {

diff --git a/testdata/run.txt b/testdata/run.txt
@@ -1,8 +1,10 @@
 # Various -run tests
 
-# Ensure os.Args[0] is set correctly
+# Ensure os.Args[0] is set correctly and that Stdin is correctly
+# attached
+stdin input
 gobin -m -run example.com/blah
-stdout '^blah$'
+cmp stdout output.golden
 
 -- go.mod --
 module example.com/blah
@@ -12,9 +14,18 @@ package main
 
 import (
 	"fmt"
+	"io"
 	"os"
 )
 
 func main() {
 	fmt.Println(os.Args[0])
+	if _, err := io.Copy(os.Stdout, os.Stdin); err != nil {
+		panic(err)
+	}
 }
+-- input --
+This is stdin
+-- output.golden --
+blah
+This is stdin