Add events on display add/remove.

atom/browser/api/atom_api_screen.cc

@@ -2,29 +2,92 @@
 // Use of this source code is governed by the MIT license that can be
 // found in the LICENSE file.
 
+#include "atom/browser/api/atom_api_screen.h"
+
+#include "atom/browser/browser.h"
 #include "atom/common/native_mate_converters/gfx_converter.h"
 #include "base/bind.h"
 #include "native_mate/dictionary.h"
+#include "native_mate/object_template_builder.h"
 #include "ui/gfx/screen.h"
 
 #include "atom/common/node_includes.h"
 
+namespace atom {
+
+namespace api {
+
+Screen::Screen(gfx::Screen* screen) : screen_(screen) {
+}
+
+Screen::~Screen() {
+}
+
+gfx::Point Screen::GetCursorScreenPoint() {
+  return screen_->GetCursorScreenPoint();
+}
+
+gfx::Display Screen::GetPrimaryDisplay() {
+  return screen_->GetPrimaryDisplay();
+}
+
+std::vector<gfx::Display> Screen::GetAllDisplays() {
+  return screen_->GetAllDisplays();
+}
+
+gfx::Display Screen::GetDisplayNearestPoint(const gfx::Point& point) {
+  return screen_->GetDisplayNearestPoint(point);
+}
+
+gfx::Display Screen::GetDisplayMatching(const gfx::Rect& match_rect) {
+  return screen_->GetDisplayMatching(match_rect);
+}
+
+void Screen::OnDisplayAdded(const gfx::Display& new_display) {
+  Emit("display-added");
+}
+
+void Screen::OnDisplayRemoved(const gfx::Display& old_display) {
+  Emit("display-removed");
+}
+
+void Screen::OnDisplayMetricsChanged(const gfx::Display& display,
+                                     uint32_t changed_metrics) {
+  Emit("display-metrics-changed");
+}
+
+mate::ObjectTemplateBuilder Screen::GetObjectTemplateBuilder(
+    v8::Isolate* isolate) {
+  return mate::ObjectTemplateBuilder(isolate)
+      .SetMethod("getCursorScreenPoint", &Screen::GetCursorScreenPoint)
+      .SetMethod("getPrimaryDisplay", &Screen::GetPrimaryDisplay)
+      .SetMethod("getAllDisplays", &Screen::GetAllDisplays)
+      .SetMethod("getDisplayNearestPoint", &Screen::GetDisplayNearestPoint)
+      .SetMethod("getDisplayMatching", &Screen::GetDisplayMatching);
+}
+
+// static
+mate::Handle<Screen> Screen::Create(v8::Isolate* isolate) {
+  gfx::Screen* screen = gfx::Screen::GetNativeScreen();
+  if (!screen) {
+    isolate->ThrowException(v8::Exception::Error(mate::StringToV8(
+        isolate, "Failed to get screen information")));
+    return mate::Handle<Screen>();
+  }
+
+  return CreateHandle(isolate, new Screen(screen));
+}
+
+}  // namespace api
+
+}  // namespace atom
+
 namespace {
 
 void Initialize(v8::Handle<v8::Object> exports, v8::Handle<v8::Value> unused,
                 v8::Handle<v8::Context> context, void* priv) {
-  auto screen = base::Unretained(gfx::Screen::GetNativeScreen());
   mate::Dictionary dict(context->GetIsolate(), exports);
-  dict.SetMethod("getCursorScreenPoint",
-                 base::Bind(&gfx::Screen::GetCursorScreenPoint, screen));
-  dict.SetMethod("getPrimaryDisplay",
-                 base::Bind(&gfx::Screen::GetPrimaryDisplay, screen));
-  dict.SetMethod("getAllDisplays",
-                 base::Bind(&gfx::Screen::GetAllDisplays, screen));
-  dict.SetMethod("getDisplayNearestPoint",
-                 base::Bind(&gfx::Screen::GetDisplayNearestPoint, screen));
-  dict.SetMethod("getDisplayMatching",
-                 base::Bind(&gfx::Screen::GetDisplayMatching, screen));
+  dict.Set("screen", atom::api::Screen::Create(context->GetIsolate()));
 }
 
 }  // namespace