jz 0.1.0 → 0.2.0

package/wasi.js

@@ -17,16 +17,35 @@
  * Create WASI import object for WebAssembly instantiation.
  * @param {object} [opts]
  * @param {function} [opts.write] - Custom write: (fd, text) => void
+ * @param {function} [opts.read] - Custom read: (fd, buf: Uint8Array) => bytesRead
  */
 export function wasi(opts = {}) {
   let mem = null
-  const write = opts.write || ((fd, text) => {
-    if (fd === 1) typeof process !== 'undefined' && process.stdout ? process.stdout.write(text) : console.log(text.replace(/\n$/, ''))
-    else typeof process !== 'undefined' && process.stderr ? process.stderr.write(text) : console.warn(text.replace(/\n$/, ''))
-  })
+  const fallbackWrite = (fd, text) => {
+    const stream = fd === 1 ? globalThis.process?.stdout : globalThis.process?.stderr
+    if (stream && typeof stream.write === 'function') {
+      try { stream.write(text); return }
+      catch {}
+    }
+    const msg = text.replace(/\n$/, '')
+    ;(fd === 1 ? console.log : console.warn)(msg)
+  }
+  const write = opts.write || fallbackWrite
 
   return {
     wasi_snapshot_preview1: {
+      fd_read(fd, iovs, iovs_len, nread) {
+        const dv = new DataView(mem.buffer)
+        let total = 0
+        for (let i = 0; i < iovs_len; i++) {
+          const ptr = dv.getUint32(iovs + i * 8, true)
+          const len = dv.getUint32(iovs + i * 8 + 4, true)
+          const buf = new Uint8Array(mem.buffer, ptr, len)
+          total += opts.read ? (opts.read(fd, buf) || 0) : 0
+        }
+        dv.setUint32(nread, total, true)
+        return 0
+      },
       fd_write(fd, iovs, iovs_len, nwritten) {
         const dv = new DataView(mem.buffer)
         let written = 0