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

souleyez/ui/interactive_selector.py

@@ -4,13 +4,14 @@ souleyez.ui.interactive_selector - Reusable interactive row selector with keyboa
 
 Provides arrow-key/vim-style navigation for selecting items from tables.
 """
+
 import sys
 from typing import Any, Callable, Dict, List, Optional, Set, Tuple
 
 import click
+from rich import box
 from rich.console import Console
 from rich.table import Table
-from rich import box
 
 from souleyez.ui.design_system import DesignSystem
 
@@ -104,9 +105,9 @@ def _get_key() -> str:
 
     # Fallback: use termios on Unix-like systems
     try:
+        import select
         import termios
         import tty
-        import select
 
         fd = sys.stdin.fileno()
         old_settings = termios.tcgetattr(fd)

souleyez/ui/log_formatter.py

@@ -2,6 +2,7 @@
 """
 Log formatting utilities for human-readable display.
 """
+
 import json
 import re
 from datetime import datetime

souleyez/ui/menu_components.py

@@ -2,8 +2,10 @@
 Reusable menu UI components for consistent layouts.
 """
 
+from typing import Dict, List
+
 import click
-from typing import List, Dict
+
 from souleyez.ui.design_system import DesignSystem
 
 

souleyez/ui/msf_auxiliary_menu.py

@@ -9,8 +9,11 @@ Features:
 - Batch execution mode
 - Quick actions (auto-enum, search)
 """
+
+from typing import Any, Dict, List, Optional, Set, Tuple
+
 import click
-from typing import Dict, List, Any, Optional, Set, Tuple
+
 from souleyez.ui.design_system import DesignSystem
 
 

souleyez/ui/pending_chains_view.py

@@ -6,25 +6,28 @@ UI for reviewing and approving/rejecting pending chain operations.
 This implements the "active orchestration" workflow where users can
 review suggested follow-up scans before execution.
 """
+
 import math
-import click
 from typing import List, Set, Tuple
+
+import click
+
 from souleyez.core.pending_chains import (
-    list_pending_chains,
-    get_pending_chain,
-    approve_chain,
-    reject_chain,
-    approve_all_pending,
-    reject_all_pending,
-    get_chain_stats,
-    CHAIN_PENDING,
     CHAIN_APPROVED,
-    CHAIN_REJECTED,
     CHAIN_EXECUTED,
+    CHAIN_PENDING,
+    CHAIN_REJECTED,
+    approve_all_pending,
+    approve_chain,
+    get_chain_stats,
+    get_pending_chain,
+    list_pending_chains,
+    reject_all_pending,
+    reject_chain,
 )
 from souleyez.core.tool_chaining import ToolChaining
-from souleyez.ui.menu_components import StandardMenu
 from souleyez.ui.design_system import DesignSystem
+from souleyez.ui.menu_components import StandardMenu
 
 
 def manage_pending_chains(engagement_id: int = None):
@@ -805,9 +808,9 @@ def _view_chain_history(engagement_id: int = None):
 
 def show_active_orchestration_help():
     """Display the Active Orchestration help guide with Rich formatting."""
+    from rich import box
     from rich.console import Console
     from rich.panel import Panel
-    from rich import box
 
     console = Console()
     DesignSystem.clear_screen()

souleyez/ui/progress_indicators.py

@@ -4,11 +4,14 @@ souleyez.ui.progress_indicators - Progress indicators for long-running operation
 
 Provides entertaining progress feedback during AI generation and other slow tasks.
 """
+
+import sys
 import threading
 import time
-import sys
+from typing import Any, Callable, Optional
+
 import click
-from typing import Callable, Any, Optional
+
 from souleyez.ui.ai_quotes import get_random_quote
 
 

souleyez/ui/recommendations_view.py

@@ -1,9 +1,11 @@
 """Smart recommendations dashboard."""
 
-import click
 from typing import Dict
-from souleyez.storage.recommendation_engine import RecommendationEngine
+
+import click
+
 from souleyez.storage.engagements import EngagementManager
+from souleyez.storage.recommendation_engine import RecommendationEngine
 from souleyez.ui.design_system import DesignSystem
 
 

souleyez/ui/rule_builder.py

@@ -2,8 +2,11 @@
 """
 souleyez.ui.rule_builder - Interactive chain rule builder for custom automation
 """
+
+from typing import Any, Dict, List, Optional
+
 import click
-from typing import Optional, Dict, List, Any
+
 from souleyez.ui.design_system import DesignSystem
 
 

souleyez/ui/setup_wizard.py

@@ -2,16 +2,18 @@
 """
 souleyez.ui.setup_wizard - First-run setup wizard for new users
 """
-import os
-import time
-import click
+
 import getpass
+import os
 import shutil
 import subprocess
+import time
 from pathlib import Path
-from typing import List, Dict, Optional
-from souleyez.ui.design_system import DesignSystem
+from typing import Dict, List, Optional
 
+import click
+
+from souleyez.ui.design_system import DesignSystem
 
 # Wizard state file
 WIZARD_STATE_FILE = Path.home() / ".souleyez" / ".wizard_completed"
@@ -471,7 +473,7 @@ def run_setup_wizard() -> bool:
     """
     try:
         # Check if user has Pro tier
