custom_widgets.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Кастомные виджеты для браузера
"""

from PyQt6.QtWidgets import (
    QLineEdit, QPushButton, QWidget, QVBoxLayout, 
    QHBoxLayout, QLabel, QListWidget, QDialog
)
from PyQt6.QtCore import pyqtSignal, Qt
from PyQt6.QtGui import QKeyEvent
import config


class SmartUrlBar(QLineEdit):
    """Умная адресная строка с автодополнением"""
    
    navigate_requested = pyqtSignal(str)
    
    def __init__(self, parent=None):
        super().__init__(parent)
        self.setPlaceholderText("Введите URL или поисковый запрос...")
        self.history = []
        self.suggestions = []
    
    def keyPressEvent(self, event: QKeyEvent):
        """Обработка нажатия клавиш"""
        if event.key() == Qt.Key.Key_Return or event.key() == Qt.Key.Key_Enter:
            self.navigate_requested.emit(self.text())
            self.add_to_history(self.text())
        else:
            super().keyPressEvent(event)
    
    def add_to_history(self, text: str):
        """Добавить в историю адресной строки"""
        if text and text not in self.history:
            self.history.insert(0, text)
            # Ограничиваем размер истории
            if len(self.history) > 100:
                self.history.pop()
    
    def get_suggestions(self, text: str) -> list:
        """Получить подсказки на основе истории"""
        if not text:
            return []
        
        text_lower = text.lower()
        suggestions = [
            item for item in self.history
            if text_lower in item.lower()
        ]
        return suggestions[:10]  # Максимум 10 подсказок


class MinimalContextMenu(QWidget):
    """Минималистичное контекстное меню"""
    
    def __init__(self, parent=None):
        super().__init__(parent)
        self.setWindowFlags(Qt.WindowType.Popup | Qt.WindowType.FramelessWindowHint)
        self.setup_ui()
    
    def setup_ui(self):
        """Настройка UI"""
        layout = QVBoxLayout()
        layout.setContentsMargins(5, 5, 5, 5)
        layout.setSpacing(2)
        
        self.setLayout(layout)
        self.setStyleSheet("""
            QWidget {
                background-color: #2b2b2b;
                color: #ffffff;
                border: 1px solid #555;
            }
            QPushButton {
                border: none;
                padding: 8px;
                text-align: left;
            }
            QPushButton:hover {
                background-color: #3b3b3b;
            }
        """)
    
    def add_action(self, text: str, callback):
        """Добавить действие в меню"""
        btn = QPushButton(text)
        btn.clicked.connect(callback)
        btn.clicked.connect(self.hide)
        self.layout().addWidget(btn)


class HistoryDialog(QDialog):
    """Диалог истории посещений"""
    
    url_selected = pyqtSignal(str)
    
    def __init__(self, history_manager, parent=None):
        super().__init__(parent)
        self.history_manager = history_manager
        self.setWindowTitle("История посещений")
        self.setModal(True)
        self.resize(600, 400)
        self.setup_ui()
        self.load_history()
    
    def setup_ui(self):
        """Настройка UI"""
        layout = QVBoxLayout()
        
        # Поиск
        search_layout = QHBoxLayout()
        self.search_input = QLineEdit()
        self.search_input.setPlaceholderText("Поиск в истории...")
        self.search_input.textChanged.connect(self.on_search)
        search_layout.addWidget(self.search_input)
        
        clear_btn = QPushButton("Очистить всё")
        clear_btn.clicked.connect(self.clear_all_history)
        search_layout.addWidget(clear_btn)
        
        layout.addLayout(search_layout)
        
        # Список истории
        self.history_list = QListWidget()
        self.history_list.itemDoubleClicked.connect(self.on_item_selected)
        layout.addWidget(self.history_list)
        
        # Статистика
        self.stats_label = QLabel()
        layout.addWidget(self.stats_label)
        
        # Кнопки
        buttons_layout = QHBoxLayout()
        close_btn = QPushButton("Закрыть")
        close_btn.clicked.connect(self.close)
        buttons_layout.addStretch()
        buttons_layout.addWidget(close_btn)
        
        layout.addLayout(buttons_layout)
        
        self.setLayout(layout)
    
    def load_history(self):
        """Загрузить историю"""
        self.history_list.clear()
        entries = self.history_manager.get_recent(100)
        
        for entry in entries:
            title = entry.title or "Без названия"
            item_text = f"{title}\n{entry.url}"
            self.history_list.addItem(item_text)
        
        # Обновить статистику
        stats = self.history_manager.get_stats()
        self.stats_label.setText(
            f"Всего записей: {stats['total_entries']} | "
            f"Хранится до: {config.HISTORY_MAX_HOURS} часов"
        )
    
    def on_search(self, query: str):
        """Поиск в истории"""
        self.history_list.clear()
        
        if not query:
            self.load_history()
            return
        
        results = self.history_manager.search(query)
        
        for entry in results[:100]:
            title = entry.title or "Без названия"
            item_text = f"{title}\n{entry.url}"
            self.history_list.addItem(item_text)
    
    def on_item_selected(self, item):
        """Обработка выбора элемента"""
        # Извлекаем URL из текста элемента
        text = item.text()
        lines = text.split('\n')
        if len(lines) >= 2:
            url = lines[1]
            self.url_selected.emit(url)
            self.close()
    
    def clear_all_history(self):
        """Очистить всю историю"""
        self.history_manager.clear_all()
        self.load_history()
        self.stats_label.setText("✓ История очищена")


class StatusBarWidget(QWidget):
    """Виджет статус-бара с дополнительной информацией"""
    
    def __init__(self, parent=None):
        super().__init__(parent)
        self.setup_ui()
    
    def setup_ui(self):
        """Настройка UI"""
        layout = QHBoxLayout()
        layout.setContentsMargins(5, 2, 5, 2)
        
        self.status_label = QLabel("Готов")
        self.security_label = QLabel("⚠️ Без защиты")
        self.tabs_label = QLabel("Вкладок: 0")
        
        layout.addWidget(self.status_label)
        layout.addStretch()
        layout.addWidget(self.tabs_label)
        layout.addWidget(self.security_label)
        
        self.setLayout(layout)
    
    def set_status(self, text: str):
        """Установить текст статуса"""
        self.status_label.setText(text)
    
    def set_tabs_count(self, count: int):
        """Установить количество вкладок"""
        self.tabs_label.setText(f"Вкладок: {count}")
    
    def set_security_status(self, secure: bool):
        """Установить статус безопасности"""
        if secure:
            self.security_label.setText("🔒 HTTPS")
        else:
            self.security_label.setText("⚠️ HTTP")