Library drag-in

src/bedit_gui/views/graph_editor_widget.py

@@ -1,15 +1,17 @@
 from __future__ import annotations
 
+import json
 from enum import Enum
 from itertools import pairwise
 from math import hypot
 
 from PySide6.QtCore import QEvent, QObject, QPointF, QRectF, QSize, QTimer, Qt, Signal
-from PySide6.QtGui import QActionGroup, QBrush, QColor, QCursor, QKeySequence, QMouseEvent, QPainter, QPainterPath, QPen, QShortcut, QWheelEvent
+from PySide6.QtGui import QActionGroup, QBrush, QColor, QCursor, QDragEnterEvent, QDragMoveEvent, QDropEvent, QKeySequence, QMouseEvent, QPainter, QPainterPath, QPen, QShortcut, QWheelEvent
 from PySide6.QtWidgets import QGraphicsEllipseItem, QGraphicsItem, QGraphicsPathItem, QGraphicsPixmapItem, QGraphicsScene, QGraphicsSceneMouseEvent, QGraphicsTextItem, QGraphicsView, QMenu, QWidget
 
 from bedit_core.models import BondCausality, BondConnection, BondPort, Component, ComponentID, Connection, ConnectionID, GraphImplementation, Port, PortID, SignalConnection, SignalDirection, SignalPort
 from bedit_gui.models import Graph, GraphComponentLabel, GraphConnection, Icon
+from bedit_gui.services.component_clipboard import COMPONENTS_MIME
 from bedit_gui.ui.generated.ui_graph_editor_widget import Ui_graphEditorWidget
 from bedit_gui.utils.icon import get_pixmap_bounding_box, render_icon
 
@@ -291,6 +293,7 @@ class GraphEditorWidget(QWidget):
     connection_points_change_requested = Signal(object, object, object, str)
     connection_add_requested = Signal(object, object)
     connections_delete_requested = Signal(object, object)
+    component_drop_requested = Signal(object, object, object)
 
     def __init__(self, parent: QWidget | None = None, snap_to_grid_size: int = 4) -> None:
         super().__init__(parent)
@@ -314,6 +317,7 @@ class GraphEditorWidget(QWidget):
         self.ui.graphicsView.setScene(self.scene)
         self.ui.graphicsView.setTransformationAnchor(QGraphicsView.ViewportAnchor.AnchorUnderMouse)
         self.ui.graphicsView.viewport().installEventFilter(self)
+        self.ui.graphicsView.viewport().setAcceptDrops(True)
         self._mode_actions = QActionGroup(self)
         self._mode_actions.setExclusive(True)
         self._mode_actions.addAction(self.ui.actionMouseMode)
@@ -736,6 +740,25 @@ class GraphEditorWidget(QWidget):
         self.ui.graphicsView.centerOn(bounds.center())
 
     def eventFilter(self, watched: QObject, event: QEvent) -> bool:
+        if watched is self.ui.graphicsView.viewport() and event.type() in (QEvent.Type.DragEnter, QEvent.Type.DragMove):
+            assert isinstance(event, (QDragEnterEvent, QDragMoveEvent))
+            if self._component is not None and event.mimeData().hasFormat(COMPONENTS_MIME):
+                event.setDropAction(Qt.DropAction.CopyAction)
+                event.accept()
+                return True
+        if watched is self.ui.graphicsView.viewport() and event.type() == QEvent.Type.Drop:
+            assert isinstance(event, QDropEvent)
+            if self._component is not None and event.mimeData().hasFormat(COMPONENTS_MIME):
+                try:
+                    payload = json.loads(bytes(event.mimeData().data(COMPONENTS_MIME)).decode("utf-8"))
+                except (UnicodeDecodeError, json.JSONDecodeError):
+                    return True
+                if isinstance(payload, dict):
+                    position = self._snap_position(self.ui.graphicsView.mapToScene(event.position().toPoint()))
+                    self.component_drop_requested.emit(self._component, payload, position)
+                    event.setDropAction(Qt.DropAction.CopyAction)
+                    event.accept()
+                return True
         if watched is self.ui.graphicsView.viewport() and event.type() == QEvent.Type.MouseMove:
             assert isinstance(event, QMouseEvent)
             self._update_connection_preview(self.ui.graphicsView.mapToScene(event.position().toPoint()))