package handlers

import (
	"encoding/json"
	"fmt"
	"log"
	"net/http"

	"github.com/play-with-docker/play-with-docker/config"
	"github.com/play-with-docker/play-with-docker/recaptcha"
)

type NewSessionResponse struct {
	SessionId string `json:"session_id"`
	Hostname  string `json:"hostname"`
}

func NewSession(rw http.ResponseWriter, req *http.Request) {
	req.ParseForm()
	if !recaptcha.IsHuman(req, rw) {
		// User it not a human
		rw.WriteHeader(http.StatusForbidden)
		return
	}

	reqDur := req.Form.Get("session-duration")
	stack := req.Form.Get("stack")

	if stack != "" {
		if ok, err := stackExists(stack); err != nil {
			log.Printf("Error retrieving stack: %s", err)
			rw.WriteHeader(http.StatusInternalServerError)
			return
		} else if !ok {
			log.Printf("Stack [%s] could not be found", stack)
			rw.WriteHeader(http.StatusBadRequest)
			return
		}

	}
	duration := config.GetDuration(reqDur)
	s, err := core.SessionNew(duration, stack, "")
	if err != nil {
		log.Println(err)
		//TODO: Return some error code
	} else {
		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
		}

		if stack != "" {
			http.Redirect(rw, req, fmt.Sprintf("http://%s/p/%s", hostname, s.Id), http.StatusFound)
			return
		}
		http.Redirect(rw, req, fmt.Sprintf("http://%s/p/%s", hostname, s.Id), http.StatusFound)
	}
}

func stackExists(stack string) (bool, error) {
	resp, err := http.Head("https://raw.githubusercontent.com/play-with-docker/stacks/master" + stack)
	if err != nil {
		return false, err
	}
	defer resp.Body.Close()

	return resp.StatusCode == 200, nil

}