antispamwizard.h

Go to the documentation of this file.

/*  -*- mode: C++ -*-
    This file is part of KMail.
    Copyright (c) 2003 Andreas Gungl <a.gungl@gmx.de>

    KMail is free software; you can redistribute it and/or modify it
    under the terms of the GNU General Public License, version 2, as
    published by the Free Software Foundation.

    KMail is distributed in the hope that it will be useful, but
    WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
    General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with this program; if not, write to the Free Software
    Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA

    In addition, as a special exception, the copyright holders give
    permission to link the code of this program with any edition of
    the Qt library by Trolltech AS, Norway (or with modified versions
    of Qt that use the same license as Qt), and distribute linked
    combinations including the two.  You must obey the GNU General
    Public License in all respects for all of the code used other than
    Qt.  If you modify this file, you may extend this exception to
    your version of the file, but you are not obligated to do so.  If
    you do not wish to do so, delete this exception statement from
    your version.
*/
#ifndef KMAIL_ANTISPAMWIZARD_H
#define KMAIL_ANTISPAMWIZARD_H

#include <KAssistantDialog>
#include <KSharedConfig>

#include <QList>
#include <akonadi/collection.h>

class QLabel;
class QTextEdit;
class QCheckBox;
class QBoxLayout;
class QListWidget;

namespace MailCommon {
class FolderTreeWidget;
class FolderRequester;
}

namespace KMail {

class ASWizInfoPage;
class ASWizSpamRulesPage;
class ASWizVirusRulesPage;
class ASWizSummaryPage;

//---------------------------------------------------------------------------
class AntiSpamWizard : public KAssistantDialog
{
    Q_OBJECT

public:
    enum WizardMode { AntiSpam, AntiVirus };

    AntiSpamWizard( WizardMode mode,
                    QWidget * parent);

protected:
    class SpamToolConfig
    {
    public:
        SpamToolConfig() {}
        SpamToolConfig( const QString &toolId, int configVersion, int prio,
                        const QString &name, const QString &exec,
                        const QString &url, const QString &filter,
                        const QString &detection, const QString &spam,
                        const QString &ham,const QString &noSpam,
                        const QString &header, const QString &pattern,
                        const QString &pattern2, const QString &serverPattern,
                        bool detectionOnly, bool regExp, bool bayesFilter,
                        bool tristateDetection, WizardMode type );

        int getVersion() const { return mVersion; }
        int getPrio() const { return mPrio; }
        QString getId()  const { return mId; }
        QString getVisibleName()  const { return mVisibleName; }
        QString getExecutable() const { return mExecutable; }
        QString getWhatsThisText() const { return mWhatsThisText; }
        QString getFilterName() const { return mFilterName; }
        QString getDetectCmd() const { return mDetectCmd; }
        QString getSpamCmd() const { return mSpamCmd; }
        QString getHamCmd() const { return mHamCmd; }
        QString getNoSpamCmd() const { return mNoSpamCmd; }
        QString getDetectionHeader() const { return mDetectionHeader; }
        QString getDetectionPattern() const { return mDetectionPattern; }
        QString getDetectionPattern2() const { return mDetectionPattern2; }
        QString getServerPattern() const { return mServerPattern; }
        bool isServerBased() const;
        bool isDetectionOnly() const { return mDetectionOnly; }
        bool isUseRegExp() const { return mUseRegExp; }
        bool useBayesFilter() const { return mSupportsBayesFilter; }
        bool hasTristateDetection() const { return mSupportsUnsure; }
        WizardMode getType() const { return mType; }
        // convenience methods for types
        bool isSpamTool() const { return ( mType == AntiSpam ); }
        bool isVirusTool() const { return ( mType == AntiVirus ); }

    private:
        // used to identifiy configs for the same tool
        QString mId;
        // The version of the config data, used for merging and
        // detecting newer configs
        int mVersion;
        // the priority of the tool in the list presented to the user
        int mPrio;
        // the name as shown by the checkbox in the dialog page
        QString mVisibleName;
        // the command to check the existence of the tool
        QString mExecutable;
        // the What's This help text (e.g. url for the tool)
        QString mWhatsThisText;
        // name for the created filter in the filter list
        QString mFilterName;
        // pipe through cmd used to detect spam messages
        QString mDetectCmd;
        // pipe through cmd to let the tool learn a spam message
        QString mSpamCmd;
        // pipe through cmd to let the tool learn a ham message
        QString mHamCmd;
        // pipe through cmd to let the tool delete the spam markup
        QString mNoSpamCmd;
        // by which header are messages marked as spam
        QString mDetectionHeader;
        // what header pattern is used to mark spam messages
        QString mDetectionPattern;
        // what header pattern is used to mark unsure messages
        QString mDetectionPattern2;
        // what header pattern is used in the account to check for a certain server
        QString mServerPattern;
        // filter cannot search actively but relies on pattern by regExp or contain rule
        bool mDetectionOnly;
        // filter searches for the pattern by regExp or contain rule
        bool mUseRegExp;
        // can the tool learn spam and ham, has it a bayesian algorithm
        bool mSupportsBayesFilter;
        // differentiate between ham, spam and a third "unsure" state
        bool mSupportsUnsure;
        // Is the tool AntiSpam or AntiVirus
        WizardMode mType;
    };

