/*
 * Oak Video Editor - Non-Linear Video Editor
 * Copyright (C) 2025 Olive CE Team
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

#ifndef OAK_NODEVIEWCONTEXT_H
#define OAK_NODEVIEWCONTEXT_H

#include <QGraphicsRectItem>
#include <QGraphicsTextItem>
#include <QHash>

#include "engineeventbridge.h"
#include "oakutil/oaknode.h"
#include "nodeviewcommon.h"
#include "nodeviewedge.h"

namespace olive
{

class NodeViewContext : public QObject, public QGraphicsRectItem {
	Q_OBJECT
public:
	NodeViewContext(oak::Node context, QGraphicsItem *item = nullptr);

	virtual ~NodeViewContext() override;

	oak::Node get_context() const
	{
		return context_;
	}

	void update_rect();

	void set_flow_direction(NodeViewCommon::FlowDirection dir);

	void set_curved_edges(bool e);

	void get_selected_for_deletion(QVector<OakEngineNode *> &nodes,
								  QVector<OakEngineNode *> &contexts,
								  QVector<NodeViewEdge *> &edges) const;

	void select(const QVector<OakEngineNode *> &nodes);

	QVector<NodeViewItem *> get_selected_items() const;

	QPointF map_scene_pos_to_node_pos_in_context(const QPointF &pos) const;

	NodeViewItem *get_item_from_map(OakEngineNode *node) const
	{
		return item_map_.value(node);
	}

	virtual void paint(QPainter *painter,
					   const QStyleOptionGraphicsItem *option,
					   QWidget *widget = nullptr) override;

public:
	// Not slots: they are invoked from EngineEventBridge lambdas / directly,
	// never as connect() targets.
	void add_child(OakEngineNode *node);

	void set_child_position(OakEngineNode *node, const QPointF &pos);

	void remove_child(OakEngineNode *node);

	void child_input_connected(OakEngineNode *output, const oak::Input &input);

	bool child_input_disconnected(OakEngineNode *output, const oak::Input &input);

signals:
	void item_about_to_be_deleted(NodeViewItem *item);

protected:
	virtual QVariant itemChange(QGraphicsItem::GraphicsItemChange change,
								const QVariant &value) override;

	virtual void mousePressEvent(QGraphicsSceneMouseEvent *event) override;

private:
	void add_node_internal(OakEngineNode *node, NodeViewItem *item);

	void add_edge_internal(OakEngineNode *output, const oak::Input &input,
						 NodeViewItem *from, NodeViewItem *to);

	oak::Node context_;

	QString lbl_;

	NodeViewCommon::FlowDirection flow_dir_;

	bool curved_edges_;

	int last_titlebar_height_;

	QMap<OakEngineNode *, NodeViewItem *> item_map_;

	QVector<NodeViewEdge *> edges_;

	EngineEventBridge *bridge_ = nullptr;

	QHash<OakEngineNode *, QVector<int64_t>> node_subs_;

private:
	// Ordinary member functions (NOT slots): invoked from lambdas only.
	void group_added_node(OakEngineNode *node, OakEngineNode *group);

	void group_removed_node(OakEngineNode *node, OakEngineNode *group);
};

}

#endif // OAK_NODEVIEWCONTEXT_H