/***************************************************************************
           filter_kmail_maildir.cpp  -  Kmail maildir mail import
                            -------------------
   begin                : April 06 2005
   copyright            : (C) 2005 by Danny Kukawka
   email                : danny.kukawka@web.de
***************************************************************************/

/***************************************************************************
 *                                                                         *
 *   This program is free software; you can redistribute it and/or modify  *
 *   it under the terms of the GNU General Public License as published by  *
 *   the Free Software Foundation; either version 2 of the License, or     *
 *   (at your option) any later version.                                   *
 *                                                                         *
 ***************************************************************************/

#include "filter_kmail_maildir.h"

#include <config.h>
#include <tdelocale.h>
#include <tdefiledialog.h>

/** Default constructor. */
FilterKMail_maildir::FilterKMail_maildir( void ) :
        Filter( i18n( "Import KMail Maildirs and Folder Structure" ),
                "Danny Kukawka",
                i18n( "<p><b>KMail import filter</b></p>"
                      "<p>Select the base directory of the KMail mailfolder you want to import.</p>"
                      "<p><b>Note:</b> Never select your current local KMail maildir (usually "
                      "~/Mail or ~/.trinity/share/apps/kmail/mail ): in this case, KMailCVT may become stuck "
                      "in a continuous loop. </p>"
                      "<p>This filter does not import KMail mailfolders with mbox files.</p>"
                      "<p>Since it is possible to recreate the folder structure, the folders "
                      "will be stored under: \"KMail-Import\" in your local folder.</p>" ) )
{}

/** Destructor. */
FilterKMail_maildir::~FilterKMail_maildir( void )
{
}

/** Recursive import of KMail maildir. */
void FilterKMail_maildir::import( FilterInfo *info )
{

    TQString _homeDir = TQDir::homeDirPath();

    KFileDialog *kfd;
    kfd = new KFileDialog( _homeDir, "", 0, "tdefiledialog", true );
    kfd->setMode( KFile::Directory | KFile::LocalOnly );
    kfd->exec();
    mailDir = kfd->selectedFile();
    delete kfd;

    if ( mailDir.isEmpty() ) {
        info->alert( i18n( "No directory selected." ) );
    }
    /**
     * If the user only select homedir no import needed because 
     * there should be no files and we surely import wrong files.
     */
    else if ( mailDir == TQDir::homeDirPath() || mailDir == ( TQDir::homeDirPath() + "/" ) ) {
        info->addLog( i18n( "No files found for import." ) );
    } else {
        info->setOverall(0);

        /** Recursive import of the MailArchives */
        TQDir dir(mailDir);
        TQStringList rootSubDirs = dir.entryList("*", TQDir::Dirs | TQDir::Hidden, TQDir::Name);
        int currentDir = 1, numSubDirs = rootSubDirs.size();
        for(TQStringList::Iterator filename = rootSubDirs.begin() ; filename != rootSubDirs.end() ; ++filename, ++currentDir) {
            if(info->shouldTerminate()) break;
            if(!(*filename == "." || *filename == "..")) {
                info->setCurrent(0);
                importDirContents(info, dir.filePath(*filename));
                info->setOverall((int) ((float) currentDir / numSubDirs * 100));
                info->setCurrent(100);
            }
        }
    }

    info->addLog( i18n("Finished importing emails from %1").arg( mailDir ));
    if (count_duplicates > 0) {
        info->addLog( i18n("1 duplicate message not imported", "%n duplicate messages not imported", count_duplicates));
    }
    if (info->shouldTerminate()) info->addLog( i18n("Finished import, canceled by user."));
    count_duplicates = 0;
    info->setCurrent(100);
    info->setOverall(100);
}

/**
 * Import of a directory contents.
 * @param info Information storage for the operation.
 * @param dirName The name of the directory to import.
 */
void FilterKMail_maildir::importDirContents( FilterInfo *info, const TQString& dirName)
{

    /** Here Import all archives in the current dir */
    importFiles(info, dirName);

    /** If there are subfolders, we import them one by one */

    TQDir subfolders(dirName);
    TQStringList subDirs = subfolders.entryList("*", TQDir::Dirs | TQDir::Hidden, TQDir::Name);
    for(TQStringList::Iterator filename = subDirs.begin() ; filename != subDirs.end() ; ++filename) {
        if(info->shouldTerminate()) return;
        if(!(*filename == "." || *filename == "..")) {
            importDirContents(info, subfolders.filePath(*filename));
        }
    }
}


