package httpserver

import (
	"context"
	"encoding/json"
	"github.com/julienschmidt/httprouter"
	"net/http"
	"net/url"
)

type Ctx struct {
	httpContext context.Context
	writer      http.ResponseWriter
	request     *http.Request

	routerParams httprouter.Params

	IHttpResponse
}

func NewCtx(writer http.ResponseWriter, request *http.Request) *Ctx {
	ctx := &Ctx{
		httpContext: request.Context(),
		writer:      writer,
		request:     request,

		IHttpResponse: NewHttpResponse(writer),
	}

	return ctx
}

func (ctx *Ctx) _setRouterParams(params httprouter.Params) {
	ctx.routerParams = params
}

func (ctx *Ctx) Context() context.Context {
	return ctx.httpContext
}

func (ctx *Ctx) WithValue(key, val interface{}) {
	ctx.httpContext = context.WithValue(ctx.httpContext, key, val)
}

func (ctx *Ctx) Form() url.Values {
	r := ctx.request

	contentType := r.Header.Get("Content-Type")

	form := url.Values{}
	switch contentType {
	case "application/json":
		var fm map[string]string

		decoder := json.NewDecoder(r.Body)
		decoder.Decode(&fm)

		for k, v := range fm {
			form.Add(k, v)
		}

	default:
		r.ParseForm()
		form = r.Form
	}

	return form
}

func (ctx *Ctx) Value(key interface{}) interface{} {
	return ctx.httpContext.Value(key)
}

func (ctx *Ctx) SetResponseHandler(response IHttpResponse) {
	ctx.IHttpResponse = response
}

func (ctx *Ctx) Writer() http.ResponseWriter {
	return ctx.writer
}

func (ctx *Ctx) Request() *http.Request {
	return ctx.request
}

func (ctx *Ctx) Param(name string) string {
	return ctx.routerParams.ByName(name)
}

func (ctx *Ctx) WriteHeader(code int) {
	ctx.writer.WriteHeader(code)
}

func (ctx *Ctx) Header() http.Header {
	return ctx.writer.Header()
}

func (ctx *Ctx) Write(resp []byte) (int, error) {
	return ctx.writer.Write(resp)
}

func (ctx *Ctx) WriteStr(resp string) (int, error) {
	return ctx.writer.Write([]byte(resp))
}