mailimporter/filter_kmail_maildir.cpp

   1 /***************************************************************************
   2            filter_kmail_maildir.cxx  -  Kmail maildir mail import
   3                             -------------------
   4    begin                : April 06 2005
   5    copyright            : (C) 2005 by Danny Kukawka
   6    email                : danny.kukawka@web.de
   7 ***************************************************************************/
   8
   9 /***************************************************************************
  10  *                                                                         *
  11  *   This program is free software; you can redistribute it and/or modify  *
  12  *   it under the terms of the GNU General Public License as published by  *
  13  *   the Free Software Foundation; either version 2 of the License, or     *
  14  *   (at your option) any later version.                                   *
  15  *                                                                         *
  16  ***************************************************************************/
  17 /* Copyright (c) 2012 Montel Laurent <montel@kde.org>                      */
  18
  19
  20 #include "filter_kmail_maildir.h"
  21
  22 #include <klocale.h>
  23 #include <kfiledialog.h>
  24 #include <QPointer>
  25
  26 using namespace MailImporter;
  27
  28 /** Default constructor. */
  29 FilterKMail_maildir::FilterKMail_maildir() :
  30     Filter( i18n( "Import KMail Maildirs and Folder Structure" ),
  31             "Danny Kukawka",
  32             i18n( "<p><b>KMail import filter</b></p>"
  33                   "<p>Select the base directory of the KMail mailfolder you want to import.</p>"
  34                   "<p><b>Note:</b> Never select your current local KMail maildir (usually "
  35                   "~/Mail or ~/.kde/share/apps/kmail/mail ): in this case, KMailCVT may become stuck "
  36                   "in a continuous loop. </p>"
  37                   "<p>This filter does not import KMail mailfolders with mbox files.</p>"
  38                   "<p>Since it is possible to recreate the folder structure, the folders "
  39                   "will be stored under: \"KMail-Import\" in your local folder.</p>" ) )
  40 {
  41 }
  42
  43 /** Destructor. */
  44 FilterKMail_maildir::~FilterKMail_maildir()
  45 {
  46 }
  47
  48 /** Recursive import of KMail maildir. */
  49 void FilterKMail_maildir::import()
  50 {
  51     setCountDuplicates(0);
  52     const QString homeDir = QDir::homePath();
  53
  54     QPointer<KFileDialog> kfd = new KFileDialog( homeDir, "", 0 );
  55     kfd->setMode( KFile::Directory | KFile::LocalOnly );
  56     if (kfd->exec()) {
  57         const QString maildir = kfd->selectedFile();
  58         importMails( maildir );
  59     }
  60     delete kfd;
  61 }
  62
  63
  64 void FilterKMail_maildir::processDirectory( const QString &path)
  65 {
  66     QDir dir(path);
  67     const QStringList rootSubDirs = dir.entryList(QStringList("*"), QDir::Dirs | QDir::Hidden, QDir::Name);
  68     QStringList::ConstIterator end = rootSubDirs.constEnd();
  69     for (QStringList::ConstIterator filename = rootSubDirs.constBegin() ; filename != end ; ++filename ) {
  70         if(filterInfo()->shouldTerminate())
  71             break;
  72         if(!(*filename == QLatin1String( "." ) || *filename == QLatin1String( ".." ))) {
  73             filterInfo()->setCurrent(0);
  74             importDirContents(dir.filePath(*filename));
  75             filterInfo()->setOverall((int) ((float) mImportDirDone / mTotalDir * 100));
  76             filterInfo()->setCurrent(100);
  77             mImportDirDone++;
  78         }
  79     }
  80 }
  81
  82 void FilterKMail_maildir::importMails( const QString &maildir )
  83 {
  84     setMailDir(maildir);
  85     if ( mailDir().isEmpty() ) {
  86         filterInfo()->alert( i18n( "No directory selected." ) );
  87         return;
  88     }
  89     /**
  90    * If the user only select homedir no import needed because
  91    * there should be no files and we surely import wrong files.
  92    */
  93     else if ( mailDir() == QDir::homePath() || mailDir() == ( QDir::homePath() + '/' ) ) {
  94         filterInfo()->addErrorLogEntry( i18n( "No files found for import." ) );
  95     } else {
  96         filterInfo()->setOverall(0);
  97         mImportDirDone = 0;
  98
  99         /** Recursive import of the MailArchives */
 100         QDir dir(mailDir());
 101         mTotalDir = Filter::countDirectory( dir, true /*search hidden directory*/ );
 102         processDirectory( mailDir());
 103
 104         filterInfo()->addInfoLogEntry( i18n("Finished importing emails from %1", mailDir() ));
 105         if (countDuplicates() > 0) {
 106             filterInfo()->addInfoLogEntry( i18np("1 duplicate message not imported", "%1 duplicate messages not imported", countDuplicates()));
 107         }
 108     }
 109     if (filterInfo()->shouldTerminate())
 110         filterInfo()->addInfoLogEntry( i18n("Finished import, canceled by user."));
 111     filterInfo()->setCurrent(100);
 112     filterInfo()->setOverall(100);
 113 }
 114
 115 /**
 116  * Import of a directory contents.
 117  * @param info Information storage for the operation.
 118  * @param dirName The name of the directory to import.
 119  */
 120 void FilterKMail_maildir::importDirContents( const QString &dirName)
 121 {
 122
 123     /** Here Import all archives in the current dir */
 124     importFiles(dirName);
 125
 126     /** If there are subfolders, we import them one by one */
 127
 128     processDirectory( dirName );
 129 }
 130
 131 /**
 132  * Import the files within a Folder.
 133  * @param info Information storage for the operation.
 134  * @param dirName The name of the directory to import.
 135  */
 136 void FilterKMail_maildir::importFiles( const QString &dirName)
 137 {
 138
 139     QDir dir(dirName);
 140     QString _path;
 141     bool generatedPath = false;
 142
 143     QDir importDir (dirName);
 144     const QStringList files = importDir.entryList(QStringList("[^\\.]*"), QDir::Files, QDir::Name);
 145     int currentFile = 1, numFiles = files.size();
 146     QStringList::ConstIterator filesEnd( files.constEnd() );
 147
 148     for ( QStringList::ConstIterator mailFile = files.constBegin(); mailFile != filesEnd; ++mailFile, ++currentFile) {
 149         if(filterInfo()->shouldTerminate()) {
 150             return;
 151         }
 152
 153         QString temp_mailfile = *mailFile;
 154         if (!(temp_mailfile.endsWith(QLatin1String(".index")) || temp_mailfile.endsWith(QLatin1String(".index.ids")) ||
 155               temp_mailfile.endsWith(QLatin1String(".index.sorted")) || temp_mailfile.endsWith(QLatin1String(".uidcache")) )) {
 156             if(!generatedPath) {
 157                 _path = "KMail-Import";
 158                 QString _tmp = dir.filePath(*mailFile);
 159                 _tmp = _tmp.remove( mailDir(), Qt::CaseSensitive );
 160                 const QStringList subFList = _tmp.split( QLatin1Char('/'), QString::SkipEmptyParts );
 161                 QStringList::ConstIterator end( subFList.end() );
 162                 for ( QStringList::ConstIterator it = subFList.constBegin(); it != end; ++it ) {
 163                     QString _cat = *it;
 164                     if(!(_cat == *mailFile)) {
 165                         if(_cat.startsWith('.') && _cat.endsWith(".directory")) {
 166                             _cat.remove(0,1);
 167                             _cat.remove((_cat.length() - 10), 10);
 168                         } else if (_cat.startsWith('.')) {
 169                             _cat = _cat.remove(0 , 1);
 170                         }
 171                         _path += '/' + _cat;
 172                     }
 173                 }
 174                 if(_path.endsWith("cur"))
 175                     _path.remove(_path.length() - 4 , 4);
 176                 QString _info = _path;
 177                 filterInfo()->addInfoLogEntry(i18n("Import folder %1...", _info.remove(0,12)));
 178                 filterInfo()->setFrom(_info);
 179                 filterInfo()->setTo(_path);
 180                 generatedPath = true;
 181             }
 182
 183             if(filterInfo()->removeDupMessage()) {
 184                 if(! addMessage( _path, dir.filePath(*mailFile) )) {
 185                     filterInfo()->addErrorLogEntry( i18n( "Could not import %1, duplicated message", *mailFile ) );
 186                 }
 187                 filterInfo()->setCurrent((int) ((float) currentFile / numFiles * 100));
 188             } else {
 189                 if(! addMessage_fastImport( _path, dir.filePath(*mailFile) )) {
 190                     filterInfo()->addErrorLogEntry( i18n("Could not import %1", *mailFile ) );
 191                 }
 192                 filterInfo()->setCurrent((int) ((float) currentFile / numFiles * 100));
 193             }
 194         }
 195     }
 196 }
 197