set the close protocol on the app's main widget

1 files changed, 25 insertions, 1 deletions

diff --git a/otk/application.cc b/otk/application.cc
index 16b9e0f5..0a250ea3 100644
--- a/otk/application.cc
+++ b/otk/application.cc
@@ -1,7 +1,10 @@
 #include "application.hh"
 #include "eventhandler.hh"
+#include "widget.hh"
 
 extern "C" {
+#include <X11/Xlib.h>
+  
 #ifdef HAVE_STDLIB_H
 # include <stdlib.h>
 #endif
@@ -12,7 +15,7 @@ extern "C" {
 namespace otk {
 
 OtkApplication::OtkApplication(int argc, char **argv)
-  : OtkEventDispatcher(), _dockable(false)
+  : OtkEventDispatcher(), _main_widget(0), _dockable(false)
 {
   argc = argc;
   argv = argv;
@@ -52,10 +55,31 @@ void OtkApplication::loadStyle(void)
 
 void OtkApplication::exec(void)
 {
+  if (!_main_widget) {
+    std::cerr << "No main widget set. You must create a main OtkWidget for " <<
+      "the OtkApplication before calling OtkApplication::exec().\n";
+    ::exit(1);
+  }
   while (1) {
     dispatchEvents();
     _timer_manager->fire(); // fire pending events
   }
 }
 
+bool OtkApplication::setMainWidget(const OtkWidget *main_widget)
+{
+  // ignore it if it has already been set
+  if (_main_widget) return false;
+
+  _main_widget = main_widget;
+
+  // set WM Protocols on the window
+  Atom protocols[2];
+  protocols[0] = XInternAtom(OBDisplay::display, "WM_PROTOCOLS", false);
+  protocols[1] = XInternAtom(OBDisplay::display, "WM_DELETE_WINDOW", false);
+  XSetWMProtocols(OBDisplay::display, _main_widget->getWindow(), protocols, 2);
+
+  return true;
+}
+
 }