transfermanager.cpp

/*
    This file is part of the Kapotah project.
    Copyright (C) 2010 Shantanu Tushar <shaan7in@gmail.com>
    Copyright (C) 2010 Sudhendu Kumar <sudhendu.kumar.roy@gmail.com>

    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 "transfermanager.h"
#include "transfer.h"
#include "incomingtransfer.h"
#include "outgoingtransfer.h"
#include <xml/transferxmlparser.h>
#include <messagedispatcher.h>
#include <QDateTime>

using namespace Kapotah;

template<> TransferManager *Kapotah::Singleton<TransferManager>::m_instance = 0;

TransferManager::TransferManager() : m_seqNo(0)
{
    connect(MessageDispatcher::instance(), SIGNAL(gotNewTransfer(QString,QHostAddress)),
            SLOT(handleIncomingTransfer(QString,QHostAddress)));
}

Transfer* TransferManager::addTransfer (Transfer::TransferType type, QList< TransferFile > files,
                                        QHostAddress peer, bool isSearchResponse)
{
    TransferXmlData data;
    data.files = files;
    return addTransfer(type, data, peer, isSearchResponse);
}

Transfer* TransferManager::addTransfer (Transfer::TransferType type, TransferXmlData data, QHostAddress peer,
    bool isSearchResponse)
{
    Transfer *transfer = 0;

    if (type == Transfer::Incoming) {
        transfer = new IncomingTransfer (data.files, data.totalSize, data.totalNumFiles, data.totalNumDirs,
                                         peer, data.id, isSearchResponse, this);
    } else if (type == Transfer::Outgoing) {
        transfer = new OutgoingTransfer (data.files, peer, isSearchResponse, this);
    }

    transfer->setItemNames(data.items);
    m_transfersList.append (transfer);

    connect (transfer, SIGNAL (done()), SLOT (onTransferFinished()));
    emit newTransferAdded (transfer);
    return transfer;
}

QString TransferManager::newId (QString path)
{
    QString id = QString::number(m_seqNo++) + QString::number (QDateTime::currentMSecsSinceEpoch());

    if (m_paths.contains (id)) {
        qFatal ("THIS IS NASTY, I MEAN IT, STOP _NOW_");
    }

    m_paths[id] = path;

    return id;
}

QString TransferManager::pathForId (QString id)
{
    return m_paths[id];
}

void TransferManager::handleIncomingTransfer (QString transfer, QHostAddress peer)
{
    TransferXmlParser parser;
    TransferXmlData *data = static_cast<TransferXmlData*>(parser.parseXml(transfer));
    Transfer *newTransfer = addTransfer (Transfer::Incoming, *data, peer, data->isSearchResponse);
    delete data;
}

void TransferManager::onTransferFinished()
{
    emit transferFinished(qobject_cast<Transfer*>(sender()));
}

void TransferManager::emitNewTransferAdded (Transfer* transfer)
{
    emit newTransferAdded(transfer);
}

#include "transfermanager.moc"