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

souleyez/ui/template_selector.py

@@ -1,4 +1,5 @@
 """Template selection interface for deliverables - Interactive table style."""
+
 import click
 from typing import Dict, List, Optional, Any
 from rich.console import Console
@@ -10,6 +11,7 @@ from souleyez.ui.design_system import DesignSystem
 # Key codes
 try:
     import readchar
+
     KEY_UP = readchar.key.UP
     KEY_DOWN = readchar.key.DOWN
     KEY_PAGE_UP = readchar.key.PAGE_UP
@@ -18,51 +20,52 @@ try:
     KEY_ENTER = readchar.key.ENTER
     KEY_SPACE = readchar.key.SPACE
 except (ImportError, AttributeError):
-    KEY_UP = '\x1b[A'
-    KEY_DOWN = '\x1b[B'
-    KEY_PAGE_UP = '\x1b[5~'
-    KEY_PAGE_DOWN = '\x1b[6~'
-    KEY_ESCAPE = '\x1b'
-    KEY_ENTER = '\r'
-    KEY_SPACE = ' '
+    KEY_UP = "\x1b[A"
+    KEY_DOWN = "\x1b[B"
+    KEY_PAGE_UP = "\x1b[5~"
+    KEY_PAGE_DOWN = "\x1b[6~"
+    KEY_ESCAPE = "\x1b"
+    KEY_ENTER = "\r"
+    KEY_SPACE = " "
 
 
 def _get_key() -> str:
     """Read a single keypress."""
     try:
         import readchar
+
         return readchar.readkey()
     except ImportError:
         pass
 
     try:
         ch = click.getchar()
-        if ch == '\x1b' or (len(ch) > 1 and ch.startswith('\x1b')):
+        if ch == "\x1b" or (len(ch) > 1 and ch.startswith("\x1b")):
             if len(ch) >= 3:
-                if ch == '\x1b[A':
+                if ch == "\x1b[A":
                     return KEY_UP
-                elif ch == '\x1b[B':
+                elif ch == "\x1b[B":
                     return KEY_DOWN
-                elif ch in ('\x1b[5~', '\x1b[5'):
+                elif ch in ("\x1b[5~", "\x1b[5"):
                     return KEY_PAGE_UP
-                elif ch in ('\x1b[6~', '\x1b[6'):
+                elif ch in ("\x1b[6~", "\x1b[6"):
                     return KEY_PAGE_DOWN
-            elif ch == '\x1b':
+            elif ch == "\x1b":
                 try:
                     ch2 = click.getchar()
-                    if ch2 == '[':
+                    if ch2 == "[":
                         ch3 = click.getchar()
-                        if ch3 == 'A':
+                        if ch3 == "A":
                             return KEY_UP
-                        elif ch3 == 'B':
+                        elif ch3 == "B":
                             return KEY_DOWN
-                        elif ch3 in ('5', '6'):
+                        elif ch3 in ("5", "6"):
                             click.getchar()
-                            return KEY_PAGE_UP if ch3 == '5' else KEY_PAGE_DOWN
-                    return ''
+                            return KEY_PAGE_UP if ch3 == "5" else KEY_PAGE_DOWN
+                    return ""
                 except Exception:
                     return KEY_ESCAPE
-            return ''
+            return ""
         return ch
     except Exception:
         return KEY_ESCAPE
@@ -71,17 +74,36 @@ def _get_key() -> str:
 class TemplateSelector:
     """Interactive template selector with keyboard navigation."""
 
