1 files changed, 15 insertions, 10 deletions

diff --git a/src/FbTk/Menu.cc b/src/FbTk/Menu.cc
index ded5dab..d44612e 100644
--- a/src/FbTk/Menu.cc
+++ b/src/FbTk/Menu.cc
@@ -220,7 +220,7 @@ int Menu::insert(MenuItem *item, int pos) {
         menuitems.insert(menuitems.begin() + pos, item);
         fixMenuItemIndices();
         if (m_active_index >= pos)
-            m_active_index++;
+            setActive(m_active_index + 1);
     }
     m_need_update = true; // we need to redraw the menu
     return menuitems.size();
@@ -274,7 +274,7 @@ int Menu::remove(unsigned int index) {
         m_which_sub--;
 
     if (static_cast<unsigned int>(m_active_index) > index)
-        m_active_index--;
+        setActive(m_active_index-1);
 
     m_need_update = true; // we need to redraw the menu
 
@@ -335,7 +335,7 @@ void Menu::cycleItems(bool reverse) {
 void Menu::setActiveIndex(int new_index) {
     // clear the items and close any open submenus
     int old_active_index = m_active_index;
-    m_active_index = new_index;
+    setActive(new_index);
     if (validIndex(old_active_index) &&
         menuitems[old_active_index] != 0) {
         if (menuitems[old_active_index]->submenu()) {
@@ -361,7 +361,7 @@ void Menu::enterSubmenu() {
 
     drawSubmenu(m_active_index);
     submenu->grabInputFocus();
-    submenu->m_active_index = -1; // so we land on 0 after nextItem()
+    submenu->setActive(-1); // so we land on 0 after nextItem()
     submenu->cycleItems(false);
 }
 
@@ -386,11 +386,11 @@ void Menu::updateMenu() {
         for (size_t i = 1; i < menuitems.size(); i++) {
             if (validIndex(m_active_index + i) &&
                 menuitems[m_active_index + i]->isEnabled()) {
-                m_active_index += i;
+                setActive(m_active_index + i);
                 break;
             } else if (validIndex(m_active_index - i) &&
                        menuitems[m_active_index - i]->isEnabled()) {
-                m_active_index -= i;
+                setActive(m_active_index - i);
                 break;
             }
         }
@@ -622,7 +622,7 @@ void Menu::internal_hide(bool first) {
     // if we have an active index we need to redraw it
     // as non active
     int old = m_active_index;
-    m_active_index = -1;
+    setActive(-1);
     clearItem(old); // clear old area from highlight
 
     if (shown == this) {
@@ -932,7 +932,7 @@ void Menu::buttonReleaseEvent(XButtonEvent &re) {
                 menuitems[w]->click(re.button, re.time, re.state);
             } else {
                 int old = m_active_index;
-                m_active_index = w;
+                setActive(w);
                 clearItem(old);
             }
             clearItem(w);
@@ -983,7 +983,7 @@ void Menu::motionNotifyEvent(XMotionEvent &me) {
 
         if (itmp->isEnabled()) {
             int old = m_active_index;
-            m_active_index = w;
+            setActive(w);
             clearItem(w);
             clearItem(old);
 
@@ -1148,7 +1148,7 @@ void Menu::leaveNotifyEvent(XCrossingEvent &ce) {
     if (validIndex(m_which_sub) && m_active_index != m_which_sub &&
         menuitems[m_which_sub]->submenu()->isVisible()) {
         int old = m_active_index;
-        m_active_index = m_which_sub;
+        setActive(m_which_sub);
         clearItem(m_active_index);
         clearItem(old);
         menuitems[m_which_sub]->submenu()->stopHide();
@@ -1280,6 +1280,8 @@ void Menu::clearItem(int index, bool clear, int search_index) {
                true, false, item_x, item_y,
                item_w, item_h);
 
+        item->setActive(highlight);
+
     if (search_index < (int)m_matches.size())
         drawLine(index, m_type_ahead.stringSize());
 }
@@ -1313,6 +1315,9 @@ void Menu::highlightItem(int index) {
                         item_x, item_y,
                         item_w, item_h);
 
+    MenuItem *item = menuitems[index];
+    if (item)
+                item->setActive(true);
 }
 
 void Menu::resetTypeAhead() {