// Ryzom - MMORPG Framework <http://dev.ryzom.com/projects/ryzom/>
// Copyright (C) 2010  Winch Gate Property Limited
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU Affero 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 Affero General Public License for more details.
//
// You should have received a copy of the GNU Affero General Public License
// along with this program.  If not, see <http://www.gnu.org/licenses/>.

#include "stdpch.h"
#include "installdialog.h"
#include "configfile.h"
#include "utils.h"

#include "nel/misc/system_info.h"
#include "nel/misc/common.h"

#ifdef DEBUG_NEW
	#define new DEBUG_NEW
#endif

CInstallDialog::CInstallDialog():QDialog()
{
	setWindowFlags(windowFlags() & ~Qt::WindowContextHelpButtonHint);

	setupUi(this);

	oldDirectoryRadioButton->setVisible(false);

	// enable download radio button by default
	internetRadioButton->setChecked(true);

	m_oldDirectory = CConfigFile::getInstance()->getOldInstallationDirectory();

	// found a previous installation
	if (CConfigFile::getInstance()->areRyzomDataInstalledIn(m_oldDirectory))
	{
		oldDirectoryRadioButton->setText(tr("Old installation: %1").arg(m_oldDirectory));
		oldDirectoryRadioButton->setVisible(true);
		oldDirectoryRadioButton->setChecked(true);
	}

	updateAnotherLocationText();

	// update default destination
	onDestinationDefaultButtonClicked();

	// check whether OS architecture is 32 or 64 bits
	if (CConfigFile::has64bitsOS())
	{
		clientArchGroupBox->setVisible(true);
		clientArch64RadioButton->setChecked(true);
	}
	else
	{
		clientArchGroupBox->setVisible(false);
		clientArch32RadioButton->setChecked(true);
	}

	const CServer &server = CConfigFile::getInstance()->getServer();

	internetRadioButton->setText(tr("Internet (%1 to download)").arg(qBytesToHumanReadable(server.dataCompressedSize)));
	destinationGroupBox->setTitle(tr("Files will be installed to (requires %1):").arg(qBytesToHumanReadable(server.dataUncompressedSize)));

	connect(anotherLocationBrowseButton, SIGNAL(clicked()), SLOT(onAnotherLocationBrowseButtonClicked()));
	connect(destinationDefaultButton, SIGNAL(clicked()), SLOT(onDestinationDefaultButtonClicked()));
	connect(destinationBrowseButton, SIGNAL(clicked()), SLOT(onDestinationBrowseButtonClicked()));

	// TODO: if found a folder with initial data, get its total size and check if at least that size is free on the disk

	// by default, advanced parameters are hidden
	onShowAdvancedParameters(Qt::Unchecked);

	connect(advancedCheckBox, SIGNAL(stateChanged(int)), SLOT(onShowAdvancedParameters(int)));

	raise();
}

CInstallDialog::~CInstallDialog()
{
}

void CInstallDialog::onShowAdvancedParameters(int state)
{
	advancedFrame->setVisible(state != Qt::Unchecked);

	adjustSize();
}

void CInstallDialog::onAnotherLocationBrowseButtonClicked()
{
	QString directory;
	
	for(;;)
	{
		directory = QFileDialog::getExistingDirectory(this, tr("Please choose directory where Ryzom is currently installed."));

		if (directory.isEmpty()) return;

		if (CConfigFile::getInstance()->isRyzomInstalledIn(directory)) break;

		QMessageBox::StandardButton res = QMessageBox::warning(this, tr("Unable to find Ryzom"), tr("Unable to find Ryzom in selected directory. Please choose another one or cancel."));
	}

	m_anotherDirectory = directory;

	// if we browse to a Ryzom installation, we want to use it
	anotherLocationRadioButton->setChecked(true);

	updateAnotherLocationText();
}

void CInstallDialog::onDestinationDefaultButtonClicked()
{
	m_dstDirectory = CConfigFile::getNewInstallationDirectory();

	updateDestinationText();
}

void CInstallDialog::onDestinationBrowseButtonClicked()
{
	QString directory = QFileDialog::getExistingDirectory(this, tr("Please choose directory to install Ryzom in"), m_dstDirectory);

	if (directory.isEmpty()) return;

	m_dstDirectory = directory;

	updateDestinationText();
}

void CInstallDialog::updateAnotherLocationText()
{
	anotherLocationRadioButton->setText(tr("Another location: %1").arg(m_anotherDirectory.isEmpty() ? tr("Undefined"):m_anotherDirectory));
}

void CInstallDialog::updateDestinationText()
{
	destinationLabel->setText(m_dstDirectory);
}

void CInstallDialog::accept()
{
	// check free disk space
	qint64 freeSpace = NLMISC::CSystemInfo::availableHDSpace(m_dstDirectory.toUtf8().constData());

	const CServer &server = CConfigFile::getInstance()->getServer();

	if (freeSpace < server.dataUncompressedSize)
	{
		QMessageBox::StandardButton res = QMessageBox::warning(this, tr("Not enough free disk space"), tr("You don't have enough free space on this disk, please make more space or choose a directory on another disk."));
		return;
	}

	// create directory if doesn't exist
	bool succeedsToWrite = QDir().mkpath(m_dstDirectory);

	// if unable to create directory, don't expect to write a file in it
	if (succeedsToWrite)
	{
		// check if directory is writable by current user
		QFile file(m_dstDirectory + "/writable_test_for_ryzom_installer.txt");

		if (file.open(QFile::WriteOnly))
		{
			file.close();

			// remove it
			file.remove();
		}
		else
		{
			succeedsToWrite = false;
		}
	}

	if (!succeedsToWrite)
	{
		QMessageBox::StandardButton res = QMessageBox::warning(this, tr("Unable to write in directory"), tr("You don't have the permission to write in this directory with your current user account, please choose another directory."));
		return;
	}

	// if reinstalling in same directory, don't check if directory is empty
	if (m_dstDirectory != CConfigFile::getInstance()->getNewInstallationDirectory())
	{
		if (!isDirectoryEmpty(m_dstDirectory, true))
		{
			QMessageBox::StandardButton res = QMessageBox::warning(this, tr("Directory not empty"), tr("This directory is not empty, please choose another one."));
			return;
		}
	}

	if (oldDirectoryRadioButton->isChecked())
	{
		CConfigFile::getInstance()->setSrcServerDirectory(m_oldDirectory);
	}
	else if (anotherLocationRadioButton->isChecked())
	{
		CConfigFile::getInstance()->setSrcServerDirectory(m_anotherDirectory);
	}
	else
	{
		CConfigFile::getInstance()->setSrcServerDirectory("");
	}

	CConfigFile::getInstance()->setInstallationDirectory(m_dstDirectory);
	CConfigFile::getInstance()->setUse64BitsClient(clientArch64RadioButton->isChecked());
	CConfigFile::getInstance()->save();

	QDialog::accept();
}