pentool 1.0.0__py3-none-any.whl

pentool/tui/widgets/menu_bar.py

@@ -0,0 +1,238 @@
+"""MenuBar — горизонтальная полоса с каскадными выпадающими меню."""
+
+from __future__ import annotations
+
+
+
+from textual import on
+from textual.app import ComposeResult
+from textual.message import Message
+from textual.widget import Widget
+from textual.widgets import Static
+from pathlib import Path
+
+_CSS = (Path(__file__).parent / "menu_bar.tcss").read_text(encoding="utf-8")
+
+
+class MenuItem(Static):
+    """Пункт выпадающего меню."""
+
+    DEFAULT_CSS = _CSS
+
+    class Selected(Message):
+        """Пользователь выбрал пункт меню."""
+        def __init__(self, action: str) -> None:
+            super().__init__()
+            self.action = action
+
+    def __init__(
+        self,
+        label: str,
+        action: str = "",
+        shortcut: str = "",
+        disabled: bool = False,
+        **kwargs,
+    ) -> None:
+        self._action = action
+        self._shortcut = shortcut
+        self._is_separator = label == "---"
+        self._is_disabled = disabled or self._is_separator
+
+        if self._is_separator:
+            display_text = "─" * 20
+        elif shortcut:
+            display_text = f"{label:<22}{shortcut}"
+        else:
+            display_text = label
+
+        super().__init__(display_text, **kwargs)
+
+        if self._is_separator:
+            self.add_class("-separator")
+        if self._is_disabled:
+            self.add_class("-disabled")
+
+    def on_click(self) -> None:
+        if self._is_disabled or not self._action:
+            return
+        self.post_message(self.Selected(self._action))
+
+
+class MenuDropdown(Widget):
+    """Выпадающий список пунктов меню."""
+
+    DEFAULT_CSS = _CSS
+
+    def __init__(self, items: list[tuple], **kwargs) -> None:
+        super().__init__(**kwargs)
+        self._items = items
+
+    def compose(self) -> ComposeResult:
+        for item in self._items:
+            if item[0] == "---":
+                yield MenuItem("---")
+            elif len(item) == 2:
+                yield MenuItem(item[0], action=item[1])
+            elif len(item) == 3:
+                yield MenuItem(item[0], action=item[1], shortcut=item[2])
+
+    def show(self, x: int, y: int) -> None:
+        self.styles.offset = (x, y)
+        self.display = True
+
+    def hide(self) -> None:
+        self.display = False
+
+    @on(MenuItem.Selected)
+    def on_menu_item_selected(self, event: MenuItem.Selected) -> None:
+        event.stop()
+        self.hide()
+        # Закрыть остальные дропдауны через MenuBar
+        try:
+            self.app.query_one("#menu-bar", MenuBar)._close_all()
+        except Exception:
+            pass
+        action = event.action
+        if action.startswith("app.action_"):
+            # run_action принимает имя без префикса "action_"
+            action_name = action[len("app.action_"):]
+            self.app.run_worker(self.app.run_action(action_name), exclusive=False)
+
+
+class MenuHeader(Static):
+    """Заголовок одного меню (кликабельный)."""
+
+    DEFAULT_CSS = _CSS
+
+    class Activated(Message):
+        """MenuHeader был нажат."""
+        def __init__(self, header: MenuHeader) -> None:
+            super().__init__()
+            self.header = header
+
+    def __init__(self, label: str, menu_id: str, **kwargs) -> None:
+        super().__init__(label, **kwargs)
+        self._menu_id = menu_id
+
+    @property
+    def menu_id(self) -> str:
+        return self._menu_id
+
+    def on_click(self) -> None:
+        self.post_message(self.Activated(self))
+
+
+class MenuBar(Static):
+    """Горизонтальная полоса с выпадающими меню — аналог десктопных приложений."""
+
+    DEFAULT_CSS = _CSS
+
+    # Структура: [(label, menu_id, [(item_label, action, shortcut?), ...])]
+    MENUS: list[tuple[str, str, list]] = [
+        ("PenTool", "burp", [
+            ("About PenTool", "app.action_about"),
+            ("Check for updates...", "app.action_check_updates"),
+        ]),
+        ("Project", "project", [
+            ("New Project",  "app.action_new_project",  "Ctrl+N"),
+            ("Open...",      "app.action_open_project",  "Ctrl+O"),
+            ("Save",         "app.action_save_project",  "Ctrl+S"),
+            ("Recent",       "app.action_recent_projects"),
+            ("---", ""),
+            ("Exit",         "app.action_quit",          "Ctrl+Q"),
+        ]),
+        ("Intruder", "intruder", [
+            ("New Attack",   "app.action_intruder_new"),
+            ("Start Attack", "app.action_intruder_start"),
+            ("Stop Attack",  "app.action_intruder_stop"),
+        ]),
+        ("Repeater", "repeater", [
+            ("Send to Repeater", "app.action_send_to_repeater"),
+            ("New Tab",          "app.action_repeater_new_tab"),
+        ]),
+        ("View", "view", [
+            ("Filter settings",        "app.action_filter_settings"),
+            ("Toggle Dark/Light theme","app.action_toggle_theme"),
+            ("Word wrap",              "app.action_toggle_word_wrap"),
+        ]),
+        ("Help", "help", [
+            ("Documentation",      "app.action_documentation"),
+            ("Keyboard shortcuts", "app.action_keyboard_shortcuts"),
+            ("---", ""),
+            ("CA Certificate",     "app.action_open_ca_cert"),
+            ("---", ""),
+            ("About",              "app.action_about"),
+        ]),
+    ]
+
+    def __init__(self, **kwargs) -> None:
+        super().__init__(**kwargs)
+        self._active_menu: str | None = None
+        self._dropdowns: dict[str, MenuDropdown] = {}
+
+    def compose(self) -> ComposeResult:
+        for label, menu_id, items in self.MENUS:
+            yield MenuHeader(label, menu_id, id=f"menu-header-{menu_id}")
+
+        # Выпадающие списки монтируются вне MenuBar (через app.mount)
+        # чтобы они поверх всего контента
+
+    def on_mount(self) -> None:
+        # Создаём дропдауны и монтируем в app
+        for label, menu_id, items in self.MENUS:
+            dropdown = MenuDropdown(items, id=f"menu-dropdown-{menu_id}")
+            self._dropdowns[menu_id] = dropdown
+            self.app.mount(dropdown)
+
+    def on_unmount(self) -> None:
+        for dropdown in self._dropdowns.values():
+            try:
+                dropdown.remove()
+            except Exception:
+                pass
+
+    @on(MenuHeader.Activated)
+    def on_menu_header_activated(self, event: MenuHeader.Activated) -> None:
+        menu_id = event.header.menu_id
+        if self._active_menu == menu_id:
+            self._close_all()
+            return
+
+        self._close_all()
+        self._open_menu(menu_id, event.header)
+
+    def _open_menu(self, menu_id: str, header: MenuHeader) -> None:
+        dropdown = self._dropdowns.get(menu_id)
+        if dropdown is None:
+            return
+
+        # Вычисляем позицию под заголовком
+        try:
+            region = header.content_region
+            x = region.x
+            y = region.y + 1
+        except Exception:
+            x, y = 0, 1
+
+        dropdown.show(x, y)
+        header.add_class("-active")
+        self._active_menu = menu_id
+
+    def _close_all(self) -> None:
+        for dropdown in self._dropdowns.values():
+            dropdown.hide()
+        for label, menu_id, _ in self.MENUS:
+            try:
+                header = self.query_one(f"#menu-header-{menu_id}", MenuHeader)
+                header.remove_class("-active")
+            except Exception:
+                pass
+        self._active_menu = None
+
+    def on_key(self, event) -> None:
+        if event.key == "escape":
+            self._close_all()
+
+    def on_click(self, event) -> None:
+        # Клик вне MenuBar — закрыть (обрабатывается через app)
+        pass

