package http

import (
	"fmt"
	"io/ioutil"
	"net/http"
	"os"
	"strings"

	"github.com/Shugyousha/stasher/input"
	"github.com/Shugyousha/stasher/registry"
	"github.com/Shugyousha/stasher/work"
)

func init() {
	registry.Inputregistry["http"] = New
}

type HTTPInput struct {
	retchan chan *work.Work
	prefix  string
	port    string
}

func New(conf map[string]string) input.Input {
	prefix := conf["prefix"]
	if prefix == "" {
		fmt.Fprintf(os.Stderr, "Need a prefix when setting up http input. Exiting.\n")
		os.Exit(1)
	}
	prefix = strings.Replace(prefix, "\"", "", -1)

	port := conf["port"]
	if port == "" {
		fmt.Fprintf(os.Stderr, "Need a port number when setting up http input. Exiting.\n")
		os.Exit(1)
	}
	port = strings.Replace(port, "\"", "", -1)

	if port[0] != ':' {
		port = ":" + port
	}

	return &HTTPInput{prefix: prefix, port: port}
}

func (hi *HTTPInput) httphandler(w http.ResponseWriter, r *http.Request) {
	if r.Method != "POST" {
		fmt.Printf("Expected POST method was: %q\n", r.Method)
		return
	}
	all, err := ioutil.ReadAll(r.Body)
	wrk := &work.Work{Data: all}
	if err != nil {
		fmt.Printf("Error when reading HTTP request body: %q\n", err)
		wrk.SetError(err)
	}

	hi.retchan <- wrk
}

func (hi *HTTPInput) Start() chan *work.Work {
	hi.retchan = make(chan *work.Work, 100)

	go func() {
		http.HandleFunc("/"+hi.prefix, hi.httphandler)
		err := http.ListenAndServe(hi.port, nil)
		fmt.Printf("Error when serving HTTP: %q\n", err)
		close(hi.retchan)
	}()

	return hi.retchan
}