souleyez 2.43.26__py3-none-any.whl → 2.43.34__py3-none-any.whl

souleyez/ui/wazuh_vulns_view.py

@@ -18,10 +18,10 @@ console = Console()
 
 # Severity colors
 SEVERITY_COLORS = {
-    'Critical': 'red',
-    'High': 'yellow',
-    'Medium': 'white',
-    'Low': 'bright_black'
+    "Critical": "red",
+    "High": "yellow",
+    "Medium": "white",
+    "Low": "bright_black",
 }
 
 
@@ -50,14 +50,24 @@ def show_wazuh_vulns_view(engagement_id: int, engagement_name: str = "") -> None
 
         # Header
         click.echo("\n┌" + "─" * (width - 2) + "┐")
-        click.echo("│" + click.style(" WAZUH VULNERABILITIES ".center(width - 2), bold=True, fg='cyan') + "│")
+        click.echo(
+            "│"
+            + click.style(
+                " WAZUH VULNERABILITIES ".center(width - 2), bold=True, fg="cyan"
+            )
+            + "│"
+        )
         click.echo("└" + "─" * (width - 2) + "┘")
         click.echo()
 
         # Check if Wazuh is configured
         config = WazuhConfig.get_config(engagement_id)
-        if not config or not config.get('enabled'):
-            click.echo(click.style("  ⚠️  Wazuh is not configured for this engagement.", fg='yellow'))
+        if not config or not config.get("enabled"):
+            click.echo(
+                click.style(
+                    "  ⚠️  Wazuh is not configured for this engagement.", fg="yellow"
+                )
+            )
             click.echo()
             click.echo("  Configure Wazuh in Settings → Integrations → Wazuh SIEM")
             click.echo()
@@ -65,7 +75,7 @@ def show_wazuh_vulns_view(engagement_id: int, engagement_name: str = "") -> None
             click.echo()
             click.echo("  [q] Back")
             click.echo()
-            if click.getchar().lower() == 'q':
+            if click.getchar().lower() == "q":
                 return
             continue
 
@@ -74,42 +84,60 @@ def show_wazuh_vulns_view(engagement_id: int, engagement_name: str = "") -> None
         sync_status = sync.get_sync_status()
 
         # Sync status line at top
-        if sync_status.get('synced'):
-            last_sync = sync_status.get('last_sync_at', 'Unknown')
-            stale = sync_status.get('is_stale')
-            status_color = 'yellow' if stale else 'green'
-            status_text = '(stale)' if stale else '(fresh)'
-            click.echo(f"  Last sync: {last_sync} " + click.style(status_text, fg=status_color) +
-                      f" | Count: {sync_status.get('last_sync_count', 0)}")
+        if sync_status.get("synced"):
+            last_sync = sync_status.get("last_sync_at", "Unknown")
+            stale = sync_status.get("is_stale")
+            status_color = "yellow" if stale else "green"
+            status_text = "(stale)" if stale else "(fresh)"
+            click.echo(
+                f"  Last sync: {last_sync} "
+                + click.style(status_text, fg=status_color)
+                + f" | Count: {sync_status.get('last_sync_count', 0)}"
+            )
             click.echo()
         else:
-            click.echo(click.style("  Never synced - press [s] to sync now", fg='yellow'))
+            click.echo(
+                click.style("  Never synced - press [s] to sync now", fg="yellow")
+            )
             click.echo()
 
         # Get summary
         summary = vulns_manager.get_summary(engagement_id)
         mapping_stats = host_mapper.get_mapping_stats(engagement_id)
 
-        total = summary.get('total', 0)
-        verified = summary.get('verified', 0)
-        by_sev = summary.get('by_severity', {})
-        mapped = mapping_stats.get('mapped', 0)
-        unmapped = mapping_stats.get('unmapped', 0)
+        total = summary.get("total", 0)
+        verified = summary.get("verified", 0)
+        by_sev = summary.get("by_severity", {})
+        mapped = mapping_stats.get("mapped", 0)
+        unmapped = mapping_stats.get("unmapped", 0)
 
         # Get all vulnerabilities with filters
         filter_kwargs = {}
         if severity_filter:
-            filter_kwargs['severity'] = severity_filter
+            filter_kwargs["severity"] = severity_filter
         if host_filter:
-            filter_kwargs['agent_ip'] = host_filter
+            filter_kwargs["agent_ip"] = host_filter
 
-        all_vulns = vulns_manager.list_vulnerabilities(engagement_id, limit=1000, **filter_kwargs)
+        all_vulns = vulns_manager.list_vulnerabilities(
+            engagement_id, limit=1000, **filter_kwargs
+        )
 
         # Display table with summary header
         page, total_pages = _display_vulns_table(
-            console, all_vulns, selected_ids, page, page_size,
-            view_all, severity_filter, host_filter, width, by_sev, total,
-            mapped, unmapped, verified
+            console,
+            all_vulns,
+            selected_ids,
+            page,
+            page_size,
+            view_all,
+            severity_filter,
+            host_filter,
+            width,
+            by_sev,
+            total,
+            mapped,
+            unmapped,
+            verified,
         )
 
         # Menu
@@ -128,30 +156,30 @@ def show_wazuh_vulns_view(engagement_id: int, engagement_name: str = "") -> None
         try:
             choice = input("  Select option: ").strip().lower()
 
-            if choice == 'q':
+            if choice == "q":
                 return
-            elif choice == 's':
+            elif choice == "s":
                 _do_sync(engagement_id)
-            elif choice == 'm':
+            elif choice == "m":
                 _mapping_menu(engagement_id)
-            elif choice == 'i':
+            elif choice == "i":
                 # Interactive mode
                 _interactive_mode(engagement_id, all_vulns, selected_ids)
-            elif choice == 't':
+            elif choice == "t":
                 view_all = not view_all
                 if not view_all:
                     page = 0
-            elif choice == 'n' and not view_all and page < total_pages - 1:
+            elif choice == "n" and not view_all and page < total_pages - 1:
                 page += 1
-            elif choice == 'p' and not view_all and page > 0:
+            elif choice == "p" and not view_all and page > 0:
                 page -= 1
-            elif choice == 'f':
+            elif choice == "f":
                 severity_filter = _select_severity_filter()
                 page = 0
-            elif choice == 'h':
+            elif choice == "h":
                 host_filter = _select_host_filter(engagement_id)
                 page = 0
-            elif choice == 'c':
+            elif choice == "c":
                 severity_filter = None
                 host_filter = None
                 page = 0
@@ -161,10 +189,10 @@ def show_wazuh_vulns_view(engagement_id: int, engagement_name: str = "") -> None
                 if 0 <= vuln_idx < len(all_vulns):
                     _show_vuln_detail(all_vulns[vuln_idx])
                 else:
-                    click.echo(click.style("  Invalid number", fg='red'))
+                    click.echo(click.style("  Invalid number", fg="red"))
                     click.pause()
             else:
-                click.echo(click.style("Invalid option", fg='red'))
+                click.echo(click.style("Invalid option", fg="red"))
                 click.pause()
 
         except (KeyboardInterrupt, EOFError):
@@ -172,10 +200,20 @@ def show_wazuh_vulns_view(engagement_id: int, engagement_name: str = "") -> None
 
 
 def _display_vulns_table(
-    console: Console, vulns: List[Dict], selected_ids: set,
-    page: int, page_size: int, view_all: bool,
-    severity_filter: Optional[str], host_filter: Optional[str],
-    width: int, by_sev: Dict, total: int, mapped: int, unmapped: int, verified: int
+    console: Console,
+    vulns: List[Dict],
+    selected_ids: set,
+    page: int,
+    page_size: int,
+    view_all: bool,
+    severity_filter: Optional[str],
+    host_filter: Optional[str],
+    width: int,
+    by_sev: Dict,
+    total: int,
+    mapped: int,
+    unmapped: int,
+    verified: int,
 ) -> tuple:
     """Display vulnerabilities table with summary header.
 
@@ -183,15 +221,19 @@ def _display_vulns_table(
     """
     # Count by severity from current vulns
     severity_counts = {
-        'Critical': sum(1 for v in vulns if v.get('severity') == 'Critical'),
-        'High': sum(1 for v in vulns if v.get('severity') == 'High'),
-        'Medium': sum(1 for v in vulns if v.get('severity') == 'Medium'),
-        'Low': sum(1 for v in vulns if v.get('severity') == 'Low')
+        "Critical": sum(1 for v in vulns if v.get("severity") == "Critical"),
+        "High": sum(1 for v in vulns if v.get("severity") == "High"),
+        "Medium": sum(1 for v in vulns if v.get("severity") == "Medium"),
+        "Low": sum(1 for v in vulns if v.get("severity") == "Low"),
     }
 
     # Summary header
     click.echo("═" * width)