/**
 * Extract the X-Status flag from a mailfile 
 * @param info Information storage for the operation.
 * @param fileName The full path to the file to import
 */
TQString FilterKMail_maildir::getMessageFlags( FilterInfo *info, const TQString& fileName ) {
	
	TQString status_flag = "";
       
	TQFile mailfile( fileName );
        if (! mailfile.open( IO_ReadOnly ) ) {
            info->alert( i18n("Unable to open %1, skipping").arg( fileName ) );
	    return status_flag;
        } else {

            TQByteArray input(MAX_LINE);

            while ( !mailfile.atEnd() && status_flag.isEmpty()) {
                TQCString seperate;

                while ( ! mailfile.atEnd() && mailfile.readLine(input.data(),MAX_LINE) ) {

			if ((seperate = input.data()).left(10) == "X-Status: ") {
				status_flag = seperate;
				status_flag.remove("X-Status: ");
				status_flag = status_flag.stripWhiteSpace();
				// tqDebug("status_flag: %s", status_flag.latin1() );
				break;		
			}
		}
	    }
	    mailfile.close();
	}
	return status_flag;
}


/**
 * Import the files within a Folder.
 * @param info Information storage for the operation.
 * @param dirName The name of the directory to import.
 */
void FilterKMail_maildir::importFiles( FilterInfo *info, const TQString& dirName)
{

    TQDir dir(dirName);
    TQString _path;
    bool generatedPath = false;

    TQDir importDir (dirName);
    TQStringList files = importDir.entryList("[^\\.]*", TQDir::Files, TQDir::Name);
    int currentFile = 1, numFiles = files.size();
    for ( TQStringList::Iterator mailFile = files.begin(); mailFile != files.end(); ++mailFile, ++currentFile) {
        if(info->shouldTerminate()) return;

        TQString temp_mailfile = *mailFile;

        if (!(temp_mailfile.endsWith(".index") || temp_mailfile.endsWith(".index.ids") ||
                temp_mailfile.endsWith(".index.sorted") || temp_mailfile.endsWith(".uidcache") )) {
            if(!generatedPath) {
                _path = "KMail-Import";
                TQString _tmp = dir.filePath(*mailFile);
                _tmp = _tmp.remove( mailDir ,TRUE);
                TQStringList subFList = TQStringList::split("/",_tmp,FALSE);
                for ( TQStringList::Iterator it = subFList.begin(); it != subFList.end(); ++it ) {
                    TQString _cat = *it;
                    if(!(_cat == *mailFile)) {
                        if(_cat.startsWith(".") && _cat.endsWith(".directory")) {
                            _cat.remove(0,1);
                            _cat.remove((_cat.length() - 10), 10);
                        } else if (_cat.startsWith(".")) {
                            _cat = _cat.remove(0 , 1);
                        }
                        _path += "/" + _cat;
                    }
                }
                if(_path.endsWith("cur"))
                    _path.remove(_path.length() - 4 , 4);
                TQString _info = _path;
                info->addLog(i18n("Import folder %1...").arg(_info.remove(0,12)));
                info->setFrom(_info);
                info->setTo(_path);
                generatedPath = true;
            }

	    TQString msg_flag = getMessageFlags(info, dir.filePath(*mailFile)); 

            if(info->removeDupMsg) {
                if(! addMessage( info, _path, dir.filePath(*mailFile), msg_flag )) {
                    info->addLog( i18n("Could not import %1").arg( *mailFile ) );
                }
                info->setCurrent((int) ((float) currentFile / numFiles * 100));
            } else {
                if(! addMessage_fastImport( info, _path, dir.filePath(*mailFile), msg_flag )) {
                    info->addLog( i18n("Could not import %1").arg( *mailFile ) );
                }
                info->setCurrent((int) ((float) currentFile / numFiles * 100));
            }
        }
    }
}