-    CURSOR = '▶'
-    NO_CURSOR = ' '
-    RADIO_EMPTY = '○'
-    RADIO_SELECTED = '●'
+    CURSOR = "▶"
+    NO_CURSOR = " "
+    RADIO_EMPTY = "○"
+    RADIO_SELECTED = "●"
 
     # Category filters
     CATEGORIES = [
-        ('all', 'All', None),
-        ('compliance', '🏛️ Compliance', ['hipaa', 'pci-dss', 'nist', 'owasp', 'soc2', 'iso27001', 'cis', 'cmmc', 'gdpr', 'glba']),
-        ('pentest', '🎯 Pentest', ['ptes', 'internal', 'external', 'webapp', 'redteam', 'cloud', 'ad']),
-        ('industry', '🏭 Industry', ['nerc-cip', 'hitrust', 'ffiec']),
+        ("all", "All", None),
+        (
+            "compliance",
+            "🏛️ Compliance",
+            [
+                "hipaa",
+                "pci-dss",
+                "nist",
+                "owasp",
+                "soc2",
+                "iso27001",
+                "cis",
+                "cmmc",
+                "gdpr",
+                "glba",
+            ],
+        ),
+        (
+            "pentest",
+            "🎯 Pentest",
+            ["ptes", "internal", "external", "webapp", "redteam", "cloud", "ad"],
+        ),
+        ("industry", "🏭 Industry", ["nerc-cip", "hitrust", "ffiec"]),
     ]
 
     def __init__(self, templates: List[Dict], engagement_id: int):
@@ -96,12 +118,12 @@ class TemplateSelector:
         self.selected_template = None
         self.selected_idx = None  # Currently selected row (for inline options)
         self.show_inline_options = False
-        self.current_filter = 'all'
+        self.current_filter = "all"
 
     def run(self) -> Optional[Dict]:
         """Run the interactive selector. Returns selected template or None."""
         if not self.templates:
-            click.echo(click.style("  No templates available.", fg='yellow'))
+            click.echo(click.style("  No templates available.", fg="yellow"))
             click.pause()
             return None
 
@@ -122,29 +144,44 @@ class TemplateSelector:
         self.selected_idx = None
         self.show_inline_options = False
 
-        if filter_key == 'all':
+        if filter_key == "all":
             self.templates = self.all_templates
         else:
             # Find the filter
             for key, label, frameworks in self.CATEGORIES:
                 if key == filter_key and frameworks:
-                    self.templates = [t for t in self.all_templates if t.get('framework') in frameworks]
+                    self.templates = [
+                        t
+                        for t in self.all_templates
+                        if t.get("framework") in frameworks
+                    ]
                     break
 
     def _get_category(self, template: Dict) -> str:
         """Get display category for template."""
-        framework = template.get('framework', '')
-        compliance = ['hipaa', 'pci-dss', 'nist', 'owasp', 'soc2', 'iso27001', 'cis', 'cmmc', 'gdpr', 'glba']
-        pentest = ['ptes', 'internal', 'external', 'webapp', 'redteam', 'cloud', 'ad']
-        industry = ['nerc-cip', 'hitrust', 'ffiec']
+        framework = template.get("framework", "")
+        compliance = [
+            "hipaa",
+            "pci-dss",
+            "nist",
+            "owasp",
+            "soc2",
+            "iso27001",
+            "cis",
+            "cmmc",
+            "gdpr",
+            "glba",
+        ]
+        pentest = ["ptes", "internal", "external", "webapp", "redteam", "cloud", "ad"]
+        industry = ["nerc-cip", "hitrust", "ffiec"]
 
         if framework in compliance:
-            return '🏛️'
+            return "🏛️"
         elif framework in pentest:
-            return '🎯'
+            return "🎯"
         elif framework in industry:
-            return '🏭'
-        return '📋'
+            return "🏭"
+        return "📋"
 
     def _render(self):
         """Render the table."""
@@ -154,7 +191,15 @@ class TemplateSelector:
         # Title
         click.echo()
         click.echo("┌" + "─" * (width - 2) + "┐")
-        click.echo("│" + click.style(" 🎯 INITIALIZE DELIVERABLES FROM TEMPLATE ".center(width - 2), bold=True, fg='cyan') + "│")
+        click.echo(
+            "│"
+            + click.style(
+                " 🎯 INITIALIZE DELIVERABLES FROM TEMPLATE ".center(width - 2),
+                bold=True,
+                fg="cyan",
+            )
+            + "│"
+        )
         click.echo("└" + "─" * (width - 2) + "┘")
         click.echo()
 
