/***

  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/>.

***/

#include "trackviewitem.h"

#include <QDebug>
#include <QHBoxLayout>
#include <QMessageBox>
#include <QMouseEvent>
#include <QPainter>
#include <QtMath>

#include "core.h"
#include "timeline/timelineundogeneral.h"
#include "ui/icons/icons.h"
#include "widget/menu/menu.h"

namespace olive
{

TrackViewItem::TrackViewItem(Track *track, QWidget *parent)
	: QWidget(parent)
	, track_(track)
{
	QHBoxLayout *layout = new QHBoxLayout(this);
	layout->setSpacing(0);
	layout->setContentsMargins(0, 0, 0, 0);

	stack_ = new QStackedWidget();
	layout->addWidget(stack_);

	label_ = new ClickableLabel();
	connect(label_, &ClickableLabel::MouseDoubleClicked, this,
			&TrackViewItem::LabelClicked);
	connect(track_, &Track::LabelChanged, this, &TrackViewItem::UpdateLabel);
	connect(track_, &Track::IndexChanged, this, &TrackViewItem::UpdateLabel);
	UpdateLabel();
	stack_->addWidget(label_);

	line_edit_ = new FocusableLineEdit();
	connect(line_edit_, &FocusableLineEdit::Confirmed, this,
			&TrackViewItem::LineEditConfirmed);
	connect(line_edit_, &FocusableLineEdit::Cancelled, this,
			&TrackViewItem::LineEditCancelled);
	stack_->addWidget(line_edit_);

	mute_button_ = CreateMSLButton(Qt::red);
	mute_button_->setChecked(track->IsMuted());
	UpdateMuteButton(track->IsMuted());
	connect(mute_button_, &QPushButton::toggled, track_, &Track::SetMuted);
	connect(mute_button_, &QPushButton::toggled, this,
			&TrackViewItem::UpdateMuteButton);
	layout->addWidget(mute_button_);

	/*solo_button_ = CreateMSLButton(tr("S"), Qt::yellow);
  layout->addWidget(solo_button_);*/

	lock_button_ = CreateMSLButton(Qt::gray);
	lock_button_->setChecked(track->IsLocked());
	UpdateLockButton(track->IsLocked());
	connect(lock_button_, &QPushButton::toggled, track_, &Track::SetLocked);
	connect(lock_button_, &QPushButton::toggled, this,
			&TrackViewItem::UpdateLockButton);
	layout->addWidget(lock_button_);

	setMinimumHeight(mute_button_->height());
	setContextMenuPolicy(Qt::CustomContextMenu);

	connect(track, &Track::MutedChanged, mute_button_,
			&QPushButton::setChecked);
	connect(this, &QWidget::customContextMenuRequested, this,
			&TrackViewItem::ShowContextMenu);
}

QPushButton *TrackViewItem::CreateMSLButton(const QColor &checked_color) const
{
	QPushButton *button = new QPushButton();
	button->setSizePolicy(QSizePolicy::Fixed, QSizePolicy::Expanding);
	button->setCheckable(true);
	button->setStyleSheet(
		QStringLiteral("QPushButton::checked { background: %1; }")
			.arg(checked_color.name()));

	int size = button->sizeHint().height();
	size = qRound(size * 0.75);
	button->setFixedSize(size, size);

	return button;
}

void TrackViewItem::LabelClicked()
{
	stack_->setCurrentWidget(line_edit_);
	line_edit_->setFocus();
	line_edit_->selectAll();
}

void TrackViewItem::LineEditConfirmed()
{
	line_edit_->blockSignals(true);

	track_->SetLabel(line_edit_->text());
	UpdateLabel();

	stack_->setCurrentWidget(label_);

	line_edit_->blockSignals(false);
}

void TrackViewItem::LineEditCancelled()
{
	line_edit_->blockSignals(true);

	stack_->setCurrentWidget(label_);

	line_edit_->blockSignals(false);
}

void TrackViewItem::UpdateLabel()
{
	label_->setText(track_->GetLabelOrName());
}

void TrackViewItem::ShowContextMenu(const QPoint &p)
{
	Menu m(this);

	QAction *delete_action = m.addAction(tr("&Delete"));
	connect(delete_action, &QAction::triggered, this,
			&TrackViewItem::DeleteTrack, Qt::QueuedConnection);

	m.addSeparator();

	QAction *delete_unused_action = m.addAction(tr("Delete All &Empty"));
	connect(delete_unused_action, &QAction::triggered, this,
			&TrackViewItem::DeleteAllEmptyTracks, Qt::QueuedConnection);

	m.exec(mapToGlobal(p));
}

void TrackViewItem::DeleteTrack()
{
	emit AboutToDeleteTrack(track_);
	Core::instance()->undo_stack()->push(
		new TimelineRemoveTrackCommand(track_),
		tr("Deleted Track \"%1\"").arg(track_->GetLabelOrName()));
}

void TrackViewItem::DeleteAllEmptyTracks()
{
	Sequence *sequence = track_->sequence();
	QVector<Track *> tracks_to_remove;
	QStringList track_names_to_remove;

	foreach (Track *t, sequence->GetTracks()) {
		if (t->Blocks().isEmpty()) {
			tracks_to_remove.append(t);
			track_names_to_remove.append(t->GetLabelOrName());
		}
	}

	if (tracks_to_remove.isEmpty()) {
		QMessageBox::information(this, tr("Delete All Empty"),
								 tr("No tracks are currently empty"));
	} else {
		if (QMessageBox::question(
				this, tr("Delete All Empty"),
				tr("This will delete the following tracks:\n\n%1\n\nDo you wish to continue?")
					.arg(track_names_to_remove.join('\n')),
				QMessageBox::Ok | QMessageBox::Cancel) == QMessageBox::Ok) {
			MultiUndoCommand *command = new MultiUndoCommand();
			foreach (Track *track, tracks_to_remove) {
				command->add_child(new TimelineRemoveTrackCommand(track));
			}
			Core::instance()->undo_stack()->push(
				command, tr("Deleted All Empty Tracks"));
		}
	}
}

void TrackViewItem::UpdateMuteButton(bool e)
{
	mute_button_->setIcon(e ? icon::EyeClosed : icon::EyeOpened);
}

void TrackViewItem::UpdateLockButton(bool e)
{
	lock_button_->setIcon(e ? icon::LockClosed : icon::LockOpened);
}

}