PrismLauncher/application/dialogs/NewInstanceDialog.cpp

/* Copyright 2013-2018 MultiMC Contributors
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#include "MultiMC.h"
#include "NewInstanceDialog.h"
#include "ui_NewInstanceDialog.h"

#include <BaseVersion.h>
#include <icons/IconList.h>
#include <tasks/Task.h>
#include <InstanceList.h>

#include "VersionSelectDialog.h"
#include "ProgressDialog.h"
#include "IconPickerDialog.h"
#include "ChooseFtbPackDialog.h"

#include <QLayout>
#include <QPushButton>
#include <QFileDialog>
#include <QValidator>

#include <meta/Index.h>
#include <meta/VersionList.h>

class UrlValidator : public QValidator
{
public:
	using QValidator::QValidator;

	State validate(QString &in, int &pos) const
	{
		const QUrl url(in);
		if (url.isValid() && !url.isRelative() && !url.isEmpty())
		{
			return Acceptable;
		}
		else if (QFile::exists(in))
		{
			return Acceptable;
		}
		else
		{
			return Intermediate;
		}
	}
};

NewInstanceDialog::NewInstanceDialog(const QString & initialGroup, const QString & url, QWidget *parent)
	: QDialog(parent), ui(new Ui::NewInstanceDialog)
{
	ui->setupUi(this);
	resize(minimumSizeHint());
	layout()->setSizeConstraint(QLayout::SetFixedSize);

	auto vlist = ENV.metadataIndex()->get("net.minecraft");
	if(vlist->isLoaded())
	{
		setSelectedVersion(vlist->getRecommended());
	}
	else
	{
		vlist->load(Net::Mode::Online);
		auto task = vlist->getLoadTask();
		if(vlist->isLoaded())
		{
			setSelectedVersion(vlist->getRecommended());
		}
		if(task)
		{
			connect(task.get(), &Task::succeeded, this, &NewInstanceDialog::versionListUpdated);
		}
	}

	InstIconKey = "default";
	ui->iconButton->setIcon(MMC->icons()->getIcon(InstIconKey));

	ui->modpackEdit->setValidator(new UrlValidator(ui->modpackEdit));

	ui->instNameTextBox->setAlignment(Qt::AlignHCenter);

	connect(ui->modpackEdit, &QLineEdit::textChanged, this, &NewInstanceDialog::updateDialogState);
	connect(ui->modpackBox, &QRadioButton::clicked, this, &NewInstanceDialog::updateDialogState);

	connect(ui->versionBox, &QRadioButton::clicked, this, &NewInstanceDialog::updateDialogState);
	connect(ui->versionTextBox, &QLineEdit::textChanged, this, &NewInstanceDialog::updateDialogState);

	connect(ui->ftbBox, &QRadioButton::clicked, this, &NewInstanceDialog::updateDialogState);

	auto groups = MMC->instances()->getGroups().toSet();
	auto groupList = QStringList(groups.toList());
	groupList.sort(Qt::CaseInsensitive);
	groupList.removeOne("");
	groupList.push_front(initialGroup);
	groupList.push_front("");
	ui->groupBox->addItems(groupList);
	int index = groupList.indexOf(initialGroup);
	if(index == -1)
	{
		index = 0;
	}
	ui->groupBox->setCurrentIndex(index);
	ui->groupBox->lineEdit()->setPlaceholderText(tr("No group"));
	ui->buttonBox->setFocus();

	originalPlaceholderText = ui->instNameTextBox->placeholderText();
	if(!url.isEmpty())
	{
		ui->modpackBox->setChecked(true);
		ui->modpackEdit->setText(url);
	}

	ftbPackDownloader = new FtbPackDownloader();

	connect(ftbPackDownloader, &FtbPackDownloader::ready, this, &NewInstanceDialog::ftbPackDataDownloadSuccessfully);
	connect(ftbPackDownloader, &FtbPackDownloader::packFetchFailed, this, &NewInstanceDialog::ftbPackDataDownloadFailed);

	ftbPackDownloader->fetchModpacks(false);

	updateDialogState();
}

void NewInstanceDialog::versionListUpdated()
{
	if(!m_versionSetByUser)
	{
		auto vlist = ENV.metadataIndex()->get("net.minecraft");
		setSelectedVersion(vlist->getRecommended());
	}
}

NewInstanceDialog::~NewInstanceDialog()
{
	delete ui;
}

void NewInstanceDialog::updateDialogState()
{
	QString suggestedName;
	if(ui->versionBox->isChecked())
	{
		suggestedName = ui->versionTextBox->text();
	}
	else if (ui->modpackBox->isChecked())
	{
		auto url = QUrl::fromUserInput(ui->modpackEdit->text());
		QFileInfo fi(url.fileName());
		suggestedName = fi.completeBaseName();
	}
	else if (ui->ftbBox->isChecked())
	{
		if(ftbPackDownloader->isValidPackSelected()) {
			suggestedName = ftbPackDownloader->getSuggestedInstanceName();
			ui->labelFtbPack->setText(selectedPack.name);
		}

	}

	ftbModpackRequested = ui->ftbBox->isChecked();

	if(suggestedName.isEmpty())
	{
		ui->instNameTextBox->setPlaceholderText(originalPlaceholderText);
	}
	else
	{
		ui->instNameTextBox->setPlaceholderText(suggestedName);
	}
	bool allowOK = !instName().isEmpty() && (
				(ui->versionBox->isChecked() && m_selectedVersion) ||
				(ui->modpackBox->isChecked() && ui->modpackEdit->hasAcceptableInput()) ||
				(ui->ftbBox->isChecked() && ftbPackDownloader && ftbPackDownloader->isValidPackSelected() )
				);
	ui->buttonBox->button(QDialogButtonBox::Ok)->setEnabled(allowOK);
}

void NewInstanceDialog::setSelectedVersion(BaseVersionPtr version)
{
	m_selectedVersion = version;

	if (m_selectedVersion)
	{
		ui->versionTextBox->setText(version->descriptor());
	}
	else
	{
		ui->versionTextBox->setText("");
	}
}

QString NewInstanceDialog::instName() const
{
	auto result = ui->instNameTextBox->text();
	if(result.size())
	{
		return result.trimmed();
	}
	result = ui->instNameTextBox->placeholderText();
	if(result.size() && result != originalPlaceholderText)
	{
		return result.trimmed();
	}
	return QString();
}

QString NewInstanceDialog::instGroup() const
{
	return ui->groupBox->currentText();
}
QString NewInstanceDialog::iconKey() const
{
	return InstIconKey;
}
QUrl NewInstanceDialog::modpackUrl() const
{
	if (ui->modpackBox->isChecked())
	{
		const QUrl url(ui->modpackEdit->text());
		if (url.isValid() && !url.isRelative() && !url.host().isEmpty())
		{
			return url;
		}
		else
		{
			return QUrl::fromLocalFile(ui->modpackEdit->text());
		}
	}
	else
	{
		return QUrl();
	}
}

BaseVersionPtr NewInstanceDialog::selectedVersion() const
{
	return m_selectedVersion;
}

void NewInstanceDialog::on_btnChangeVersion_clicked()
{
	VersionSelectDialog vselect(ENV.metadataIndex()->get("net.minecraft").get(), tr("Change Minecraft version"), this);
	vselect.exec();
	if (vselect.result() == QDialog::Accepted)
	{
		BaseVersionPtr version = vselect.selectedVersion();
		if (version)
		{
			m_versionSetByUser = true;
			setSelectedVersion(version);
		}
	}
}

void NewInstanceDialog::on_iconButton_clicked()
{
	IconPickerDialog dlg(this);
	dlg.execWithSelection(InstIconKey);

	if (dlg.result() == QDialog::Accepted)
	{
		InstIconKey = dlg.selectedIconKey;
		ui->iconButton->setIcon(MMC->icons()->getIcon(InstIconKey));
	}
}

void NewInstanceDialog::on_instNameTextBox_textChanged(const QString &arg1)
{
	updateDialogState();
}

void NewInstanceDialog::on_modpackBtn_clicked()
{
	const QUrl url = QFileDialog::getOpenFileUrl(this, tr("Choose modpack"), modpackUrl(), tr("Zip (*.zip)"));
	if (url.isValid())
	{
		if (url.isLocalFile())
		{
			ui->modpackEdit->setText(url.toLocalFile());
		}
		else
		{
			ui->modpackEdit->setText(url.toString());
		}
	}
}

bool NewInstanceDialog::isFtbModpackRequested() {
	return ftbModpackRequested;
}

FtbPackDownloader *NewInstanceDialog::getFtbPackDownloader() {
	return ftbPackDownloader;
}

void NewInstanceDialog::on_btnChooseFtbPack_clicked() {
	ChooseFtbPackDialog dl(ftbPackDownloader->getModpacks());
	dl.exec();
	if(dl.result() == QDialog::Accepted) {
		selectedPack = dl.getSelectedModpack();
		ftbPackDownloader->selectPack(selectedPack, dl.getSelectedVersion());
	}
	updateDialogState();
}

void NewInstanceDialog::ftbPackDataDownloadSuccessfully() {
	ui->packDataDownloadStatus->setText(tr("(Pack data download complete)"));
	// ui->labelFtbPack->setText(tr("Disabled for now... not completed!"));

	// Disable for PR
	ui->ftbBox->setEnabled(true);
}

void NewInstanceDialog::ftbPackDataDownloadFailed() {
	ui->packDataDownloadStatus->setText(tr("(Pack data download failed)"));
}