messagecomposer/aliasesexpandjob.cpp

   1 /*
   2  * This file is part of KMail.
   3  *
   4  * Copyright (c) 2010 KDAB
   5  *
   6  * Author: Tobias Koenig <tokoe@kde.org>
   7  *
   8  * This program is free software; you can redistribute it and/or modify
   9  * it under the terms of the GNU General Public License as published by
  10  * the Free Software Foundation; either version 2 of the License, or
  11  * (at your option) any later version.
  12  *
  13  * This program is distributed in the hope that it will be useful,
  14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16  * GNU General Public License for more details.
  17  *
  18  * You should have received a copy of the GNU General Public License along
  19  * with this program; if not, write to the Free Software Foundation, Inc.,
  20  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
  21  */
  22
  23 #include "aliasesexpandjob.h"
  24 #include "aliasesexpandjob_p.h"
  25
  26 #include <akonadi/contact/contactgroupexpandjob.h>
  27 #include <akonadi/contact/contactgroupsearchjob.h>
  28 #include <akonadi/contact/contactsearchjob.h>
  29
  30 #include <KPIMUtils/Email>
  31
  32 #include <messagecore/stringutil.h>
  33
  34 using namespace MessageComposer;
  35
  36 DistributionListExpandJob::DistributionListExpandJob( const QString &name, QObject *parent )
  37   : KJob( parent ), mListName( name ), mIsEmpty( false )
  38 {
  39 }
  40
  41 DistributionListExpandJob::~DistributionListExpandJob()
  42 {
  43 }
  44
  45 void DistributionListExpandJob::start()
  46 {
  47   if ( mListName.isEmpty() ) {
  48     emitResult();
  49     return;
  50   }
  51
  52   Akonadi::ContactGroupSearchJob *job = new Akonadi::ContactGroupSearchJob( this );
  53   job->setQuery( Akonadi::ContactGroupSearchJob::Name, mListName );
  54   connect( job, SIGNAL(result(KJob*)), SLOT(slotSearchDone(KJob*)) );
  55 }
  56
  57 QString DistributionListExpandJob::addresses() const
  58 {
  59   return mEmailAddresses.join( QLatin1String( ", " ) );
  60 }
  61
  62 bool DistributionListExpandJob::isEmpty() const
  63 {
  64   return mIsEmpty;
  65 }
  66
  67 void DistributionListExpandJob::slotSearchDone( KJob *job )
  68 {
  69   if ( job->error() ) {
  70     setError( job->error() );
  71     setErrorText( job->errorText() );
  72     emitResult();
  73     return;
  74   }
  75
  76   const Akonadi::ContactGroupSearchJob *searchJob = qobject_cast<Akonadi::ContactGroupSearchJob*>( job );
  77
  78   const KABC::ContactGroup::List groups = searchJob->contactGroups();
  79   if ( groups.isEmpty() ) {
  80     emitResult();
  81     return;
  82   }
  83
  84   Akonadi::ContactGroupExpandJob *expandJob = new Akonadi::ContactGroupExpandJob( groups.first() );
  85   connect( expandJob, SIGNAL(result(KJob*)), SLOT(slotExpansionDone(KJob*)) );
  86   expandJob->start();
  87 }
  88
  89 void DistributionListExpandJob::slotExpansionDone( KJob *job )
  90 {
  91   if ( job->error() ) {
  92     setError( job->error() );
  93     setErrorText( job->errorText() );
  94     emitResult();
  95     return;
  96   }
  97
  98   const Akonadi::ContactGroupExpandJob *expandJob = qobject_cast<Akonadi::ContactGroupExpandJob*>( job );
  99
 100   const KABC::Addressee::List contacts = expandJob->contacts();
 101
 102   foreach ( const KABC::Addressee &contact, contacts )
 103     mEmailAddresses << contact.fullEmail();
 104
 105   mIsEmpty = mEmailAddresses.isEmpty();
 106
 107   emitResult();
 108 }
 109
 110
 111 AliasesExpandJob::AliasesExpandJob( const QString &recipients, const QString &defaultDomain, QObject *parent )
 112   : KJob( parent ),
 113     mRecipients( KPIMUtils::splitAddressList( recipients ) ),
 114     mDefaultDomain( defaultDomain ),
 115     mDistributionListExpansionJobs( 0 ),
 116     mNicknameExpansionJobs( 0 )
 117 {
 118 }
 119
 120 AliasesExpandJob::~AliasesExpandJob()
 121 {
 122 }
 123
 124 void AliasesExpandJob::start()
 125 {
 126   // At first we try to expand the recipient to a distribution list
 127   // or nick name and save the results in a map for later lookup
 128   foreach ( const QString &recipient, mRecipients ) {
 129     // check for distribution list
 130     DistributionListExpandJob *expandJob = new DistributionListExpandJob( recipient, this );
 131     expandJob->setProperty( "recipient", recipient );
 132     connect( expandJob, SIGNAL(result(KJob*)), SLOT(slotDistributionListExpansionDone(KJob*)) );
 133     mDistributionListExpansionJobs++;
 134     expandJob->start();
 135
 136     // check for nick name
 137     Akonadi::ContactSearchJob *searchJob = new Akonadi::ContactSearchJob( this );
 138     searchJob->setProperty( "recipient", recipient );
 139     searchJob->setQuery( Akonadi::ContactSearchJob::NickName, recipient.toLower() );
 140     connect( searchJob, SIGNAL(result(KJob*)), SLOT(slotNicknameExpansionDone(KJob*)) );
 141     mNicknameExpansionJobs++;
 142     searchJob->start();
 143   }
 144
 145   if ( mDistributionListExpansionJobs == 0 && mNicknameExpansionJobs == 0 )
 146     emitResult();
 147 }
 148
 149 QString AliasesExpandJob::addresses() const
 150 {
 151   return mEmailAddresses;
 152 }
 153
 154 QStringList AliasesExpandJob::emptyDistributionLists() const
 155 {
 156   return mEmptyDistributionLists;
 157 }
 158
 159 void AliasesExpandJob::slotDistributionListExpansionDone( KJob *job )
 160 {
 161   if ( job->error() ) {
 162     setError( job->error() );
 163     setErrorText( job->errorText() );
 164     emitResult();
 165     return;
 166   }
 167
 168   const DistributionListExpandJob *expandJob = qobject_cast<DistributionListExpandJob*>( job );
 169   const QString recipient = expandJob->property( "recipient" ).toString();
 170
 171   DistributionListExpansionResult result;
 172   result.addresses = expandJob->addresses();
 173   result.isEmpty = expandJob->isEmpty();
 174
 175   mDistListExpansionResults.insert( recipient, result );
 176
 177   mDistributionListExpansionJobs--;
 178   if ( mDistributionListExpansionJobs == 0 && mNicknameExpansionJobs == 0 )
 179     finishExpansion();
 180 }
 181
 182 void AliasesExpandJob::slotNicknameExpansionDone( KJob *job )
 183 {
 184   if ( job->error() ) {
 185     setError( job->error() );
 186     setErrorText( job->errorText() );
 187     emitResult();
 188     return;
 189   }
 190
 191   const Akonadi::ContactSearchJob *searchJob = qobject_cast<Akonadi::ContactSearchJob*>( job );
 192   const KABC::Addressee::List contacts = searchJob->contacts();
 193   const QString recipient = searchJob->property( "recipient" ).toString();
 194
 195   foreach ( const KABC::Addressee &contact, contacts ) {
 196     if ( contact.nickName().toLower() == recipient.toLower() ) {
 197       mNicknameExpansionResults.insert( recipient, contact.fullEmail() );
 198       break;
 199     }
 200   }
 201
 202   mNicknameExpansionJobs--;
 203   if ( mDistributionListExpansionJobs == 0 && mNicknameExpansionJobs == 0 )
 204     finishExpansion();
 205 }
 206
 207 void AliasesExpandJob::finishExpansion()
 208 {
 209   foreach ( const QString &recipient, mRecipients ) {
 210     if( recipient.isEmpty() )
 211       continue;
 212     if ( !mEmailAddresses.isEmpty() )
 213       mEmailAddresses += QLatin1String( ", " );
 214
 215     const QString receiver = recipient.trimmed();
 216
 217     // take prefetched expand distribution list results
 218     const DistributionListExpansionResult result = mDistListExpansionResults.value( recipient );
 219
 220     if ( result.isEmpty ) {
 221       mEmailAddresses += receiver;
 222       mEmptyDistributionLists << receiver;
 223       continue;
 224     }
 225
 226     if ( !result.addresses.isEmpty() ) {
 227       mEmailAddresses += result.addresses;
 228       continue;
 229     }
 230
 231     // take prefetched expand nick name results
 232     if ( !mNicknameExpansionResults.value( recipient ).isEmpty() ) {
 233       mEmailAddresses += mNicknameExpansionResults.value( recipient );
 234       continue;
 235     }
 236
 237     // check whether the address is missing the domain part
 238     QString displayName, addrSpec, comment;
 239     KPIMUtils::splitAddress( receiver, displayName, addrSpec, comment );
 240     if ( !addrSpec.contains( QLatin1Char('@') ) ) {
 241       if ( !mDefaultDomain.isEmpty() )
 242         mEmailAddresses += KPIMUtils::normalizedAddress( displayName, addrSpec + QLatin1Char( '@' ) +
 243                                                             mDefaultDomain, comment );
 244       else
 245         mEmailAddresses += MessageCore::StringUtil::guessEmailAddressFromLoginName( addrSpec );
 246     } else
 247       mEmailAddresses += receiver;
 248   }
 249
 250   emitResult();
 251 }
 252
 253 #include "aliasesexpandjob.moc"
 254 #include "aliasesexpandjob_p.moc"