package testclient

import (
	"encoding/json"
	"net/http"
)

type CustomResponseWriter struct {
	body       []byte
	statusCode int
	header     http.Header
}

func newCustomResponseWriter() *CustomResponseWriter {
	return &CustomResponseWriter{
		header: http.Header{},
	}
}

func (w *CustomResponseWriter) Header() http.Header {
	return w.header
}

func (w *CustomResponseWriter) Write(b []byte) (int, error) {
	w.body = b
	return 0, nil
}

func (w *CustomResponseWriter) WriteHeader(statusCode int) {
	w.statusCode = statusCode
}

func InterceptBody[T any](w *CustomResponseWriter, in T) (*T, error) {
	err := json.Unmarshal(w.body, &in)
	if err != nil {
		return nil, err
	}
	return &in, nil
}