/***

  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 "viewertexteditor.h"

#include <QAbstractTextDocumentLayout>
#include <QColorDialog>
#include <QKeyEvent>
#include <QHBoxLayout>
#include <QPainter>
#include <QScrollBar>
#include <QTextBlock>
#include <QtMath>

#include "common/qtutils.h"
#include "ui/icons/icons.h"

namespace olive
{

#define super QTextEdit

ViewerTextEditor::ViewerTextEditor(double scale, QWidget *parent)
	: super(parent)
	, transparent_clone_(nullptr)
	, block_update_toolbar_signal_(false)
	, forced_default_(false)
{
	// Ensure default text color is white
	QPalette p = palette();
	p.setColor(QPalette::Text, Qt::white);
	setPalette(p);

	document()->setDefaultStyleSheet(QStringLiteral("body { color: white; }"));

	// Ensure cursor is visible at this scale
	setCursorWidth(std::ceil(1.0 / scale));

	viewport()->setAutoFillBackground(false);
	setHorizontalScrollBarPolicy(Qt::ScrollBarAlwaysOff);
	setVerticalScrollBarPolicy(Qt::ScrollBarAlwaysOff);
	connect(horizontalScrollBar(), &QScrollBar::rangeChanged, this,
			&ViewerTextEditor::lock_scroll_bar_maximum_to_zero);
	connect(verticalScrollBar(), &QScrollBar::rangeChanged, this,
			&ViewerTextEditor::lock_scroll_bar_maximum_to_zero);

	// Force DPI to the same one that we're using in the actual render
	dpi_force_ = QImage(1, 1, QImage::Format_RGBA8888_Premultiplied);

	const int dpm = 3780;
	dpi_force_.setDotsPerMeterX(dpm);
	dpi_force_.setDotsPerMeterY(dpm);
	document()->documentLayout()->setPaintDevice(&dpi_force_);

	connect(this, &QTextEdit::currentCharFormatChanged, this,
			&ViewerTextEditor::format_changed);
	connect(document(), &QTextDocument::contentsChanged, this,
			&ViewerTextEditor::document_changed, Qt::QueuedConnection);

	setAcceptRichText(false);
}

void ViewerTextEditor::connect_tool_bar(ViewerTextEditorToolBar *toolbar)
{
	connect(toolbar, &ViewerTextEditorToolBar::family_changed, this,
			&ViewerTextEditor::set_family);
	connect(toolbar, &ViewerTextEditorToolBar::size_changed, this,
			&ViewerTextEditor::setFontPointSize);
	connect(toolbar, &ViewerTextEditorToolBar::style_changed, this,
			&ViewerTextEditor::set_style);
	connect(toolbar, &ViewerTextEditorToolBar::underline_changed, this,
			&ViewerTextEditor::setFontUnderline);
	connect(toolbar, &ViewerTextEditorToolBar::strikethrough_changed, this,
			&ViewerTextEditor::set_font_strikethrough);
	connect(toolbar, &ViewerTextEditorToolBar::color_changed, this,
			&ViewerTextEditor::setTextColor);
	connect(toolbar, &ViewerTextEditorToolBar::small_caps_changed, this,
			&ViewerTextEditor::set_small_caps);
	connect(toolbar, &ViewerTextEditorToolBar::stretch_changed, this,
			&ViewerTextEditor::set_font_stretch);
	connect(toolbar, &ViewerTextEditorToolBar::kerning_changed, this,
			&ViewerTextEditor::set_font_kerning);
	connect(toolbar, &ViewerTextEditorToolBar::line_height_changed, this,
			&ViewerTextEditor::set_line_height);
	connect(toolbar, &ViewerTextEditorToolBar::alignment_changed, this,
			[this](Qt::Alignment a) {
				this->setAlignment(a);

				// Ensure no buttons are checked that shouldn't be
				static_cast<ViewerTextEditorToolBar *>(sender())->set_alignment(
					a);
			});

	update_tool_bar(toolbar, this->currentCharFormat(),
				  this->textCursor().blockFormat(), this->alignment());

	toolbars_.append(toolbar);
}

void ViewerTextEditor::paint(QPainter *p, Qt::Alignment valign)
{
	QAbstractTextDocumentLayout::PaintContext ctx;

	QRect clip = this->rect();
	p->setClipRect(clip, Qt::IntersectClip);
	ctx.clip = clip;

	ctx.cursorPosition = this->textCursor().position();

	if (this->textCursor().hasSelection()) {
		QAbstractTextDocumentLayout::Selection selection;
		selection.cursor = this->textCursor();

		QPalette::ColorGroup cg = this->hasFocus() ? QPalette::Active :
													 QPalette::Inactive;
		QBrush b = ctx.palette.brush(cg, QPalette::Highlight);
		QColor bc = b.color();
		bc.setAlpha(128);
		b.setColor(bc);
		selection.format.setBackground(b);
		QStyleOption opt;
		opt.initFrom(this);
		if (this->style()->styleHint(QStyle::SH_RichText_FullWidthSelection,
									 &opt, this)) {
			selection.format.setProperty(QTextFormat::FullWidthSelection, true);
		}
		ctx.selections.append(selection);
	}

	switch (valign) {
	case Qt::AlignTop:
		// Do nothing
		break;
	case Qt::AlignVCenter:
		p->translate(0, clip.height() / 2 - document()->size().height() / 2);
		break;
	case Qt::AlignBottom:
		p->translate(0, clip.height() - document()->size().height());
		break;
	}

	const bool use_transparent_clone = true;
	if (transparent_clone_ && use_transparent_clone) {
		transparent_clone_->setPageSize(this->document()->pageSize());
		transparent_clone_->documentLayout()->draw(p, ctx);
	} else {
		document()->documentLayout()->draw(p, ctx);
	}
}

void ViewerTextEditor::paintEvent(QPaintEvent *e)
{
	// Disable painting
}

void ViewerTextEditor::update_tool_bar(ViewerTextEditorToolBar *toolbar,
									 const QTextCharFormat &f,
									 const QTextBlockFormat &b,
									 Qt::Alignment alignment)
{
	QStringList families = f.fontFamilies().toStringList();
	QString family;
	if (families.isEmpty()) {
		family = qApp->font().family();
	} else {
		family = families.first();
	}

	QString style = f.fontStyleName().toString();
	QStringList styles = QFontDatabase().styles(family);
	if (!styles.isEmpty() && (style.isEmpty() || !styles.contains(style))) {
		// There seems to be no better way to find the "regular" style outside of this heuristic.
		// Feel free to add more if a font isn't working right.
		style = QStringLiteral("Regular");
		if (!styles.contains(style)) {
			style = QStringLiteral("Normal");
		}
		if (!styles.contains(style)) {
			style = QStringLiteral("R");
		}
		if (!styles.contains(style)) {
			style = styles.first();
		}
	}

	toolbar->set_font_family(family);
	toolbar->set_font_size(f.fontPointSize());
	toolbar->set_style(style);
	toolbar->set_underline(f.fontUnderline());
	toolbar->set_strikethrough(f.fontStrikeOut());
	toolbar->set_alignment(alignment);
	toolbar->set_color(f.foreground().color());
	toolbar->set_small_caps(f.fontCapitalization() == QFont::SmallCaps);
	toolbar->set_stretch(f.fontStretch() == 0 ? 100 : f.fontStretch());
	toolbar->set_kerning(f.fontLetterSpacing() == 0.0 ? 100 :
													   f.fontLetterSpacing());
	toolbar->set_line_height(b.lineHeight() == 0.0 ? 100 : b.lineHeight());
}

void ViewerTextEditor::format_changed(const QTextCharFormat &f)
{
	if (!block_update_toolbar_signal_) {
		foreach (ViewerTextEditorToolBar *toolbar, toolbars_) {
			update_tool_bar(toolbar, f, textCursor().blockFormat(),
						  this->alignment());
		}
	}

	if (!(document()->blockCount() == 1 &&
		  document()->firstBlock().text().isEmpty())) {
		default_fmt_ = f;
	}
}

void ViewerTextEditor::set_family(const QString &s)
{
	ViewerTextEditorToolBar *toolbar =
		static_cast<ViewerTextEditorToolBar *>(sender());

	QTextCharFormat f;
	f.setFontFamilies({ s });

	apply_style(&f, s, toolbar->get_font_style_name());

	merge_char_format(f);
}

void ViewerTextEditor::set_style(const QString &s)
{
	ViewerTextEditorToolBar *toolbar =
		static_cast<ViewerTextEditorToolBar *>(sender());

	QTextCharFormat f;

	apply_style(&f, toolbar->get_font_family(), s);

	merge_char_format(f);
}

void ViewerTextEditor::set_font_strikethrough(bool e)
{
	QTextCharFormat f;
	f.setFontStrikeOut(e);
	merge_char_format(f);
}

void ViewerTextEditor::set_small_caps(bool e)
{
	QTextCharFormat f;
	f.setFontCapitalization(e ? QFont::SmallCaps : QFont::MixedCase);
	merge_char_format(f);
}

void ViewerTextEditor::set_font_stretch(int i)
{
	QTextCharFormat f;
	f.setFontStretch(i);
	merge_char_format(f);
}

void ViewerTextEditor::set_font_kerning(qreal i)
{
	QTextCharFormat f;
	f.setFontLetterSpacing(i);
	merge_char_format(f);
}

void ViewerTextEditor::merge_char_format(const QTextCharFormat &fmt)
{
	// mergeCurrentCharFormat throws a currentCharFormatChanged signal that updates the toolbar,
	// this can be undesirable if the user is currently typing a font
	block_update_toolbar_signal_ = true;
	mergeCurrentCharFormat(fmt);
	//default_fmt_ = this->currentCharFormat();
	block_update_toolbar_signal_ = false;
}

void ViewerTextEditor::apply_style(QTextCharFormat *format,
								  const QString &family, const QString &style)
{
	// NOTE: Windows appears to require setting weight and italic manually, while macOS and Linux are
	//       perfectly fine with just the style name
	format->setFontWeight(QFontDatabase().weight(family, style));
	format->setFontItalic(QFontDatabase().italic(family, style));

	format->setFontStyleName(style);
}

void ViewerTextEditor::set_line_height(qreal i)
{
	QTextBlockFormat f = this->textCursor().blockFormat();
	f.setLineHeight(i, QTextBlockFormat::ProportionalHeight);
	this->textCursor().setBlockFormat(f);
}

void ViewerTextEditor::lock_scroll_bar_maximum_to_zero()
{
	static_cast<QScrollBar *>(sender())->setMaximum(0);
}

void ViewerTextEditor::document_changed()
{
	if (document()->blockCount() == 1 &&
		document()->firstBlock().text().isEmpty()) {
		if (!forced_default_) {
			QTextCursor c(document()->firstBlock());
			c.setBlockCharFormat(default_fmt_);
			forced_default_ = true;
		}
	} else {
		if (default_fmt_.isEmpty()) {
			default_fmt_ = document()->firstBlock().charFormat();
		}
		forced_default_ = false;
	}

	// HACK: We want to show the text cursor and selections without necessarily rendering the text,
	//       because the text is already being rendered underneath the gizmo (and rendering twice will
	//       alter the overall look of the text while editing). This is something that Qt does not
	//       support by default, and while it could be solved by subclassing QAbstractTextDocumentLayout,
	//       this seems like overkill since 99% of QTextDocumentLayout's (the subclass used by default)
	//       functionality and would need to be fully implemented ourselves. So instead, we clone the
	//       document, set all of its colors to rgba(0,0,0,0) to make them transparent, and draw that
	//       document instead. The result is cursor and selections being rendered without the text.
	//       While this isn't the fastest code, nor is it the cleanest solution, it definitely works.
	delete transparent_clone_;
	transparent_clone_ = document()->clone(this);
	transparent_clone_->documentLayout()->setPaintDevice(&dpi_force_);
	transparent_clone_->documentLayout()->setProperty(
		"cursorWidth", document()->documentLayout()->property("cursorWidth"));

	QTextCursor cursor(transparent_clone_);
	cursor.select(QTextCursor::Document);

	QTextCharFormat fmt;
	fmt.setForeground(QColor(0, 0, 0, 0));
	cursor.mergeCharFormat(fmt);
}

ViewerTextEditorToolBar::ViewerTextEditorToolBar(QWidget *parent)
	: QWidget(parent)
	, painted_(false)
	, drag_enabled_(true)
{
	QVBoxLayout *outer_layout = new QVBoxLayout(this);

	const int advanced_slider_width =
		QtUtils::q_font_metrics_width(fontMetrics(), QStringLiteral("9999.9%"));

	{
		QHBoxLayout *row_layout = new QHBoxLayout();
		row_layout->setSpacing(0);
		outer_layout->addLayout(row_layout);

		font_combo_ = new QFontComboBox();
		connect(
			font_combo_, &QFontComboBox::currentTextChanged, this,
			&ViewerTextEditorToolBar::update_font_style_list_and_emit_family_changed);
		row_layout->addWidget(font_combo_);

		font_sz_slider_ = new FloatSlider();
		font_sz_slider_->set_minimum(0.1);
		font_sz_slider_->set_maximum(9999.9);
		font_sz_slider_->set_decimal_places(1);
		font_sz_slider_->set_alignment(Qt::AlignCenter);
		font_sz_slider_->setFixedWidth(advanced_slider_width);
		connect(font_sz_slider_, &FloatSlider::value_changed, this,
				&ViewerTextEditorToolBar::size_changed);
		font_sz_slider_->set_ladder_element_count(2);
		row_layout->addWidget(font_sz_slider_);

		style_combo_ = new QComboBox();
		connect(style_combo_, &QComboBox::currentTextChanged, this,
				&ViewerTextEditorToolBar::style_changed);
		row_layout->addWidget(style_combo_);

		underline_btn_ = new QPushButton();
		connect(underline_btn_, &QPushButton::clicked, this,
				&ViewerTextEditorToolBar::underline_changed);
		underline_btn_->setCheckable(true);
		underline_btn_->setIcon(icon::text_underline);
		row_layout->addWidget(underline_btn_);

		strikethrough_btn_ = new QPushButton();
		connect(strikethrough_btn_, &QPushButton::clicked, this,
				&ViewerTextEditorToolBar::strikethrough_changed);
		strikethrough_btn_->setCheckable(true);
		strikethrough_btn_->setIcon(icon::text_strikethrough);
		row_layout->addWidget(strikethrough_btn_);

		add_spacer(row_layout);

		color_btn_ = new QPushButton();
		color_btn_->setAutoFillBackground(true);
		connect(color_btn_, &QPushButton::clicked, this, [this] {
			QColor c = color_btn_->property("color").value<QColor>();

			QColorDialog cd(c, this);
			if (cd.exec() == QDialog::Accepted) {
				c = cd.selectedColor();
				set_color(c);
				emit color_changed(c);
			}
		});
		row_layout->addWidget(color_btn_);

		row_layout->addStretch();
	}

	{
		QHBoxLayout *row_layout = new QHBoxLayout();
		row_layout->setSpacing(0);
		outer_layout->addLayout(row_layout);

		align_left_btn_ = new QPushButton();
		align_left_btn_->setCheckable(true);
		align_left_btn_->setIcon(icon::text_align_left);
		connect(align_left_btn_, &QPushButton::clicked, this,
				[this] { emit alignment_changed(Qt::AlignLeft); });
		row_layout->addWidget(align_left_btn_);

		align_center_btn_ = new QPushButton();
		align_center_btn_->setCheckable(true);
		align_center_btn_->setIcon(icon::text_align_center);
		connect(align_center_btn_, &QPushButton::clicked, this,
				[this] { emit alignment_changed(Qt::AlignHCenter); });
		row_layout->addWidget(align_center_btn_);

		align_right_btn_ = new QPushButton();
		align_right_btn_->setCheckable(true);
		align_right_btn_->setIcon(icon::text_align_right);
		connect(align_right_btn_, &QPushButton::clicked, this,
				[this] { emit alignment_changed(Qt::AlignRight); });
		row_layout->addWidget(align_right_btn_);

		align_justify_btn_ = new QPushButton();
		align_justify_btn_->setCheckable(true);
		align_justify_btn_->setIcon(icon::text_align_justify);
		connect(align_justify_btn_, &QPushButton::clicked, this,
				[this] { emit alignment_changed(Qt::AlignJustify); });
		row_layout->addWidget(align_justify_btn_);

		add_spacer(row_layout);

		align_top_btn_ = new QPushButton();
		align_top_btn_->setCheckable(true);
		align_top_btn_->setIcon(icon::text_align_top);
		connect(align_top_btn_, &QPushButton::clicked, this,
				[this] { emit vertical_alignment_changed(Qt::AlignTop); });
		row_layout->addWidget(align_top_btn_);

		align_middle_btn_ = new QPushButton();
		align_middle_btn_->setCheckable(true);
		align_middle_btn_->setIcon(icon::text_align_middle);
		connect(align_middle_btn_, &QPushButton::clicked, this,
				[this] { emit vertical_alignment_changed(Qt::AlignVCenter); });
		row_layout->addWidget(align_middle_btn_);

		align_bottom_btn_ = new QPushButton();
		align_bottom_btn_->setCheckable(true);
		align_bottom_btn_->setIcon(icon::text_align_bottom);
		connect(align_bottom_btn_, &QPushButton::clicked, this,
				[this] { emit vertical_alignment_changed(Qt::AlignBottom); });
		row_layout->addWidget(align_bottom_btn_);

		add_spacer(row_layout);

		small_caps_btn_ = new QPushButton();
		small_caps_btn_->setIcon(icon::text_small_caps);
		small_caps_btn_->setCheckable(true);
		connect(small_caps_btn_, &QPushButton::clicked, this,
				&ViewerTextEditorToolBar::small_caps_changed);
		row_layout->addWidget(small_caps_btn_);

		add_spacer(row_layout);

		row_layout->addWidget(
			new QLabel(tr("Stretch: "))); // FIXME: Procure icon

		stretch_slider_ = new IntegerSlider();
		stretch_slider_->set_minimum(0);
		stretch_slider_->SetDefaultValue(100);
		stretch_slider_->setFixedWidth(advanced_slider_width);
		stretch_slider_->set_format(tr("%1%"));
		connect(stretch_slider_, &IntegerSlider::value_changed, this,
				&ViewerTextEditorToolBar::stretch_changed);
		row_layout->addWidget(stretch_slider_);

		row_layout->addWidget(
			new QLabel(tr("Kerning: "))); // FIXME: Procure icon

		kerning_slider_ = new FloatSlider();
		kerning_slider_->set_minimum(0);
		kerning_slider_->SetDefaultValue(100);
		kerning_slider_->set_decimal_places(1);
		kerning_slider_->setFixedWidth(advanced_slider_width);
		kerning_slider_->set_format(tr("%1%"));
		connect(kerning_slider_, &FloatSlider::value_changed, this,
				&ViewerTextEditorToolBar::kerning_changed);
		row_layout->addWidget(kerning_slider_);

		row_layout->addWidget(
			new QLabel(tr("Line Height: "))); // FIXME: Procure icon

		line_height_slider_ = new FloatSlider();
		line_height_slider_->set_minimum(0);
		line_height_slider_->SetDefaultValue(100);
		line_height_slider_->set_decimal_places(1);
		line_height_slider_->setFixedWidth(advanced_slider_width);
		line_height_slider_->set_format(tr("%1%"));
		connect(line_height_slider_, &FloatSlider::value_changed, this,
				&ViewerTextEditorToolBar::line_height_changed);
		row_layout->addWidget(line_height_slider_);

		row_layout->addStretch();
	}

	setAutoFillBackground(true);

	resize(sizeHint());
}

void ViewerTextEditorToolBar::set_alignment(Qt::Alignment a)
{
	align_left_btn_->setChecked(a == Qt::AlignLeft);
	align_center_btn_->setChecked(a == Qt::AlignHCenter);
	align_right_btn_->setChecked(a == Qt::AlignRight);
	align_justify_btn_->setChecked(a == Qt::AlignJustify);
}

void ViewerTextEditorToolBar::set_vertical_alignment(Qt::Alignment a)
{
	align_top_btn_->setChecked(a == Qt::AlignTop);
	align_middle_btn_->setChecked(a == Qt::AlignVCenter);
	align_bottom_btn_->setChecked(a == Qt::AlignBottom);
}

void ViewerTextEditorToolBar::set_color(const QColor &c)
{
	color_btn_->setProperty("color", c);
	color_btn_->setStyleSheet(
		QStringLiteral("QPushButton { background: %1; }").arg(c.name()));
}

void ViewerTextEditorToolBar::closeEvent(QCloseEvent *event)
{
	event->ignore();
}

void ViewerTextEditorToolBar::paintEvent(QPaintEvent *event)
{
	if (!painted_) {
		emit first_paint();
		painted_ = true;
	}
	QWidget::paintEvent(event);
}

void ViewerTextEditorToolBar::add_spacer(QLayout *l)
{
	const int spacing = this->fontMetrics().height() / 4;
	QWidget *a = new QWidget();
	a->setFixedSize(spacing, 1);
	l->addWidget(a);

	l->addWidget(QtUtils::create_vertical_line());

	QWidget *b = new QWidget();
	b->setFixedSize(spacing, 1);
	l->addWidget(b);
}

void ViewerTextEditorToolBar::update_font_style_list(const QString &family)
{
	QString temp = style_combo_->currentText();

	style_combo_->blockSignals(true);
	style_combo_->clear();
	QStringList l = QFontDatabase().styles(family);
	foreach (const QString &style, l) {
		style_combo_->addItem(style);
	}
	style_combo_->setCurrentText(temp);
	style_combo_->blockSignals(false);
}

void ViewerTextEditorToolBar::update_font_style_list_and_emit_family_changed(
	const QString &family)
{
	// Ensures correct ordering of commands
	update_font_style_list(family);
	emit family_changed(family);
}

void ViewerTextEditorToolBar::mousePressEvent(QMouseEvent *event)
{
	QWidget::mousePressEvent(event);

	if (event->button() == Qt::LeftButton && drag_enabled_) {
		drag_anchor_ = event->pos();
	}
}

void ViewerTextEditorToolBar::mouseMoveEvent(QMouseEvent *event)
{
	QWidget::mouseMoveEvent(event);

	if ((event->buttons() & Qt::LeftButton) && drag_enabled_) {
		this->move(mapToParent(QPoint(event->pos() - drag_anchor_)));
	}
}

void ViewerTextEditorToolBar::mouseReleaseEvent(QMouseEvent *event)
{
	QWidget::mouseReleaseEvent(event);
}

}