From adc0597451c3369dc24a3c66049214c5f7c1db00 Mon Sep 17 00:00:00 2001
From: Charles Lowell <cowboyd@thefrontside.net>
Date: Tue, 17 Aug 2010 06:31:18 -0500
Subject: [PATCH] expose SetCallAsFunctionHandler() to ObjectTemplate

---
 ext/v8/v8_script.cpp   |  2 ++
 ext/v8/v8_template.cpp | 24 +++++++++++++++++-------
 2 files changed, 19 insertions(+), 7 deletions(-)

diff --git a/ext/v8/v8_script.cpp b/ext/v8/v8_script.cpp
index 63dca31..c610742 100644
--- a/ext/v8/v8_script.cpp
+++ b/ext/v8/v8_script.cpp
@@ -14,12 +14,14 @@ namespace {
   }
 
   VALUE Compile(VALUE self, VALUE source, VALUE source_name) {
+    HandleScope scope;
     Local<String> src(rr_rb2v8(source)->ToString());
     Local<String> src_name(rr_rb2v8(source_name)->ToString());
     return rr_v8_ref_create(self, Script::Compile(src, src_name));
   }
 
   VALUE Run(VALUE self) {
+    HandleScope scope;
     Local<Script> script(V8_Ref_Get<Script>(self));
     Local<Value> result(script->Run());
     return result.IsEmpty() ? Qnil : rr_v82rb(result);
diff --git a/ext/v8/v8_template.cpp b/ext/v8/v8_template.cpp
index 82d8bca..276fdda 100644
--- a/ext/v8/v8_template.cpp
+++ b/ext/v8/v8_template.cpp
@@ -38,6 +38,13 @@ namespace {
     return Qnil;
   }
 
+  Handle<Value> RubyInvocationCallback(const Arguments& args) {
+    VALUE code = (VALUE)External::Unwrap(args.Data());
+    VALUE rb_args = rr_v82rb(args);
+    VALUE result = rb_funcall(code, rb_intern("call"), 1, rb_args);
+    return rr_rb2v8(result);
+  }
+
   namespace Obj {
 
     /**
@@ -238,17 +245,19 @@ namespace {
       );
       return Qnil;
     }
+    VALUE SetCallAsFunctionHandler(VALUE self) {
+      HandleScope scope;
+      VALUE code = rb_block_proc();
+      if (NIL_P(code)) {
+        return Qnil;
+      }
+      obj(self)->SetCallAsFunctionHandler(RubyInvocationCallback, rr_v8_external_create(code));
+      return Qnil;
+    }
   }
 
   namespace Func {
 
-    Handle<Value> RubyInvocationCallback(const Arguments& args) {
-      VALUE code = (VALUE)External::Unwrap(args.Data());
-      VALUE rb_args = rr_v82rb(args);
-      VALUE result = rb_funcall(code, rb_intern("call"), 1, rb_args);
-      return rr_rb2v8(result);
-    }
-
     VALUE New(VALUE function_template) {
       HandleScope handles;
       VALUE code = rb_block_proc();
@@ -307,6 +316,7 @@ void rr_init_template() {
   rr_define_method(ObjectTemplateClass, "NewInstance", Obj::NewInstance, 0);
   rr_define_method(ObjectTemplateClass, "SetNamedPropertyHandler", Obj::SetNamedPropertyHandler, 5);
   rr_define_method(ObjectTemplateClass, "SetIndexedPropertyHandler", Obj::SetIndexedPropertyHandler, 5);
+  rr_define_method(ObjectTemplateClass, "SetCallAsFunctionHandler", Obj::SetCallAsFunctionHandler, 0);
 
   FunctionTemplateClass = rr_define_class("FunctionTemplate", Template);
   rr_define_singleton_method(FunctionTemplateClass, "New", Func::New, 0);