hidemenus hide custommenus and fix custommenu memleak

2 files changed, 9 insertions, 3 deletions

diff --git a/src/FbCommands.cc b/src/FbCommands.cc
index d53afd2..20d45e0 100644
--- a/src/FbCommands.cc
+++ b/src/FbCommands.cc
@@ -269,6 +269,8 @@ void HideMenuCmd::execute() {
         screen->rootMenu().hide();
     if (screen->workspaceMenu().isVisible())
         screen->workspaceMenu().hide();
+    if (FbTk::Menu::shownMenu())
+        FbTk::Menu::shownMenu()->hide();
 }
 
 ShowCustomMenuCmd::ShowCustomMenuCmd(const string &arguments) : custom_menu_file(arguments) {}
@@ -277,8 +279,9 @@ void ShowCustomMenuCmd::execute() {
     BScreen *screen = Fluxbox::instance()->mouseScreen();
     if (screen == 0)
         return;
-    ::showMenu(*screen, *MenuCreator::createFromFile(custom_menu_file,
-                            screen->screenNumber(), true));
+    m_menu = MenuCreator::createFromFile(custom_menu_file,
+            screen->screenNumber(), true);
+    ::showMenu(*screen, **m_menu);
 }
 
 void ShowRootMenuCmd::execute() {
diff --git a/src/FbCommands.hh b/src/FbCommands.hh
index 3bc5074..314d339 100644
--- a/src/FbCommands.hh
+++ b/src/FbCommands.hh
@@ -28,6 +28,9 @@
 
 #include "Command.hh"
 
+#include "FbTk/RefCount.hh"
+#include "FbTk/Menu.hh"
+
 #include <string>
 
 namespace FbCommands {
@@ -124,7 +127,7 @@ public:
     void execute();
 private:
    std::string custom_menu_file;
-    
+   FbTk::RefCount<FbTk::Menu> m_menu;    
 };
 
 class ShowRootMenuCmd: public FbTk::Command {