/***

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

#include <QCheckBox>
#include <QGridLayout>
#include <QLabel>
#include <QPushButton>
#include <QVBoxLayout>
#include <QWidget>

#include "node/node.h"
#include "nodeparamviewarraywidget.h"
#include "nodeparamviewconnectedlabel.h"
#include "nodeparamviewkeyframecontrol.h"
#include "nodeparamviewitembase.h"
#include "nodeparamviewwidgetbridge.h"
#include "widget/clickablelabel/clickablelabel.h"
#include "widget/collapsebutton/collapsebutton.h"
#include "widget/keyframeview/keyframeview.h"

namespace olive
{

enum NodeParamViewCheckBoxBehavior {
	kNoCheckBoxes,
	kCheckBoxesOn,
	kCheckBoxesOnNonConnected
};

class NodeParamViewItemBody : public QWidget {
	Q_OBJECT
public:
	NodeParamViewItemBody(Node *node,
						  NodeParamViewCheckBoxBehavior create_checkboxes,
						  QWidget *parent = nullptr);

	void SetTimeTarget(ViewerOutput *target);

	void Retranslate();

	int GetElementY(NodeInput c) const;

	// Set the timebase of any timebased widgets contained here
	void SetTimebase(const rational &timebase);

	void SetInputChecked(const NodeInput &input, bool e);

signals:
	void RequestSelectNode(Node *node);

	void ArrayExpandedChanged(bool e);

	void InputCheckedChanged(const NodeInput &input, bool e);

	void RequestEditTextInViewer();

private:
	void CreateWidgets(QGridLayout *layout, Node *node, const QString &input,
					   int element, int row_index);

	void UpdateUIForEdgeConnection(const NodeInput &input);

	void PlaceWidgetsFromBridge(QGridLayout *layout,
								NodeParamViewWidgetBridge *bridge, int row);

	void InputArraySizeChangedInternal(Node *node, const QString &input,
									   int size);

	struct InputUI {
		InputUI();

		QLabel *main_label;
		NodeParamViewWidgetBridge *widget_bridge;
		NodeParamViewConnectedLabel *connected_label;
		NodeParamViewKeyframeControl *key_control;
		QGridLayout *layout;
		int row;
		QPushButton *extra_btn;
		QCheckBox *optional_checkbox;

		NodeParamViewArrayButton *array_insert_btn;
		NodeParamViewArrayButton *array_remove_btn;
	};

	QHash<NodeInput, InputUI> input_ui_map_;

	struct ArrayUI {
		QWidget *widget;
		int count;
		NodeParamViewArrayButton *append_btn;
	};

	void SetTimeTargetOnInputUI(const InputUI &ui);
	void SetTimebaseOnInputUI(const InputUI &ui);

	Node *node_;

	QHash<NodeInputPair, ArrayUI> array_ui_;

	QHash<NodeInputPair, CollapseButton *> array_collapse_buttons_;

	rational timebase_;

	ViewerOutput *time_target_;

	NodeParamViewCheckBoxBehavior create_checkboxes_;

	QHash<NodeInputPair, NodeInputPair> input_group_lookup_;

	/**
   * @brief The column to place the keyframe controls in
   *
   * Serves as an effective "maximum column" index because the keyframe button is always aligned
   * to the right edge.
   */
	static const int kKeyControlColumn;

	static const int kArrayInsertColumn;
	static const int kArrayRemoveColumn;
	static const int kExtraButtonColumn;

	static const int kOptionalCheckBox;
	static const int kArrayCollapseBtnColumn;
	static const int kLabelColumn;
	static const int kWidgetStartColumn;
	static const int kMaxWidgetColumn;

private slots:
	void EdgeChanged(Node *output, const NodeInput &input);

	void ArrayCollapseBtnPressed(bool checked);

	void InputArraySizeChanged(const QString &input, int old_sz, int size);

	void ArrayAppendClicked();

	void ArrayInsertClicked();

	void ArrayRemoveClicked();

	void ToggleArrayExpanded();

	void ReplaceWidgets(const NodeInput &input);

	void ShowSpeedDurationDialogForNode();

	void OptionalCheckBoxClicked(bool e);
};

class NodeParamViewItem : public NodeParamViewItemBase {
	Q_OBJECT
public:
	NodeParamViewItem(Node *node,
					  NodeParamViewCheckBoxBehavior create_checkboxes,
					  QWidget *parent = nullptr);

	void SetTimeTarget(ViewerOutput *target)
	{
		time_target_ = target;

		body_->SetTimeTarget(target);
	}

	void SetTimebase(const rational &timebase)
	{
		timebase_ = timebase;

		body_->SetTimebase(timebase);
	}

	Node *GetContext() const
	{
		return ctx_;
	}

	void SetContext(Node *ctx)
	{
		ctx_ = ctx;
	}

	Node *GetNode() const
	{
		return node_;
	}

	int GetElementY(const NodeInput &c) const;

	void SetInputChecked(const NodeInput &input, bool e);

	KeyframeView::NodeConnections &GetKeyframeConnections()
	{
		return keyframe_connections_;
	}

	void SetKeyframeConnections(const KeyframeView::NodeConnections &c)
	{
		keyframe_connections_ = c;
	}

signals:
	void RequestSelectNode(Node *node);

	void ArrayExpandedChanged(bool e);

	void InputCheckedChanged(const NodeInput &input, bool e);

	void RequestEditTextInViewer();

	void InputArraySizeChanged(const QString &input, int old_size,
							   int new_size);

protected slots:
	virtual void Retranslate() override;

private:
	NodeParamViewItemBody *body_;
	QLabel *message_label_;
	QPushButton *message_clear_button_;
	QWidget *message_container_;

	Node *node_;

	NodeParamViewCheckBoxBehavior create_checkboxes_;

	Node *ctx_;

	ViewerOutput *time_target_;

	rational timebase_;

	KeyframeView::NodeConnections keyframe_connections_;

private slots:
	void RecreateBody();
	void UpdateMessagePanel();
	void ClearMessages();
};

}

#endif // NODEPARAMVIEWITEM_H
