From 34e22a8cc0e9828f4d0afeec94c7dd9206ea9f6d Mon Sep 17 00:00:00 2001
From: Jeff Lindsay <progrium@gmail.com>
Date: Sun, 20 May 2018 13:56:57 -0500
Subject: [PATCH] basic idea of runnable api sketched out

Signed-off-by: Jeff Lindsay <progrium@gmail.com>
---
 _examples/ssh-docker/docker.go   |   2 +
 _examples/ssh-docker/runnable.go | 155 +++++++++++++++++++++++++++++++
 2 files changed, 157 insertions(+)
 create mode 100644 _examples/ssh-docker/runnable.go

diff --git a/_examples/ssh-docker/docker.go b/_examples/ssh-docker/docker.go
index 6c2b347..4942e51 100644
--- a/_examples/ssh-docker/docker.go
+++ b/_examples/ssh-docker/docker.go
@@ -92,6 +92,7 @@ func dockerRun(cfg *container.Config, sess ssh.Session) (status int64, cleanup f
 	if err != nil {
 		return
 	}
+
 	if cfg.Tty {
 		_, winCh, _ := sess.Pty()
 		go func() {
@@ -107,6 +108,7 @@ func dockerRun(cfg *container.Config, sess ssh.Session) (status int64, cleanup f
 			}
 		}()
 	}
+
 	resultC, errC := docker.ContainerWait(ctx, res.ID, container.WaitConditionNotRunning)
 	select {
 	case err = <-errC:
diff --git a/_examples/ssh-docker/runnable.go b/_examples/ssh-docker/runnable.go
new file mode 100644
index 0000000..7a130c9
--- /dev/null
+++ b/_examples/ssh-docker/runnable.go
@@ -0,0 +1,155 @@
+package main
+
+import (
+	"context"
+	"io"
+	"strings"
+	"sync"
+
+	"github.com/docker/docker/api/types"
+	"github.com/docker/docker/api/types/container"
+	"github.com/docker/docker/client"
+	"github.com/gliderlabs/ssh"
+)
+
+type Runnable interface {
+	Env() []string
+	Setenv(k string, v string)
+	Stdin() io.Writer
+	Stdout() io.Reader
+	Stderr() io.Reader
+	ExitStatus() int
+	Run() error
+	IsPty() bool
+	Resize(width, height int)
+}
+
+func Example() {
+	ssh.Handle(func(sess ssh.Session) {
+		client, err := client.NewEnvClient()
+		if err != nil {
+			panic(err)
+		}
+		runnable, err := NewDockerRunnable(client, &container.Config{})
+		err = Attach(sess, runnable)
+		if err != nil {
+			panic(err)
+		}
+		sess.Exit(runnable.ExitStatus())
+	})
+}
+
+func Attach(sess ssh.Session, runnable Runnable) error {
+	ptyReq, winCh, _ := sess.Pty()
+
+	// IO
+	go func() {
+		io.Copy(runnable.Stdin(), sess)
+	}()
+	var wg sync.WaitGroup
+	wg.Add(1)
+	go func() {
+		defer wg.Done()
+		io.Copy(sess, runnable.Stdout())
+	}()
+	if !runnable.IsPty() {
+		wg.Add(1)
+		go func() {
+			defer wg.Done()
+			io.Copy(sess.Stderr(), runnable.Stderr())
+		}()
+	}
+	// Environment
+	for _, v := range sess.Environ() {
+		kvp := strings.SplitN(v, "=", 2)
+		if len(kvp) > 1 {
+			runnable.Setenv(kvp[0], kvp[1])
+		}
+	}
+
+	// Interactive
+	if runnable.IsPty() {
+		// Window resizing
+		go func() {
+			for win := range winCh {
+				runnable.Resize(win.Width, win.Height)
+			}
+		}()
+		// Terminal type
+		runnable.Setenv("TERM", ptyReq.Term)
+		// Terminal modes... TODO
+	}
+
+	err := runnable.Run()
+	if err != nil {
+		return err
+	}
+	return nil
+}
+
+type DockerRunnable struct {
+	client     *client.Client
+	env        []string
+	exitStatus int
+
+	containerCfg *container.Config
+	containerID  string
+
+	sync.Mutex
+}
+
+func NewDockerRunnable(client *client.Client, cfg *container.Config) (runnable *DockerRunnable, err error) {
+	run := &DockerRunnable{
+		client:       client,
+		containerCfg: cfg,
+	}
+	ctx := context.Background()
+	res, err := client.ContainerCreate(ctx, cfg, nil, nil, "")
+	if err != nil {
+		return nil, err
+	}
+	run.containerID = res.ID
+	return run, nil
+}
+
+func (dr *DockerRunnable) Env() []string {
+	dr.Lock()
+	defer dr.Unlock()
+	return dr.env
+}
+func (dr *DockerRunnable) Setenv(k string, v string) {
+	dr.Lock()
+	defer dr.Unlock()
+	dr.env = append(dr.env, strings.Join([]string{k, v}, "="))
+}
+func (dr *DockerRunnable) Stdin() io.Writer {
+	return nil
+}
+func (dr *DockerRunnable) Stdout() io.Reader {
+	return nil
+}
+func (dr *DockerRunnable) Stderr() io.Reader {
+	return nil
+}
+func (dr *DockerRunnable) ExitStatus() int {
+	dr.Lock()
+	defer dr.Unlock()
+	return dr.exitStatus
+}
+func (dr *DockerRunnable) Run() error {
+	return nil
+}
+func (dr *DockerRunnable) IsPty() bool {
+	dr.Lock()
+	defer dr.Unlock()
+	return dr.containerCfg.Tty
+}
+func (dr *DockerRunnable) Resize(width, height int) {
+	dr.Lock()
+	defer dr.Unlock()
+	ctx := context.Background()
+	dr.client.ContainerResize(ctx, dr.containerID, types.ResizeOptions{
+		Height: uint(height),
+		Width:  uint(width),
+	})
+}