-        from souleyez.auth import get_current_user, Tier
+        from souleyez.auth import Tier, get_current_user
 
         user = get_current_user()
         is_pro = user and user.tier == Tier.PRO
@@ -881,10 +883,10 @@ def _wizard_create_engagement() -> dict:
 def _wizard_tool_check() -> dict:
     """Check installed tools using the centralized tool_checker module."""
     from souleyez.utils.tool_checker import (
-        check_tool_version,
         EXTERNAL_TOOLS,
-        get_install_command,
+        check_tool_version,
         detect_distro,
+        get_install_command,
         get_tool_version,
         get_upgrade_command,
     )

souleyez/ui/shortcuts.py

@@ -6,8 +6,8 @@ across all views to prevent conflicts and improve consistency.
 """
 
 from dataclasses import dataclass
-from typing import Dict, Optional, Callable
 from enum import Enum
+from typing import Callable, Dict, Optional
 
 
 class ShortcutContext(Enum):

souleyez/ui/splunk_gap_analysis_view.py

@@ -5,14 +5,17 @@ souleyez.ui.splunk_gap_analysis_view - Splunk Gap Analysis View
 Displays comparison between Splunk (passive, synced from Wazuh) and
 scan (active) vulnerability detection to identify gaps in coverage.
 """
+
+from dataclasses import dataclass, field
+from typing import Any, Dict, List, Optional
+
 import click
+from rich import box
 from rich.console import Console
 from rich.table import Table
-from rich import box
-from dataclasses import dataclass, field
-from typing import List, Dict, Any, Optional
-from souleyez.ui.design_system import DesignSystem
+
 from souleyez.storage.database import get_db
+from souleyez.ui.design_system import DesignSystem
 
 console = Console()
 

souleyez/ui/splunk_vulns_view.py

@@ -5,10 +5,13 @@ souleyez.ui.splunk_vulns_view - Splunk Vulnerabilities View
 Displays vulnerabilities synced from Wazuh to Splunk with
 filtering and display capabilities.
 """
+
+from typing import Dict, List, Optional
+
 import click
 from rich.console import Console
 from rich.table import Table
-from typing import List, Dict, Optional
+
 from souleyez.ui.design_system import DesignSystem
 
 console = Console()

souleyez/ui/team_dashboard.py

@@ -1,17 +1,19 @@
 """Team collaboration dashboard."""
 
+from typing import Dict, List, Optional
+
 import click
-from typing import Optional, List, Dict
-from souleyez.storage.team_collaboration import TeamCollaboration
-from souleyez.storage.engagements import EngagementManager
+
 from souleyez.storage.deliverables import DeliverableManager
+from souleyez.storage.engagements import EngagementManager
+from souleyez.storage.team_collaboration import TeamCollaboration
 from souleyez.ui.design_system import DesignSystem
 from souleyez.ui.interactive_selector import (
-    _get_key,
-    KEY_UP,
     KEY_DOWN,
     KEY_ENTER,
     KEY_ESCAPE,
+    KEY_UP,
+    _get_key,
 )
 
 

souleyez/ui/template_selector.py

@@ -1,7 +1,8 @@
 """Template selection interface for deliverables - Interactive table style."""
 
+from typing import Any, Dict, List, Optional
+
 import click
-from typing import Dict, List, Optional, Any
 from rich.console import Console
 from rich.table import Table
 

souleyez/ui/terminal.py

@@ -2,10 +2,11 @@
 """
 souleyez.ui.terminal - Terminal configuration and input handling
 """
-import sys
-import readline
+
 import atexit
 import os
+import readline
+import sys
 
 
 def init_readline():

souleyez/ui/timeline_view.py

@@ -1,9 +1,10 @@
 """Timeline and velocity tracking UI for deliverables."""
 
 import click
-from souleyez.storage.timeline_tracker import TimelineTracker
+
 from souleyez.storage.deliverables import DeliverableManager
 from souleyez.storage.engagements import EngagementManager
+from souleyez.storage.timeline_tracker import TimelineTracker
 from souleyez.ui.design_system import DesignSystem
 
 

souleyez/ui/tool_setup.py

@@ -5,24 +5,26 @@ souleyez.ui.tool_setup - Tool installation wizard for Ubuntu/Debian systems
 Helps users install pentesting tools that aren't available via apt on Ubuntu.
 Handles PATH configuration for go and pipx installed tools.
 """
