@wasm-fmt/gofmt 0.0.1

package/.node-version

@@ -0,0 +1 @@
+v18.17.0

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2023 wasm-fmt
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/README.md

@@ -0,0 +1,16 @@
+# Build from source
+
+```bash
+# 1. clone this repo
+git clone https://github.com/wasm-fmt/gofmt.git
+
+# 2. install TinyGo https://tinygo.org/getting-started/install/
+
+# 3. build
+npm run go_wasm # copy `wasm_exec.js`
+npm run patch
+npm run build
+
+# 4. test
+npm run test
+```

package/go.mod

@@ -0,0 +1,3 @@
+module github.com/wasm-fmt/gofmt
+
+go 1.20

package/go_wasm.js

@@ -0,0 +1,410 @@
+// Copyright 2018 The Go Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style
+// license that can be found in the LICENSE file.
+//
+// This file has been modified for use by the TinyGo compiler.
+
+	const encoder = new TextEncoder("utf-8");
+	const decoder = new TextDecoder("utf-8");
+	var logLine = [];
+
+	export class Go {
+		constructor() {
+			this._callbackTimeouts = new Map();
+			this._nextCallbackTimeoutID = 1;
+
+			const mem = () => {
+				// The buffer may change when requesting more memory.
+				return new DataView(this._inst.exports.memory.buffer);
+			}
+
+			const setInt64 = (addr, v) => {
+				mem().setUint32(addr + 0, v, true);
+				mem().setUint32(addr + 4, Math.floor(v / 4294967296), true);
+			}
+
+			const getInt64 = (addr) => {
+				const low = mem().getUint32(addr + 0, true);
+				const high = mem().getInt32(addr + 4, true);
+				return low + high * 4294967296;
+			}
+
+			const loadValue = (addr) => {
+				const f = mem().getFloat64(addr, true);
+				if (f === 0) {
+					return undefined;
+				}
+				if (!isNaN(f)) {
+					return f;
+				}
+
+				const id = mem().getUint32(addr, true);
+				return this._values[id];
+			}
+
+			const storeValue = (addr, v) => {
+				const nanHead = 0x7FF80000;
+
+				if (typeof v === "number") {
+					if (isNaN(v)) {
+						mem().setUint32(addr + 4, nanHead, true);
+						mem().setUint32(addr, 0, true);
+						return;
+					}
+					if (v === 0) {
+						mem().setUint32(addr + 4, nanHead, true);
+						mem().setUint32(addr, 1, true);
+						return;
+					}
+					mem().setFloat64(addr, v, true);
+					return;
+				}
+
+				switch (v) {
+					case undefined:
+						mem().setFloat64(addr, 0, true);
+						return;
+					case null:
+						mem().setUint32(addr + 4, nanHead, true);
+						mem().setUint32(addr, 2, true);
+						return;
+					case true:
+						mem().setUint32(addr + 4, nanHead, true);
+						mem().setUint32(addr, 3, true);
+						return;
+					case false:
+						mem().setUint32(addr + 4, nanHead, true);
+						mem().setUint32(addr, 4, true);
+						return;
+				}
+
+				let id = this._ids.get(v);
+				if (id === undefined) {
+					id = this._idPool.pop();
+					if (id === undefined) {
+						id = this._values.length;
+					}
+					this._values[id] = v;
+					this._goRefCounts[id] = 0;
+					this._ids.set(v, id);
+				}
+				this._goRefCounts[id]++;
+				let typeFlag = 1;
+				switch (typeof v) {
+					case "string":
+						typeFlag = 2;
+						break;
+					case "symbol":
+						typeFlag = 3;
+						break;
+					case "function":
+						typeFlag = 4;
+						break;
+				}
+				mem().setUint32(addr + 4, nanHead | typeFlag, true);
+				mem().setUint32(addr, id, true);
+			}
+
+			const loadSlice = (array, len, cap) => {
+				return new Uint8Array(this._inst.exports.memory.buffer, array, len);
+			}
+
+			const loadSliceOfValues = (array, len, cap) => {
+				const a = new Array(len);
+				for (let i = 0; i < len; i++) {
+					a[i] = loadValue(array + i * 8);
+				}
+				return a;
+			}
+
+			const loadString = (ptr, len) => {
+				return decoder.decode(new DataView(this._inst.exports.memory.buffer, ptr, len));
+			}
+
+			const timeOrigin = Date.now() - performance.now();
+			this.importObject = {
+				wasi_snapshot_preview1: {
+					// https://github.com/WebAssembly/WASI/blob/main/phases/snapshot/docs.md#fd_write
+					fd_write: function(fd, iovs_ptr, iovs_len, nwritten_ptr) {
+						let nwritten = 0;
+						if (fd == 1) {
+							for (let iovs_i=0; iovs_i<iovs_len;iovs_i++) {
+								let iov_ptr = iovs_ptr+iovs_i*8; // assuming wasm32
+								let ptr = mem().getUint32(iov_ptr + 0, true);
+								let len = mem().getUint32(iov_ptr + 4, true);
+								nwritten += len;
+								for (let i=0; i<len; i++) {
+									let c = mem().getUint8(ptr+i);
+									if (c == 13) { // CR
+										// ignore
+									} else if (c == 10) { // LF
+										// write line
+										let line = decoder.decode(new Uint8Array(logLine));
+										logLine = [];
+										console.log(line);
+									} else {
+										logLine.push(c);
+									}
+								}
+							}
+						} else {
+							console.error('invalid file descriptor:', fd);
+						}
+						mem().setUint32(nwritten_ptr, nwritten, true);
+						return 0;
+					},
+					fd_close: () => 0,      // dummy
+					fd_fdstat_get: () => 0, // dummy
+					fd_seek: () => 0,       // dummy
+					"proc_exit": (code) => {
+						if (global.process) {
+							// Node.js
+							process.exit(code);
+						} else {
+							// Can't exit in a browser.
+							throw 'trying to exit with code ' + code;
+						}
+					},
+					random_get: (bufPtr, bufLen) => {
+						crypto.getRandomValues(loadSlice(bufPtr, bufLen));
+						return 0;
+					},
+				},
+				env: {
+					// func ticks() float64
+					"runtime.ticks": () => {
+						return timeOrigin + performance.now();
+					},
+
+					// func sleepTicks(timeout float64)
+					"runtime.sleepTicks": (timeout) => {
+						// Do not sleep, only reactivate scheduler after the given timeout.
+						setTimeout(this._inst.exports.go_scheduler, timeout);
+					},
+
+					// func finalizeRef(v ref)
+					"syscall/js.finalizeRef": (v_addr) => {
+						const id = mem().getUint32(v_addr, true);
+						this._goRefCounts[id]--;
+						if (this._goRefCounts[id] === 0) {
+							const v = this._values[id];
+							this._values[id] = null;
+							this._ids.delete(v);
+							this._idPool.push(id);
+						}
+					},
+
+					// func stringVal(value string) ref
+					"syscall/js.stringVal": (ret_ptr, value_ptr, value_len) => {
+						const s = loadString(value_ptr, value_len);
+						storeValue(ret_ptr, s);
+					},
+
+					// func valueGet(v ref, p string) ref
+					"syscall/js.valueGet": (retval, v_addr, p_ptr, p_len) => {
+						let prop = loadString(p_ptr, p_len);
+						let value = loadValue(v_addr);
+						let result = Reflect.get(value, prop);
+						storeValue(retval, result);
+					},
+
+					// func valueSet(v ref, p string, x ref)
+					"syscall/js.valueSet": (v_addr, p_ptr, p_len, x_addr) => {
+						const v = loadValue(v_addr);
+						const p = loadString(p_ptr, p_len);
+						const x = loadValue(x_addr);
+						Reflect.set(v, p, x);
+					},
+
+					// func valueDelete(v ref, p string)
+					"syscall/js.valueDelete": (v_addr, p_ptr, p_len) => {
+						const v = loadValue(v_addr);
+						const p = loadString(p_ptr, p_len);
+						Reflect.deleteProperty(v, p);
+					},
+
+					// func valueIndex(v ref, i int) ref
+					"syscall/js.valueIndex": (ret_addr, v_addr, i) => {
+						storeValue(ret_addr, Reflect.get(loadValue(v_addr), i));
+					},
+
+					// valueSetIndex(v ref, i int, x ref)
+					"syscall/js.valueSetIndex": (v_addr, i, x_addr) => {
+						Reflect.set(loadValue(v_addr), i, loadValue(x_addr));
+					},
+
+					// func valueCall(v ref, m string, args []ref) (ref, bool)
+					"syscall/js.valueCall": (ret_addr, v_addr, m_ptr, m_len, args_ptr, args_len, args_cap) => {
+						const v = loadValue(v_addr);
+						const name = loadString(m_ptr, m_len);
+						const args = loadSliceOfValues(args_ptr, args_len, args_cap);
+						try {
+							const m = Reflect.get(v, name);
+							storeValue(ret_addr, Reflect.apply(m, v, args));
+							mem().setUint8(ret_addr + 8, 1);
+						} catch (err) {
+							storeValue(ret_addr, err);
+							mem().setUint8(ret_addr + 8, 0);
+						}
+					},
+
+					// func valueInvoke(v ref, args []ref) (ref, bool)
+					"syscall/js.valueInvoke": (ret_addr, v_addr, args_ptr, args_len, args_cap) => {
+						try {
+							const v = loadValue(v_addr);
+							const args = loadSliceOfValues(args_ptr, args_len, args_cap);
+							storeValue(ret_addr, Reflect.apply(v, undefined, args));
+							mem().setUint8(ret_addr + 8, 1);
+						} catch (err) {
+							storeValue(ret_addr, err);
+							mem().setUint8(ret_addr + 8, 0);
+						}
+					},
+
+					// func valueNew(v ref, args []ref) (ref, bool)
+					"syscall/js.valueNew": (ret_addr, v_addr, args_ptr, args_len, args_cap) => {
+						const v = loadValue(v_addr);
+						const args = loadSliceOfValues(args_ptr, args_len, args_cap);
+						try {
+							storeValue(ret_addr, Reflect.construct(v, args));
+							mem().setUint8(ret_addr + 8, 1);
+						} catch (err) {
+							storeValue(ret_addr, err);
+							mem().setUint8(ret_addr+ 8, 0);
+						}
+					},
+
+					// func valueLength(v ref) int
+					"syscall/js.valueLength": (v_addr) => {
+						return loadValue(v_addr).length;
+					},
+
+					// valuePrepareString(v ref) (ref, int)
+					"syscall/js.valuePrepareString": (ret_addr, v_addr) => {
+						const s = String(loadValue(v_addr));
+						const str = encoder.encode(s);
+						storeValue(ret_addr, str);
+						setInt64(ret_addr + 8, str.length);
+					},
+
+					// valueLoadString(v ref, b []byte)
+					"syscall/js.valueLoadString": (v_addr, slice_ptr, slice_len, slice_cap) => {
+						const str = loadValue(v_addr);
+						loadSlice(slice_ptr, slice_len, slice_cap).set(str);
+					},
+
+					// func valueInstanceOf(v ref, t ref) bool
+					"syscall/js.valueInstanceOf": (v_addr, t_addr) => {
+ 						return loadValue(v_addr) instanceof loadValue(t_addr);
+					},
+
+					// func copyBytesToGo(dst []byte, src ref) (int, bool)
+					"syscall/js.copyBytesToGo": (ret_addr, dest_addr, dest_len, dest_cap, source_addr) => {
+						let num_bytes_copied_addr = ret_addr;
+						let returned_status_addr = ret_addr + 4; // Address of returned boolean status variable
+
+						const dst = loadSlice(dest_addr, dest_len);
+						const src = loadValue(source_addr);
+						if (!(src instanceof Uint8Array || src instanceof Uint8ClampedArray)) {
+							mem().setUint8(returned_status_addr, 0); // Return "not ok" status
+							return;
+						}
+						const toCopy = src.subarray(0, dst.length);
+						dst.set(toCopy);
+						setInt64(num_bytes_copied_addr, toCopy.length);
+						mem().setUint8(returned_status_addr, 1); // Return "ok" status
+					},
+
+					// copyBytesToJS(dst ref, src []byte) (int, bool)
+					// Originally copied from upstream Go project, then modified:
+					//   https://github.com/golang/go/blob/3f995c3f3b43033013013e6c7ccc93a9b1411ca9/misc/wasm/wasm_exec.js#L404-L416
+					"syscall/js.copyBytesToJS": (ret_addr, dest_addr, source_addr, source_len, source_cap) => {
+						let num_bytes_copied_addr = ret_addr;
+						let returned_status_addr = ret_addr + 4; // Address of returned boolean status variable
+
+						const dst = loadValue(dest_addr);
+						const src = loadSlice(source_addr, source_len);
+						if (!(dst instanceof Uint8Array || dst instanceof Uint8ClampedArray)) {
+							mem().setUint8(returned_status_addr, 0); // Return "not ok" status
+							return;
+						}
+						const toCopy = src.subarray(0, dst.length);
+						dst.set(toCopy);
+						setInt64(num_bytes_copied_addr, toCopy.length);
+						mem().setUint8(returned_status_addr, 1); // Return "ok" status
+					},
+				}
+			};
+		}
+
+		storeString(str) {
+			const addr = this._inst.exports.getBuffer();
+			const buf = this._inst.exports.memory.buffer;
+
+			const mem = new Uint8Array(buf);
+			const view = mem.subarray(addr);
+			return encoder.encodeInto(str, view).written;
+		}
+
+		loadString(len) {
+			const addr = this._inst.exports.getBuffer();
+			const buf = this._inst.exports.memory.buffer;
+
+			return decoder.decode(new DataView(buf, addr, len));
+		}
+
+		async run(instance) {
+			this._inst = instance;
+			this._values = [ // JS values that Go currently has references to, indexed by reference id
+				NaN,
+				0,
+				null,
+				true,
+				false,
+				"undefined" != typeof globalThis ? globalThis : global || self,
+				this,
+			];
+			this._goRefCounts = []; // number of references that Go has to a JS value, indexed by reference id
+			this._ids = new Map();  // mapping from JS values to reference ids
+			this._idPool = [];      // unused ids that have been garbage collected
+			this.exited = false;    // whether the Go program has exited
+
+			const mem = new DataView(this._inst.exports.memory.buffer)
+
+			while (true) {
+				const callbackPromise = new Promise((resolve) => {
+					this._resolveCallbackPromise = () => {
+						if (this.exited) {
+							throw new Error("bad callback: Go program has already exited");
+						}
+						setTimeout(resolve, 0); // make sure it is asynchronous
+					};
+				});
+				this._inst.exports._start();
+				if (this.exited) {
+					break;
+				}
+				await callbackPromise;
+			}
+		}
+
+		_resume() {
+			if (this.exited) {
+				throw new Error("Go program has already exited");
+			}
+			this._inst.exports.resume();
+			if (this.exited) {
+				this._resolveExitPromise();
+			}
+		}
+
+		_makeFuncWrapper(id) {
+			const go = this;
+			return function () {
+				const event = { id: id, this: this, args: arguments };
+				go._pendingEvent = event;
+				go._resume();
+				return event.result;
+			};
+		}
+	}

