package webcontext

import (
	"context"
	"net/http"
)

// WebContext wraps the standard request and response writer
type WebContext struct {
	request        *http.Request
	responseWriter http.ResponseWriter
}

// NewWebContext creates a new WebContext from http.ResponseWriter and *http.Request
func NewWebContext(w http.ResponseWriter, r *http.Request) *WebContext {
	return &WebContext{
		request:        r,
		responseWriter: w,
	}
}

// Context returns the request's context
func (c *WebContext) Context() context.Context {
	return c.request.Context()
}

// WithContext returns a shallow copy of c with its context changed to ctx
func (c *WebContext) WithContext(ctx context.Context) *WebContext {
	c2 := new(WebContext)
	*c2 = *c
	c2.request = c2.request.WithContext(ctx)
	return c2
}

// ResponseWriter returns the response writer
func (c *WebContext) ResponseWriter() http.ResponseWriter {
	return c.responseWriter
}

// SetResponseWriter sets the response writer for the context
func (c *WebContext) SetResponseWriter(w http.ResponseWriter) {
	c.responseWriter = w
}

// Request returns the HTTP request
func (c *WebContext) Request() *http.Request {
	return c.request
}

// GetRequestID returns the request ID from the context
func (c *WebContext) GetRequestID() string {
	if id := c.request.Context().Value(requestIDKey); id != nil {
		return id.(string)
	}
	return ""
}

// SetRequestID stores the request ID in the context
func (c *WebContext) SetRequestID(id string) {
	ctx := context.WithValue(c.request.Context(), requestIDKey, id)
	c.request = c.request.WithContext(ctx)
}