#pragma once

#include <warppipe/warppipe.hpp>

#include <QString>
#include <QWidget>

namespace QtNodes {
class BasicGraphicsScene;
class GraphicsView;
}  // namespace QtNodes

class WarpGraphModel;
class QLabel;
class QTimer;

class GraphEditorWidget : public QWidget {
  Q_OBJECT

public:
  explicit GraphEditorWidget(warppipe::Client *client,
                             QWidget *parent = nullptr);

  int nodeCount() const;
  int linkCount() const;

  void setDebugScreenshotDir(const QString &dir);

Q_SIGNALS:
  void graphReady();

private slots:
  void onRefreshTimer();
  void onContextMenuRequested(const QPoint &pos);
  void scheduleSaveLayout();

private:
  void showCanvasContextMenu(const QPoint &screenPos, const QPointF &scenePos);
  void showNodeContextMenu(const QPoint &screenPos, uint32_t pwNodeId);
  void createVirtualNode(bool isSink, const QPointF &scenePos);
  void captureDebugScreenshot(const QString &event);

  warppipe::Client *m_client = nullptr;
  WarpGraphModel *m_model = nullptr;
  QtNodes::BasicGraphicsScene *m_scene = nullptr;
  QtNodes::GraphicsView *m_view = nullptr;
  QTimer *m_refreshTimer = nullptr;
  QTimer *m_saveTimer = nullptr;
  QString m_layoutPath;
  QString m_debugScreenshotDir;
  bool m_graphReady = false;
};