Merge branch 'pr/9'

Author: Hywan

src/instance.rs

@@ -8,8 +8,8 @@ use rutie::{
     rubysys::{class, value::ValueType},
     types::{Argc, Value},
     util::str_to_cstring,
-    wrappable_struct, AnyException, AnyObject, Array, Exception, Fixnum, Float, Module, NilClass,
-    Object, RString, Symbol,
+    wrappable_struct, AnyException, AnyObject, Array, Boolean, Exception, Fixnum, Float, Module,
+    NilClass, Object, RString, Symbol,
 };
 use std::{mem, rc::Rc};
 use wasmer_runtime::{self as runtime, imports, Export};
@@ -27,6 +27,11 @@ impl ExportedFunctions {
         Self { instance }
     }
 
+    /// Check that an exported function exists.
+    pub fn respond_to_missing(&self, method_name: &str) -> bool {
+        self.instance.dyn_func(method_name).is_ok()
+    }
+
     /// Call an exported function on the given WebAssembly instance.
     pub fn method_missing(
         &self,
@@ -164,6 +169,22 @@ wrappable_struct!(
 
 class!(RubyExportedFunctions);
 
+#[rustfmt::skip]
+methods!(
+    RubyExportedFunctions,
+    itself,
+
+    // Glue code to call the `ExportedFunctions.respond_to` method.
+    fn ruby_exported_functions_method_exists(symbol: Symbol, _include_private: Boolean) -> Boolean {
+        unwrap_or_raise(|| {
+            let symbol = symbol?;
+            let instance = itself.get_data(&*EXPORTED_FUNCTIONS_WRAPPER);
+
+            Ok(Boolean::new(instance.respond_to_missing(symbol.to_str())))
+        })
+    }
+);
+
 /// Glue code to call the `ExportedFunctions.method_missing` method.
 pub extern "C" fn ruby_exported_functions_method_missing(
     argc: Argc,

src/lib.rs

@@ -34,6 +34,12 @@ pub extern "C" fn Init_wasmer() {
     wasmer_module
         .define_nested_class("ExportedFunctions", Some(&exported_functions_data_class))
         .define(|itself| {
+            // Declare the `respond_to_missing?` method.
+            itself.def(
+                "respond_to_missing?",
+                instance::ruby_exported_functions_method_exists,
+            );
+
             // Declare the `method_missing` method.
             itself.def(
                 "method_missing",

tests/instance_test.rb

@@ -36,12 +36,16 @@ def test_invalid_module
   end
 
   def test_basic_sum
-    assert_equal 3, Wasmer::Instance.new(self.bytes).exports.sum(1, 2)
+    exports = Wasmer::Instance.new(self.bytes).exports
+    assert exports.respond_to?(:sum)
+    assert_equal 3, exports.sum(1, 2)
   end
 
   def test_call_unknown_function
+    exports = Wasmer::Instance.new(self.bytes).exports
+    assert !exports.respond_to?(:foo)
     error = assert_raises(RuntimeError) {
-      Wasmer::Instance.new(self.bytes).exports.foo
+      exports.foo
     }
     assert_equal "Function `foo` does not exist.", error.message
   end