-import click
+
+import os
 import shutil
 import subprocess
 import sys
-import os
 from pathlib import Path
-from typing import List, Dict, Optional
+from typing import Dict, List, Optional
 
+import click
+
+from souleyez.ui.design_system import DesignSystem
 from souleyez.utils.tool_checker import (
+    EXTERNAL_TOOLS,
+    check_tool,
     detect_distro,
-    get_missing_tools,
-    get_tools_by_category,
     get_category_name,
-    check_tool,
+    get_missing_tools,
     get_tool_stats,
-    EXTERNAL_TOOLS,
+    get_tools_by_category,
 )
-from souleyez.ui.design_system import DesignSystem
 
 
 def _reset_terminal():
@@ -46,35 +48,35 @@ def _reset_terminal():
 PREREQUISITES = {
     "build-deps": {
         "check": None,  # Always install to ensure all deps present
-        "install": "sudo apt install -y build-essential python3-dev libxml2-dev libxslt1-dev libuv1-dev libffi-dev libssl-dev rustc cargo",
+        "install": "sudo apt-get -o DPkg::Lock::Timeout=120 install -y build-essential python3-dev libxml2-dev libxslt1-dev libuv1-dev libffi-dev libssl-dev rustc cargo",
         "description": "Build dependencies for Python packages with native extensions",
         "always_install": True,  # Flag to always run this
     },
     "pipx": {
         "check": "pipx",
-        "install": "sudo apt install -y pipx && pipx ensurepath",
+        "install": "sudo apt-get -o DPkg::Lock::Timeout=120 install -y pipx && pipx ensurepath",
         "description": "Python application installer (for theHarvester, NetExec, etc.)",
         "path_additions": ["~/.local/bin"],
     },
     "golang": {
         "check": "go",
-        "install": "sudo apt install -y golang-go",
+        "install": "sudo apt-get -o DPkg::Lock::Timeout=120 install -y golang-go",
         "description": "Go programming language (for nuclei, ffuf)",
         "path_additions": ["~/go/bin"],
     },
     "ruby": {
         "check": "gem",
-        "install": "sudo apt install -y ruby-full ruby-dev build-essential",
+        "install": "sudo apt-get -o DPkg::Lock::Timeout=120 install -y ruby-full ruby-dev build-essential",
         "description": "Ruby programming language (for wpscan)",
     },
     "snap": {
         "check": "snap",
-        "install": "sudo apt install -y snapd",
+        "install": "sudo apt-get -o DPkg::Lock::Timeout=120 install -y snapd",
         "description": "Snap package manager (for enum4linux)",
     },
     "git": {
         "check": "git",
-        "install": "sudo apt install -y git",
+        "install": "sudo apt-get -o DPkg::Lock::Timeout=120 install -y git",
         "description": "Git version control (for exploitdb, Responder)",
     },
 }
@@ -371,9 +373,17 @@ def _ensure_msfdb_initialized(console):
     console.print("  [dim]Initializing MSF database (this may take a minute)...[/dim]")
 
     try:
-        # Use sudo for msfdb init (required on Kali and some other distros)
+        # Distro-aware msfdb init:
+        # - Kali/Parrot (apt install): requires sudo
+        # - Ubuntu/Debian (omnibus installer): must run as normal user, NOT root
+        distro = detect_distro()
+        if distro in ("kali", "parrot"):
+            msfdb_cmd = ["sudo", msfdb_path, "init"]
+        else:
+            msfdb_cmd = [msfdb_path, "init"]
+
         result = subprocess.run(
-            ["sudo", msfdb_path, "init"],
+            msfdb_cmd,
             capture_output=True,
             timeout=300,  # 5 minute timeout
         )
@@ -749,7 +759,7 @@ def _install_apt_tools(console, tools: List[Dict]) -> bool:
         return True
 
     console.print(f"    Packages: {', '.join(packages)}")
