From fea69e5bab87f163771c3642effa53a96cb9381d Mon Sep 17 00:00:00 2001
From: Francesc Campoy <campoy@golang.org>
Date: Mon, 14 Oct 2013 14:14:08 -0700
Subject: [PATCH] go.tools/present: add support for non Go code execution

R=adg, r
CC=golang-dev
https://golang.org/cl/14615043
---
 playground/socket/socket.go | 39 ++++++++++++++++++++++++++++++++++++-
 1 file changed, 38 insertions(+), 1 deletion(-)

diff --git a/playground/socket/socket.go b/playground/socket/socket.go
index e24d21c9..b8bde032 100644
--- a/playground/socket/socket.go
+++ b/playground/socket/socket.go
@@ -140,7 +140,13 @@ func startProcess(id, body string, out chan<- *Message, opt *Options) *process {
 		out:  out,
 		done: make(chan struct{}),
 	}
-	if err := p.start(body, opt); err != nil {
+	var err error
+	if path, args := shebang(body); path != "" {
+		err = p.startProcess(path, args, body)
+	} else {
+		err = p.start(body, opt)
+	}
+	if err != nil {
 		p.end(err)
 		return nil
 	}
@@ -157,6 +163,37 @@ func (p *process) Kill() {
 	<-p.done // block until process exits
 }
 
+// shebang looks for a shebang ('#!') at the beginning of the passed string.
+// If found, it returns the path and args after the shebang.
+func shebang(body string) (path string, args []string) {
+	body = strings.TrimSpace(body)
+	if !strings.HasPrefix(body, "#!") {
+		return "", nil
+	}
+	if i := strings.Index(body, "\n"); i >= 0 {
+		body = body[:i]
+	}
+	fs := strings.Fields(body[2:])
+	return fs[0], fs[1:]
+}
+
+// startProcess starts a given program given its path and passing the given body
+// to the command standard input.
+func (p *process) startProcess(path string, args []string, body string) error {
+	cmd := &exec.Cmd{
+		Path:   path,
+		Args:   args,
+		Stdin:  strings.NewReader(body),
+		Stdout: &messageWriter{id: p.id, kind: "stdout", out: p.out},
+		Stderr: &messageWriter{id: p.id, kind: "stderr", out: p.out},
+	}
+	if err := cmd.Start(); err != nil {
+		return err
+	}
+	p.run = cmd
+	return nil
+}
+
 // start builds and starts the given program, sending its output to p.out,
 // and stores the running *exec.Cmd in the run field.
 func (p *process) start(body string, opt *Options) error {