pentool/tui/widgets/module_tabs.py

@@ -0,0 +1,73 @@
+"""Горизонтальная панель вкладок для навигации по модулям (Burp-стиль)."""
+
+from __future__ import annotations
+
+from textual.app import ComposeResult
+from textual.reactive import reactive
+from textual.widget import Widget
+from textual.widgets import Tab, Tabs
+
+from pentool.tui.widgets.menu import BASIC_MODULES, MODULES, ModuleSelected
+from pathlib import Path
+
+_CSS = (Path(__file__).parent / "module_tabs.tcss").read_text(encoding="utf-8")
+
+
+class ModuleTabs(Widget):
+    """Горизонтальные вкладки навигации по модулям.
+
+    Постит тот же ModuleSelected message что и SideMenu —
+    app.on_module_selected не требует изменений.
+    """
+
+    DEFAULT_CSS = _CSS
+
+    active_module: reactive[str] = reactive("proxy")
+
+    def compose(self) -> ComposeResult:
+        tabs = [Tab(label, id=f"tab-{mod_id}") for mod_id, label, _ in MODULES]
+        yield Tabs(*tabs, id="module-tabs-inner")
+
+    def on_tabs_tab_activated(self, event: Tabs.TabActivated) -> None:
+        if event.tab is None:
+            return
+        tab_id = event.tab.id or ""
+        if tab_id.startswith("tab-"):
+            module_id = tab_id[4:]
+            self.active_module = module_id
+            self.post_message(ModuleSelected(module_id))
+
+    def select_module(self, module_id: str) -> None:
+        """Программно переключить вкладку (без постинга события).
+
+        Args:
+            module_id: ID модуля из MODULES.
+        """
+        self.active_module = module_id
+        try:
+            tabs = self.query_one("#module-tabs-inner", Tabs)
+            tabs.active = f"tab-{module_id}"
+        except Exception:
+            pass
+
+    def set_mode(self, mode: str) -> None:
+        """Переключить режим: 'basic' показывает только базовые вкладки.
+
+        Args:
+            mode: 'basic' | 'advanced'
+        """
+        try:
+            tabs = self.query_one("#module-tabs-inner", Tabs)
+            for mod_id, _, _ in MODULES:
+                tab_id = f"tab-{mod_id}"
+                try:
+                    tab = tabs.query_one(f"#{tab_id}", Tab)
+                    if mode == "basic" and mod_id not in BASIC_MODULES:
+                        tab.styles.display = "none"
+                    else:
+                        tab.styles.display = "block"
+                except Exception:
+                    pass
+            tabs.refresh()
+        except Exception:
+            pass

