Add mate::TrackableObject

atom/browser/api/atom_api_web_contents.h

@@ -8,7 +8,7 @@
 #include <string>
 #include <vector>
 
-#include "atom/browser/api/event_emitter.h"
+#include "atom/browser/api/trackable_object.h"
 #include "atom/browser/common_web_contents_delegate.h"
 #include "content/public/browser/browser_plugin_guest_delegate.h"
 #include "content/public/common/favicon_url.h"
@@ -46,7 +46,7 @@ struct SetSizeParams {
   scoped_ptr<gfx::Size> normal_size;
 };
 
-class WebContents : public mate::EventEmitter,
+class WebContents : public mate::TrackableObject,
                     public content::BrowserPluginGuestDelegate,
                     public CommonWebContentsDelegate,
                     public content::WebContentsObserver,

atom/browser/api/trackable_object.cc

@@ -0,0 +1,39 @@
+// 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/browser/api/trackable_object.h"
+
+namespace mate {
+
+atom::IDWeakMap TrackableObject::weak_map_;
+
+// static
+TrackableObject* TrackableObject::FromWeakMapID(v8::Isolate* isolate,
+                                                int32_t id) {
+  v8::MaybeLocal<v8::Object> object = weak_map_.Get(isolate, id);
+  if (object.IsEmpty())
+    return nullptr;
+
+  TrackableObject* self = nullptr;
+  mate::ConvertFromV8(isolate, object.ToLocalChecked(), &self);
+  return self;
+}
+
+// static
+void TrackableObject::ReleaseAllWeakReferences() {
+  weak_map_.Clear();
+}
+
+TrackableObject::TrackableObject() : weak_map_id_(0) {
+}
+
+TrackableObject::~TrackableObject() {
+  weak_map_.Remove(weak_map_id_);
+}
+
+void TrackableObject::AfterInit(v8::Isolate* isolate) {
+  weak_map_id_ = weak_map_.Add(isolate, GetWrapper(isolate));
+}
+
+}  // namespace mate

atom/browser/api/trackable_object.h

@@ -0,0 +1,43 @@
+// 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_BROWSER_API_TRACKABLE_OBJECT_H_
+#define ATOM_BROWSER_API_TRACKABLE_OBJECT_H_
+
+#include "atom/browser/api/event_emitter.h"
+#include "atom/common/id_weak_map.h"
+
+namespace mate {
+
+// All instances of TrackableObject will be kept in a weak map and can be got
+// from its ID.
+class TrackableObject : public mate::EventEmitter {
+ public:
+  // Find out the TrackableObject from its ID in weak map:
+  static TrackableObject* FromWeakMapID(v8::Isolate* isolate, int32_t id);
+
+  // Releases all weak references in weak map, called when app is terminating.
+  static void ReleaseAllWeakReferences();
+
+  // mate::Wrappable:
+  void AfterInit(v8::Isolate* isolate) override;
+
+  // The ID in weak map.
+  int32_t weak_map_id() const { return weak_map_id_; }
+
+ protected:
+  TrackableObject();
+  ~TrackableObject() override;
+
+ private:
+  static atom::IDWeakMap weak_map_;
+
+  int32_t weak_map_id_;
+
+  DISALLOW_COPY_AND_ASSIGN(TrackableObject);
+};
+
+}  // namespace mate
+
+#endif  // ATOM_BROWSER_API_TRACKABLE_OBJECT_H_