-    cmd = f"sudo apt install -y {' '.join(packages)}"
+    cmd = f"sudo apt-get -o DPkg::Lock::Timeout=120 install -y {' '.join(packages)}"
     success, _, _ = _run_command(cmd, console)
 
     if success:
@@ -767,20 +777,63 @@ def _install_pipx_tool(console, tool: Dict) -> bool:
 
     console.print(f"    {name}...", end=" ")
 
-    # pipx install commands
-    success, _, stderr = _run_command(cmd, console, capture=True)
+    # For netexec and other complex builds, show message about output capture
+    if "NetExec" in cmd or "netexec" in cmd.lower():
+        console.print()
+        console.print(
+            "      [dim]Building (output captured, this may take a while)...[/dim]"
+        )
 
-    if success:
-        console.print("[green]✓[/green]")
-        return True
-    else:
-        # Check if already installed
-        if "already exists" in stderr or "already installed" in stderr.lower():
-            console.print("[green]✓ (already installed)[/green]")
+    # pipx install commands - use subprocess directly to ensure full capture
+    # Some tools (like netexec with Rust deps) can corrupt terminal with progress bars
+    try:
+        result = subprocess.run(
+            cmd,
+            shell=True,
+            capture_output=True,
+            text=True,
+            timeout=600,  # 10 minute timeout
+        )
+        success = result.returncode == 0
+        stderr = result.stderr
+
+        if success:
+            console.print(
+                "    [green]✓[/green]"
+                if "NetExec" not in cmd
+                else "      [green]✓[/green]"
+            )
             return True
-        console.print(f"[red]✗[/red]")
-        if stderr:
-            console.print(f"      [dim]{stderr[:80]}[/dim]")
+        else:
+            # Check if already installed
+            if "already exists" in stderr or "already installed" in stderr.lower():
+                console.print(
+                    "    [green]✓ (already installed)[/green]"
+                    if "NetExec" not in cmd
+                    else "      [green]✓ (already installed)[/green]"
+                )
+                return True
+            console.print(
+                "    [red]✗[/red]" if "NetExec" not in cmd else "      [red]✗[/red]"
+            )
+            if stderr:
+                # Truncate and clean stderr for display
+                clean_stderr = stderr.replace("\n", " ")[:80]
+                console.print(f"      [dim]{clean_stderr}[/dim]")
+            return False
+    except subprocess.TimeoutExpired:
+        console.print(
+            "    [red]✗ timeout[/red]"
+            if "NetExec" not in cmd
+            else "      [red]✗ timeout[/red]"
+        )
+        return False
+    except Exception as e:
+        console.print(
+            f"    [red]✗[/red] {str(e)[:50]}"
+            if "NetExec" not in cmd
+            else f"      [red]✗[/red] {str(e)[:50]}"
+        )
         return False
 
 
@@ -980,11 +1033,19 @@ def _install_other_tool(console, tool: Dict) -> bool:
         console.print(
             "      [dim]Installing postgresql and downloading Metasploit installer...[/dim]"
         )
+        console.print(
+            "      [dim]Output is captured to prevent terminal corruption...[/dim]"
+        )
 
-        # Run the full install command (includes postgresql setup)
+        # Run the full install command with captured output to prevent terminal corruption
+        # Metasploit installer outputs progress bars and escape sequences that can corrupt Rich console
         try:
             result = subprocess.run(
-                cmd, shell=True, timeout=1800  # 30 minute timeout for metasploit
+                cmd,
+                shell=True,
+                capture_output=True,  # Capture to prevent terminal corruption
+                text=True,
+                timeout=1800,  # 30 minute timeout for metasploit
             )
             if result.returncode == 0:
                 console.print("      [green]✓ Metasploit installed[/green]")

souleyez/ui/tutorial.py

@@ -9,10 +9,13 @@ This tutorial guides new users through:
 4. Running reconnaissance scans
 5. Exploring the Command Center (dashboard) interactively
 """
-import click
+
 import time
 from pathlib import Path
-from souleyez.ui.tutorial_state import get_tutorial_state, TutorialStep
+
+import click
+
+from souleyez.ui.tutorial_state import TutorialStep, get_tutorial_state
 
 
 def clear_screen():
@@ -453,9 +456,9 @@ def _show_tutorial_complete():
 def _cleanup_tutorial_data():
     """Clean up tutorial engagement, jobs, and related data."""
     try:
-        from souleyez.storage.engagements import EngagementManager
-        from souleyez.engine.background import list_jobs, delete_job, kill_job
         from souleyez.core.tool_chaining import ToolChaining
+        from souleyez.engine.background import delete_job, kill_job, list_jobs
+        from souleyez.storage.engagements import EngagementManager
 
         # Disable auto-chaining (it's a PRO feature, tutorial enabled it for demo)
         try:

souleyez/ui/tutorial_state.py

@@ -5,10 +5,11 @@ souleyez.ui.tutorial_state - Tutorial state manager for interactive guided tutor
 This module tracks tutorial progress across different UI screens,
 allowing contextual hints to be displayed in dashboard, job queue, etc.
 """
