[FEATURE] Use ReadableStream for Response (#15)

* feat: uses ReadableStream for Response

* chore: rebuilds other examples

serve.go

@@ -1,57 +1,92 @@
 package wasmhttp
 
 import (
+	"context"
 	"fmt"
 	"net/http"
 	"strings"
 	"syscall/js"
 
-	promise "github.com/nlepage/go-js-promise"
+	"github.com/nlepage/go-wasm-http-server/internal/safejs"
+)
+
+var (
+	wasmhttp = safejs.Safe(js.Global().Get("wasmhttp"))
 )
 
 // Serve serves HTTP requests using handler or http.DefaultServeMux if handler is nil.
-func Serve(handler http.Handler) func() {
-	var h = handler
+func Serve(handler http.Handler) (func(), error) {
+	h := handler
 	if h == nil {
 		h = http.DefaultServeMux
 	}
 
-	var prefix = js.Global().Get("wasmhttp").Get("path").String()
+	prefix, err := wasmhttp.GetString("path")
+	if err != nil {
+		return nil, err
+	}
+
 	for strings.HasSuffix(prefix, "/") {
 		prefix = strings.TrimSuffix(prefix, "/")
 	}
 
 	if prefix != "" {
-		var mux = http.NewServeMux()
+		mux := http.NewServeMux()
 		mux.Handle(prefix+"/", http.StripPrefix(prefix, h))
 		h = mux
 	}
 
-	var cb = js.FuncOf(func(_ js.Value, args []js.Value) interface{} {
-		var resPromise, resolve, reject = promise.New()
+	handlerValue, err := safejs.FuncOf(func(_ safejs.Value, args []safejs.Value) interface{} {
+		res, err := NewResponse()
+		if err != nil {
+			panic(err)
+		}
 
 		go func() {
+			ctx, cancel := context.WithCancel(res.Context())
+
 			defer func() {
-				if r := recover(); r != nil {
-					if err, ok := r.(error); ok {
-						reject(fmt.Sprintf("wasmhttp: panic: %+v\n", err))
-					} else {
-						reject(fmt.Sprintf("wasmhttp: panic: %v\n", r))
-					}
+				cancel()
+			}()
+
+			defer func() {
+				if err := res.Close(); err != nil {
+					panic(err)
 				}
 			}()
 
-			var res = NewResponseRecorder()
+			defer func() {
+				if r := recover(); r != nil {
+					var errStr string
+					if err, ok := r.(error); ok {
+						errStr = err.Error()
+					} else {
+						errStr = fmt.Sprintf("%s", r)
+					}
+					res.WriteError(errStr)
+				}
+			}()
 
-			h.ServeHTTP(res, Request(args[0]))
+			req, err := Request(safejs.Unsafe(args[0]))
+			if err != nil {
+				res.WriteError(err.Error())
+				return
+			}
 
-			resolve(res.JSResponse())
+			req = req.WithContext(ctx)
+
+			h.ServeHTTP(res, req)
 		}()
 
-		return resPromise
+		return res.JSValue()
 	})
+	if err != nil {
+		return nil, err
+	}
 
-	js.Global().Get("wasmhttp").Call("setHandler", cb)
+	if _, err = wasmhttp.Call("setHandler", handlerValue); err != nil {
+		return nil, err
+	}
 
-	return cb.Release
+	return handlerValue.Release, nil
 }