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{}

// 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
}

var _ js.Wrapper = ResponseWriter{}

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

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

	if len(rw.header) != 0 {
		headers := make(map[string]interface{}, len(rw.header))
		for k := range rw.header {
			headers[k] = rw.header.Get(k)
		}
		init.Set("headers", headers)
	}

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

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