package/lib.d.ts

@@ -0,0 +1 @@
+export declare function format(input: string): Promise<string>;

package/lib.js

@@ -0,0 +1,33 @@
+import { Go } from "./go_wasm.js";
+const go = new Go();
+
+let mod;
+
+export async function format(input) {
+  if (!mod) {
+    const url = new URL("lib.wasm", import.meta.url);
+
+    if (url.protocol === "file:") {
+      const fs = await import("node:fs");
+      const bytes = fs.readFileSync(url);
+      mod = new WebAssembly.Module(bytes);
+    } else if ("instantiateStreaming" in WebAssembly) {
+      mod = await WebAssembly.compileStreaming(fetch(url), go.importObject);
+    } else {
+      const response = await fetch(url);
+      const bytes = await response.arrayBuffer();
+      mod = new WebAssembly.Module(bytes);
+    }
+  }
+
+  const inst = new WebAssembly.Instance(mod, go.importObject);
+  go.run(inst);
+
+  const input_len = go.storeString(input);
+  const output_len = inst.exports.format(input_len);
+  if (output_len < 0) {
+    throw new Error(go.loadString(-output_len));
+  }
+
+  return go.loadString(output_len);
+}

package/package.json

@@ -0,0 +1,23 @@
+{
+  "name": "@wasm-fmt/gofmt",
+  "author": "magic-akari <akari.ccino@gamil.com>",
+  "version": "0.0.1",
+  "description": "wasm based gofmt",
+  "main": "lib.js",
+  "types": "lib.d.ts",
+  "type": "module",
+  "scripts": {
+    "go_wasm": "cp $(tinygo env TINYGOROOT)/targets/wasm_exec.js ./go_wasm.js",
+    "patch": "git apply ./go_wasm.patch",
+    "build": "tinygo build -o=lib.wasm -target=wasm -no-debug -stack-size=24kb ./src/lib.go",
+    "test": "node --test"
+  },
+  "engines": {
+    "node": ">=16.17.0"
+  },
+  "keywords": [],
+  "license": "MIT",
+  "publishConfig": {
+    "access": "public"
+  }
+}

