feat: support runtime.wasmWrite

Author: x1unix

web/src/lib/go/wrapper/wrapper.ts

@@ -1,6 +1,6 @@
 import { StackReader } from '../stack'
 import { MemoryInspector } from '../debug'
-import { Bool, GoStringType, Int32 } from '../types'
+import { Bool, GoStringType, Int32, Int64 } from '../types'
 import { type GoInstance, type ImportObject, type PendingEvent } from './interface'
 import { type Func, Ref, RefType } from '../pkg/syscall/js'
 import { MemoryView } from '../memory/view'
@@ -58,17 +58,23 @@ export type CallImportHandler = (sp: number, stack: StackReader, mem: MemoryView
  */
 const defaultCmdline = ['js']
 
+export interface WasmWriter {
+  writeSync: (fd: number, data: Uint8Array) => void
+}
+
 export interface Options {
   debug?: boolean
   debugCalls?: string[]
   globalValue?: any
+  stdoutHandler?: WasmWriter
 }
 
 export class GoWrapper {
   private _inspector: MemoryInspector | null = null
   private _memView: MemoryView | null = null
   private readonly _debug: boolean = false
   private readonly _debugCalls: Set<string>
+  private readonly _stdout?: Options['stdoutHandler']
   private _globalValue: object
   private readonly go: GoInstance
 
@@ -104,11 +110,12 @@ export class GoWrapper {
     return this.go._inst!.exports
   }
 
-  constructor(parent: GoInstance, { debug = false, debugCalls, globalValue }: Options = {}) {
+  constructor(parent: GoInstance, { debug = false, debugCalls, globalValue, stdoutHandler }: Options = {}) {
     this.go = parent
     this._debug = debug
     this._debugCalls = new Set(debugCalls ?? [])
     this._globalValue = globalValue?.Go === GoWrapper ? globalValue : wrapGlobal()
+    this._stdout = stdoutHandler
 
     this.patchImportObject()
   }
@@ -148,6 +155,17 @@ export class GoWrapper {
       this.valueCall(sp, reader)
     })
 
+    if (this._stdout) {
+      this.exportFunction('runtime.wasmWrite', (sp, reader, mem) => {
+        reader.skipHeader()
+        const fd = reader.next<number>(Int64)
+        const bufPtr = reader.next<number>(Int64)
+        const len = reader.next<number>(Int32)
+        const chunk = mem.memory.buffer.slice(bufPtr, bufPtr + len)
+        this._stdout?.writeSync(fd, new Uint8Array(chunk, 0, len))
+      })
+    }
+
     const wasmExitFunc = getImportNamespace(this.go)['runtime.wasmExit']
     this.exportFunction('runtime.wasmExit', (sp, reader) => {
       reader.skipHeader()

web/src/workers/go/go.worker.ts

@@ -33,6 +33,7 @@ class WorkerHandler implements GoExecutor {
 
     const go = new GoWrapper(new globalThis.Go(), {
       globalValue: wrapGlobal(mocks, globalThis),
+      stdoutHandler: fs,
     })
 
     const { instance } = await WebAssembly.instantiate(image, go.importObject)