pentool/tui/widgets/payload_drop_zone.py

@@ -0,0 +1,66 @@
+"""PayloadDropZone — зона для перетаскивания файлов с payload'ами."""
+
+from __future__ import annotations
+
+from textual.message import Message
+from textual.widget import Widget
+from pathlib import Path
+
+_CSS = (Path(__file__).parent / "payload_drop_zone.tcss").read_text(encoding="utf-8")
+
+
+class PayloadDropZone(Widget):
+    """Визуальная зона drag & drop для payload-файлов.
+
+    При отсутствии нативного DragDrop (textual-filedrop) — показывает
+    подсказку. Интегрируется с кнопкой "Load from file".
+
+    Сообщения:
+        PayloadDropZone.PayloadsLoaded — payload'ы загружены из файла.
+    """
+
+    DEFAULT_CSS = _CSS
+
+    class PayloadsLoaded(Message):
+        """Payload'ы успешно загружены из файла."""
+        def __init__(self, payloads: list[str], source_path: str = "") -> None:
+            super().__init__()
+            self.payloads = payloads
+            self.source_path = source_path
+
+    def __init__(self, **kwargs) -> None:
+        super().__init__(**kwargs)
+        self.update_text()
+
+    def update_text(self, count: int = 0) -> None:
+        pass  # count аргумент сохранён для обратной совместимости API
+
+    def render(self) -> str:
+        return "  Drop .txt / .yaml payload file here\n  (or use 'Load from file' button)"
+
+    def load_from_path(self, path: str) -> None:
+        """Загрузить payload'ы из файла и отправить сообщение."""
+        from pentool.api.intruder_api import load_payloads_from_file
+        payloads = load_payloads_from_file(path)
+        if payloads:
+            self.post_message(self.PayloadsLoaded(payloads, source_path=path))
+
+    def on_click(self) -> None:
+        """Клик по зоне — открыть FileSelectorDialog как альтернативу DragDrop."""
+        from pentool.tui.dialogs.file_selector import FileSelectorDialog, FileSelectorMode
+
+        def _on_selected(path: str | None) -> None:
+            if path:
+                self.load_from_path(path)
+
+        try:
+            self.app.push_screen(
+                FileSelectorDialog(
+                    mode=FileSelectorMode.OPEN,
+                    filter_ext=["*.txt", "*.yaml", "*.yml"],
+                    title="Select payload file",
+                ),
+                _on_selected,
+            )
+        except Exception:
+            pass