/*
	(c) Copyright  2012 - 2018 Anton Sviridenko
	https://picapica.im

	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 <https://www.gnu.org/licenses/>.
*/
#include "filetransfercontroller.h"
#include "globals.h"
#include "skynet.h"
#include "mainwindow.h"
#include "msguirouter.h"
#include <QFileDialog>
#include <QFileInfo>

FileTransferController::FileTransferController(QObject *parent) :
	QObject(parent)
{
	connect(skynet, SIGNAL(IncomingFileRequestReceived(QByteArray, quint64, QString)),
	        this, SLOT(file_request(QByteArray, quint64, QString)));

	connect(skynet, SIGNAL(FileProgress(QByteArray, quint64, quint64)),
	        this, SLOT(file_progress(QByteArray, quint64, quint64)));

	connect(skynet, SIGNAL(OutgoingFileRequestAccepted(QByteArray)), this, SLOT(file_accepted_by_peer(QByteArray)));

	connect(skynet, SIGNAL(OutgoingFileRequestDenied(QByteArray)), this, SLOT(file_denied_by_peer(QByteArray)));

	connect(skynet, SIGNAL(c2cClosed(QByteArray)), this, SLOT(file_peer_disconnected(QByteArray)));
}

void FileTransferController::file_peer_disconnected(QByteArray peer_id)
{
	if (ftdialogs_in.contains(peer_id))
	{
		ftdialogs_in[peer_id]->peerDisconnected();
		ftdialogs_in.remove(peer_id);
	}

	if (ftdialogs_out.contains(peer_id))
	{
		ftdialogs_out[peer_id]->peerDisconnected();
		ftdialogs_out.remove(peer_id);
	}
}

void FileTransferController::file_finished_incoming(QByteArray peer_id)
{
	if (ftdialogs_in.contains(peer_id))

		ftdialogs_in.remove(peer_id);
}

void FileTransferController::file_finished_outgoing(QByteArray peer_id)
{
	if (ftdialogs_out.contains(peer_id))
	{
		ftdialogs_out.remove(peer_id);
		qDebug() << "outgoing file transfer finished, removing dialog pointer\n";
	}
}

void FileTransferController::file_progress(QByteArray peer_id, quint64 bytes_sent, quint64 bytes_received)
{
	if (bytes_received > 0)
	{
		if (ftdialogs_in.contains(peer_id))
		{
			ftdialogs_in[peer_id]->update(bytes_received);
		}
	}
	else if (bytes_sent > 0)
	{
		if (ftdialogs_out.contains(peer_id))
		{
			ftdialogs_out[peer_id]->update(bytes_sent);
		}
	}
}

void FileTransferController::file_request(QByteArray peer_id, quint64 file_size, QString filename)
{
	FileTransferDialog *ftd = new FileTransferDialog(peer_id, filename, file_size, FileTransferDialog::RECEIVING);

	ftdialogs_in[peer_id] = ftd;
	fnames[peer_id] = filename;

	connect(ftd, SIGNAL(acceptedFile(QByteArray)), this, SLOT(file_accepted_by_me(QByteArray)));
	connect(ftd, SIGNAL(deniedFile(QByteArray)), this, SLOT(file_denied_by_me(QByteArray)));

	ftd->show();
}

void FileTransferController::file_accepted_by_peer(QByteArray peer_id)
{
	connect(skynet, SIGNAL(OutgoingFilePaused(QByteArray)), this, SLOT(file_paused_outgoing(QByteArray)));
	connect(skynet, SIGNAL(OutgoingFileResumed(QByteArray)), this, SLOT(file_resumed_outgoing(QByteArray)));
	connect(skynet, SIGNAL(OutgoingFileCancelled(QByteArray)), this, SLOT(file_cancelled_outgoing(QByteArray)));
	connect(skynet, SIGNAL(OutgoingFileIoError(QByteArray)), this, SLOT(file_ioerror_outgoing(QByteArray)));
	connect(skynet, SIGNAL(OutgoingFileFinished(QByteArray)), this, SLOT(file_finished_outgoing(QByteArray)));

	ftdialogs_out[peer_id]->setTransferStatus(FileTransferDialog::SENDINGFILE);
}

void FileTransferController::file_denied_by_peer(QByteArray peer_id)
{
	ftdialogs_out[peer_id]->setTransferStatus(FileTransferDialog::DENIED);
}

void FileTransferController::file_paused_incoming(QByteArray peer_id)
{
	ftdialogs_in[peer_id]->pausedByPeer();
}

void FileTransferController::file_resumed_incoming(QByteArray peer_id)
{
	ftdialogs_in[peer_id]->resumedByPeer();
}

void FileTransferController::file_cancelled_incoming(QByteArray peer_id)
{
	ftdialogs_in[peer_id]->cancelledByPeer();
	ftdialogs_in.remove(peer_id);
}

void FileTransferController::file_ioerror_incoming(QByteArray peer_id)
{
	ftdialogs_in[peer_id]->ioError();
	ftdialogs_in.remove(peer_id);
}

void FileTransferController::file_paused_outgoing(QByteArray peer_id)
{
	ftdialogs_out[peer_id]->pausedByPeer();
}

