package hws

import "net/http"

// Wraps the http.ResponseWriter, adding a statusCode field
type wrappedWriter struct {
	http.ResponseWriter
	statusCode int
}

// Extends WriteHeader to the ResponseWriter to add the status code
func (w *wrappedWriter) WriteHeader(statusCode int) {
	w.ResponseWriter.WriteHeader(statusCode)
	w.statusCode = statusCode
}

func (w *wrappedWriter) Unwrap() http.ResponseWriter {
	return w.ResponseWriter
}