Files
play-with-docker/handlers/new_session.go
2017-05-15 16:09:13 -03:00

70 lines
1.9 KiB
Go

package handlers
import (
"encoding/json"
"fmt"
"log"
"net/http"
"path"
"github.com/play-with-docker/play-with-docker/config"
"github.com/play-with-docker/play-with-docker/services"
)
type NewSessionResponse struct {
SessionId string `json:"session_id"`
Hostname string `json:"hostname"`
}
func NewSession(rw http.ResponseWriter, req *http.Request) {
req.ParseForm()
if !services.IsHuman(req, rw) {
// User it not a human
rw.WriteHeader(http.StatusForbidden)
return
}
reqDur := req.Form.Get("session-duration")
stack := req.Form.Get("stack")
duration := services.GetDuration(reqDur)
s, err := services.NewSession(duration)
if err != nil {
log.Println(err)
//TODO: Return some error code
} else {
s.StackFile = stack
if stack != "" {
go deployStack(s)
}
hostname := fmt.Sprintf("%s.%s", config.PWDCName, req.Host)
// If request is not a form, return sessionId in the body
if req.Header.Get("X-Requested-With") == "XMLHttpRequest" {
resp := NewSessionResponse{SessionId: s.Id, Hostname: hostname}
rw.Header().Set("Content-Type", "application/json")
json.NewEncoder(rw).Encode(resp)
return
}
http.Redirect(rw, req, fmt.Sprintf("http://%s/p/%s", hostname, s.Id), http.StatusFound)
}
}
func deployStack(s *services.Session) {
i, err := services.NewInstance(s, services.InstanceConfig{})
if err != nil {
log.Printf("Error creating instance for stack [%s]: %s\n", s.StackFile, err)
}
err = i.UploadFromURL("https://raw.githubusercontent.com/play-with-docker/stacks/master" + s.StackFile)
if err != nil {
log.Printf("Error uploading stack file [%s]: %s\n", s.StackFile, err)
}
fileName := path.Base(s.StackFile)
code, err := services.Exec(i.Name, []string{"docker-compose", "-f", "/var/run/pwd/uploads/" + fileName, "up", "-d"})
if err != nil {
log.Printf("Error executing stack [%s]: %s\n", s.StackFile, err)
}
log.Printf("Stack execution finished with code %d\n", code)
}