/***

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

#include <iostream>

namespace olive
{

CLIProgressDialog::CLIProgressDialog(const QString &title, QObject *parent)
	: QObject(parent)
	, title_(title)
	, progress_(-1)
	, drawn_(false)
{
	set_progress(0);
}

void CLIProgressDialog::update()
{
	if (drawn_) {
		// We've been here before, do a carriage return back to the start of the terminal line
		std::cout << "\r";
	} else {
		drawn_ = true;
	}

	// FIXME: Get real column count
	int columns = 80;

	int title_columns = columns / 2 - 1;

	// Print "title" text
	QString sized_title = title_;

	if (title_.size() > title_columns) {
		sized_title =
			title_.left(title_columns - 3).append(QStringLiteral("..."));
	} else {
		sized_title = title_;
	}

	std::cout << sized_title.toUtf8().constData();

	// Pad out the rest of the title area if necessary
	for (int i = sized_title.size(); i < title_columns; i++) {
		std::cout << " ";
	}

	// Percentage counter " 100% " is 5 characters + the enclosing brackets [] are 2 characters
	int progress_bar_columns = columns / 2 - 7;

	std::cout << "[";

	// Get UI bar progress
	int bar_prog = qRound(progress_ * progress_bar_columns);

	// Draw filled in bar
	for (int i = 0; i < bar_prog; i++) {
		std::cout << "=";
	}

	// Draw empty space
	for (int i = bar_prog; i < progress_bar_columns; i++) {
		std::cout << " ";
	}

	std::cout << "] ";

	// Pad the percentage so single/double/triple digit values all occupy the
	// same width. Note progress_ is normalized (0.0-1.0), so the percentage
	// must be computed before comparing against 10/100.
	const int percent = qRound(progress_ * 100.0);

	if (percent < 100) {
		std::cout << " ";
	}

	if (percent < 10) {
		std::cout << " ";
	}

	std::cout << percent << "% " << std::endl << std::flush;
}

void CLIProgressDialog::set_progress(double p)
{
	if (progress_ != p) {
		progress_ = p;

		update();
	}
}

}