void FileTransferController::file_resumed_outgoing(QByteArray peer_id)
{
	ftdialogs_out[peer_id]->resumedByPeer();

}

void FileTransferController::file_cancelled_outgoing(QByteArray peer_id)
{
	ftdialogs_out[peer_id]->cancelledByPeer();
	ftdialogs_out.remove(peer_id);
}

void FileTransferController::file_ioerror_outgoing(QByteArray peer_id)
{
	ftdialogs_out[peer_id]->ioError();
	ftdialogs_out.remove(peer_id);
}

void FileTransferController::file_accepted_by_me(QByteArray peer_id)
{
	QString filePathToSave = QFileDialog::getSaveFileName(ftdialogs_in[peer_id], tr("Save received file to"),
	                         fnames[peer_id], tr("Any file (* *.*)"));

	skynet->AcceptFile(peer_id, filePathToSave);

	connect(ftdialogs_in[peer_id], SIGNAL(cancelledFile(QByteArray, FileTransferDialog*)), this, SLOT(file_cancelled_by_me(QByteArray, FileTransferDialog*)));
	connect(ftdialogs_in[peer_id], SIGNAL(pausedFile(QByteArray, FileTransferDialog*)), this, SLOT(file_paused_by_me(QByteArray, FileTransferDialog*)));
	connect(ftdialogs_in[peer_id], SIGNAL(resumedFile(QByteArray, FileTransferDialog*)), this, SLOT(file_resumed_by_me(QByteArray, FileTransferDialog*)));

	connect(skynet, SIGNAL(IncomingFilePaused(QByteArray)), this, SLOT(file_paused_incoming(QByteArray)));
	connect(skynet, SIGNAL(IncomingFileResumed(QByteArray)), this, SLOT(file_resumed_incoming(QByteArray)));
	connect(skynet, SIGNAL(IncomingFileCancelled(QByteArray)), this, SLOT(file_cancelled_incoming(QByteArray)));
	connect(skynet, SIGNAL(IncomingFileIoError(QByteArray)), this, SLOT(file_ioerror_incoming(QByteArray)));
	connect(skynet, SIGNAL(IncomingFileFinished(QByteArray)), this, SLOT(file_finished_incoming(QByteArray)));

	ftdialogs_in[peer_id]->setTransferStatus(FileTransferDialog::RECEIVINGFILE);
}

void FileTransferController::file_denied_by_me(QByteArray peer_id)
{
	skynet->DenyFile(peer_id);

	//delete ftdialogs_in[peer_id];

	ftdialogs_in.remove(peer_id);
}

void FileTransferController::file_cancelled_by_me(QByteArray peer_id, FileTransferDialog *from)
{
	bool sending = true;

	if (from->getTransferDirection() == FileTransferDialog::RECEIVING)
	{
		ftdialogs_in.remove(peer_id);

		sending = false;
	}
	else
	{
		ftdialogs_out.remove(peer_id);
	}

	skynet->CancelFile(peer_id, sending);
}

void FileTransferController::file_paused_by_me(QByteArray peer_id, FileTransferDialog *from)
{
	bool sending = true;

	if (from->getTransferDirection() == FileTransferDialog::RECEIVING)
		sending = false;

	skynet->PauseFile(peer_id, sending);
}

void FileTransferController::file_resumed_by_me(QByteArray peer_id, FileTransferDialog *from)
{
	bool sending = true;

	if (from->getTransferDirection() == FileTransferDialog::RECEIVING)
		sending = false;

	skynet->ResumeFile(peer_id, sending);
}

void FileTransferController::send_file(QByteArray peer_id)
{

	if (ftdialogs_out.contains(peer_id))
	{
		msguirouter->notification(tr("There is already file transfer in progress. Wait until it is finished."), true);
		return;
	}

	QString filepath = QFileDialog::getOpenFileName(0, tr("Select file to send"), QString(), tr("Any file (* *.*)"));

	if (filepath.isNull())
		return;

	QFileInfo fi(filepath);

	if (!fi.exists())
	{
		msguirouter->notification(tr("File \"%1\" does not exist.").arg(filepath), true);
		return;
	}

	if (fi.size() == 0)
	{
		msguirouter->notification(tr("Cannot send empty file."), true);
	}

	skynet->SendFile(peer_id, filepath);

	FileTransferDialog *ftd = new FileTransferDialog(peer_id, fi.fileName(), fi.size(), FileTransferDialog::SENDING);

	ftdialogs_out[peer_id] = ftd;

	connect(ftdialogs_out[peer_id], SIGNAL(cancelledFile(QByteArray, FileTransferDialog*)), this, SLOT(file_cancelled_by_me(QByteArray, FileTransferDialog*)));
	connect(ftdialogs_out[peer_id], SIGNAL(pausedFile(QByteArray, FileTransferDialog*)), this, SLOT(file_paused_by_me(QByteArray, FileTransferDialog*)));
	connect(ftdialogs_out[peer_id], SIGNAL(resumedFile(QByteArray, FileTransferDialog*)), this, SLOT(file_resumed_by_me(QByteArray, FileTransferDialog*)));

	ftd->show();
}