/***

  Olive - Non-Linear Video Editor
  Copyright (C) 2022 Olive Team
  Modifications Copyright (C) 2025 mikesolar

  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 NODEGROUP_H
#define NODEGROUP_H

#include "node/node.h"

namespace olive
{

class NodeGroup : public Node {
	Q_OBJECT
public:
	NodeGroup();

	NODE_DEFAULT_FUNCTIONS(NodeGroup)

	virtual QString Name() const override;
	virtual QString id() const override;
	virtual QVector<CategoryID> Category() const override;
	virtual QString Description() const override;

	virtual void Retranslate() override;

	virtual bool LoadCustom(QXmlStreamReader *reader,
							SerializedData *data) override;
	virtual void SaveCustom(QXmlStreamWriter *writer) const override;
	virtual void PostLoadEvent(SerializedData *data) override;

	QString AddInputPassthrough(const NodeInput &input,
								const QString &force_id = QString());

	void RemoveInputPassthrough(const NodeInput &input);

	Node *GetOutputPassthrough() const
	{
		return output_passthrough_;
	}

	void SetOutputPassthrough(Node *node);

	using InputPassthrough = QPair<QString, NodeInput>;
	using InputPassthroughs = QVector<InputPassthrough>;
	const InputPassthroughs &GetInputPassthroughs() const
	{
		return input_passthroughs_;
	}

	bool ContainsInputPassthrough(const NodeInput &input) const;

	virtual QString GetInputName(const QString &id) const override;

	static NodeInput ResolveInput(NodeInput input);
	static bool GetInner(NodeInput *input);

	QString GetIDOfPassthrough(const NodeInput &input) const
	{
		for (auto it = input_passthroughs_.cbegin();
			 it != input_passthroughs_.cend(); it++) {
			if (it->second == input) {
				return it->first;
			}
		}
		return QString();
	}

	NodeInput GetInputFromID(const QString &id) const
	{
		for (auto it = input_passthroughs_.cbegin();
			 it != input_passthroughs_.cend(); it++) {
			if (it->first == id) {
				return it->second;
			}
		}
		return NodeInput();
	}

signals:
	void InputPassthroughAdded(olive::NodeGroup *group,
							   const olive::NodeInput &input);

	void InputPassthroughRemoved(olive::NodeGroup *group,
								 const olive::NodeInput &input);

	void OutputPassthroughChanged(olive::NodeGroup *group, olive::Node *output);

private:
	InputPassthroughs input_passthroughs_;

	Node *output_passthrough_;
};

class NodeGroupAddInputPassthrough : public UndoCommand {
public:
	NodeGroupAddInputPassthrough(NodeGroup *group, const NodeInput &input,
								 const QString &force_id = QString())
		: group_(group)
		, input_(input)
		, actually_added_(false)
	{
	}

	virtual Project *GetRelevantProject() const override
	{
		return group_->project();
	}

protected:
	virtual void redo() override;

	virtual void undo() override;

private:
	NodeGroup *group_;

	NodeInput input_;

	QString force_id_;

	bool actually_added_;
};

class NodeGroupSetOutputPassthrough : public UndoCommand {
public:
	NodeGroupSetOutputPassthrough(NodeGroup *group, Node *output)
		: group_(group)
		, new_output_(output)
	{
	}

	virtual Project *GetRelevantProject() const override
	{
		return group_->project();
	}

protected:
	virtual void redo() override;

	virtual void undo() override;

private:
	NodeGroup *group_;

	Node *new_output_;
	Node *old_output_;
};

}

#endif // NODEGROUP_H
