wasm-http-server/internal/whutil/response_writer.go

package whutil

import (
	"bytes"
	"net/http"
	"syscall/js"
)

// ResponseWriter implements http.ResponseWriter
type ResponseWriter struct {
	header     http.Header
	buf        *bytes.Buffer
	statusCode int
}

// NewResponseWriter creates a new ResponseWriter
func NewResponseWriter() *ResponseWriter {
	return &ResponseWriter{
		header:     http.Header{},
		buf:        bytes.NewBuffer(nil),
		statusCode: 0,
	}
}

var _ http.ResponseWriter = (*ResponseWriter)(nil)

// Header implements http.ResponseWriter.Header
func (rw *ResponseWriter) Header() http.Header {
	return rw.header
}

// Write implements http.ResponseWriter.Write
func (rw *ResponseWriter) Write(p []byte) (int, error) {
	return rw.buf.Write(p)
}

// WriteHeader implements http.ResponseWriter.WriteHeader
func (rw *ResponseWriter) WriteHeader(statusCode int) {
	rw.statusCode = statusCode
}

// JSResponse builds and returns the equivalent JS Response
func (rw *ResponseWriter) JSResponse() js.Value {
	init := js.Global().Get("Object").New()

	if rw.statusCode != 0 {
		init.Set("status", rw.statusCode)
	}

	if len(rw.header) != 0 {
		init.Set("headers", map[string][]string(rw.header))
	}

	jsBody := js.Global().Get("Uint8Array").New(rw.buf.Len())
	js.CopyBytesToJS(jsBody, rw.buf.Bytes())

	return js.Global().Get("Response").New(jsBody, init)
}