---

import sys
import keyboard
from PyQt6.QtCore import Qt, QPoint
from PyQt6.QtGui import QPainter, QPen, QColor
from PyQt6.QtWidgets import QApplication, QMainWindow, QWidget, QVBoxLayout, QHBoxLayout, QLabel, QSpinBox, QLineEdit

class DrawingOverlay(QWidget):
    def __init__(self):
        super().__init__()
        # Делаем окно полностью прозрачным и кликабельным насквозь
        self.setWindowFlags(
            Qt.WindowType.FramelessWindowHint | 
            Qt.WindowType.WindowStaysOnTopHint | 
            Qt.WindowType.SubWindow
        )
        self.setAttribute(Qt.WidgetAttribute.WA_TranslucentBackground, True)
        self.setAttribute(Qt.WidgetAttribute.WA_TransparentForMouseEvents, True)
        
        self.start_point = None
        self.end_point = None
        self.line_thickness = 5
        self.is_active = False

    def resize_to_screen(self):
        # Растягиваем окно на весь экран
        screen = QApplication.primaryScreen().geometry()
        self.setGeometry(screen)

    def paintEvent(self, event):
        if self.is_active and self.start_point and self.end_point:
            painter = QPainter(self)
            painter.setRenderHint(QPainter.RenderHint.Antialiasing)
            
            # Настройка кисти (Цвет: Красный, Толщина из настроек)
            pen = QPen(QColor(255, 0, 0, 255), self.line_thickness)
            pen.setCapStyle(Qt.PenCapStyle.RoundCap)
            painter.setPen(pen)
            
            # Рисуем линию
            painter.drawLine(self.start_point, self.end_point)

class SettingsWindow(QMainWindow):
    def __init__(self):
        super().__init__()
        self.setWindowTitle("Настройки Линии")
        self.setFixedSize(300, 150)
        self.setWindowFlags(Qt.WindowType.WindowStaysOnTopHint) # Настройки тоже поверх всего

        # Создаем оверлей для рисования
        self.overlay = DrawingOverlay()
        self.overlay.resize_to_screen()
        self.overlay.show()

        self.init_ui()
        
        # Дефолтная клавиша - F4
        self.current_hotkey = "F4"
        keyboard.add_hotkey(self.current_hotkey, self.toggle_drawing_mode)

        # Переменные для логики кликов
        self.click_count = 0

    def init_ui(self):
        central_widget = QWidget()
        self.setCentralWidget(central_widget)
        layout = QVBoxLayout(central_widget)

        # Настройка толщины
        thick_layout = QHBoxLayout()
        thick_label = QLabel("Толщина линии (px):")
        self.thick_spin = QSpinBox()
        self.thick_spin.setRange(1, 50)
        self.thick_spin.setValue(5)
        self.thick_spin.valueChanged.connect(self.change_thickness)
        thick_layout.addWidget(thick_label)
        thick_layout.addWidget(self.thick_spin)
        layout.addLayout(thick_layout)

        # Настройка горячей клавиши
        hk_layout = QHBoxLayout()
        hk_label = QLabel("Горячая клавиша:")
        self.hk_input = QLineEdit(r"F4")
        self.hk_input.editingFinished.connect(self.update_hotkey)
        hk_layout.addWidget(hk_label)
        hk_layout.addWidget(self.hk_input)
        layout.addLayout(hk_layout)
        
        # Инструкция
        self.status_label = QLabel("Статус: Ожидание. Нажмите F4")
        self.status_label.setStyleSheet("color: gray;")
        layout.addWidget(self.status_label)

    def change_thickness(self, value):
        self.overlay.line_thickness = value
        self.overlay.update()

    def update_hotkey(self):
        new_key = self.hk_input.text().strip()
        if new_key and new_key != self.current_hotkey:
            try:
                keyboard.remove_hotkey(self.current_hotkey)
                self.current_hotkey = new_key
                keyboard.add_hotkey(self.current_hotkey, self.toggle_drawing_mode)
                self.status_label.setText(f"Клавиша изменена на {new_key}")
            except Exception:
                self.status_label.setText("Ошибка: Неверная клавиша!")

    def toggle_drawing_mode(self):
        # Переключение режима активации по горячей клавише
        if not self.overlay.is_active:
            self.overlay.is_active = True
            self.click_count = 0
            self.overlay.start_point = None
            self.overlay.end_point = None
            # Делаем окно оверлея кликабельным, чтобы поймать координаты
            self.overlay.setAttribute(Qt.WidgetAttribute.WA_TransparentForMouseEvents, False)
            # Меняем поведение мыши оверлея, чтобы ловить клики вручную
            self.overlay.mousePressEvent = self.handle_overlay_click
            self.status_label.setText("Режим рисования! Кликните точку 1")
            self.status_label.setStyleSheet("color: green;")
        else:
            self.reset_overlay()

    def handle_overlay_click(self, event):
        if event.button() == Qt.MouseButton.LeftButton:
            if self.click_count == 0:
                self.overlay.start_point = event.position().toPoint()
                self.click_count += 1
                self.status_label.setText("Точка 1 задана. Кликните точку 2")
            elif self.click_count == 1:
                self.overlay.end_point = event.position().toPoint()
                self.overlay.update() # Перерисовываем экран, чтобы линия появилась
                
                # Возвращаем окну сквозную прозрачность, чтобы оно не мешало кликать дальше
                self.overlay.setAttribute(Qt.WidgetAttribute.WA_TransparentForMouseEvents, True)
                self.status_label.setText("Линия построена! Нажмите F4 для сброса")
                self.status_label.setStyleSheet("color: blue;")
                self.click_count += 1

    def reset_overlay(self):
        self.overlay.is_active = False
        self.overlay.start_point = None
        self.overlay.end_point = None
        self.overlay.setAttribute(Qt.WidgetAttribute.WA_TransparentForMouseEvents, True)
        self.overlay.update()
        self.status_label.setText(f"Сброшено. Нажмите {self.current_hotkey}")
        self.status_label.setStyleSheet("color: gray;")

    def closeEvent(self, event):
        self.overlay.close()
        keyboard.unhook_all()
        event.accept()

if __name__ == "__main__":
    app = QApplication(sys.argv)
    window = SettingsWindow()
    window.show()
    sys.exit(app.exec())