From a7da5d9ed8c5137c04585bcc0a1610d13c0a15f4 Mon Sep 17 00:00:00 2001
From: itdominator <1itdominator@gmail.com>
Date: Fri, 1 Apr 2022 01:13:56 -0500
Subject: [PATCH] Fix broken menus

---
 src/signal_classes/Controller.py | 14 ++++++++------
 1 file changed, 8 insertions(+), 6 deletions(-)

diff --git a/src/signal_classes/Controller.py b/src/signal_classes/Controller.py
index bb4480f..c447326 100644
--- a/src/signal_classes/Controller.py
+++ b/src/signal_classes/Controller.py
@@ -155,12 +155,14 @@ class Controller(ProcessorMixin, Menu, Controller_Data):
         elif "Favorites" in group:
             desktop_objects = self.favorites
         else:
-            for option in self.menu_data[group]:
-                keys = option.keys()
-                if "comment" in keys and len(option["comment"]) > 0 :
-                    desktop_objects.append( f"{option['title']} || {option['comment']}" )
-                else:
-                    desktop_objects.append( f"{option['title']} || {option['fileName'].replace('.desktop', '')}" )
+            for key in self.menu_data[group]:
+                option = self.flat_menu_data[key]
+                keys   = option.keys()
+                if "comment" in keys and len(option["comment"]) > 0:
+                    if query.lower() in option["comment"].lower():
+                        desktop_objects.append( f"{key} || {option['comment']}" )
+                elif query.lower() in key.lower() or query.lower() in option["fileName"].lower():
+                        desktop_objects.append( f"{key} || {option['fileName'].replace('.desktop', '')}" )
 
         return desktop_objects