Use native_mate to simplify id_weak_map api.

atom/common/api/atom_api_id_weak_map.cc

@@ -1,5 +1,4 @@
 // Copyright (c) 2013 GitHub, Inc. All rights reserved.
-// Copyright (c) 2012 Intel Corp. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
@@ -10,6 +9,9 @@
 #include "atom/common/v8/native_type_conversions.h"
 #include "atom/common/v8/node_common.h"
 #include "base/logging.h"
+#include "native_mate/constructor.h"
+#include "native_mate/function_template.h"
+#include "native_mate/object_template_builder.h"
 
 namespace atom {
 
@@ -22,11 +24,48 @@ IDWeakMap::IDWeakMap()
 IDWeakMap::~IDWeakMap() {
 }
 
-bool IDWeakMap::Has(int key) const {
+// static
+void IDWeakMap::WeakCallback(v8::Isolate* isolate,
+                             v8::Persistent<v8::Object>* value,
+                             IDWeakMap* self) {
+  v8::HandleScope handle_scope(isolate);
+  v8::Local<v8::Object> local = v8::Local<v8::Object>::New(isolate, *value);
+  self->Remove(
+      FromV8Value(local->GetHiddenValue(v8::String::New("IDWeakMapKey"))));
+}
+
+int32_t IDWeakMap::Add(v8::Isolate* isolate, v8::Handle<v8::Object> object) {
+  int32_t key = GetNextID();
+  object->SetHiddenValue(mate::StringToV8(isolate, "IDWeakMapKey"),
+                         mate::Converter<int32_t>::ToV8(isolate, key));
+
+  map_[key] = new RefCountedPersistent<v8::Object>(object);
+  map_[key]->MakeWeak(this, WeakCallback);
+  return key;
+}
+
+v8::Handle<v8::Value> IDWeakMap::Get(int32_t key) {
+  if (!Has(key)) {
+    node::ThrowError("Invalid key");
+    return v8::Undefined();
+  }
+
+  return map_[key]->NewHandle();
+}
+
+bool IDWeakMap::Has(int32_t key) const {
   return map_.find(key) != map_.end();
 }
 
-void IDWeakMap::Erase(int key) {
+std::vector<int32_t> IDWeakMap::Keys() const {
+  std::vector<int32_t> keys;
+  keys.reserve(map_.size());
+  for (auto it = map_.begin(); it != map_.end(); ++it)
+    keys.push_back(it->first);
+  return keys;
+}
+
+void IDWeakMap::Remove(int32_t key) {
   if (Has(key))
     map_.erase(key);
   else
@@ -37,102 +76,21 @@ int IDWeakMap::GetNextID() {
   return ++next_id_;
 }
 
-// static
-void IDWeakMap::WeakCallback(v8::Isolate* isolate,
-                             v8::Persistent<v8::Object>* value,
-                             IDWeakMap* self) {
-  v8::HandleScope handle_scope(isolate);
-  v8::Local<v8::Object> local = v8::Local<v8::Object>::New(isolate, *value);
-  self->Erase(
-      FromV8Value(local->GetHiddenValue(v8::String::New("IDWeakMapKey"))));
-}
+void IDWeakMap::Initialize(v8::Handle<v8::Object> exports) {
+  v8::Local<v8::FunctionTemplate> constructor = mate::CreateConstructor(
+      v8::Isolate::GetCurrent(),
+      "IDWeakMap",
+      base::Bind(&mate::NewOperatorFactory<IDWeakMap>));
 
-// static
-void IDWeakMap::New(const v8::FunctionCallbackInfo<v8::Value>& args) {
-  (new IDWeakMap)->Wrap(args.This());
-}
+  mate::ObjectTemplateBuilder builder(
+      v8::Isolate::GetCurrent(), constructor->PrototypeTemplate());
+  builder.SetMethod("add", &IDWeakMap::Add)
+         .SetMethod("get", &IDWeakMap::Get)
+         .SetMethod("has", &IDWeakMap::Has)
+         .SetMethod("keys", &IDWeakMap::Keys)
+         .SetMethod("remove", &IDWeakMap::Remove);
 
-// static
-void IDWeakMap::Add(const v8::FunctionCallbackInfo<v8::Value>& args) {
-  if (!args[0]->IsObject())
-    return node::ThrowTypeError("Bad argument");
-
-  IDWeakMap* self = Unwrap<IDWeakMap>(args.This());
-
-  int key = self->GetNextID();
-  v8::Local<v8::Object> v8_value = args[0]->ToObject();
-  v8_value->SetHiddenValue(v8::String::New("IDWeakMapKey"), ToV8Value(key));
-
-  self->map_[key] = new RefCountedPersistent<v8::Object>(v8_value);
-  self->map_[key]->MakeWeak(self, WeakCallback);
-
-  args.GetReturnValue().Set(key);
-}
-
-// static
-void IDWeakMap::Get(const v8::FunctionCallbackInfo<v8::Value>& args) {
-  int key;
-  if (!FromV8Arguments(args, &key))
-    return node::ThrowTypeError("Bad argument");
-
-  IDWeakMap* self = Unwrap<IDWeakMap>(args.This());
-  if (!self->Has(key))
-    return node::ThrowError("Invalid key");
-
-  args.GetReturnValue().Set(self->map_[key]->NewHandle());
-}
-
-// static
-void IDWeakMap::Has(const v8::FunctionCallbackInfo<v8::Value>& args) {
-  int key;
-  if (!FromV8Arguments(args, &key))
-    return node::ThrowTypeError("Bad argument");
-
-  IDWeakMap* self = Unwrap<IDWeakMap>(args.This());
-  args.GetReturnValue().Set(self->Has(key));
-}
-
-// static
-void IDWeakMap::Keys(const v8::FunctionCallbackInfo<v8::Value>& args) {
-  IDWeakMap* self = Unwrap<IDWeakMap>(args.This());
-
-  v8::Local<v8::Array> keys = v8::Array::New(self->map_.size());
-
-  int i = 0;
-  for (auto el = self->map_.begin(); el != self->map_.end(); ++el) {
-    keys->Set(i, ToV8Value(el->first));
-    ++i;
-  }
-
-  args.GetReturnValue().Set(keys);
-}
-
-// static
-void IDWeakMap::Remove(const v8::FunctionCallbackInfo<v8::Value>& args) {
-  int key;
-  if (!FromV8Arguments(args, &key))
-    return node::ThrowTypeError("Bad argument");
-
-  IDWeakMap* self = Unwrap<IDWeakMap>(args.This());
-  if (!self->Has(key))
-    return node::ThrowError("Invalid key");
-
-  self->Erase(key);
-}
-
-// static
-void IDWeakMap::Initialize(v8::Handle<v8::Object> target) {
-  v8::Local<v8::FunctionTemplate> t = v8::FunctionTemplate::New(New);
-  t->InstanceTemplate()->SetInternalFieldCount(1);
-  t->SetClassName(v8::String::NewSymbol("IDWeakMap"));
-
-  NODE_SET_PROTOTYPE_METHOD(t, "add", Add);
-  NODE_SET_PROTOTYPE_METHOD(t, "get", Get);
-  NODE_SET_PROTOTYPE_METHOD(t, "has", Has);
-  NODE_SET_PROTOTYPE_METHOD(t, "keys", Keys);
-  NODE_SET_PROTOTYPE_METHOD(t, "remove", Remove);
-
-  target->Set(v8::String::NewSymbol("IDWeakMap"), t->GetFunction());
+  exports->Set(v8::String::NewSymbol("IDWeakMap"), constructor->GetFunction());
 }
 
 }  // namespace api