Merge pull request #15 from sjrd/js-interop

Implement use-site JS interop.

Author: tanishiking

Diff for: build.sbt

@@ -1,3 +1,4 @@
+import org.scalajs.linker.interface.ESVersion
 import org.scalajs.linker.interface.OutputPatterns
 
 val scalaV = "2.13.12"
@@ -9,7 +10,10 @@ inThisBuild(Def.settings(
     "-feature",
     "-deprecation",
     "-Xfatal-warnings",
-  )
+  ),
+  scalaJSLinkerConfig ~= {
+    _.withESFeatures(_.withESVersion(ESVersion.ES2016))
+  },
 ))
 
 lazy val cli = project

Diff for: cli/src/main/scala/TestSuites.scala

@@ -9,6 +9,7 @@ object TestSuites {
     TestSuite("testsuite.core.virtualdispatch.VirtualDispatch", "virtualDispatch"),
     TestSuite("testsuite.core.interfacecall.InterfaceCall", "interfaceCall"),
     TestSuite("testsuite.core.asinstanceof.AsInstanceOfTest", "asInstanceOf"),
+    TestSuite("testsuite.core.jsinterop.JSInteropTest", "jsInterop"),
     TestSuite("testsuite.core.hijackedclassesdispatch.HijackedClassesDispatchTest", "hijackedClassesDispatch"),
     TestSuite("testsuite.core.hijackedclassesmono.HijackedClassesMonoTest", "hijackedClassesMono"),
     TestSuite("testsuite.core.hijackedclassesupcast.HijackedClassesUpcastTest", "hijackedClassesUpcast"),

Diff for: loader.mjs

@@ -13,6 +13,14 @@ function stringHashCode(s) {
   return res;
 }
 
+const linkingInfo = Object.freeze({
+  "esVersion": 6,
+  "assumingES6": true,
+  "productionMode": false,
+  "linkerVersion": "1.15.0",
+  "fileLevelThis": this
+});
+
 const scalaJSHelpers = {
   // BinaryOp.===
   is: Object.is,
@@ -79,6 +87,54 @@ const scalaJSHelpers = {
       return 0;
     return 42; // for any JS object
   },
+
+  // JS interop
+  jsGlobalRefGet: (globalRefName) => (new Function("return " + globalRefName))(),
+  jsGlobalRefSet: (globalRefName, v) => {
+    var argName = globalRefName === 'v' ? 'w' : 'v';
+    (new Function(argName, globalRefName + " = " + argName))(v);
+  },
+  jsGlobalRefTypeof: (globalRefName) => (new Function("return typeof " + globalRefName))(),
+  jsNewArray: () => [],
+  jsArrayPush: (a, v) => (a.push(v), a),
+  jsArraySpreadPush: (a, vs) => (a.push(...vs), a),
+  jsNewObject: () => ({}),
+  jsObjectPush: (o, p, v) => (o[p] = v, o),
+  jsSelect: (o, p) => o[p],
+  jsSelectSet: (o, p, v) => o[p] = v,
+  jsNew: (constr, args) => new constr(...args),
+  jsFunctionApply: (f, args) => f(...args),
+  jsMethodApply: (o, m, args) => o[m](...args),
+  jsDelete: (o, p) => { delete o[p]; },
+  jsIsTruthy: (x) => !!x,
+  jsLinkingInfo: () => linkingInfo,
+
+  // Excruciating list of all the JS operators
+  jsUnaryPlus: (a) => +a,
+  jsUnaryMinus: (a) => -a,
+  jsUnaryTilde: (a) => ~a,
+  jsUnaryBang: (a) => !a,
+  jsUnaryTypeof: (a) => typeof a,
+  jsStrictEquals: (a, b) => a === b,
+  jsNotStrictEquals: (a, b) => a !== b,
+  jsPlus: (a, b) => a + b,
+  jsMinus: (a, b) => a - b,
+  jsTimes: (a, b) => a * b,
+  jsDivide: (a, b) => a / b,
+  jsModulus: (a, b) => a % b,
+  jsBinaryOr: (a, b) => a | b,
+  jsBinaryAnd: (a, b) => a & b,
+  jsBinaryXor: (a, b) => a ^ b,
+  jsShiftLeft: (a, b) => a << b,
+  jsArithmeticShiftRight: (a, b) => a >> b,
+  jsLogicalShiftRight: (a, b) => a >>> b,
+  jsLessThan: (a, b) => a < b,
+  jsLessEqual: (a, b) => a <= b,
+  jsGreaterThan: (a, b) => a > b,
+  jsGreaterEqual: (a, b) => a >= b,
+  jsIn: (a, b) => a in b,
+  jsInstanceof: (a, b) => a instanceof b,
+  jsExponent: (a, b) => a ** b,
 }
 
 export async function load(wasmFileName) {

Diff for: sample/src/main/scala/Sample.scala

@@ -2,6 +2,7 @@ package sample
 
 import scala.annotation.tailrec
 
+import scala.scalajs.js
 import scala.scalajs.js.annotation._
 
 //
@@ -16,6 +17,9 @@ object Main {
     val loopFib = fib(new LoopFib {}, i)
     val recFib = fib(new RecFib {}, i)
     val tailrecFib = fib(new TailRecFib {}, i)
+    js.Dynamic.global.console.log(s"loopFib: $loopFib -- recFib: $recFib -- tailrecFib: $tailrecFib")
+    val date = new js.Date(0)
+    js.Dynamic.global.console.log(date)
     loopFib == recFib && loopFib == tailrecFib
   }
   def fib(fib: Fib, n: Int): Int = fib.fib(n)

Diff for: test-suite/src/main/scala/testsuite/core/JSInteropTest.scala

@@ -0,0 +1,103 @@
+package testsuite.core.jsinterop
+
+import scala.scalajs.js
+import scala.scalajs.js.annotation._
+
+object JSInteropTest {
+  def main(): Unit = { val _ = test() }
+
+  @JSExportTopLevel("jsInterop")
+  def test(): Boolean = {
+    testBasicTopLevel() &&
+      testBasicStatic() &&
+      testBasicInstance() &&
+      testArray() &&
+      testObject() &&
+      testOperators() &&
+      testLinkingInfo()
+  }
+
+  def testBasicTopLevel(): Boolean = {
+    ("" + js.undefined) == "undefined" &&
+      js.eval("3 + 4").asInstanceOf[Int] == 7 &&
+      js.isUndefined(())
+  }
+
+  def testBasicStatic(): Boolean = {
+    js.Math.PI == 3.1415926535897932 &&
+      js.Math.abs(-5.6) == 5.6 &&
+      js.Math.clz32(6548) == 19 &&
+      js.Math.min(5, 8, 2, 12, 3) == 2
+  }
+
+  def testBasicInstance(): Boolean = {
+    val d = new js.Date(1710190169564.0)
+    d.getTime() == 1710190169564.0 &&
+      d.getUTCFullYear() == 2024 && {
+        d.setTime(0.0)
+        d.getTime() == 0.0
+      }
+  }
+
+  def testArray(): Boolean = {
+    val a = js.Array(1, 5, 3)
+    a.length == 3 &&
+      a(0) == 1 &&
+      a(2) == 3 && {
+        a(0) = 65
+        a.push(78)
+        a.length == 4 &&
+          a(0) == 65 &&
+          a(3) == 78
+      }
+  }
+
+  def testObject(): Boolean = {
+    val o = new js.Object().asInstanceOf[js.Dynamic]
+    js.isUndefined(o.foo) && {
+      o.foo = 5
+      o.hasOwnProperty("foo").asInstanceOf[Boolean] &&
+        o.foo.asInstanceOf[Int] == 5 && {
+          js.special.delete(o, "foo")
+          !o.hasOwnProperty("foo").asInstanceOf[Boolean]
+        }
+    }
+  }
+
+  def testOperators(): Boolean = {
+    val x = 5.asInstanceOf[js.Dynamic]
+    val y = 11.asInstanceOf[js.Dynamic]
+    same(+x, 5) &&
+      same(-x, -5) &&
+      same(~x, -6) &&
+      same(!x, false) &&
+      same(js.typeOf(x), "number") && // regular typeof
+      same(js.typeOf(js.Dynamic.global.Date), "function") && // typeof global ref
+      same(x + y, 16) &&
+      same(x - y, -6) &&
+      same(x * y, 55) &&
+      same(x / y, 0.45454545454545453) &&
+      same(y % x, 1) &&
+      same(x << 3.asInstanceOf[js.Dynamic], 40) &&
+      same(x >> 1.asInstanceOf[js.Dynamic], 2) &&
+      same(x >>> 2.asInstanceOf[js.Dynamic], 1) &&
+      same(x & y, 1) &&
+      same(x | y, 15) &&
+      same(x ^ y, 14) &&
+      same(x < y, true) &&
+      same(x <= y, true) &&
+      same(x > y, false) &&
+      same(x >= y, false) &&
+      same(x && y, 11) &&
+      same(x || y, 5) &&
+      same(x ** 3.asInstanceOf[js.Dynamic], 125)
+  }
+
+  def testLinkingInfo(): Boolean = {
+    val linkingInfo = scala.scalajs.runtime.linkingInfo
+    linkingInfo.esVersion >= 6 &&
+      linkingInfo.assumingES6 == true
+  }
+
+  def same(a: js.Any, b: js.Any): Boolean = js.special.strictEquals(a, b)
+}

Diff for: wasm/src/main/scala/Compiler.scala

@@ -7,9 +7,8 @@ import org.scalajs.ir
 import org.scalajs.ir.Trees._
 import org.scalajs.ir.Types._
 
-import org.scalajs.linker.frontend.LinkerFrontendImpl
-import org.scalajs.linker.interface.{IRFile, ModuleInitializer}
-import org.scalajs.linker.standard.{LinkedClass, SymbolRequirement}
+import org.scalajs.linker.interface._
+import org.scalajs.linker.standard._
 
 import org.scalajs.logging.{Level, ScalaConsoleLogger}
 
@@ -30,10 +29,12 @@ object Compiler {
     implicit val context: WasmContext = new WasmContext(module)
     println("compiling...  ")
 
-    val config = LinkerFrontendImpl
-      .Config()
+    val config = StandardConfig()
+      .withESFeatures(_.withESVersion(ESVersion.ES2016)) // to be able to link `**`
+      .withSemantics(_.optimized) // because that's the only thing we actually support at the moment
       .withOptimizer(false)
-    val linkerFrontend = LinkerFrontendImpl(config)
+
+    val linkerFrontend = StandardLinkerFrontend(config)
 
     /* The symbol requirements of our back-end.
      * The symbol requirements tell the LinkerFrontend that we need these
@@ -75,6 +76,9 @@ object Compiler {
       onlyModule.topLevelExports.foreach { tle =>
         builder.transformTopLevelExport(tle)
       }
+
+      context.complete()
+
       val textOutput = new converters.WasmTextWriter().write(module)
       FS.writeFileSync(s"./target/$outputName.wat", textOutput.getBytes().toTypedArray)
 

Diff for: wasm/src/main/scala/converters/WasmBinaryWriter.scala

@@ -67,6 +67,8 @@ final class WasmBinaryWriter(module: WasmModule) {
     writeSection(fullOutput, SectionFunction)(writeFunctionSection(_))
     writeSection(fullOutput, SectionGlobal)(writeGlobalSection(_))
     writeSection(fullOutput, SectionExport)(writeExportSection(_))
+    if (module.startFunction.isDefined)
+      writeSection(fullOutput, SectionStart)(writeStartSection(_))
     writeSection(fullOutput, SectionCode)(writeCodeSection(_))
 
     fullOutput.result()
@@ -146,6 +148,10 @@ final class WasmBinaryWriter(module: WasmModule) {
     }
   }
 
+  private def writeStartSection(buf: Buffer): Unit = {
+    writeFuncIdx(buf, module.startFunction.get)
+  }
+
   private def writeCodeSection(buf: Buffer): Unit = {
     buf.vec(module.definedFunctions) { func =>
       buf.byteLengthSubSection(writeFunc(_, func))

Diff for: wasm/src/main/scala/converters/WasmTextWriter.scala

@@ -29,6 +29,7 @@ class WasmTextWriter {
         module.definedFunctions.foreach(writeFunction)
         module.globals.foreach(writeGlobal)
         module.exports.foreach(writeExport)
+        module.startFunction.foreach(writeStart)
       }
     )
     // context.gcTypes
@@ -194,6 +195,14 @@ class WasmTextWriter {
     }
   )
 
+  private def writeStart(startFunction: WasmFunctionName)(implicit b: WatBuilder): Unit = {
+    b.newLineList(
+      "start", {
+        b.appendElement(startFunction.show)
+      }
+    )
+  }
+
   private def writeImmediate(i: WasmImmediate, instr: WasmInstr)(implicit b: WatBuilder): Unit = {
     val str = i match {
       case WasmImmediate.I64(v)          => v.toString()

Diff for: wasm/src/main/scala/ir2wasm/Preprocessor.scala

@@ -22,21 +22,10 @@ object Preprocessor {
   }
 
   private def preprocess(clazz: LinkedClass)(implicit ctx: WasmContext): Unit = {
-    clazz.kind match {
-      case ClassKind.ModuleClass | ClassKind.Class | ClassKind.Interface |
-          ClassKind.HijackedClass =>
-        collectMethods(clazz)
-      case ClassKind.JSClass | ClassKind.JSModuleClass | ClassKind.NativeJSModuleClass |
-          ClassKind.AbstractJSType | ClassKind.NativeJSClass =>
-        println(s"${clazz.kind} ${clazz.fullName}")
-        ???
-    }
-  }
-
-  private def collectMethods(clazz: LinkedClass)(implicit ctx: WasmContext): Unit = {
     val infos = clazz.methods.filterNot(_.flags.namespace.isConstructor).map { method =>
       makeWasmFunctionInfo(clazz, method)
     }
+
     ctx.putClassInfo(
       clazz.name.name,
       new WasmClassInfo(
@@ -46,7 +35,9 @@ object Preprocessor {
         clazz.fields.collect { case f: IRTrees.FieldDef => Names.WasmFieldName(f.name.name) },
         clazz.superClass.map(_.name),
         clazz.interfaces.map(_.name),
-        clazz.ancestors
+        clazz.ancestors,
+        clazz.jsNativeLoadSpec,
+        clazz.jsNativeMembers.map(m => m.name.name -> m.jsNativeLoadSpec).toMap
       )
     )
   }

Diff for: wasm/src/main/scala/ir2wasm/WasmBuilder.scala

@@ -28,7 +28,7 @@ class WasmBuilder {
       case ClassKind.Class         => transformClass(clazz)
       case ClassKind.HijackedClass => transformHijackedClass(clazz)
       case ClassKind.Interface     => transformInterface(clazz)
-      case _                       => ???
+      case _                       => ()
     }
   }