+
+import json
 from enum import Enum, auto
 from pathlib import Path
-from typing import Optional, Dict, Any
-import json
+from typing import Any, Dict, Optional
 
 
 class TutorialStep(Enum):

souleyez/ui/wazuh_vulns_view.py

@@ -5,12 +5,15 @@ souleyez.ui.wazuh_vulns_view - Wazuh Vulnerabilities View
 Displays vulnerabilities discovered by Wazuh agents with sync,
 filtering, and status management capabilities.
 """
+
+from typing import Dict, List, Optional
+
 import click
 from rich.console import Console
 from rich.table import Table
-from typing import List, Dict, Optional
+
+from souleyez.integrations.wazuh import WazuhConfig, WazuhHostMapper, WazuhVulnSync
 from souleyez.storage.wazuh_vulns import WazuhVulnsManager
-from souleyez.integrations.wazuh import WazuhVulnSync, WazuhConfig, WazuhHostMapper
 from souleyez.ui.design_system import DesignSystem
 from souleyez.ui.interactive_selector import interactive_select
 

souleyez/ui/wordlist_browser.py

@@ -5,6 +5,7 @@ souleyez.ui.wordlist_browser - Interactive wordlist browser with keyboard naviga
 Discovers wordlists from common directories (SecLists, Kali, etc.) and provides
 an interactive browser for selection.
 """
+
 import os
 from pathlib import Path
 from typing import Any, Dict, List, Optional, Tuple
@@ -15,11 +16,11 @@ from rich.table import Table
 
 from souleyez.ui.design_system import DesignSystem
 from souleyez.ui.interactive_selector import (
-    _get_key,
-    KEY_UP,
     KEY_DOWN,
     KEY_ENTER,
     KEY_ESCAPE,
+    KEY_UP,
+    _get_key,
 )
 
 # Common wordlist directories to scan
@@ -322,8 +323,8 @@ class WordlistBrowser:
 
     def _render(self):
         """Render the browser UI."""
-        from rich.table import Table
         from rich import box
+        from rich.table import Table
 
         DesignSystem.clear_screen()
         width = DesignSystem.get_terminal_width()

souleyez/ui.py

@@ -5,11 +5,12 @@
 # - We group by tool, and show ALL entries (no pagination)
 # - Export pulls an entry by list index (1-based) in current view
 
-import sys
 import os
 import subprocess
-from typing import List, Dict, Any
-from .utils import nmap_installed, detect_local_subnet
+import sys
+from typing import Any, Dict, List
+
+from .utils import detect_local_subnet, nmap_installed
 
 # Ensure enqueue_job is available for the TUI enqueue flow.
 # If the background engine isn't available in this environment, provide a friendly stub
@@ -34,11 +35,11 @@ except Exception:
         )
 
 
-from .engine.background import list_jobs, get_job
+from .engine.background import get_job, list_jobs
 from .engine.loader import discover_plugins
 from .engine.manager import run_scan_sync
 from .scanner import run_nmap
-from .storage.db import get_scans, get_scan
+from .storage.db import get_scan, get_scans
 
 # (history.py kept for export helpers if needed in future, not used for reads)
 
@@ -336,8 +337,9 @@ def history_export(scans: List[Dict[str, Any]]):
         print("Invalid index.")
         return
     # Export using a simple inline writer (avoid legacy)
+    import csv
+    import json
     from pathlib import Path
-    import json, csv
 
     export_dir = Path.home() / ".souleyez" / "exports"
     export_dir.mkdir(parents=True, exist_ok=True)
@@ -707,7 +709,11 @@ def view_job_live(job_id: int, refresh_interval: float = 1.0, max_lines: int = 3
     - tries to locate log path via storage.get_scan(scan_id) if result_scan_id exists
     - falls back to ~/.souleyez/artifacts/<job_id>.log when missing
     """
-    import os, sys, time, json, select
+    import json
+    import os
+    import select
+    import sys
+    import time
 
     # lookup job record
     j = None