create binding to idweakmap

atom/common/api/atom_api_v8_util.cc

@@ -5,7 +5,6 @@
 #include "atom/common/api/object_life_monitor.h"
 #include "atom/common/id_weak_map.h"
 #include "atom/common/node_includes.h"
-#include "native_mate/handle.h"
 #include "native_mate/dictionary.h"
 #include "v8/include/v8-profiler.h"
 
@@ -48,11 +47,6 @@ void TakeHeapSnapshot(v8::Isolate* isolate) {
   isolate->GetHeapProfiler()->TakeHeapSnapshot();
 }
 
-mate::Handle<atom::IDWeakMap> CreateWeakMap(v8::Isolate* isolate) {
-  auto handle = mate::CreateHandle(isolate, new atom::IDWeakMap);
-  return handle;
-}
-
 void Initialize(v8::Local<v8::Object> exports, v8::Local<v8::Value> unused,
                 v8::Local<v8::Context> context, void* priv) {
   mate::Dictionary dict(context->GetIsolate(), exports);
@@ -63,7 +57,6 @@ void Initialize(v8::Local<v8::Object> exports, v8::Local<v8::Value> unused,
   dict.SetMethod("getObjectHash", &GetObjectHash);
   dict.SetMethod("setDestructor", &SetDestructor);
   dict.SetMethod("takeHeapSnapshot", &TakeHeapSnapshot);
-  dict.SetMethod("createWeakMap", &CreateWeakMap);
 }
 
 }  // namespace

atom/common/api/atom_api_weak_map.cc

@@ -0,0 +1,79 @@
+// Copyright (c) 2015 GitHub, Inc.
+// Use of this source code is governed by the MIT license that can be
+// found in the LICENSE file.
+
+#include "atom/common/api/atom_api_weak_map.h"
+
+#include "atom/common/node_includes.h"
+#include "native_mate/constructor.h"
+#include "native_mate/dictionary.h"
+
+namespace atom {
+
+namespace api {
+
+WeakMap::WeakMap() {
+  id_weak_map_.reset(new atom::IDWeakMap);
+}
+
+WeakMap::~WeakMap() {
+}
+
+void WeakMap::Set(v8::Isolate* isolate,
+                  int32_t id,
+                  v8::Local<v8::Object> object) {
+  id_weak_map_->Set(isolate, id, object);
+}
+
+v8::Local<v8::Object> WeakMap::Get(v8::Isolate* isolate, int32_t id) {
+  return id_weak_map_->Get(isolate, id).ToLocalChecked();
+}
+
+bool WeakMap::Has(int32_t id) {
+  return id_weak_map_->Has(id);
+}
+
+void WeakMap::Remove(int32_t id) {
+  id_weak_map_->Remove(id);
+}
+
+bool WeakMap::IsDestroyed() const {
+  return !id_weak_map_;
+}
+
+// static
+void WeakMap::BuildPrototype(v8::Isolate* isolate,
+                             v8::Local<v8::ObjectTemplate> prototype) {
+  mate::ObjectTemplateBuilder(isolate, prototype)
+      .SetMethod("set", &WeakMap::Set)
+      .SetMethod("get", &WeakMap::Get)
+      .SetMethod("has", &WeakMap::Has)
+      .SetMethod("remove", &WeakMap::Remove);
+}
+
+// static
+mate::Wrappable* WeakMap::Create(v8::Isolate* isolate) {
+  return new WeakMap;
+}
+
+}  // namespace api
+
+}  // namespace atom
+
+namespace {
+
+using atom::api::WeakMap;
+
+void Initialize(v8::Local<v8::Object> exports, v8::Local<v8::Value> unused,
+                v8::Local<v8::Context> context, void* priv) {
+  v8::Isolate* isolate = context->GetIsolate();
+  v8::Local<v8::Function> constructor = mate::CreateConstructor<WeakMap>(
+      isolate, "WeakMap", base::Bind(&WeakMap::Create));
+  mate::Dictionary weak_map(isolate, constructor);
+  mate::Dictionary dict(isolate, exports);
+  dict.Set("WeakMap", weak_map);
+}
+
+}  // namespace
+
+NODE_MODULE_CONTEXT_AWARE_BUILTIN(atom_common_weak_map, Initialize)

atom/common/api/atom_api_weak_map.h

@@ -0,0 +1,46 @@
+// Copyright (c) 2015 GitHub, Inc.
+// Use of this source code is governed by the MIT license that can be
+// found in the LICENSE file.
+
+#ifndef ATOM_COMMON_API_ATOM_API_WEAK_MAP_H_
+#define ATOM_COMMON_API_ATOM_API_WEAK_MAP_H_
+
+#include "atom/common/id_weak_map.h"
+#include "native_mate/object_template_builder.h"
+#include "native_mate/handle.h"
+
+namespace atom {
+
+namespace api {
+
+class WeakMap : public mate::Wrappable {
+ public:
+  static mate::Wrappable* Create(v8::Isolate* isolate);
+
+  static void BuildPrototype(v8::Isolate* isolate,
+                             v8::Local<v8::ObjectTemplate> prototype);
+
+ protected:
+  WeakMap();
+  virtual ~WeakMap();
+
+  // mate::Wrappable:
+  bool IsDestroyed() const override;
+
+ private:
+  // Api for IDWeakMap.
+  void Set(v8::Isolate* isolate, int32_t id, v8::Local<v8::Object> object);
+  v8::Local<v8::Object> Get(v8::Isolate* isolate, int32_t id);
+  bool Has(int32_t id);
+  void Remove(int32_t id);
+
+  scoped_ptr<IDWeakMap> id_weak_map_;
+
+  DISALLOW_COPY_AND_ASSIGN(WeakMap);
+};
+
+}  // namespace api
+
+}  // namespace atom
+
+#endif  // ATOM_COMMON_API_ATOM_API_WEAK_MAP_H_