souleyez 2.43.29__py3-none-any.whl → 3.0.0__py3-none-any.whl

souleyez/ui/shortcuts.py

@@ -12,16 +12,18 @@ from enum import Enum
 
 class ShortcutContext(Enum):
     """Context where shortcuts are available."""
-    GLOBAL = "global"              # Available everywhere
-    DASHBOARD = "dashboard"        # Dashboard/Command Center
-    DETAILED_VIEW = "detailed"     # Detailed view mode
-    MENU = "menu"                  # Interactive menus
-    VIEW = "view"                  # Specific views (hosts, findings, etc.)
+
+    GLOBAL = "global"  # Available everywhere
+    DASHBOARD = "dashboard"  # Dashboard/Command Center
+    DETAILED_VIEW = "detailed"  # Detailed view mode
+    MENU = "menu"  # Interactive menus
+    VIEW = "view"  # Specific views (hosts, findings, etc.)
 
 
 @dataclass
 class Shortcut:
     """Represents a keyboard shortcut."""
+
     key: str
     description: str
     context: ShortcutContext
@@ -50,182 +52,228 @@ class ShortcutManager:
         # GLOBAL SHORTCUTS - Available everywhere
         # ═══════════════════════════════════════════════════════════
 
-        cls.register(Shortcut(
-            key='q',
-            description='Quit/Exit',
-            context=ShortcutContext.GLOBAL,
-            action='quit'
-        ))
-
-        cls.register(Shortcut(
-            key='?',
-            description='Help',
-            context=ShortcutContext.GLOBAL,
-            action='help'
-        ))
-
-        cls.register(Shortcut(
-            key='m',
-            description='Menu',
-            context=ShortcutContext.GLOBAL,
-            action='menu'
-        ))
+        cls.register(
+            Shortcut(
+                key="q",
+                description="Quit/Exit",
+                context=ShortcutContext.GLOBAL,
+                action="quit",
+            )
+        )
+
+        cls.register(
+            Shortcut(
+                key="?",
+                description="Help",
+                context=ShortcutContext.GLOBAL,
+                action="help",
+            )
+        )
+
+        cls.register(
+            Shortcut(
+                key="m",
+                description="Menu",
+                context=ShortcutContext.GLOBAL,
+                action="menu",
+            )
+        )
 
         # ═══════════════════════════════════════════════════════════
         # DASHBOARD / COMMAND CENTER SHORTCUTS
         # ═══════════════════════════════════════════════════════════
 
