BridgeJS: Support @JS class in imported function signatures

Author: kateinoigakukun

Plugins/BridgeJS/Sources/BridgeJSCore/ImportTS.swift

@@ -958,17 +958,7 @@ extension BridgeType {
         case .unsafePointer:
             return LoweringParameterInfo(loweredParameters: [("pointer", .pointer)])
         case .swiftHeapObject(let className):
-            switch context {
-            case .importTS:
-                throw BridgeJSCoreError(
-                    """
-                    swiftHeapObject '\(className)' is not supported in TypeScript imports.
-                    Swift classes can only be used in @JS protocols where Swift owns the instance.
-                    """
-                )
-            case .exportSwift:
-                return LoweringParameterInfo(loweredParameters: [("pointer", .pointer)])
-            }
+            return LoweringParameterInfo(loweredParameters: [("pointer", .pointer)])
         case .swiftProtocol:
             throw BridgeJSCoreError("swiftProtocol is not supported in imported signatures")
         case .caseEnum:
@@ -1038,18 +1028,8 @@ extension BridgeType {
             return LiftingReturnInfo(valueToLift: .i32)
         case .unsafePointer:
             return LiftingReturnInfo(valueToLift: .pointer)
-        case .swiftHeapObject(let className):
-            switch context {
-            case .importTS:
-                throw BridgeJSCoreError(
-                    """
-                    swiftHeapObject '\(className)' cannot be returned from imported TypeScript functions.
-                    JavaScript cannot create Swift heap objects.
-                    """
-                )
-            case .exportSwift:
-                return LiftingReturnInfo(valueToLift: .pointer)
-            }
+        case .swiftHeapObject:
+            return LiftingReturnInfo(valueToLift: .pointer)
         case .swiftProtocol:
             throw BridgeJSCoreError("swiftProtocol is not supported in imported signatures")
         case .caseEnum:

Plugins/BridgeJS/Sources/BridgeJSLink/JSGlueGen.swift

@@ -581,7 +581,7 @@ struct IntrinsicJSFragment: Sendable {
         return IntrinsicJSFragment(
             parameters: ["pointer"],
             printCode: { arguments, scope, printer, cleanupCode in
-                return ["\(name).__construct(\(arguments[0]))"]
+                return ["_exports['\(name)'].__construct(\(arguments[0]))"]
             }
         )
     }
@@ -1985,14 +1985,7 @@ struct IntrinsicJSFragment: Sendable {
         case .jsValue: return .jsValueLiftParameter
         case .unsafePointer: return .identity
         case .swiftHeapObject(let name):
-            switch context {
-            case .importTS:
-                throw BridgeJSLinkError(
-                    message: "swiftHeapObject '\(name)' can only be used in protocol exports, not in \(context)"
-                )
-            case .exportSwift:
-                return .swiftHeapObjectLiftParameter(name)
-            }
+            return .swiftHeapObjectLiftParameter(name)
         case .swiftProtocol: return .jsObjectLiftParameter
         case .void:
             throw BridgeJSLinkError(
@@ -2074,15 +2067,8 @@ struct IntrinsicJSFragment: Sendable {
         case .jsObject: return .jsObjectLowerReturn
         case .jsValue: return .jsValueLowerReturn(context: context)
         case .unsafePointer: return .identity
-        case .swiftHeapObject(let name):
-            switch context {
-            case .importTS:
-                throw BridgeJSLinkError(
-                    message: "swiftHeapObject '\(name)' can only be used in protocol exports, not in \(context)"
-                )
-            case .exportSwift:
-                return .swiftHeapObjectLowerReturn
-            }
+        case .swiftHeapObject:
+            return .swiftHeapObjectLowerReturn
         case .swiftProtocol: return .jsObjectLowerReturn
         case .void: return .void
         case .nullable(let wrappedType, let kind):

Plugins/BridgeJS/Tests/BridgeJSToolTests/Inputs/MacroSwift/SwiftClass.swift

@@ -18,3 +18,5 @@
 
 @JS public class PublicGreeter {}
 @JS package class PackageGreeter {}
+
+@JSFunction func jsRoundTripGreeter(greeter: Greeter) throws(JSException) -> Greeter

Plugins/BridgeJS/Tests/BridgeJSToolTests/__Snapshots__/BridgeJSCodegenTests/SwiftClass.json

@@ -141,5 +141,34 @@
 
     ]
   },
+  "imported" : {
+    "children" : [
+      {
+        "functions" : [
+          {
+            "name" : "jsRoundTripGreeter",
+            "parameters" : [
+              {
+                "name" : "greeter",
+                "type" : {
+                  "swiftHeapObject" : {
+                    "_0" : "Greeter"
+                  }
+                }
+              }
+            ],
+            "returnType" : {
+              "swiftHeapObject" : {
+                "_0" : "Greeter"
+              }
+            }
+          }
+        ],
+        "types" : [
+
+        ]
+      }
+    ]
+  },
   "moduleName" : "TestModule"
 }

Plugins/BridgeJS/Tests/BridgeJSToolTests/__Snapshots__/BridgeJSCodegenTests/SwiftClass.swift

@@ -134,4 +134,22 @@ fileprivate func _bjs_PackageGreeter_wrap(_ pointer: UnsafeMutableRawPointer) ->
 fileprivate func _bjs_PackageGreeter_wrap(_ pointer: UnsafeMutableRawPointer) -> Int32 {
     fatalError("Only available on WebAssembly")
 }
-#endif
+#endif
+
+#if arch(wasm32)
+@_extern(wasm, module: "TestModule", name: "bjs_jsRoundTripGreeter")
+fileprivate func bjs_jsRoundTripGreeter(_ greeter: UnsafeMutableRawPointer) -> UnsafeMutableRawPointer
+#else
+fileprivate func bjs_jsRoundTripGreeter(_ greeter: UnsafeMutableRawPointer) -> UnsafeMutableRawPointer {
+    fatalError("Only available on WebAssembly")
+}
+#endif
+
+func _$jsRoundTripGreeter(_ greeter: Greeter) throws(JSException) -> Greeter {
+    let greeterPointer = greeter.bridgeJSLowerParameter()
+    let ret = bjs_jsRoundTripGreeter(greeterPointer)
+    if let error = _swift_js_take_exception() {
+        throw error
+    }
+    return Greeter.bridgeJSLiftReturn(ret)
+}

Plugins/BridgeJS/Tests/BridgeJSToolTests/__Snapshots__/BridgeJSLinkTests/Protocol.js

@@ -513,7 +513,7 @@ export async function createInstantiator(options, swift) {
             }
             TestModule["bjs_MyViewControllerDelegate_onHelperUpdated"] = function bjs_MyViewControllerDelegate_onHelperUpdated(self, helper) {
                 try {
-                    swift.memory.getObject(self).onHelperUpdated(Helper.__construct(helper));
+                    swift.memory.getObject(self).onHelperUpdated(_exports['Helper'].__construct(helper));
                 } catch (error) {
                     setException(error);
                 }

Plugins/BridgeJS/Tests/BridgeJSToolTests/__Snapshots__/BridgeJSLinkTests/SwiftClass.d.ts

@@ -31,6 +31,7 @@ export type Exports = {
     takeGreeter(greeter: Greeter): void;
 }
 export type Imports = {
+    jsRoundTripGreeter(greeter: Greeter): Greeter;
 }
 export function createInstantiator(options: {
     imports: Imports;

Plugins/BridgeJS/Tests/BridgeJSToolTests/__Snapshots__/BridgeJSLinkTests/SwiftClass.js

@@ -38,6 +38,7 @@ export async function createInstantiator(options, swift) {
         addImports: (importObject, importsContext) => {
             bjs = {};
             importObject["bjs"] = bjs;
+            const imports = options.getImports(importsContext);
             bjs["swift_js_return_string"] = function(ptr, len) {
                 const bytes = new Uint8Array(memory.buffer, ptr, len);
                 tmpRetString = textDecoder.decode(bytes);
@@ -214,6 +215,16 @@ export async function createInstantiator(options, swift) {
                 const obj = PublicGreeter.__construct(pointer);
                 return swift.memory.retain(obj);
             };
+            const TestModule = importObject["TestModule"] = importObject["TestModule"] || {};
+            TestModule["bjs_jsRoundTripGreeter"] = function bjs_jsRoundTripGreeter(greeter) {
+                try {
+                    let ret = imports.jsRoundTripGreeter(_exports['Greeter'].__construct(greeter));
+                    return ret.pointer;
+                } catch (error) {
+                    setException(error);
+                    return 0
+                }
+            }
         },
         setInstance: (i) => {
             instance = i;

Tests/BridgeJSRuntimeTests/Generated/BridgeJS.swift

@@ -10666,4 +10666,22 @@ func _$jsRoundTripOptionalStringUndefined(_ name: JSUndefinedOr<String>) throws(
         throw error
     }
     return JSUndefinedOr<String>.bridgeJSLiftReturnFromSideChannel()
+}
+
+#if arch(wasm32)
+@_extern(wasm, module: "BridgeJSRuntimeTests", name: "bjs_SwiftClassSupportImports_jsRoundTripGreeter_static")
+fileprivate func bjs_SwiftClassSupportImports_jsRoundTripGreeter_static(_ greeter: UnsafeMutableRawPointer) -> UnsafeMutableRawPointer
+#else
+fileprivate func bjs_SwiftClassSupportImports_jsRoundTripGreeter_static(_ greeter: UnsafeMutableRawPointer) -> UnsafeMutableRawPointer {
+    fatalError("Only available on WebAssembly")
+}
+#endif
+
+func _$SwiftClassSupportImports_jsRoundTripGreeter(_ greeter: Greeter) throws(JSException) -> Greeter {
+    let greeterPointer = greeter.bridgeJSLowerParameter()
+    let ret = bjs_SwiftClassSupportImports_jsRoundTripGreeter_static(greeterPointer)
+    if let error = _swift_js_take_exception() {
+        throw error
+    }
+    return Greeter.bridgeJSLiftReturn(ret)
 }

Tests/BridgeJSRuntimeTests/Generated/JavaScript/BridgeJS.json

@@ -15310,6 +15310,45 @@
         "types" : [
 
         ]
+      },
+      {
+        "functions" : [
+
+        ],
+        "types" : [
+          {
+            "getters" : [
+
+            ],
+            "methods" : [
+
+            ],
+            "name" : "SwiftClassSupportImports",
+            "setters" : [
+
+            ],
+            "staticMethods" : [
+              {
+                "name" : "jsRoundTripGreeter",
+                "parameters" : [
+                  {
+                    "name" : "greeter",
+                    "type" : {
+                      "swiftHeapObject" : {
+                        "_0" : "Greeter"
+                      }
+                    }
+                  }
+                ],
+                "returnType" : {
+                  "swiftHeapObject" : {
+                    "_0" : "Greeter"
+                  }
+                }
+              }
+            ]
+          }
+        ]
       }
     ]
   },