diff --git a/check.anko b/check.anko index 42168fd..6598ad0 100644 --- a/check.anko +++ b/check.anko @@ -2,6 +2,8 @@ e = 5 v = 53 println(e + v) +println("cock", Cmd("ls").Stdout()) + if v < 55 { println("it fucking works") } diff --git a/src/tool/script/main.go b/src/tool/script/main.go index 7614c66..579435b 100644 --- a/src/tool/script/main.go +++ b/src/tool/script/main.go @@ -18,8 +18,11 @@ import ( "github.com/surdeus/goscript/vm" "github.com/surdeus/gomtool/src/mtool" "os/exec" + //"bytes" ) +type Output []byte + type Command struct { input io.Reader *exec.Cmd @@ -36,6 +39,10 @@ var ( //flag *mtool.Flags ) +func (o Output) String() string { + return string(o) +} + func Run(flagSet *mtool.Flags) { printVersion := flagSet.Bool("v", false, "prints out the version and then exits") @@ -81,6 +88,33 @@ func (cmd *Command)IO(input io.Reader, output io.Writer) *Command { return cmd } +func (cmd *Command) Stdout() Output { + r, w := io.Pipe() + cmd = cmd.IO(os.Stdin, w) + + ret := []byte{} + + go func() { + buf := make([]byte, 512) + for { + n, err := r.Read(buf) + ret = append(ret, buf[:n]...) + if err == io.EOF { + break + } else if err != nil { + panic(err) + } + } + }() + + err := cmd.Run() + if err != nil { + panic(err) + } + + return Output(ret) +} + func (cmd *Command) Run() error { input := cmd.input output := cmd.output