package/src/lib.go

@@ -0,0 +1,31 @@
+package main
+
+import "go/format"
+
+const buf_len = 8192
+
+var buf [buf_len]byte
+
+//go:export getBuffer
+func GetBuffer() *byte {
+	return &buf[0]
+}
+
+//go:export format
+func Format(input_len uint) int {
+	input := buf[:input_len]
+	output, err := format.Source(input)
+	if err != nil {
+		return -copy(buf[:], []byte(err.Error()))
+	}
+	result := len(output)
+
+	if result > buf_len {
+		return -copy(buf[:], []byte("Buffer out of memory"))
+	}
+
+	copy(buf[:], output)
+	return result
+}
+
+func main() {}

package/test.js

@@ -0,0 +1,31 @@
+import assert from "node:assert";
+import fs from "node:fs/promises";
+import test from "node:test";
+import { format } from "./lib.js";
+
+const files = (await fs.readdir("testdata"))
+  .filter((f) => f.endsWith(".input"))
+  .map((f) => {
+    return {
+      input_name: f,
+      golden_name: f.replace(".input", ".golden"),
+    };
+  });
+
+for (const { input_name, golden_name } of files) {
+  await test(
+    `format ${input_name}`,
+    { skip: input_name[0] === "." },
+    async () => {
+      const [input, expected] = await Promise.all(
+        [input_name, golden_name].map((f) =>
+          fs.readFile(`testdata/${f}`, "utf-8")
+        )
+      );
+
+      const actual = await format(input);
+
+      assert.strictEqual(actual, expected);
+    }
+  );
+}