kopete/protocols/wlm/wlmtransfermanager.cpp

   1 /*
   2     wlmtransfermanager.cpp - Kopete Wlm Protocol
   3
   4     Copyright (c) 2008      by Tiago Salem Herrmann <tiagosh@gmail.com>
   5     Kopete    (c) 2002-2003 by the Kopete developers <kopete-devel@kde.org>
   6
   7     *************************************************************************
   8     *                                                                       *
   9     * This library is free software; you can redistribute it and/or         *
  10     * modify it under the terms of the GNU General Public                   *
  11     * License as published by the Free Software Foundation; either          *
  12     * version 2 of the License, or (at your option) any later version.      *
  13     *                                                                       *
  14     *************************************************************************
  15 */
  16
  17 #include <QObject>
  18 #include <kmessagebox.h>
  19 #include <kcodecs.h>
  20 #include <klocale.h>
  21
  22 #include <kdebug.h>
  23
  24 #include "wlmtransfermanager.h"
  25 #include "wlmcontact.h"
  26 #include "kopetecontact.h"
  27 #include "kopeteuiglobal.h"
  28
  29
  30 WlmTransferManager::WlmTransferManager (WlmAccount * account1) :
  31 m_account (account1),
  32 nextID (1)
  33 {
  34     QObject::connect (&account ()->server ()->cb,
  35                       SIGNAL (incomingFileTransfer
  36                               (MSN::SwitchboardServerConnection *,
  37                                const MSN::fileTransferInvite &)), this,
  38                       SLOT (incomingFileTransfer
  39                             (MSN::SwitchboardServerConnection *,
  40                              const MSN::fileTransferInvite &)));
  41
  42     QObject::connect (&account ()->server ()->cb,
  43                       SIGNAL (gotFileTransferProgress
  44                               (MSN::SwitchboardServerConnection *,
  45                                const unsigned int &,
  46                                const unsigned long long &)), this,
  47                       SLOT (gotFileTransferProgress
  48                             (MSN::SwitchboardServerConnection *,
  49                              const unsigned int &,
  50                              const unsigned long long &)));
  51
  52     QObject::connect (&account ()->server ()->cb,
  53                       SIGNAL (gotFileTransferFailed (MSN::SwitchboardServerConnection *,
  54                               const unsigned int &, const MSN::fileTransferError&)),
  55                       this,
  56                       SLOT (gotFileTransferFailed (MSN::SwitchboardServerConnection *,
  57                               const unsigned int &, const MSN::fileTransferError&)));
  58
  59     QObject::connect (&account ()->server ()->cb,
  60                       SIGNAL (gotFileTransferSucceeded
  61                               (MSN::SwitchboardServerConnection *,
  62                                const unsigned int &)), this,
  63                       SLOT (gotFileTransferSucceeded (MSN::SwitchboardServerConnection *,
  64                               const unsigned int &)));
  65
  66     QObject::connect (&account ()->server ()->cb,
  67                       SIGNAL (slotfileTransferInviteResponse
  68                               (MSN::SwitchboardServerConnection *,
  69                                const unsigned int &, const bool &)), this,
  70                       SLOT (fileTransferInviteResponse
  71                             (MSN::SwitchboardServerConnection * ,
  72                              const unsigned int &, const bool &)));
  73
  74     connect (Kopete::TransferManager::transferManager (),
  75              SIGNAL (accepted (Kopete::Transfer *, const QString &)),
  76              this, SLOT (slotAccepted (Kopete::Transfer *, const QString &)));
  77     connect (Kopete::TransferManager::transferManager (),
  78              SIGNAL (refused (const Kopete::FileTransferInfo &)),
  79              this, SLOT (slotRefused (const Kopete::FileTransferInfo &)));
  80 }
  81
  82 void
  83 WlmTransferManager::fileTransferInviteResponse (MSN::SwitchboardServerConnection * conn,
  84                                                 const unsigned int &sessionID,
  85                                                 const bool & response)
  86 {
  87     if(!transferSessions.count(sessionID))
  88         return;
  89
  90     if (response)
  91     {
  92         transferSessionData tfd = transferSessions[sessionID];
  93         Kopete::ContactPtrList chatmembers;
  94         chatmembers.append (account ()->contacts ()[tfd.to]);
  95         WlmChatSession *_manager =
  96             qobject_cast <WlmChatSession *>(Kopete::ChatSessionManager::self ()->
  97               findChatSession (account ()->myself (), chatmembers,
  98                                account ()->protocol ()));
  99         if (!_manager)
 100         {
 101             _manager =
 102                 new WlmChatSession (account ()->protocol (),
 103                                     account ()->myself (), chatmembers);
 104         }
 105     }
 106     else
 107     {
 108         transferSessionData tfd = transferSessions[sessionID];
 109         if(tfd.internalID)
 110             Kopete::TransferManager::transferManager()->
 111                 cancelIncomingTransfer(tfd.internalID);
 112         else
 113             tfd.ft->slotError(KIO::ERR_ABORTED, i18n("File transfer cancelled."));
 114     }
 115 }
 116
 117 WlmTransferManager::~WlmTransferManager ()
 118 {
 119 }
 120
 121 void
 122 WlmTransferManager::incomingFileTransfer (MSN::SwitchboardServerConnection * conn,
 123                                           const MSN::fileTransferInvite & ft)
 124 {
 125     Kopete::Contact * contact =
 126         account ()->contacts ()[ft.userPassport.c_str ()];
 127
 128     if(!contact)
 129             return;
 130
 131     if (ft.type == MSN::FILE_TRANSFER_WITH_PREVIEW
 132         || ft.type == MSN::FILE_TRANSFER_WITHOUT_PREVIEW)
 133     {
 134         QPixmap preview;
 135         if (ft.type == MSN::FILE_TRANSFER_WITH_PREVIEW)
 136         {
 137             preview.
 138                 loadFromData (KCodecs::
 139                               base64Decode (QString (ft.preview.c_str ()).
 140                                             toAscii ()));
 141         }
 142         transferSessionData tsd;
 143         tsd.from = ft.userPassport.c_str ();
 144         tsd.to = account ()->myself ()->contactId ();
 145         tsd.ft = NULL;
 146         tsd.internalID = 0;
 147         account ()->chatManager ()->createChat (conn);
 148         WlmChatSession *chat = account ()->chatManager ()->chatSessions[conn];
 149         if(chat)
 150             chat->setCanBeDeleted (false);
 151
 152         tsd.internalID =
 153             Kopete::TransferManager::transferManager ()->
 154             askIncomingTransfer (contact,
 155                                  ft.filename.c_str (),
 156                                  ft.filesize,
 157                                  "",
 158                                  QString::number (ft.sessionId),
 159                                  preview);
 160         transferSessions[ft.sessionId] = tsd;
 161     }
 162 }
 163
 164 void
 165 WlmTransferManager::gotFileTransferProgress (MSN::SwitchboardServerConnection * conn,
 166                                              const unsigned int &sessionID,
 167                                              const unsigned long long
 168                                              &transferred)
 169 {
 170     if(!transferSessions.count(sessionID))
 171         return;
 172
 173     Kopete::Transfer * transfer = transferSessions[sessionID].ft;
 174     if (transfer)
 175         transfer->slotProcessed (transferred);
 176 }
 177
 178 void
 179 WlmTransferManager::slotAccepted (Kopete::Transfer * ft,
 180                                   const QString & filename)
 181 {
 182     Kopete::ContactPtrList chatmembers;
 183
 184     // grab contactId from the sender
 185     unsigned int sessionID = ft->info ().internalId ().toUInt ();
 186
 187     if(!transferSessions.count(sessionID))
 188         return;
 189
 190     QString from = transferSessions[sessionID].from;
 191
 192     if (from.isEmpty ())
 193         return;
 194
 195     // find an existent session, or create a new one
 196     chatmembers.append (account ()->contacts ()[from]);
 197     WlmChatSession *_manager = qobject_cast <WlmChatSession *>
 198         (Kopete::ChatSessionManager::self ()->
 199           findChatSession (account ()->myself (), chatmembers,
 200                            account ()->protocol ()));
 201
 202     if (!_manager)
 203     {
 204         _manager =
 205             new WlmChatSession (account ()->protocol (),
 206                                 account ()->myself (), chatmembers);
 207     }
 208
 209     MSN::SwitchboardServerConnection * conn = _manager->getChatService ();
 210     if (!conn)
 211         return;
 212
 213     _manager->setCanBeDeleted (false);
 214     transferSessions[sessionID].ft = ft;
 215
 216     connect (ft, SIGNAL (transferCanceled ()), this, SLOT (slotCanceled ()));
 217
 218     conn->fileTransferResponse (sessionID, filename.toLatin1 ().data (),
 219                                 true);
 220 }
 221
 222 void
 223 WlmTransferManager::slotRefused (const Kopete::FileTransferInfo & fti)
 224 {
 225     Kopete::ContactPtrList chatmembers;
 226     chatmembers.append (fti.contact ());
 227     WlmChatSession *_manager = qobject_cast <WlmChatSession *>
 228         (Kopete::ChatSessionManager::self ()->
 229           findChatSession (account ()->myself (), chatmembers,
 230                            account ()->protocol ()));
 231
 232     if (!_manager)
 233         return;
 234
 235     MSN::SwitchboardServerConnection * conn = _manager->getChatService ();
 236     if (!conn)
 237         return;
 238
 239     conn->fileTransferResponse (fti.internalId ().toUInt (), "", false);
 240 }
 241
 242 void
 243 WlmTransferManager::gotFileTransferFailed (MSN::SwitchboardServerConnection * conn,
 244                                             const unsigned int &sessionID,
 245                                             const MSN::fileTransferError & error)
 246 {
 247     if(!transferSessions.count(sessionID))
 248         return;
 249
 250     transferSessionData tsd = transferSessions[sessionID];
 251     if (tsd.internalID)
 252     {
 253         Kopete::TransferManager::transferManager ()->
 254             cancelIncomingTransfer(tsd.internalID);
 255         if(tsd.ft)
 256             tsd.ft->slotError(KIO::ERR_ABORTED, i18n("File transfer cancelled."));
 257     }
 258     else
 259     {
 260         if (tsd.ft)
 261         {
 262             tsd.ft->slotError(KIO::ERR_ABORTED, i18n("File transfer cancelled."));
 263         }
 264     }
 265     transferSessions.remove (sessionID);
 266
 267 }
 268
 269 void
 270 WlmTransferManager::gotFileTransferSucceeded (MSN::SwitchboardServerConnection * conn,
 271                                             const unsigned int &sessionID)
 272 {
 273     Kopete::Transfer * transfer = transferSessions[sessionID].ft;
 274     if (transfer)
 275     {
 276         Kopete::ContactPtrList chatmembers;
 277         if (transfer->info ().direction () ==
 278             Kopete::FileTransferInfo::Incoming)
 279             chatmembers.append (account ()->
 280                                 contacts ()[transferSessions[sessionID].
 281                                             from]);
 282         else
 283             chatmembers.append (account ()->
 284                                 contacts ()[transferSessions[sessionID].to]);
 285
 286         WlmChatSession *_manager = qobject_cast <WlmChatSession *>
 287             (Kopete::ChatSessionManager::self ()->
 288               findChatSession (account ()->myself (), chatmembers,
 289                                account ()->protocol ()));
 290         if (_manager)
 291             _manager->raiseView ();
 292         transfer->slotComplete ();
 293         transferSessions.remove (sessionID);
 294     }
 295 }
 296
 297 void
 298 WlmTransferManager::slotCanceled ()
 299 {
 300     kDebug (14210) << k_funcinfo;
 301     Kopete::Transfer * ft = qobject_cast < Kopete::Transfer * >(sender ());
 302     if (!ft)
 303         return;
 304     unsigned int sessionID = 0;
 305     QMap < unsigned int, transferSessionData >::iterator i =
 306         transferSessions.begin ();
 307     for (; i != transferSessions.end (); ++i)
 308         if (i.value ().ft == ft)
 309             sessionID = i.key ();
 310
 311     if (!sessionID)
 312         return;
 313
 314     transferSessionData session = transferSessions[sessionID];
 315
 316     Kopete::ContactPtrList chatmembers;
 317     if (ft->info ().direction () == Kopete::FileTransferInfo::Incoming)
 318         chatmembers.append (account ()->contacts ()[session.from]);
 319     else
 320         chatmembers.append (account ()->contacts ()[session.to]);
 321
 322     WlmChatSession *_manager = qobject_cast <WlmChatSession *>
 323         (Kopete::ChatSessionManager::self ()->
 324           findChatSession (account ()->myself (), chatmembers,
 325                            account ()->protocol ()));
 326
 327     if (!_manager)
 328         return;
 329     _manager->raiseView ();
 330
 331     MSN::SwitchboardServerConnection * conn = _manager->getChatService ();
 332
 333     if (!conn)
 334         return;
 335
 336     if (sessionID)
 337     {
 338         transferSessions.remove (sessionID);
 339         conn->cancelFileTransfer (sessionID);
 340     }
 341 }
 342
 343 #include "wlmtransfermanager.moc"