package main

import (
	"bytes"
	"encoding/json"
	"flag"
	"io"
	"log"
	"net/http"
	"net/http/httputil"
	"net/url"
	"os"
	"strconv"
	"text/template"
)

func main() {
	fs := flag.NewFlagSet(os.Args[0], flag.ContinueOnError)
	p := fs.Int("p", 28080, "port")
	t := fs.String("t", "{{ . }}", "template")
	y := fs.String("y", "http://localhost:41912", "target")
	if err := fs.Parse(os.Args[1:]); err != nil {
		panic(err)
	}

	tmpl, err := template.New("").Parse(*t)
	if err != nil {
		panic(err)
	}

	u, err := url.Parse(*y)
	if err != nil {
		panic(err)
	}
	proxy := httputil.NewSingleHostReverseProxy(u)

	handle := func(w http.ResponseWriter, r *http.Request) error {
		var v interface{}
		if err := json.NewDecoder(r.Body).Decode(&v); err != nil {
			return err
		}
		buff := bytes.NewBuffer(nil)
		if err := tmpl.Execute(buff, v); err != nil {
			return err
		}
		r.Body = io.NopCloser(buff)
		r.ContentLength = int64(len(buff.Bytes()))
		proxy.ServeHTTP(w, r)
		return nil
	}
	log.Println("listening on", *p)
	if err := http.ListenAndServe(":"+strconv.Itoa(*p), http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		if err := handle(w, r); err != nil {
			http.Error(w, err.Error(), http.StatusBadRequest)
		}
	})); err != nil {
		panic(err)
	}
}