-        cls.register(Shortcut(
-            key='c',
-            description='Command Center',
-            context=ShortcutContext.DASHBOARD,
-            action='command_center'
-        ))
-
-        cls.register(Shortcut(
-            key='o',
-            description='OSINT/Discovery View',
-            context=ShortcutContext.DASHBOARD,
-            action='osint_discovery'
-        ))
-
-        cls.register(Shortcut(
-            key='v',
-            description='Toggle View (Command Center ↔ Detailed)',
-            context=ShortcutContext.DASHBOARD,
-            action='toggle_view'
-        ))
-
-        cls.register(Shortcut(
-            key='i',
-            description='Intelligence Dashboard',
-            context=ShortcutContext.DASHBOARD,
-            action='intelligence'
-        ))
-
-        cls.register(Shortcut(
-            key='e',
-            description='Evidence Vault',
-            context=ShortcutContext.DASHBOARD,
-            action='evidence'
-        ))
-
-        cls.register(Shortcut(
-            key='r',
-            description='Reports & Deliverables',
-            context=ShortcutContext.DASHBOARD,
-            action='reports'
-        ))
-
-        cls.register(Shortcut(
-            key='a',
-            description='Auto-chaining Toggle',
-            context=ShortcutContext.DASHBOARD,
-            action='auto_chain'
-        ))
-
-        cls.register(Shortcut(
-            key='x',
-            description='Execute AI Recommendation',
-            context=ShortcutContext.DASHBOARD,
-            action='ai_execute'
-        ))
-
-        cls.register(Shortcut(
-            key='1',
-            description='Quick Action #1',
-            context=ShortcutContext.DASHBOARD,
-            action='quick_action_1'
-        ))
-
-        cls.register(Shortcut(
-            key='2',
-            description='Quick Action #2',
-            context=ShortcutContext.DASHBOARD,
-            action='quick_action_2'
-        ))
-
-        cls.register(Shortcut(
-            key='3',
-            description='Quick Action #3',
-            context=ShortcutContext.DASHBOARD,
-            action='quick_action_3'
-        ))
+        cls.register(
+            Shortcut(
+                key="c",
+                description="Command Center",
+                context=ShortcutContext.DASHBOARD,
+                action="command_center",
+            )
+        )
+
+        cls.register(
+            Shortcut(
+                key="o",
+                description="OSINT/Discovery View",
+                context=ShortcutContext.DASHBOARD,
+                action="osint_discovery",
+            )
+        )
+
+        cls.register(
+            Shortcut(
+                key="v",
+                description="Toggle View (Command Center ↔ Detailed)",
+                context=ShortcutContext.DASHBOARD,
+                action="toggle_view",
+            )
+        )
+
+        cls.register(
+            Shortcut(
+                key="i",
+                description="Intelligence Dashboard",
+                context=ShortcutContext.DASHBOARD,
+                action="intelligence",
+            )
+        )
+
+        cls.register(
+            Shortcut(
+                key="e",
+                description="Evidence Vault",
+                context=ShortcutContext.DASHBOARD,
+                action="evidence",
+            )
+        )
+
+        cls.register(
+            Shortcut(
+                key="r",
+                description="Reports & Deliverables",
+                context=ShortcutContext.DASHBOARD,
+                action="reports",
+            )
+        )
+
+        cls.register(
+            Shortcut(
+                key="a",
+                description="Auto-chaining Toggle",
+                context=ShortcutContext.DASHBOARD,
+                action="auto_chain",
+            )
+        )
+
+        cls.register(
+            Shortcut(
+                key="x",
+                description="Execute AI Recommendation",
+                context=ShortcutContext.DASHBOARD,
+                action="ai_execute",
+            )
+        )
+
+        cls.register(
+            Shortcut(
+                key="1",
+                description="Quick Action #1",
+                context=ShortcutContext.DASHBOARD,
+                action="quick_action_1",
+            )
+        )
+
+        cls.register(
+            Shortcut(
+                key="2",
+                description="Quick Action #2",
+                context=ShortcutContext.DASHBOARD,
+                action="quick_action_2",
+            )
+        )
+
+        cls.register(
+            Shortcut(
+                key="3",
+                description="Quick Action #3",
+                context=ShortcutContext.DASHBOARD,
+                action="quick_action_3",
+            )
+        )
 
         # ═══════════════════════════════════════════════════════════
         # DATA ACCESS SHORTCUTS
         # ═══════════════════════════════════════════════════════════
 
-        cls.register(Shortcut(
-            key='h',
-            description='Hosts',
-            context=ShortcutContext.DASHBOARD,
-            action='hosts'
-        ))
-
-        cls.register(Shortcut(
-            key='f',
-            description='Findings',
-            context=ShortcutContext.DASHBOARD,
-            action='findings'
-        ))
-
-        cls.register(Shortcut(
-            key='u',
-            description='User Credentials',
-            context=ShortcutContext.DASHBOARD,
-            action='credentials'
-        ))
-
-        cls.register(Shortcut(
-            key='j',
-            description='Jobs',
-            context=ShortcutContext.DASHBOARD,
-            action='jobs'
-        ))
+        cls.register(
+            Shortcut(
+                key="h",
+                description="Hosts",
+                context=ShortcutContext.DASHBOARD,
+                action="hosts",
+            )
+        )
+
+        cls.register(
+            Shortcut(
+                key="f",
+                description="Findings",
+                context=ShortcutContext.DASHBOARD,
+                action="findings",
+            )
+        )
+
+        cls.register(
+            Shortcut(
+                key="u",
+                description="User Credentials",
+                context=ShortcutContext.DASHBOARD,
+                action="credentials",
+            )
+        )
+
+        cls.register(
+            Shortcut(
+                key="j",
+                description="Jobs",
+                context=ShortcutContext.DASHBOARD,
+                action="jobs",
+            )
+        )
 
         # ═══════════════════════════════════════════════════════════
         # DETAILED VIEW SHORTCUTS
         # ═══════════════════════════════════════════════════════════
 
