wasm-http-server/sw.js

importScripts('https://cdn.jsdelivr.net/gh/golang/go@go1.15.7/misc/wasm/wasm_exec.js')

// FIXME use an options object with { env, argv }
const startWasm = async (wasm, WASMHTTP_HANDLER_ID, WASMHTTP_PATH, args) => {
  const go = new Go()
  go.env = {
    WASMHTTP_HANDLER_ID,
    WASMHTTP_PATH,
  }
  go.argv = [wasm, ...args]
  const { instance } = await WebAssembly.instantiateStreaming(fetch(wasm), go.importObject)
  return go.run(instance)
}

const trimStart = (s, c) => {
  let r = s
  while (r.startsWith(c)) r = r.slice(c.length)
  return r
}

const trimEnd = (s, c) => {
  let r = s
  while (r.endsWith(c)) r = r.slice(0, -c.length)
  return r
}

// addEventListener('install', (event) => {
//   event.waitUntil(skipWaiting())
// })

const running = new Set()
let nextHandlerId = 1
const handlerResolvers = {}
const handlers = []

self.wasmhttp = {
  registerHandler: (handlerId, handler) => {
    handlerResolvers[handlerId](handler)
    delete handlerResolvers[handlerId]
  },
}

console.log('aha!')

addEventListener('activate', event => {
  console.log('activate!')
  event.waitUntil(clients.claim())
})

addEventListener('message', async ({ data }) => {
  console.log('message!', data)

  if (data.type !== 'wasmhttp.register') return

  const { wasm, base, args } = data

  let path = new URL(registration.scope).pathname
  if (base && base !== '') path = `${trimEnd(path, '/')}/${trimStart(base, '/')}`

  const key = `${wasm}:${path}`

  if (!running.has(key)) {
    const handlerId = `${nextHandlerId++}`
    const handler = new Promise(resolve => handlerResolvers[handlerId] = resolve)

    startWasm(wasm, handlerId, path, args)
    running.add(key)

    // FIXME try catch
    handlers.push([path, await handler])
  }
})

addEventListener('fetch', e => {
  console.log('fetch')

  const { pathname } = new URL(e.request.url)
  const [, handler] = handlers.find(([path]) => pathname.startsWith(path)) || []
  if (!handler) return

  e.respondWith(handler(e.request))
})