jbarnette-johnson 1.0.0.200806240111

data/ext/spidermonkey/js_land_proxy.h

@@ -0,0 +1,20 @@
+#ifndef JOHNSON_SPIDERMONKEY_JS_LAND_PROXY_H
+#define JOHNSON_SPIDERMONKEY_JS_LAND_PROXY_H
+
+#include "spidermonkey.h"
+#include "runtime.h"
+
+bool js_value_is_proxy(JohnsonRuntime* runtime, jsval maybe_proxy);
+VALUE unwrap_js_land_proxy(JohnsonRuntime* runtime, jsval proxy);
+JSBool make_js_land_proxy(JohnsonRuntime* runtime, VALUE value, jsval* retval);
+
+#include "node.h"
+typedef struct {
+  VALUE klass, rklass;
+  VALUE recv;
+  ID id, oid;
+  int safe_level;
+  NODE *body;
+} METHOD;
+
+#endif

data/ext/spidermonkey/ruby_land_proxy.c

@@ -0,0 +1,537 @@
+#include "ruby_land_proxy.h"
+#include "conversions.h"
+
+DECLARE_RUBY_WRAPPER(rb_call_super, int argc; const VALUE* argv)
+DEFINE_RUBY_WRAPPER(rb_call_super, rb_call_super, ARGLIST2(argc, argv))
+
+DECLARE_RUBY_WRAPPER(rb_yield, VALUE v)
+DEFINE_RUBY_WRAPPER(rb_yield, rb_yield, ARGLIST1(v))
+
+static VALUE proxy_class = Qnil;
+
+static inline JSBool get_jsval_for_proxy(RubyLandProxy* proxy, jsval* jv)
+{
+  *jv = (jsval)(proxy->key);
+  return JS_TRUE;
+}
+
+static VALUE call_js_function_value(JohnsonRuntime* runtime, jsval target, jsval function, int argc, VALUE* argv)
+{
+  JSContext * context = johnson_get_current_context(runtime);
+  PREPARE_RUBY_JROOTS(context, argc + 2);
+
+  JROOT(target);
+  JROOT(function);
+
+  assert(JSVAL_IS_OBJECT(target));
+
+  jsval args[argc];
+  jsval result;
+
+  int i;
+  for(i = 0; i < argc; ++i)
+  {
+    JCHECK(convert_to_js(runtime, argv[i], &(args[i])));
+    JROOT(args[i]);
+  }
+
+  JCHECK(JS_CallFunctionValue(context,
+    JSVAL_TO_OBJECT(target), function, (unsigned) argc, args, &result));
+
+  JRETURN_RUBY(CONVERT_TO_RUBY(runtime, result));
+}
+
+/*
+ * call-seq:
+ *   [](name)
+ *
+ * Returns the property with +name+.
+ */
+static VALUE
+get(VALUE self, VALUE name)
+{
+  RubyLandProxy* proxy;
+  Data_Get_Struct(self, RubyLandProxy, proxy);
+
+  JSContext * context = johnson_get_current_context(proxy->runtime);
+  PREPARE_RUBY_JROOTS(context, 1);
+  
+  jsval proxy_value;
+  JCHECK(get_jsval_for_proxy(proxy, &proxy_value));
+  JROOT(proxy_value);
+
+  jsval js_value;  
+
+  switch(TYPE(name)) {
+    case T_FIXNUM:
+      JCHECK(JS_GetElement(context,
+          JSVAL_TO_OBJECT(proxy_value), NUM2INT(name), &js_value));
+      break;
+    default:
+      Check_Type(name, T_STRING);
+      JCHECK(JS_GetProperty(context,
+          JSVAL_TO_OBJECT(proxy_value), StringValueCStr(name), &js_value));
+      break;
+  }
+
+  JRETURN_RUBY(CONVERT_TO_RUBY(proxy->runtime, js_value));
+}
+
+/*
+ * call-seq:
+ *   []=(name,value)
+ *
+ * Sets the property with +name+ to +value+.
+ */
+static VALUE
+set(VALUE self, VALUE name, VALUE value)
+{
+  RubyLandProxy* proxy;
+  Data_Get_Struct(self, RubyLandProxy, proxy);
+  JSContext * context = johnson_get_current_context(proxy->runtime);
+  
+  PREPARE_RUBY_JROOTS(context, 2);
+  
+  jsval proxy_value;
+  JCHECK(get_jsval_for_proxy(proxy, &proxy_value));
+  JROOT(proxy_value);
+
+  jsval js_value;
+  JCHECK(convert_to_js(proxy->runtime, value, &js_value));
+  
+  JROOT(js_value);
+
+  switch(TYPE(name)) {
+    case T_FIXNUM:
+      JCHECK(JS_SetElement(context,
+              JSVAL_TO_OBJECT(proxy_value), NUM2INT(name), &js_value));
+      break;
+    default:
+      Check_Type(name, T_STRING);
+      JCHECK(JS_SetProperty(context,
+            JSVAL_TO_OBJECT(proxy_value), StringValueCStr(name), &js_value));
+      break;
+  }
+
+  JRETURN_RUBY(value);
+}
+
+/*
+ * call-seq:
+ *   function?
+ *
+ * Returns <code>true</code> if this is a function.
+ */
+static VALUE
+function_p(VALUE self)
+{
+  RubyLandProxy* proxy;
+  Data_Get_Struct(self, RubyLandProxy, proxy);
+  JSContext * context = johnson_get_current_context(proxy->runtime);
+  PREPARE_RUBY_JROOTS(context, 1);
+  jsval proxy_value;
+  JCHECK(get_jsval_for_proxy(proxy, &proxy_value));
+  JROOT(proxy_value);
+  JRETURN_RUBY(JS_TypeOfValue(context, proxy_value) == JSTYPE_FUNCTION ? Qtrue : Qfalse);
+}
+
+/*
+ * call-seq:
+ *   respond_to?(symbol)
+ *
+ * Returns <code>true</code> if _obj_ responds to given method.
+ */
+static VALUE
+respond_to_p(VALUE self, VALUE sym)
+{
+  RubyLandProxy* proxy;
+  Data_Get_Struct(self, RubyLandProxy, proxy);
+
+  JSContext * context = johnson_get_current_context(proxy->runtime);
+  PREPARE_RUBY_JROOTS(context, 2);
+  
+  char* name = rb_id2name(SYM2ID(sym));
+  
+  // assignment is always okay
+  if (name[strlen(name) - 1] == '=')
+    JRETURN_RUBY(Qtrue);
+  
+  jsval proxy_value;
+  JCHECK(get_jsval_for_proxy(proxy, &proxy_value));
+  JROOT(proxy_value);
+
+  JSObject *obj;
+  JSBool found;
+  
+  JCHECK(JS_ValueToObject(context, proxy_value, &obj));
+  JROOT(obj);
+
+  JCHECK(JS_HasProperty(context, obj, name, &found));
+
+  JRETURN_RUBY(found ? Qtrue : CALL_RUBY_WRAPPER(rb_call_super, 1, &sym));
+}
+
+/*
+ * call-seq:
+ *   native_call(global, *args)
+ *
+ * Call as a function with given +global+ using *args.
+ */
+static VALUE
+native_call(int argc, VALUE* argv, VALUE self)
+{
+  // FIXME: This should really call super#native_call.
+  if (!function_p(self))
+    rb_raise(rb_eRuntimeError, "This Johnson::SpiderMonkey::RubyLandProxy isn't a function.");
+
+  if (argc < 1)
+    rb_raise(rb_eArgError, "Target object required");
+
+  RubyLandProxy* proxy;
+  Data_Get_Struct(self, RubyLandProxy, proxy);
+  JSContext * context = johnson_get_current_context(proxy->runtime);
+  
+  PREPARE_RUBY_JROOTS(context, 1);
+  
+  jsval proxy_value;
+  JCHECK(get_jsval_for_proxy(proxy, &proxy_value));
+  JROOT(proxy_value);
+
+  jsval global;
+  JCHECK(convert_to_js(proxy->runtime, argv[0], &global));
+
+  JRETURN_RUBY(call_js_function_value(proxy->runtime, global, proxy_value, argc - 1, &(argv[1])));
+}
+
+static void
+destroy_id_array(JSContext* context, void* data)
+{
+  JS_DestroyIdArray(context, (JSIdArray*)data);
+}
+
+/*
+ * call-seq:
+ *   each { |obj| block }
+ *
+ * Calls <em>block</em> with each item in the collection.
+ */
+static VALUE
+each(VALUE self)
+{
+  RubyLandProxy* proxy;
+  Data_Get_Struct(self, RubyLandProxy, proxy);
+  JSContext * context = johnson_get_current_context(proxy->runtime);
+  
+  PREPARE_RUBY_JROOTS(context, 5);
+  
+  jsval proxy_value;
+  JCHECK(get_jsval_for_proxy(proxy, &proxy_value));
+  JROOT(proxy_value);
+
+  JSObject* value = JSVAL_TO_OBJECT(proxy_value);
+  JROOT(value);
+  
+  // arrays behave like you'd expect, indexes in order
+  if (JS_IsArrayObject(context, value))
+  {
+    jsuint length;
+    JCHECK(JS_GetArrayLength(context, value, &length));
+    
+    jsuint i = 0;
+    for (i = 0; i < length; ++i)
+    {
+      jsval element;
+      JCHECK(JS_GetElement(context, value, (signed) i, &element));
+      CALL_RUBY_WRAPPER(rb_yield, convert_to_ruby(proxy->runtime, element));
+    }
+  }
+  else
+  {
+    // not an array? behave like each on Hash; yield [key, value]
+    JSIdArray* ids = JS_Enumerate(context, value);
+    JCHECK(ids);
+
+    JCLEANUP(destroy_id_array, ids);
+
+    int i;
+    for (i = 0; i < ids->length; ++i)
+    {
+      jsval js_key, js_value;
+
+      JCHECK(JS_IdToValue(context, ids->vector[i], &js_key));
+      JROOT(js_key);
+
+      if (JSVAL_IS_STRING(js_key))
+      {
+        // regular properties have string keys
+        JCHECK(JS_GetProperty(context, value,
+          JS_GetStringBytes(JSVAL_TO_STRING(js_key)), &js_value));
+      }
+      else
+      {
+        // it's a numeric property, use array access
+        JCHECK(JS_GetElement(context, value,
+          JSVAL_TO_INT(js_key), &js_value));
+      }
+      JROOT(js_value);
+
+      VALUE key = CONVERT_TO_RUBY(proxy->runtime, js_key);
+      VALUE value = CONVERT_TO_RUBY(proxy->runtime, js_value);
+
+      CALL_RUBY_WRAPPER(rb_yield, rb_ary_new3(2, key, value));
+
+      JUNROOT(js_value);
+      JUNROOT(js_key);
+    }
+  }
+
+  JRETURN_RUBY(self);
+}
+
+/*
+ * call-seq:
+ *   length
+ *
+ * Returns the length of the collection.
+ */
+static VALUE
+length(VALUE self)
+{
+  RubyLandProxy* proxy;
+  Data_Get_Struct(self, RubyLandProxy, proxy);
+  JSContext * context = johnson_get_current_context(proxy->runtime);
+
+  PREPARE_RUBY_JROOTS(context, 2);
+  
+  jsval proxy_value;
+  JCHECK(get_jsval_for_proxy(proxy, &proxy_value));
+  JROOT(proxy_value);
+
+  JSObject* value = JSVAL_TO_OBJECT(proxy_value);
+  JROOT(value);
+  
+  if (JS_IsArrayObject(context, value))
+  {
+    jsuint length;
+    JCHECK(JS_GetArrayLength(context, value, &length));
+
+    JRETURN_RUBY(INT2FIX(length));
+  }
+  else
+  {
+    JSIdArray* ids = JS_Enumerate(context, value);
+    JCHECK(ids);
+    VALUE length = INT2FIX(ids->length);
+    
+    JS_DestroyIdArray(context, ids);
+
+    JRETURN_RUBY(length);
+  }
+}
+
+/*
+ * call-seq:
+ *   runtime
+ *
+ * Returns runtime.
+ */
+static VALUE
+runtime(VALUE self)
+{
+  RubyLandProxy* proxy;
+  Data_Get_Struct(self, RubyLandProxy, proxy);
+  return (VALUE)JS_GetRuntimePrivate(proxy->runtime->js);
+}
+
+/*
+ * call-seq:
+ *   function_property?(name)
+ *
+ * Returns <code>true</code> if +name+ is a function property.
+ */
+static VALUE
+function_property_p(VALUE self, VALUE name)
+{
+  Check_Type(name, T_STRING);
+  
+  RubyLandProxy* proxy;
+  Data_Get_Struct(self, RubyLandProxy, proxy);
+  JSContext * context = johnson_get_current_context(proxy->runtime);
+
+  PREPARE_RUBY_JROOTS(context, 2);
+
+  jsval proxy_value;
+  JCHECK(get_jsval_for_proxy(proxy, &proxy_value));
+  JROOT(proxy_value);
+
+  jsval js_value;  
+
+  JCHECK(JS_GetProperty(context,
+      JSVAL_TO_OBJECT(proxy_value), StringValueCStr(name), &js_value));
+
+  JROOT(js_value);
+
+  JSType type = JS_TypeOfValue(context, js_value);
+
+  JRETURN_RUBY(type == JSTYPE_FUNCTION ? Qtrue : Qfalse);
+}
+
+/*
+ * call-seq:
+ *   call_function_property(name, arguments)
+ *
+ * Calls function +name+ with +arguments+.
+ */
+static VALUE
+call_function_property(int argc, VALUE* argv, VALUE self)
+{
+  RubyLandProxy* proxy;
+  Data_Get_Struct(self, RubyLandProxy, proxy);
+  JSContext * context = johnson_get_current_context(proxy->runtime);
+
+  if (argc < 1)
+    rb_raise(rb_eArgError, "Function name required");
+
+  PREPARE_RUBY_JROOTS(context, 2);
+  
+  jsval proxy_value;
+  JCHECK(get_jsval_for_proxy(proxy, &proxy_value));
+  JROOT(proxy_value);
+
+  jsval function;
+  
+  JCHECK(JS_GetProperty(context,
+    JSVAL_TO_OBJECT(proxy_value), StringValueCStr(argv[0]), &function));
+
+  JROOT(function);
+
+  JSType funtype = JS_TypeOfValue(context, function);
+  
+  // should never be anything but a function
+  if (funtype != JSTYPE_FUNCTION)
+    JERROR("Specified property \"%s\" isn't a function.", StringValueCStr(argv[0]));
+
+  JRETURN_RUBY(call_js_function_value(proxy->runtime, proxy_value, function, argc - 1, &(argv[1])));
+}
+
+/*
+ * call-seq:
+ *   to_s
+ *
+ * Converts object to a string.
+ */
+static VALUE to_s(VALUE self)
+{
+  RubyLandProxy* proxy;
+  Data_Get_Struct(self, RubyLandProxy, proxy);
+  JSContext * context = johnson_get_current_context(proxy->runtime);
+
+  PREPARE_RUBY_JROOTS(context, 1);
+  
+  jsval proxy_value;
+  JCHECK(get_jsval_for_proxy(proxy, &proxy_value));
+  JROOT(proxy_value);
+  
+  JSString* str = JS_ValueToString(context, proxy_value);
+  JRETURN_RUBY(convert_js_string_to_ruby(proxy->runtime, str));
+}
+
+///////////////////////////////////////////////////////////////////////////
+//// INFRASTRUCTURE BELOW HERE ////////////////////////////////////////////
+///////////////////////////////////////////////////////////////////////////
+
+static void finalize(RubyLandProxy* proxy)
+{
+  // could get finalized after the context has been freed
+  if (proxy->runtime && proxy->runtime->jsids)
+  {
+    // remove this proxy from the OID map
+    jsval proxy_value;
+    get_jsval_for_proxy(proxy, &proxy_value);
+    JS_HashTableRemove(proxy->runtime->jsids, (void *)proxy_value);
+  }
+
+  if (proxy->runtime)
+  {
+    // remove our GC handle on the JS value
+    JS_RemoveRootRT(proxy->runtime->js, &(proxy->key));
+  }
+
+  free(proxy);
+}
+
+bool ruby_value_is_proxy(VALUE maybe_proxy)
+{
+  return proxy_class == CLASS_OF(maybe_proxy); 
+}
+
+JSBool unwrap_ruby_land_proxy(JohnsonRuntime* runtime, VALUE wrapped, jsval* retval)
+{
+  JSContext * context = johnson_get_current_context(runtime);
+  assert(ruby_value_is_proxy(wrapped));
+  
+  PREPARE_JROOTS(context, 0);
+
+  RubyLandProxy* proxy;
+  Data_Get_Struct(wrapped, RubyLandProxy, proxy);
+  
+  JCHECK(get_jsval_for_proxy(proxy, retval));
+  JRETURN;
+}
+
+VALUE make_ruby_land_proxy(JohnsonRuntime* runtime, jsval value, const char const* root_name)
+{
+  VALUE id = (VALUE)JS_HashTableLookup(runtime->jsids, (void *)value);
+  
+  if (id)
+  {
+    // if we already have a proxy, return it
+    return id;
+  }
+  else
+  {    
+    // otherwise make one and cache it
+    RubyLandProxy* our_proxy; 
+    VALUE proxy = Data_Make_Struct(proxy_class, RubyLandProxy, 0, finalize, our_proxy);
+
+    JSContext * context = johnson_get_current_context(runtime);
+
+    PREPARE_RUBY_JROOTS(context, 1);
+    JROOT(value);
+
+    our_proxy->runtime = runtime;
+    our_proxy->key = (void *)value;
+
+    // root the value for JS GC and lookups
+    JCHECK(JS_AddNamedRootRT(runtime->js, &(our_proxy->key), root_name));
+
+    // put the proxy OID in the id map
+    JCHECK(JS_HashTableAdd(runtime->jsids, (void *)value, (void *)proxy));
+    
+    JRETURN_RUBY(proxy);
+  }
+}
+
+void init_Johnson_SpiderMonkey_Proxy(VALUE spidermonkey)
+{
+  /* HACK:  These comments are *only* to make RDoc happy.
+  VALUE johnson = rb_define_module("Johnson");
+  VALUE spidermonkey = rb_define_module_under(johnson, "SpiderMonkey");
+  */
+
+  /* RubyLandProxy class. */
+  proxy_class = rb_define_class_under(spidermonkey, "RubyLandProxy", rb_cObject);
+
+  rb_define_method(proxy_class, "[]", get, 1);
+  rb_define_method(proxy_class, "[]=", set, 2);
+  rb_define_method(proxy_class, "function?", function_p, 0);
+  rb_define_method(proxy_class, "respond_to?", respond_to_p, 1);
+  rb_define_method(proxy_class, "each", each, 0);
+  rb_define_method(proxy_class, "length", length, 0);
+  rb_define_method(proxy_class, "to_s", to_s, 0);
+
+  rb_define_private_method(proxy_class, "native_call", native_call, -1);
+  rb_define_private_method(proxy_class, "runtime", runtime, 0);
+  rb_define_private_method(proxy_class, "function_property?", function_property_p, 1);
+  rb_define_private_method(proxy_class, "call_function_property", call_function_property, -1);
+}