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