sipjreen: Add AvatarManager class

src/sip/jreen/CMakeLists.txt

@@ -11,6 +11,7 @@ set( jabberSources
     jabber_p.cpp
     tomahawksipmessage.cpp
     tomahawksipmessagefactory.cpp
+    avatarmanager.cpp
 )
 
 set( jabberHeaders
@@ -18,6 +19,7 @@ set( jabberHeaders
     jabber_p.h
     tomahawksipmessage.h
     tomahawksipmessagefactory.h
+    avatarmanager.h
 )
 
 include_directories( . ${CMAKE_CURRENT_BINARY_DIR} ..

src/sip/jreen/avatarmanager.cpp

@@ -0,0 +1,171 @@
+#include "avatarmanager.h"
+
+
+#include "utils/tomahawkutils.h"
+
+#include <jreen/vcard.h>
+#include <jreen/vcardupdate.h>
+#include <jreen/presence.h>
+
+#include <QDir>
+#include <QDebug>
+#include <QCryptographicHash>
+#include <QPixmap>
+
+AvatarManager::AvatarManager(Jreen::Client *client) :
+    m_cacheDir(TomahawkUtils::appDataDir().absolutePath().append("/jreen/"))
+{
+    m_client = client;
+
+    m_cachedAvatars = m_cacheDir.entryList();
+
+    connect(m_client, SIGNAL(serverFeaturesReceived(QSet<QString>)), SLOT(onNewConnection()));
+    connect(m_client, SIGNAL(newPresence(Jreen::Presence)), SLOT(onNewPresence(Jreen::Presence)));
+    connect(m_client, SIGNAL(newIQ(Jreen::IQ)), SLOT(onNewIq(Jreen::IQ)));
+
+    connect(this, SIGNAL(newAvatar(QString)), SLOT(onNewAvatar(QString)));
+}
+
+AvatarManager::~AvatarManager()
+{
+}
+
+void AvatarManager::onNewConnection()
+{
+    fetchVCard( m_client->jid().bare() );
+}
+
+
+void AvatarManager::fetchVCard(const QString &jid)
+{
+    qDebug() << Q_FUNC_INFO;
+
+    Jreen::IQ iq(Jreen::IQ::Get, jid );
+    iq.addExtension(new Jreen::VCard());
+    m_client->send( iq, this, SLOT( onNewIq( Jreen::IQ, int ) ), 0 );
+}
+
+void AvatarManager::onNewPresence(const Jreen::Presence& presence)
+{
+    Jreen::VCardUpdate::Ptr update = presence.findExtension<Jreen::VCardUpdate>();
+    if(update)
+    {
+        qDebug() << "vcard: found update for " << presence.from().full();
+        if(!isCached(update->photoHash()))
+        {
+            qDebug() << presence.from().full() << "vcard: photo not cached, starting request..." << update->photoHash();
+            fetchVCard( presence.from().bare() );
+        }
+        else
+        {
+            qDebug() << presence.from().full() << "vcard: photo already cached no request necessary " << update->photoHash();
+            m_JidsAvatarHashes.insert( update->photoHash(), presence.from().bare() );
+
+            Q_ASSERT(!this->avatar(presence.from().bare()).isNull());
+            emit newAvatar(presence.from().bare());
+        }
+    }
+    else
+    {
+        qDebug() << Q_FUNC_INFO << presence.from().full() << "got no statusupdateextension";
+
+        //TODO: do we want this? might fetch avatars for broken clients
+        fetchVCard( presence.from().bare() );
+    }
+}
+
+void AvatarManager::onNewIq(const Jreen::IQ& iq, int context)
+{
+    Jreen::VCard *vcard = iq.findExtension<Jreen::VCard>().data();
+    if(vcard)
+    {
+        iq.accept();
+
+        qDebug() << Q_FUNC_INFO << "Got vcard from " << iq.from().full();
+
+        QString id = iq.from().full();
+        QString avatarHash;
+
+        const Jreen::VCard::Photo &photo = vcard->photo();
+        if (!photo.data().isEmpty()) {
+            qDebug() << "vcard: got photo data" << id;
+
+            avatarHash = QCryptographicHash::hash(photo.data(), QCryptographicHash::Sha1).toHex();
+
+            if (!m_cacheDir.exists())
+                m_cacheDir.mkpath( avatarDir( avatarHash ).absolutePath() );
+
+
+            QFile file(avatarPath(avatarHash));
+            if (file.open(QIODevice::WriteOnly)) {
+                file.write(photo.data());
+                file.close();
+            }
+
+            m_cachedAvatars.append(avatarHash);
+            m_JidsAvatarHashes.insert( avatarHash, iq.from().bare() );
+
+            Q_ASSERT(!this->avatar(iq.from().bare()).isNull());
+            emit newAvatar(iq.from().bare());
+        }
+        else
+        {
+            qDebug() << "vcard: got no photo data" << id;
+        }
+
+        // got own presence
+        if ( m_client->jid().bare() == id )
+        {
+            qDebug() << Q_FUNC_INFO << "got own vcard";
+
+            Jreen::Presence presence = m_client->presence();
+            Jreen::VCardUpdate::Ptr update = presence.findExtension<Jreen::VCardUpdate>();
+            if (update->photoHash() != avatarHash)
+            {
+                qDebug() << Q_FUNC_INFO << "Updating own presence...";
+
+                update->setPhotoHash(avatarHash);
+                m_client->send(presence);
+            }
+        }
+    }
+}
+
+QPixmap AvatarManager::avatar(const QString &jid) const
+{
+    if( isCached( avatarHash( jid ) ) )
+    {
+        return QPixmap( avatarPath( avatarHash( jid ) ) );
+    }
+    else
+    {
+        return QPixmap();
+    }
+}
+
+QString AvatarManager::avatarHash(const QString &jid) const
+{
+    //qDebug() << Q_FUNC_INFO << jid << m_JidsAvatarHashes.key(jid);
+    return m_JidsAvatarHashes.key(jid);
+}
+
+QDir AvatarManager::avatarDir(const QString &avatarHash) const
+{
+    return m_cacheDir;
+}
+
+QString AvatarManager::avatarPath(const QString &avatarHash) const
+{
+    Q_ASSERT(!avatarHash.contains("@"));
+    return avatarDir(avatarHash).absoluteFilePath(avatarHash);
+}
+
+bool AvatarManager::isCached(const QString &avatarHash) const
+{
+    return m_cachedAvatars.contains( avatarHash );
+}
+
+void AvatarManager::onNewAvatar(const QString& jid)
+{
+    qDebug() << Q_FUNC_INFO <<  "Found new Avatar..." << jid;
+}

src/sip/jreen/avatarmanager.h

@@ -0,0 +1,43 @@
+#ifndef AVATARMANAGER_H
+#define AVATARMANAGER_H
+
+#include <jreen/client.h>
+
+#include <QObject>
+#include <QDir>
+
+
+class AvatarManager : public QObject
+{
+Q_OBJECT
+
+public:
+    AvatarManager(Jreen::Client *client);
+    virtual ~AvatarManager();
+
+    QPixmap avatar(const QString &jid) const;
+
+signals:
+    void newAvatar( const QString &jid );
+
+private slots:
+    void onNewPresence( const Jreen::Presence& presence );
+    void onNewIq(const Jreen::IQ &iq, int context = 0 );
+    void onNewConnection();
+    void onNewAvatar( const QString &jid );
+
+private:
+    void fetchVCard( const QString &jid);
+    QString avatarHash(const QString &jid) const;
+    QString avatarPath(const QString &avatarHash) const;
+
+    QDir avatarDir(const QString &avatarHash) const;
+    bool isCached(const QString &avatarHash) const;
+
+    Jreen::Client *m_client;
+    QStringList m_cachedAvatars;
+    QDir m_cacheDir;
+    QMap<QString, QString> m_JidsAvatarHashes;
+};
+
+#endif // AVATARMANAGER_H