@@ -162,7 +207,9 @@ class TemplateSelector:
         filter_parts = []
         for idx, (key, label, _) in enumerate(self.CATEGORIES):
             if key == self.current_filter:
-                filter_parts.append(click.style(f"[{idx+1}] {label}", fg='cyan', bold=True))
+                filter_parts.append(
+                    click.style(f"[{idx+1}] {label}", fg="cyan", bold=True)
+                )
             else:
                 filter_parts.append(f"[{idx+1}] {label}")
         click.echo("  " + "  |  ".join(filter_parts))
@@ -173,12 +220,14 @@ class TemplateSelector:
         selected_info = ""
         if self.selected_idx is not None:
             selected_info = f"  |  {click.style('Selected:', bold=True, fg='green')} 1"
-        click.echo(f"  {click.style('Total:', bold=True)} {total} templates{selected_info}")
+        click.echo(
+            f"  {click.style('Total:', bold=True)} {total} templates{selected_info}"
+        )
         click.echo()
 
         # Calculate visible items
         page_end = min(self.page_start + self.page_size, len(self.templates))
-        visible_templates = self.templates[self.page_start:page_end]
+        visible_templates = self.templates[self.page_start : page_end]
 
         # Create table
         table = Table(
@@ -186,7 +235,7 @@ class TemplateSelector:
             header_style="bold cyan",
             box=DesignSystem.TABLE_BOX,
             padding=(0, 1),
-            expand=True
+            expand=True,
         )
 
         table.add_column(" ", width=2, justify="center", no_wrap=True)
@@ -199,21 +248,21 @@ class TemplateSelector:
 
         for idx, template in enumerate(visible_templates):
             absolute_idx = self.page_start + idx
-            is_cursor = (absolute_idx == self.cursor_pos)
-            is_selected = (absolute_idx == self.selected_idx)
+            is_cursor = absolute_idx == self.cursor_pos
+            is_selected = absolute_idx == self.selected_idx
 
             cursor = self.CURSOR if is_cursor else self.NO_CURSOR
             radio = self.RADIO_SELECTED if is_selected else self.RADIO_EMPTY
 
             category_icon = self._get_category(template)