    class ConfigReader
    {
    public:
        ConfigReader( WizardMode mode,
                      QList<SpamToolConfig> & configList );
        ~ConfigReader( );

        QList<SpamToolConfig> & getToolList() { return mToolList; }

        void readAndMergeConfig();

    private:
        QList<SpamToolConfig> & mToolList;
        KSharedConfig::Ptr mConfig;
        WizardMode mMode;

        SpamToolConfig readToolConfig( KConfigGroup & configGroup );
        SpamToolConfig createDummyConfig();

        void mergeToolConfig(const SpamToolConfig &config );
        void sortToolList();
    };

    void accept();

protected slots:
    void checkProgramsSelections();
    void checkVirusRulesSelections();
    void checkToolAvailability();
    void slotHelpClicked();
    void slotBuildSummary();

private:
    /* Check for the availability of an executible along the PATH */
    int checkForProgram( const QString &executable ) const;
    /* generic checks if any option in a page is checked */
    bool anyVirusOptionChecked() const;
    /* convenience method calling the appropriate filter manager method */
    const QString uniqueNameFor( const QString & name );
    /* convenience method to sort out new and existing filters */
    void sortFilterOnExistance( const QString & intendedFilterName,
                                QString & newFilters,
                                QString & replaceFilters );

    /* The pages in the wizard */
    ASWizInfoPage * mInfoPage;
    ASWizSpamRulesPage * mSpamRulesPage;
    ASWizVirusRulesPage * mVirusRulesPage;
    ASWizSummaryPage * mSummaryPage;

    KPageWidgetItem * mInfoPageItem;
    KPageWidgetItem * mSpamRulesPageItem;
    KPageWidgetItem * mVirusRulesPageItem;
    KPageWidgetItem * mSummaryPageItem;

    /* The configured tools and it's settings to be used in the wizard. */
    QList<SpamToolConfig> mToolList;

    /* Are any spam tools selected? */
    bool mSpamToolsUsed;
    /* Are any virus tools selected? */
    bool mVirusToolsUsed;

    WizardMode mMode;
};


//---------------------------------------------------------------------------
class ASWizPage : public QWidget
{
public:
    ASWizPage(QWidget *parent, const QString &name,
               const QString *bannerName = 0);

protected:
    QBoxLayout *mLayout;
private:
    QLabel *mBannerLabel;
};


//---------------------------------------------------------------------------
class ASWizInfoPage : public ASWizPage
{
    Q_OBJECT

public:
    ASWizInfoPage( AntiSpamWizard::WizardMode mode,
                   QWidget *parent, const QString &name );

    void setScanProgressText( const QString &toolName );
    void addAvailableTool( const QString &visibleName );
    bool isProgramSelected( const QString &visibleName ) const;

private slots:
    void processSelectionChange();

signals:
    void selectionChanged();

private:
    QTextEdit *mIntroText;
    QLabel *mScanProgressText;
    QLabel *mSelectionHint;
    QListWidget *mToolsList;
};

//---------------------------------------------------------------------------
class ASWizSpamRulesPage : public ASWizPage
{
    Q_OBJECT

public:
    ASWizSpamRulesPage(QWidget * parent, const QString &name);

    bool markAsReadSelected() const;
    bool moveSpamSelected() const;
    bool moveUnsureSelected() const;

    QString selectedUnsureCollectionName() const;
    QString selectedUnsureCollectionId() const;

    void allowUnsureFolderSelection( bool enabled );
    void allowMoveSpam( bool enabled );

    QString selectedSpamCollectionId() const;
    QString selectedSpamCollectionName() const;

protected:
    Akonadi::Collection selectedSpamCollection() const;
    Akonadi::Collection selectedUnsureCollection() const;

private slots:
    void processSelectionChange();
    void processSelectionChange( const Akonadi::Collection & );

signals:
    void selectionChanged();

private:
    QCheckBox * mMarkRules;
    QCheckBox * mMoveSpamRules;
    QCheckBox * mMoveUnsureRules;
    MailCommon::FolderRequester *mFolderReqForSpamFolder;
    MailCommon::FolderRequester *mFolderReqForUnsureFolder;
};

//-------------------------------------------------------------------------
class ASWizVirusRulesPage : public ASWizPage
{
    Q_OBJECT

public:
    ASWizVirusRulesPage(QWidget * parent, const QString &name );

    bool pipeRulesSelected() const;
    bool moveRulesSelected() const;
    bool markReadRulesSelected() const;

    QString selectedFolderName() const;

private slots:
    void processSelectionChange();
signals:
    void selectionChanged();

private:
    QCheckBox * mPipeRules;
    QCheckBox * mMoveRules;
    MailCommon::FolderTreeWidget *mFolderTree;
    QCheckBox * mMarkRules;
};

//---------------------------------------------------------------------------
class ASWizSummaryPage : public ASWizPage
{
    Q_OBJECT

public:
    ASWizSummaryPage( QWidget * parent, const QString &name );

    void setSummaryText( const QString & text );

private:
    QLabel * mSummaryText;
};


} // namespace KMail

#endif // KMAIL_ANTISPAMWIZARD_H