Kadu

/*

 * %kadu copyright begin%

 * Copyright 2009, 2009, 2010 Rafał Malinowski (rafal.przemyslaw.malinowski@gmail.com)

 * %kadu copyright end%

 *

 * 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 2 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 <QtCore/QFile>

#include <QtCore/QTimer>

#include "accounts/account.h"

#include "buddies/avatar.h"

#include "buddies/avatar-shared.h"

#include "configuration/configuration-manager.h"

#include "contacts/contact-manager.h"

#include "contacts/contact.h"

#include "misc/misc.h"

#include "protocols/protocol.h"

#include "protocols/services/avatar-service.h"

#include "debug.h"

#include "avatar-manager.h"

AvatarManager * AvatarManager::Instance = 0;

AvatarManager * AvatarManager::instance()

{

        if (!Instance)

                Instance = new AvatarManager();

        return Instance;

}

AvatarManager::AvatarManager()

{

        triggerAllAccountsRegistered();

        UpdateTimer = new QTimer(this);

        UpdateTimer->setInterval(5 * 60 * 1000); // 5 minutes

        connect(UpdateTimer, SIGNAL(timeout()), this, SLOT(updateAvatars()));

}

AvatarManager::~AvatarManager()

{

        triggerAllAccountsUnregistered();

}

void AvatarManager::itemAboutToBeAdded(Avatar item)

{

        emit avatarAboutToBeAdded(item);

}

void AvatarManager::itemAdded(Avatar item)

{

        emit avatarAdded(item);

}

void AvatarManager::itemAboutToBeRemoved(Avatar item)

{

        emit avatarAboutToBeRemoved(item);

}

void AvatarManager::itemRemoved(Avatar item)

{

        emit avatarRemoved(item);

}

AvatarService * AvatarManager::avatarService(Account account)

{

        Protocol *protocol = account.protocolHandler();

        if (!protocol)

                return 0;

        return protocol->avatarService();

}

AvatarService * AvatarManager::avatarService(Contact contact)

{

        Account account = contact.contactAccount();

        if (account.isNull())

                return 0;

        return avatarService(account);

}

void AvatarManager::accountRegistered(Account account)

{

        connect(account, SIGNAL(connected()), this, SLOT(updateAccountAvatars()));

        AvatarService *service = avatarService(account);

        if (!service)

                return;

        connect(service, SIGNAL(avatarFetched(Contact, const QByteArray &)),

                        this, SLOT(avatarFetched(Contact, const QByteArray &)));

}

void AvatarManager::accountUnregistered(Account account)

{

        disconnect(account, SIGNAL(connected()), this, SLOT(updateAccountAvatars()));

        AvatarService *service = avatarService(account);

        if (!service)

                return;

        disconnect(service, SIGNAL(avatarFetched(Contact, const QByteArray &)),

                           this, SLOT(avatarFetched(Contact, const QByteArray &)));

}

bool AvatarManager::needUpdate(Contact contact)

{

        if (!contact.contactAvatar())

                return true;

        QDateTime lastUpdated = contact.contactAvatar().lastUpdated();

        if (!lastUpdated.isValid())

                return true;

        // one hour passed

        if (lastUpdated.secsTo(QDateTime::currentDateTime()) > 60 * 60)

                return true;

        QDateTime nextUpdate = contact.contactAvatar().nextUpdate();

        if (nextUpdate > QDateTime::currentDateTime())

                return true;

        return false;

}

#include <stdio.h>

void AvatarManager::updateAvatar(Contact contact, bool force)

{

        if (!force && !needUpdate(contact))

                return;

        AvatarService *service = avatarService(contact);

        if (!service)

                return;

        service->fetchAvatar(contact);

}

void AvatarManager::avatarFetched(Contact contact, const QByteArray &data)

{

        Avatar avatar = contact.contactAvatar();

        if (!avatar)

        {

                avatar = Avatar::create();

                contact.setContactAvatar(avatar);

        }

        avatar.setLastUpdated(QDateTime::currentDateTime());

        QPixmap pixmap;

        if (!data.isEmpty())

                pixmap.loadFromData(data);

        avatar.setPixmap(pixmap);

        emit avatarUpdated(contact);

}

void AvatarManager::updateAvatars()

{

        foreach (Contact contact, ContactManager::instance()->items())

                if (!contact.ownerBuddy().isAnonymous())

                        updateAvatar(contact);

}

void AvatarManager::updateAccountAvatars()

{

        Account account(sender());

        if (!account)

                return;

        foreach (Contact contact, ContactManager::instance()->contacts(account))

                if (!contact.ownerBuddy().isAnonymous())

                        updateAvatar(contact, true);

}