package accesslog

import (
	"github.com/netbirdio/netbird/proxy/internal/responsewriter"
)

// statusWriter captures the HTTP status code and bytes written from responses.
// It embeds responsewriter.PassthroughWriter which handles all the optional
// interfaces (Hijacker, Flusher, Pusher) automatically.
type statusWriter struct {
	*responsewriter.PassthroughWriter
	status       int
	bytesWritten int64
}

func (w *statusWriter) WriteHeader(status int) {
	w.status = status
	w.PassthroughWriter.WriteHeader(status)
}

func (w *statusWriter) Write(b []byte) (int, error) {
	n, err := w.PassthroughWriter.Write(b)
	w.bytesWritten += int64(n)
	return n, err
}