summaryrefslogtreecommitdiffstats
path: root/sandbox/hyper/process/main.go
blob: 214dade97d81dcda6fac746a5d36057cff9a1b69 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
package main

import "json"
import "log"
import "http"
import "gorilla.googlecode.com/hg/gorilla/mux"
import "os"
import "fmt"
import "bytes"

import "hyper/process"

var proc = map[string]*hyper.Process{}

// TODO Retrieve Process, Write, Kill [autokill], get exit code

func RespondJSON(res http.ResponseWriter, v interface{}) os.Error {
	content, err := json.Marshal(v)
	if err == nil {
		log.Printf("< %s", content)
		res.Header().Set("Content-Type", "application/json; charset=\"utf-8\"")
		res.WriteHeader(http.StatusOK)
		res.Write(content)
	} else {
		log.Printf("%s while json.Marshal(%s)", err, v)
	}
	return err
}

func CreateProcessHandler(res http.ResponseWriter, req *http.Request) {
	if p, err := hyper.NewProcess(req); err == nil {
		id := p.Id()
		proc[id] = p
		RespondJSON(res, &map[string]string{
			"path": fmt.Sprintf("/proc/%s", id),
		})
	} else {
		log.Printf("%s", err)
		res.WriteHeader(http.StatusInternalServerError)
	}
}

func RetrieveProcess(res http.ResponseWriter, req *http.Request) {
	if p := proc[mux.Vars(req)["id"]]; p != nil {
		RespondJSON(res, p)
	} else {
		res.WriteHeader(http.StatusNotFound)
	}
}

func FeedProcess(res http.ResponseWriter, req *http.Request) {
	if p := proc[mux.Vars(req)["id"]]; p != nil {
		body := make([]byte, 4096)
		if _, err := req.Body.Read(body); err == nil {
			body = bytes.TrimRight(body, string([]byte{0}))
			p.Write(body)
			//if err := p.Write(body); err == nil {
			RespondJSON(res, true)
			//}
		}
	} else {
		res.WriteHeader(http.StatusNotFound)
	}
}

func main() {

	// Gorilla
	mux.HandleFunc("/proc", CreateProcessHandler).Methods("POST")
	mux.HandleFunc("/proc/{id}", RetrieveProcess).Methods("GET")
	mux.HandleFunc("/proc/{id}", FeedProcess).Methods("POST")

	err := http.ListenAndServe("0.0.0.0:8888", mux.DefaultRouter)
	if err != nil {
		log.Fatal("ListenAndServe: ", err.String())
	}
}