-    click.echo(click.style(f"⚠️  WAZUH VULNERABILITIES ({len(vulns)} total)", bold=True, fg='yellow'))
+    click.echo(
+        click.style(
+            f"⚠️  WAZUH VULNERABILITIES ({len(vulns)} total)", bold=True, fg="yellow"
+        )
+    )
 
     # Severity breakdown line with emojis
     sev_line = (
@@ -203,7 +245,12 @@ def _display_vulns_table(
     click.echo(sev_line)
 
     # Stats line
-    click.echo(click.style(f"  Verified: {verified}  │  Mapped: {mapped}  │  Unmapped: {unmapped}", fg='bright_black'))
+    click.echo(
+        click.style(
+            f"  Verified: {verified}  │  Mapped: {mapped}  │  Unmapped: {unmapped}",
+            fg="bright_black",
+        )
+    )
 
     # Show active filters
     if severity_filter or host_filter:
@@ -212,13 +259,13 @@ def _display_vulns_table(
             filter_parts.append(f"Severity: {severity_filter}")
         if host_filter:
             filter_parts.append(f"Host: {host_filter}")
-        click.echo(click.style(f"  🔍 Filters: {', '.join(filter_parts)}", fg='cyan'))
+        click.echo(click.style(f"  🔍 Filters: {', '.join(filter_parts)}", fg="cyan"))
 
     click.echo("─" * width)
     click.echo()
 
     if not vulns:
-        click.echo("  " + click.style("No vulnerabilities found!", fg='green'))
+        click.echo("  " + click.style("No vulnerabilities found!", fg="green"))
         click.echo("  Press [s] to sync from Wazuh.")
         click.echo()
         return 0, 1
@@ -240,7 +287,7 @@ def _display_vulns_table(
         header_style="bold cyan",
         box=DesignSystem.TABLE_BOX,
         padding=(0, 1),
-        expand=True
+        expand=True,
     )
 
     table.add_column("○", width=3, justify="center")  # Checkbox
@@ -258,37 +305,31 @@ def _display_vulns_table(
         else:
             display_idx = (page * page_size) + idx + 1
 
-        vuln_id = vuln.get('id')
+        vuln_id = vuln.get("id")
 
         # Checkbox
-        checkbox = '●' if vuln_id in selected_ids else '○'
+        checkbox = "●" if vuln_id in selected_ids else "○"
 
         # CVE
-        cve = vuln.get('cve_id', '-')
+        cve = vuln.get("cve_id", "-")
 
         # Severity with color
-        severity = vuln.get('severity', 'Medium')
-        sev_color = SEVERITY_COLORS.get(severity, 'white')
+        severity = vuln.get("severity", "Medium")
+        sev_color = SEVERITY_COLORS.get(severity, "white")
         sev_display = f"[{sev_color}]{severity}[/{sev_color}]"
 
         # Host
-        host_ip = vuln.get('host_ip') or vuln.get('agent_ip', '-')
+        host_ip = vuln.get("host_ip") or vuln.get("agent_ip", "-")
 
         # Package (truncated)
-        package = vuln.get('package_name', '-')[:24]
+        package = vuln.get("package_name", "-")[:24]
 
         # CVSS
-        cvss = vuln.get('cvss_score')
+        cvss = vuln.get("cvss_score")
         cvss_display = f"{cvss:.1f}" if cvss else "-"
 
         table.add_row(
-            checkbox,
-            str(display_idx),
-            cve,
-            sev_display,
-            host_ip,
-            package,
-            cvss_display
+            checkbox, str(display_idx), cve, sev_display, host_ip, package, cvss_display
         )
 
     console.print("  ", table)
@@ -312,60 +353,70 @@ def _display_vulns_table(
 def _interactive_mode(engagement_id: int, vulns: List[Dict], selected_ids: set) -> None:
     """Run interactive selection mode."""
     if not vulns:
-        click.echo(click.style("  No vulnerabilities to select.", fg='yellow'))
+        click.echo(click.style("  No vulnerabilities to select.", fg="yellow"))
         click.pause()
         return
 
     # Prepare items for interactive selector
     vuln_items = []
     for vuln in vulns:
-        vuln_items.append({
-            'id': vuln.get('id'),
-            'cve_id': vuln.get('cve_id', '-'),
-            'severity': vuln.get('severity', 'Medium'),
-            'host': vuln.get('host_ip') or vuln.get('agent_ip', '-'),
-            'package': vuln.get('package_name', '-')[:24],
-            'cvss': f"{vuln.get('cvss_score', 0):.1f}" if vuln.get('cvss_score') else '-',
-            'raw': vuln
-        })
+        vuln_items.append(
+            {
+                "id": vuln.get("id"),
+                "cve_id": vuln.get("cve_id", "-"),
+                "severity": vuln.get("severity", "Medium"),
+                "host": vuln.get("host_ip") or vuln.get("agent_ip", "-"),
+                "package": vuln.get("package_name", "-")[:24],
+                "cvss": (
+                    f"{vuln.get('cvss_score', 0):.1f}"
+                    if vuln.get("cvss_score")
+                    else "-"
+                ),
+                "raw": vuln,
+            }
+        )
 
     columns = [
-        {'name': 'CVE', 'key': 'cve_id', 'width': 18},
-        {'name': 'Severity', 'key': 'severity', 'width': 10},
-        {'name': 'Host', 'key': 'host', 'width': 16},
-        {'name': 'Package', 'key': 'package', 'width': 24},
-        {'name': 'CVSS', 'key': 'cvss', 'width': 6, 'justify': 'center'}
+        {"name": "CVE", "key": "cve_id", "width": 18},
+        {"name": "Severity", "key": "severity", "width": 10},
+        {"name": "Host", "key": "host", "width": 16},
+        {"name": "Package", "key": "package", "width": 24},
+        {"name": "CVSS", "key": "cvss", "width": 6, "justify": "center"},
     ]
 
     def format_cell(item: Dict, key: str) -> str:
-        if key == 'severity':
-            sev = item.get('severity', 'Medium')
-            color = SEVERITY_COLORS.get(sev, 'white')
+        if key == "severity":
+            sev = item.get("severity", "Medium")
+            color = SEVERITY_COLORS.get(sev, "white")
             return f"[{color}]{sev}[/{color}]"
-        return str(item.get(key, '-'))
+        return str(item.get(key, "-"))
 
     interactive_select(
         items=vuln_items,
         columns=columns,
         selected_ids=selected_ids,
-        get_id=lambda v: v['id'],
-        title='SELECT WAZUH VULNERABILITIES',
-        format_cell=format_cell
+        get_id=lambda v: v["id"],
+        title="SELECT WAZUH VULNERABILITIES",
+        format_cell=format_cell,
     )
 
     if selected_ids:
         _bulk_action_menu(engagement_id, vuln_items, selected_ids)
 
 
-def _bulk_action_menu(engagement_id: int, vuln_items: List[Dict], selected_ids: set) -> None:
+def _bulk_action_menu(
+    engagement_id: int, vuln_items: List[Dict], selected_ids: set
+) -> None:
     """Show bulk action menu for selected vulnerabilities."""
-    selected = [v for v in vuln_items if v['id'] in selected_ids]
+    selected = [v for v in vuln_items if v["id"] in selected_ids]
 
     if not selected:
         return
 
     click.echo()
-    click.echo(click.style(f"  Selected: {len(selected)} vulnerability(ies)", bold=True))
+    click.echo(
+        click.style(f"  Selected: {len(selected)} vulnerability(ies)", bold=True)
+    )
     click.echo("    [v] View details")
     click.echo("    [r] Mark as verified")
     click.echo("    [e] Export to file")
@@ -375,42 +426,52 @@ def _bulk_action_menu(engagement_id: int, vuln_items: List[Dict], selected_ids:
     click.echo()
 
     try:
-        choice = click.prompt("  Select option", default='q', show_default=False).strip().lower()
+        choice = (
+            click.prompt("  Select option", default="q", show_default=False)
+            .strip()
+            .lower()
+        )
 
-        if choice == 'v':
+        if choice == "v":
             # View details of first selected
-            vuln = selected[0]['raw']
+            vuln = selected[0]["raw"]
             _show_vuln_detail(vuln)
-        elif choice == 'r':
+        elif choice == "r":
             # Mark as verified
             vulns_manager = WazuhVulnsManager()
             count = 0
             for item in selected:
                 try:
-                    vulns_manager.update_vulnerability(item['id'], status='confirmed', verified_by_scan=True)
+                    vulns_manager.update_vulnerability(
+                        item["id"], status="confirmed", verified_by_scan=True
+                    )
                     count += 1
                 except Exception:
                     pass
-            click.echo(click.style(f"  ✓ Verified {count} vulnerabilities", fg='green'))
+            click.echo(click.style(f"  ✓ Verified {count} vulnerabilities", fg="green"))
             click.pause("  Press any key to continue...")
             selected_ids.clear()
-        elif choice == 'x':
+        elif choice == "x":
             # Mark as false positive
             vulns_manager = WazuhVulnsManager()
             count = 0
             for item in selected:
                 try:
-                    vulns_manager.update_vulnerability(item['id'], status='false_positive')
+                    vulns_manager.update_vulnerability(
+                        item["id"], status="false_positive"
+                    )
                     count += 1
                 except Exception:
                     pass
-            click.echo(click.style(f"  ✓ Marked {count} as false positive", fg='yellow'))
+            click.echo(
+                click.style(f"  ✓ Marked {count} as false positive", fg="yellow")
+            )
             click.pause("  Press any key to continue...")
             selected_ids.clear()
-        elif choice == 'e':
+        elif choice == "e":
             # Export selected
             _export_selected(selected)
-        elif choice == 'c':
+        elif choice == "c":
             selected_ids.clear()
 
     except (KeyboardInterrupt, EOFError):
@@ -426,20 +487,26 @@ def _export_selected(selected: List[Dict]) -> None:
 
     export_data = []
     for item in selected:
-        vuln = item.get('raw', item)
-        export_data.append({
-            'cve_id': vuln.get('cve_id'),
-            'severity': vuln.get('severity'),
-            'cvss_score': vuln.get('cvss_score'),
-            'host_ip': vuln.get('host_ip') or vuln.get('agent_ip'),
-            'package_name': vuln.get('package_name'),
-            'package_version': vuln.get('package_version')
-        })
-
-    with open(filename, 'w') as f:
+        vuln = item.get("raw", item)
+        export_data.append(
+            {
+                "cve_id": vuln.get("cve_id"),
+                "severity": vuln.get("severity"),
+                "cvss_score": vuln.get("cvss_score"),
+                "host_ip": vuln.get("host_ip") or vuln.get("agent_ip"),
+                "package_name": vuln.get("package_name"),
+                "package_version": vuln.get("package_version"),
+            }
+        )
+
+    with open(filename, "w") as f:
         json.dump(export_data, f, indent=2)
 
-    click.echo(click.style(f"  ✓ Exported {len(selected)} vulnerabilities to {filename}", fg='green'))
+    click.echo(
+        click.style(
+            f"  ✓ Exported {len(selected)} vulnerabilities to {filename}", fg="green"
+        )
+    )
     click.pause("  Press any key to continue...")
 
 
@@ -451,9 +518,9 @@ def _select_severity_filter() -> Optional[str]:
     click.echo()
 
     key = click.getchar().lower()
-    severity_map = {'1': 'Critical', '2': 'High', '3': 'Medium', '4': 'Low'}
+    severity_map = {"1": "Critical", "2": "High", "3": "Medium", "4": "Low"}
 
-    if key == 'c':
+    if key == "c":
         return None
     return severity_map.get(key)
 
@@ -463,7 +530,7 @@ def _select_host_filter(engagement_id: int) -> Optional[str]:
     click.echo()
     ip = click.prompt("  Enter host IP (partial match, or 'c' to clear)", default="")
 
-    if ip.lower() == 'c' or not ip:
+    if ip.lower() == "c" or not ip:
         return None
     return ip
 
@@ -474,7 +541,11 @@ def _do_sync(engagement_id: int) -> None:
     width = DesignSystem.get_terminal_width()
 
     click.echo("\n┌" + "─" * (width - 2) + "┐")
-    click.echo("│" + click.style(" SYNCING FROM WAZUH ".center(width - 2), bold=True, fg='blue') + "│")
+    click.echo(
+        "│"
+        + click.style(" SYNCING FROM WAZUH ".center(width - 2), bold=True, fg="blue")
+        + "│"
+    )
     click.echo("└" + "─" * (width - 2) + "┘")
     click.echo()
 
@@ -486,7 +557,7 @@ def _do_sync(engagement_id: int) -> None:
     click.echo()
 
     if result.success:
-        click.echo(click.style("  ✓ Sync complete!", fg='green', bold=True))
+        click.echo(click.style("  ✓ Sync complete!", fg="green", bold=True))
         click.echo(f"    Fetched: {result.total_fetched}")
         click.echo(f"    New: {result.new_vulns}")
         click.echo(f"    Updated: {result.updated_vulns}")
@@ -494,7 +565,11 @@ def _do_sync(engagement_id: int) -> None:
 
         if result.unmapped_agents:
             click.echo()
-            click.echo(click.style(f"  Unmapped agents ({len(result.unmapped_agents)}):", fg='yellow'))
+            click.echo(
+                click.style(
+                    f"  Unmapped agents ({len(result.unmapped_agents)}):", fg="yellow"
+                )
+            )
             for agent_ip in result.unmapped_agents[:5]:
                 click.echo(f"    - {agent_ip}")
             if len(result.unmapped_agents) > 5:
@@ -502,16 +577,18 @@ def _do_sync(engagement_id: int) -> None:
 
         if result.errors:
             click.echo()
-            click.echo(click.style(f"  Errors ({len(result.errors)}):", fg='yellow'))
+            click.echo(click.style(f"  Errors ({len(result.errors)}):", fg="yellow"))
             for err in result.errors[:3]:
                 click.echo(f"    - {err}")
     else:
-        click.echo(click.style("  ✗ Sync failed", fg='red', bold=True))
+        click.echo(click.style("  ✗ Sync failed", fg="red", bold=True))
         for err in result.errors:
-            click.echo(click.style(f"    {err}", fg='red'))
+            click.echo(click.style(f"    {err}", fg="red"))
 
     click.echo()
-    click.echo(click.style(f"  Duration: {result.duration_seconds:.1f}s", fg='bright_black'))
+    click.echo(
+        click.style(f"  Duration: {result.duration_seconds:.1f}s", fg="bright_black")
+    )
     click.echo()
     click.pause("  Press any key to continue...")
 
@@ -524,34 +601,42 @@ def _mapping_menu(engagement_id: int) -> None:
     width = DesignSystem.get_terminal_width()
 
     click.echo("\n┌" + "─" * (width - 2) + "┐")
-    click.echo("│" + click.style(" HOST MAPPING ".center(width - 2), bold=True, fg='cyan') + "│")
+    click.echo(
+        "│"
+        + click.style(" HOST MAPPING ".center(width - 2), bold=True, fg="cyan")
+        + "│"
+    )
     click.echo("└" + "─" * (width - 2) + "┘")
     click.echo()
 
     # Get mapping stats
     stats = host_mapper.get_mapping_stats(engagement_id)
-    click.echo(f"  Mapped: {stats['mapped']} | Unmapped: {stats['unmapped']} | Total: {stats['total']}")
+    click.echo(
+        f"  Mapped: {stats['mapped']} | Unmapped: {stats['unmapped']} | Total: {stats['total']}"
+    )
     click.echo()
 
     # Show unmapped agents
     unmapped = host_mapper.get_unmapped_agents(engagement_id)
     if unmapped:
-        click.echo(click.style("  Unmapped Agents:", fg='yellow'))
+        click.echo(click.style("  Unmapped Agents:", fg="yellow"))
         for agent in unmapped[:10]:
-            click.echo(f"    - {agent.get('agent_ip')} ({agent.get('agent_name', 'unknown')}) - {agent.get('vuln_count')} vulns")
+            click.echo(
+                f"    - {agent.get('agent_ip')} ({agent.get('agent_name', 'unknown')}) - {agent.get('vuln_count')} vulns"
+            )
 
         # Show suggestions
         suggestions = host_mapper.suggest_mappings(engagement_id)
         if suggestions:
             click.echo()
-            click.echo(click.style("  Suggested Mappings:", fg='cyan'))
+            click.echo(click.style("  Suggested Mappings:", fg="cyan"))
             for sug in suggestions[:5]:
                 click.echo(
                     f"    {sug['agent_ip']} → {sug['suggested_host_ip']} "
                     f"({sug.get('suggested_host_name', '-')}) - {sug['reason']}"
                 )
     else:
-        click.echo(click.style("  ✓ All agents are mapped to hosts.", fg='green'))
+        click.echo(click.style("  ✓ All agents are mapped to hosts.", fg="green"))
 
     click.echo()
     click.echo("─" * width)
@@ -563,13 +648,15 @@ def _mapping_menu(engagement_id: int) -> None:
 
     key = click.getchar().lower()
 
-    if key == '1':
+    if key == "1":
         click.echo()
         click.echo("  Auto-mapping agents to hosts...")
         mapping = host_mapper.auto_map_all(engagement_id)
 
         mapped_count = sum(1 for h in mapping.values() if h)
-        click.echo(click.style(f"  ✓ Mapped {mapped_count} agents to hosts", fg='green'))
+        click.echo(
+            click.style(f"  ✓ Mapped {mapped_count} agents to hosts", fg="green")
+        )
         click.echo()
         click.pause("  Press any key to continue...")
 
@@ -579,17 +666,22 @@ def _show_vuln_detail(vuln: Dict) -> None:
     DesignSystem.clear_screen()
     width = DesignSystem.get_terminal_width()
 
-    cve = vuln.get('cve_id', 'Unknown')
-    severity = vuln.get('severity', 'Medium')
-    sev_color = SEVERITY_COLORS.get(severity, 'white')
+    cve = vuln.get("cve_id", "Unknown")
+    severity = vuln.get("severity", "Medium")
+    sev_color = SEVERITY_COLORS.get(severity, "white")
 
     click.echo("\n┌" + "─" * (width - 2) + "┐")
-    click.echo("│" + click.style(f" {cve} ".center(width - 2), bold=True, fg='cyan') + "│")
+    click.echo(
+        "│" + click.style(f" {cve} ".center(width - 2), bold=True, fg="cyan") + "│"
+    )
     click.echo("└" + "─" * (width - 2) + "┘")
     click.echo()
 
-    click.echo(f"  Severity: " + click.style(severity, fg=sev_color, bold=True) +
-              f" | CVSS: {vuln.get('cvss_score', '-')}")
+    click.echo(
+        f"  Severity: "
+        + click.style(severity, fg=sev_color, bold=True)
+        + f" | CVSS: {vuln.get('cvss_score', '-')}"
+    )
     click.echo()
 
     click.echo(click.style("  Agent/Host:", bold=True))
@@ -609,10 +701,12 @@ def _show_vuln_detail(vuln: Dict) -> None:
     click.echo(f"    Detected: {vuln.get('detection_time', '-')}")
     click.echo(f"    Published: {vuln.get('published_date', '-')}")
     click.echo(f"    Status: {vuln.get('status', 'open')}")
-    click.echo(f"    Verified by scan: {'Yes' if vuln.get('verified_by_scan') else 'No'}")
+    click.echo(
+        f"    Verified by scan: {'Yes' if vuln.get('verified_by_scan') else 'No'}"
+    )
 
     # References
-    refs = vuln.get('reference_urls', [])
+    refs = vuln.get("reference_urls", [])
     if refs:
         click.echo()
         click.echo(click.style("  References:", bold=True))