-        cls.register(Shortcut(
-            key='n',
-            description='Minimal Mode Toggle',
-            context=ShortcutContext.DETAILED_VIEW,
-            action='minimal'
-        ))
-
-        cls.register(Shortcut(
-            key='s',
-            description='Filter Severity',
-            context=ShortcutContext.DETAILED_VIEW,
-            action='filter_severity'
-        ))
-
-        cls.register(Shortcut(
-            key='t',
-            description='Toggle Options',
-            context=ShortcutContext.DETAILED_VIEW,
-            action='toggle_menu'
-        ))
+        cls.register(
+            Shortcut(
+                key="n",
+                description="Minimal Mode Toggle",
+                context=ShortcutContext.DETAILED_VIEW,
+                action="minimal",
+            )
+        )
+
+        cls.register(
+            Shortcut(
+                key="s",
+                description="Filter Severity",
+                context=ShortcutContext.DETAILED_VIEW,
+                action="filter_severity",
+            )
+        )
+
+        cls.register(
+            Shortcut(
+                key="t",
+                description="Toggle Options",
+                context=ShortcutContext.DETAILED_VIEW,
+                action="toggle_menu",
+            )
+        )
 
         # ═══════════════════════════════════════════════════════════
         # VIEW-SPECIFIC SHORTCUTS
         # ═══════════════════════════════════════════════════════════
 
-        cls.register(Shortcut(
-            key='0',
-            description='Back/Return',
-            context=ShortcutContext.VIEW,
-            action='back'
-        ))
-
-        cls.register(Shortcut(
-            key='b',
-            description='Back/Return',
-            context=ShortcutContext.VIEW,
-            action='back'
-        ))
+        cls.register(
+            Shortcut(
+                key="0",
+                description="Back/Return",
+                context=ShortcutContext.VIEW,
+                action="back",
+            )
+        )
+
+        cls.register(
+            Shortcut(
+                key="b",
+                description="Back/Return",
+                context=ShortcutContext.VIEW,
+                action="back",
+            )
+        )
 
     @classmethod
     def register(cls, shortcut: Shortcut):
@@ -244,7 +292,7 @@ class ShortcutManager:
         """Get all shortcuts for a specific context."""
         prefix = f"{context.value}:"
         return {
-            key.split(':', 1)[1]: shortcut
+            key.split(":", 1)[1]: shortcut
             for key, shortcut in cls._shortcuts.items()
             if key.startswith(prefix)
         }
@@ -255,7 +303,9 @@ class ShortcutManager:
         return cls.get_all_for_context(ShortcutContext.GLOBAL)
 
     @classmethod
-    def format_shortcut_help(cls, context: ShortcutContext, include_global: bool = True) -> str:
+    def format_shortcut_help(
+        cls, context: ShortcutContext, include_global: bool = True
+    ) -> str:
         """
         Format shortcuts for display in help text.
 
@@ -317,8 +367,10 @@ class ShortcutManager:
                 unique_shortcuts.append((key, shortcut))
 
         # Format compactly
-        formatted = [f"[{key}] {shortcut.description.split()[0]}"
-                     for key, shortcut in sorted(unique_shortcuts)]
+        formatted = [
+            f"[{key}] {shortcut.description.split()[0]}"
+            for key, shortcut in sorted(unique_shortcuts)
+        ]
 
         return " ".join(formatted)