bookmarks.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""Bookmarks system"""

import json
import os
from PyQt6.QtWidgets import (
    QDockWidget, QWidget, QVBoxLayout, QHBoxLayout,
    QListWidget, QLineEdit, QPushButton, QLabel,
    QListWidgetItem, QMessageBox
)
from PyQt6.QtCore import Qt, pyqtSignal


class BookmarksWidget(QWidget):
    """Bookmarks panel"""
    
    bookmark_clicked = pyqtSignal(str)  # URL
    
    def __init__(self, parent=None):
        super().__init__(parent)
        self.bookmarks_file = 'bookmarks.json'
        self.bookmarks = []
        self.setup_ui()
        self.load_bookmarks()
    
    def setup_ui(self):
        """Setup UI"""
        layout = QVBoxLayout()
        layout.setContentsMargins(10, 10, 10, 10)
        layout.setSpacing(8)
        
        # Title with icon
        title_layout = QHBoxLayout()
        title_icon = QLabel("▶")
        title_icon.setStyleSheet("font-size: 14px; color: #0078d4;")
        title_text = QLabel("Bookmarks")
        title_text.setStyleSheet("font-size: 15px; font-weight: 600;")
        title_layout.addWidget(title_icon)
        title_layout.addWidget(title_text)
        title_layout.addStretch()
        layout.addLayout(title_layout)
        
        # Add bookmark section
        add_layout = QVBoxLayout()
        add_layout.setSpacing(6)
        
        self.url_input = QLineEdit()
        self.url_input.setPlaceholderText("URL...")
        add_layout.addWidget(self.url_input)
        
        self.title_input = QLineEdit()
        self.title_input.setPlaceholderText("Title...")
        add_layout.addWidget(self.title_input)
        
        btn_layout = QHBoxLayout()
        add_btn = QPushButton("+ Add")
        add_btn.setToolTip("Add bookmark")
        add_btn.clicked.connect(self.add_bookmark)
        btn_layout.addWidget(add_btn)
        
        del_btn = QPushButton("− Remove")
        del_btn.setToolTip("Remove selected bookmark")
        del_btn.clicked.connect(self.remove_bookmark)
        btn_layout.addWidget(del_btn)
        
        clear_btn = QPushButton("✕ Clear All")
        clear_btn.setToolTip("Remove all bookmarks")
        clear_btn.clicked.connect(self.clear_all_bookmarks)
        btn_layout.addWidget(clear_btn)
        
        add_layout.addLayout(btn_layout)
        layout.addLayout(add_layout)
        
        # Bookmarks list
        self.list_widget = QListWidget()
        self.list_widget.itemDoubleClicked.connect(self.on_bookmark_double_click)
        layout.addWidget(self.list_widget)
        
        self.setLayout(layout)
    
    def add_bookmark(self, url=None, title=None):
        """Add bookmark"""
        if url is None:
            url = self.url_input.text().strip()
        if title is None:
            title = self.title_input.text().strip()
        
        if not url:
            return
        
        if not title:
            title = url
        
        # Check if exists
        for bm in self.bookmarks:
            if bm['url'] == url:
                return
        
        bookmark = {'url': url, 'title': title}
        self.bookmarks.append(bookmark)
        self.save_bookmarks()
        self.refresh_list()
        
        self.url_input.clear()
        self.title_input.clear()
    
    def remove_bookmark(self):
        """Remove selected bookmark"""
        current = self.list_widget.currentRow()
        if current >= 0:
            del self.bookmarks[current]
            self.save_bookmarks()
            self.refresh_list()
    
    def on_bookmark_double_click(self, item):
        """Handle double click"""
        index = self.list_widget.row(item)
        if 0 <= index < len(self.bookmarks):
            url = self.bookmarks[index]['url']
            self.bookmark_clicked.emit(url)
    
    def refresh_list(self):
        """Refresh bookmarks list"""
        self.list_widget.clear()
        for bm in self.bookmarks:
            item = QListWidgetItem(f"▸ {bm['title']}")
            item.setToolTip(bm['url'])
            self.list_widget.addItem(item)
    
    def clear_all_bookmarks(self):
        """Clear all bookmarks"""
        from PyQt6.QtWidgets import QMessageBox
        reply = QMessageBox.question(
            self, 
            'Clear All Bookmarks',
            'Are you sure you want to remove all bookmarks?',
            QMessageBox.StandardButton.Yes | QMessageBox.StandardButton.No,
            QMessageBox.StandardButton.No
        )
        
        if reply == QMessageBox.StandardButton.Yes:
            self.bookmarks.clear()
            self.save_bookmarks()
            self.refresh_list()
    
    def load_bookmarks(self):
        """Load bookmarks from file"""
        if os.path.exists(self.bookmarks_file):
            try:
                with open(self.bookmarks_file, 'r', encoding='utf-8') as f:
                    self.bookmarks = json.load(f)
                self.refresh_list()
            except:
                self.bookmarks = []
    
    def save_bookmarks(self):
        """Save bookmarks to file"""
        with open(self.bookmarks_file, 'w', encoding='utf-8') as f:
            json.dump(self.bookmarks, f, ensure_ascii=False, indent=2)
    
    def get_current_url_title(self):
        """Get URL and title from parent browser"""
        browser = self.parent()
        while browser and not hasattr(browser, 'tabs'):
            browser = browser.parent()
        
        if browser and hasattr(browser, 'tabs'):
            tab = browser.tabs.currentWidget()
            if tab:
                return tab.get_url(), tab.get_title()
        return "", ""


class BookmarksDock(QDockWidget):
    """Bookmarks dock widget"""
    
    def __init__(self, parent=None):
        super().__init__("Bookmarks", parent)
        self.setAllowedAreas(Qt.DockWidgetArea.LeftDockWidgetArea | 
                            Qt.DockWidgetArea.RightDockWidgetArea)
        
        self.bookmarks_widget = BookmarksWidget(self)
        self.setWidget(self.bookmarks_widget)
        
        # Hide by default
        self.hide()