-            deliverable_count = len(template.get('deliverables', []))
-            framework = (template.get('framework') or '').upper()
+            deliverable_count = len(template.get("deliverables", []))
+            framework = (template.get("framework") or "").upper()
 
             row = [
                 cursor,
                 radio,
                 str(absolute_idx + 1),
-                template['name'],
+                template["name"],
                 str(deliverable_count),
                 category_icon,
                 framework,
@@ -244,7 +293,9 @@ class TemplateSelector:
         if self.show_inline_options and self.selected_idx is not None:
             # Inline options when template is selected
             template = self.templates[self.selected_idx]
-            click.echo(f"  {click.style('Selected:', bold=True, fg='green')} {template['name']} ({len(template.get('deliverables', []))} deliverables)")
+            click.echo(
+                f"  {click.style('Selected:', bold=True, fg='green')} {template['name']} ({len(template.get('deliverables', []))} deliverables)"
+            )
             click.echo()
             click.echo(f"  [v] Preview - View template details")
             click.echo(f"  [l] Load - Load this template")
@@ -266,16 +317,16 @@ class TemplateSelector:
         """Handle keypress."""
         # Handle inline options mode first
         if self.show_inline_options and self.selected_idx is not None:
-            if key == 'v':
+            if key == "v":
                 # Preview selected template
                 self._preview_template(self.selected_idx)
                 return
-            elif key == 'l':
+            elif key == "l":
                 # Load selected template
                 self.selected_template = self.templates[self.selected_idx]
                 self.running = False
                 return
-            elif key in ('b', KEY_ESCAPE):
+            elif key in ("b", KEY_ESCAPE):
                 # Back - deselect
                 self.selected_idx = None
                 self.show_inline_options = False
@@ -285,42 +336,42 @@ class TemplateSelector:
             self.show_inline_options = False
 
         # Filter keys (1-4)
-        if key == '1':
-            self._apply_filter('all')
+        if key == "1":
+            self._apply_filter("all")
             return
-        elif key == '2':
-            self._apply_filter('compliance')
+        elif key == "2":
+            self._apply_filter("compliance")
             return
-        elif key == '3':
-            self._apply_filter('pentest')
+        elif key == "3":
+            self._apply_filter("pentest")
             return
-        elif key == '4':
-            self._apply_filter('industry')
+        elif key == "4":
+            self._apply_filter("industry")
             return
 
         # Navigation - Up
-        if key in (KEY_UP, 'k'):
+        if key in (KEY_UP, "k"):
             if self.cursor_pos > 0:
                 self.cursor_pos -= 1
                 if self.cursor_pos < self.page_start:
                     self.page_start = max(0, self.page_start - self.page_size)
 
         # Navigation - Down
-        elif key in (KEY_DOWN, 'j'):
+        elif key in (KEY_DOWN, "j"):
             if self.cursor_pos < len(self.templates) - 1:
                 self.cursor_pos += 1
                 if self.cursor_pos >= self.page_start + self.page_size:
                     self.page_start += self.page_size
 
         # Page Up / Previous Page
-        elif key in (KEY_PAGE_UP, 'p', '[', '<'):
+        elif key in (KEY_PAGE_UP, "p", "[", "<"):
             current_page = self.page_start // self.page_size
             if current_page > 0:
                 self.page_start = (current_page - 1) * self.page_size
                 self.cursor_pos = self.page_start
 
         # Page Down / Next Page
-        elif key in (KEY_PAGE_DOWN, 'n', ']', '>'):
+        elif key in (KEY_PAGE_DOWN, "n", "]", ">"):
             total_pages = (len(self.templates) + self.page_size - 1) // self.page_size
             current_page = self.page_start // self.page_size
             if current_page < total_pages - 1:
@@ -328,15 +379,15 @@ class TemplateSelector:
                 self.cursor_pos = self.page_start
 
         # Preview (v key)
-        elif key == 'v':
+        elif key == "v":
             self._preview_template()
 
         # Import
-        elif key == 'i':
+        elif key == "i":
             self._import_template()
 
         # Space - Select/toggle template and show inline options
-        elif key in (KEY_SPACE, ' '):
+        elif key in (KEY_SPACE, " "):
             if self.selected_idx == self.cursor_pos:
                 # Deselect if same item
                 self.selected_idx = None
@@ -347,12 +398,12 @@ class TemplateSelector:
                 self.show_inline_options = True
 
         # Enter - Quick load (bypass inline options)
-        elif key in (KEY_ENTER, '\r', '\n'):
+        elif key in (KEY_ENTER, "\r", "\n"):
             self.selected_template = self.templates[self.cursor_pos]
             self.running = False
 
         # Cancel - q or Escape
-        elif key in (KEY_ESCAPE, 'q', '\x03'):
+        elif key in (KEY_ESCAPE, "q", "\x03"):
             self.selected_template = None
             self.running = False
 
@@ -368,26 +419,28 @@ class TemplateSelector:
         """Import a template from JSON file."""
         DesignSystem.clear_screen()
         click.echo()
-        click.echo(click.style("  📦 IMPORT TEMPLATE", bold=True, fg='cyan'))
+        click.echo(click.style("  📦 IMPORT TEMPLATE", bold=True, fg="cyan"))
         click.echo()
 
-        file_path = click.prompt("  Enter path to template JSON file", type=str, default='')
+        file_path = click.prompt(
+            "  Enter path to template JSON file", type=str, default=""
+        )
         if not file_path:
             return
 
         try:
             tm = TemplateManager()
-            with open(file_path, 'r') as f:
+            with open(file_path, "r") as f:
                 json_data = f.read()
             template_id = tm.import_template(json_data)
             template = tm.get_template(template_id)
             click.echo()
-            click.echo(click.style(f"  ✅ Imported: {template['name']}", fg='green'))
+            click.echo(click.style(f"  ✅ Imported: {template['name']}", fg="green"))
             click.pause()
             # Refresh templates list
             self.templates = tm.list_templates()
         except Exception as e:
-            click.echo(click.style(f"  ❌ Import failed: {e}", fg='red'))
+            click.echo(click.style(f"  ❌ Import failed: {e}", fg="red"))
             click.pause()
 
 
@@ -405,10 +458,21 @@ def show_template_selector(engagement_id: int) -> bool:
 
     # Sort by category then name
     def sort_key(t):
-        framework = t.get('framework', '') or ''
-        compliance = ['hipaa', 'pci-dss', 'nist', 'owasp', 'soc2', 'iso27001', 'cis', 'cmmc', 'gdpr', 'glba']
-        pentest = ['ptes', 'internal', 'external', 'webapp', 'redteam', 'cloud', 'ad']
-        industry = ['nerc-cip', 'hitrust', 'ffiec']
+        framework = t.get("framework", "") or ""
+        compliance = [
+            "hipaa",
+            "pci-dss",
+            "nist",
+            "owasp",
+            "soc2",
+            "iso27001",
+            "cis",
+            "cmmc",
+            "gdpr",
+            "glba",
+        ]
+        pentest = ["ptes", "internal", "external", "webapp", "redteam", "cloud", "ad"]
+        industry = ["nerc-cip", "hitrust", "ffiec"]
 
         if framework in compliance:
             cat = 0
@@ -418,7 +482,7 @@ def show_template_selector(engagement_id: int) -> bool:
             cat = 2
         else:
             cat = 3
-        return (cat, t['name'])
+        return (cat, t["name"])
 
     all_templates.sort(key=sort_key)
 
@@ -444,23 +508,34 @@ def show_template_selector(engagement_id: int) -> bool:
             continue
 
         try:
-            count = tm.apply_template(selected['id'], engagement_id)
+            count = tm.apply_template(selected["id"], engagement_id)
             click.echo()
-            click.echo(click.style(f"  ✅ Loaded {count} deliverables from template", fg='green'))
-
-            auto_val_count = sum(1 for d in selected['deliverables'] if d.get('auto_validate'))
+            click.echo(
+                click.style(
+                    f"  ✅ Loaded {count} deliverables from template", fg="green"
+                )
+            )
+
+            auto_val_count = sum(
+                1 for d in selected["deliverables"] if d.get("auto_validate")
+            )
             if auto_val_count > 0:
-                click.echo(click.style(f"  ✅ {auto_val_count} deliverables have auto-validation enabled", fg='green'))
+                click.echo(
+                    click.style(
+                        f"  ✅ {auto_val_count} deliverables have auto-validation enabled",
+                        fg="green",
+                    )
+                )
 
             click.echo()
-            click.echo(click.style("  💡 Next steps:", fg='cyan'))
+            click.echo(click.style("  💡 Next steps:", fg="cyan"))
             click.echo("     • Press [V] to validate and see current progress")
             click.echo("     • Start testing and link evidence as you go")
             click.echo("     • Export report when complete")
             click.pause()
             return True
         except Exception as e:
-            click.echo(click.style(f"  ❌ Failed to load template: {e}", fg='red'))
+            click.echo(click.style(f"  ❌ Failed to load template: {e}", fg="red"))
             click.pause()
             return False
 
@@ -471,46 +546,62 @@ def _show_template_preview(template: dict):
     width = DesignSystem.get_terminal_width()
 
     click.echo("\n┌" + "─" * (width - 2) + "┐")
-    click.echo("│" + click.style(" 📋 TEMPLATE PREVIEW ".center(width - 2), bold=True, fg='cyan') + "│")
+    click.echo(
+        "│"
+        + click.style(" 📋 TEMPLATE PREVIEW ".center(width - 2), bold=True, fg="cyan")
+        + "│"
+    )
     click.echo("└" + "─" * (width - 2) + "┘")
     click.echo()
 
     click.echo(f"  Name: {click.style(template['name'], bold=True)}")
-    if template.get('description'):
+    if template.get("description"):
         click.echo(f"  Description: {template['description']}")
-    if template.get('framework'):
+    if template.get("framework"):
         click.echo(f"  Framework: {template['framework'].upper()}")
     click.echo(f"  Total Deliverables: {len(template.get('deliverables', []))}")
 
-    auto_val_count = sum(1 for d in template.get('deliverables', []) if d.get('auto_validate'))
-    click.echo(f"  Auto-validation: {auto_val_count}/{len(template.get('deliverables', []))} deliverables")
+    auto_val_count = sum(
+        1 for d in template.get("deliverables", []) if d.get("auto_validate")
+    )
+    click.echo(
+        f"  Auto-validation: {auto_val_count}/{len(template.get('deliverables', []))} deliverables"
+    )
 
     click.echo()
-    click.echo(click.style("  DELIVERABLES", bold=True, fg='cyan'))
+    click.echo(click.style("  DELIVERABLES", bold=True, fg="cyan"))
     click.echo("  " + "═" * (width - 4))
     click.echo()
 
     # Group by category
     categories = {}
-    for d in template.get('deliverables', []):
-        cat = d.get('category', 'other')
+    for d in template.get("deliverables", []):
+        cat = d.get("category", "other")
         if cat not in categories:
             categories[cat] = []
         categories[cat].append(d)
 
     for category, deliverables in categories.items():
-        click.echo(click.style(f"  {category.upper().replace('_', ' ')}", bold=True, fg='cyan'))
+        click.echo(
+            click.style(f"  {category.upper().replace('_', ' ')}", bold=True, fg="cyan")
+        )
         click.echo("  " + "─" * (width - 4))
 
         for d in deliverables:
             priority_color = {
-                'critical': 'red',
-                'high': 'yellow',
-                'medium': 'white',
-                'low': 'bright_black'
-            }.get(d.get('priority', 'medium'), 'white')
-
-            auto_val_icon = '✓' if d.get('auto_validate') else ' '
-            click.echo(f"  [{auto_val_icon}] " + click.style(f"[{d.get('priority', 'medium').upper()}]", fg=priority_color) + f" {d['title']}")
+                "critical": "red",
+                "high": "yellow",
+                "medium": "white",
+                "low": "bright_black",
+            }.get(d.get("priority", "medium"), "white")
+
+            auto_val_icon = "✓" if d.get("auto_validate") else " "
+            click.echo(
+                f"  [{auto_val_icon}] "
+                + click.style(
+                    f"[{d.get('priority', 'medium').upper()}]", fg=priority_color
+                )
+                + f" {d['title']}"
+            )
 
         click.echo()

