package server

import (
	"errors"
	"io"
	"net/http"
	"unsafe"
)

// maxParams defines the maximum number of parameters per route.
const maxParams = 16

// Context represents the interface for a request & response context.
type Context interface {
	Bytes([]byte) error
	Error(status int, messages ...any) error
	Get(param string) string
	Reader(io.Reader) error
	Request() Request
	Response() Response
	SetStatus(status int)
	String(string) error
}

// context represents a request & response context.
type context struct {
	request     request
	response    response
	paramNames  [maxParams]string
	paramValues [maxParams]string
	paramCount  int
}

// newContext returns a new context from the pool.
func newContext(req *http.Request, res http.ResponseWriter) *context {
	ctx := contextPool.Get().(*context)
	ctx.request.Request = req
	ctx.response.ResponseWriter = res
	ctx.paramCount = 0
	return ctx
}

// Bytes responds with a raw byte slice.
func (ctx *context) Bytes(body []byte) error {
	_, err := ctx.response.Write(body)
	return err
}

// Error is used for sending error messages to the client.
func (ctx *context) Error(status int, messages ...any) error {
	var combined []error

	for _, msg := range messages {
		switch err := msg.(type) {
		case error:
			combined = append(combined, err)
		case string:
			combined = append(combined, errors.New(err))
		}
	}

	ctx.SetStatus(status)
	return errors.Join(combined...)
}

// Get retrieves a parameter.
func (ctx *context) Get(param string) string {
	for i := 0; i < ctx.paramCount; i++ {
		if ctx.paramNames[i] == param {
			return ctx.paramValues[i]
		}
	}

	return ""
}

// Reader sends the contents of the io.Reader without creating an in-memory copy.
func (ctx *context) Reader(reader io.Reader) error {
	_, err := io.Copy(ctx.response, reader)
	return err
}

// Request returns the HTTP request.
func (ctx *context) Request() Request {
	return &ctx.request
}

// Response returns the HTTP response.
func (ctx *context) Response() Response {
	return &ctx.response
}

// SetStatus writes the header with the given HTTP status code.
func (ctx *context) SetStatus(status int) {
	ctx.response.WriteHeader(status)
}

// String responds with the given string.
func (ctx *context) String(body string) error {
	slice := unsafe.Slice(unsafe.StringData(body), len(body))
	return ctx.Bytes(slice)
}

// addParameter adds a new parameter to the context.
func (ctx *context) addParameter(name string, value string) {
	ctx.paramNames[ctx.paramCount] = name
	ctx.paramValues[ctx.paramCount] = value
	ctx.paramCount++
}