souleyez/ui/terminal.py

@@ -31,16 +31,16 @@ def init_readline():
     atexit.register(readline.write_history_file, histfile)
 
     # Enable tab completion (can be customized later)
-    readline.parse_and_bind('tab: complete')
+    readline.parse_and_bind("tab: complete")
 
     # Set up proper keybindings for editing
     # These handle backspace, delete, arrow keys properly
-    readline.parse_and_bind('set editing-mode emacs')
+    readline.parse_and_bind("set editing-mode emacs")
 
     # Ensure proper character handling
-    if hasattr(readline, 'set_completer_delims'):
+    if hasattr(readline, "set_completer_delims"):
         # Set delimiters for word completion
-        readline.set_completer_delims(' \t\n;')
+        readline.set_completer_delims(" \t\n;")
 
     # Try to fix terminal settings
     try:
@@ -55,7 +55,7 @@ def init_readline():
 
         # Set ERASE character to handle backspace
         # VERASE is typically index 2 in the cc array
-        new_settings[6][termios.VERASE] = b'\x7f'  # DEL character
+        new_settings[6][termios.VERASE] = b"\x7f"  # DEL character
 
         # Apply the settings
         termios.tcsetattr(fd, termios.TCSANOW, new_settings)
@@ -77,7 +77,7 @@ def setup_terminal():
     init_readline()
 
     # Ensure stdout is unbuffered for immediate output
-    if hasattr(sys.stdout, 'reconfigure'):
+    if hasattr(sys.stdout, "reconfigure"):
         try:
             sys.stdout.reconfigure(line_buffering=True)
         except (AttributeError, ValueError):