agentfilterproxymodel.cpp

/*
    Copyright (c) 2007 Volker Krause <vkrause@kde.org>

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

    This library 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 Library General Public
    License for more details.

    You should have received a copy of the GNU Library General Public License
    along with this library; see the file COPYING.LIB.  If not, write to the
    Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
    02110-1301, USA.
*/

#include "agentfilterproxymodel.h"

#include "agenttypemodel.h"
#include "agentinstancemodel.h"

#include <kdebug.h>
#include <kmimetype.h>

#include <QtCore/QStringList>

#include <boost/static_assert.hpp>

using namespace Akonadi;

// ensure the role numbers are equivalent for both source models
BOOST_STATIC_ASSERT( (int)AgentTypeModel::CapabilitiesRole == (int)AgentInstanceModel::CapabilitiesRole );
BOOST_STATIC_ASSERT( (int)AgentTypeModel::MimeTypesRole == (int)AgentInstanceModel::MimeTypesRole );

class AgentFilterProxyModel::Private
{
  public:
    QStringList mimeTypes;
    QStringList capabilities;
    QStringList excludeCapabilities;
};

AgentFilterProxyModel::AgentFilterProxyModel( QObject *parent )
  : QSortFilterProxyModel( parent ),
    d( new Private )
{
  setDynamicSortFilter( true );
}

AgentFilterProxyModel::~AgentFilterProxyModel()
{
  delete d;
}

void AgentFilterProxyModel::addMimeTypeFilter( const QString &mimeType )
{
  d->mimeTypes << mimeType;
  invalidateFilter();
}

void AgentFilterProxyModel::addCapabilityFilter( const QString &capability )
{
  d->capabilities << capability;
  invalidateFilter();
}

void AgentFilterProxyModel::excludeCapabilities( const QString &capability )
{
  d->excludeCapabilities << capability;
  invalidateFilter();
}

void AgentFilterProxyModel::clearFilters()
{
  d->capabilities.clear();
  d->mimeTypes.clear();
  d->excludeCapabilities.clear();
  invalidateFilter();
}

bool AgentFilterProxyModel::filterAcceptsRow( int row, const QModelIndex& ) const
{
  const QModelIndex index = sourceModel()->index( row, 0 );

  // First see if the name matches a set regexp filter.
  if ( !filterRegExp().isEmpty() ) {
      if ( index.data( AgentTypeModel::IdentifierRole ).toString().contains( filterRegExp() ))
           return true;
      else if ( index.data().toString().contains( filterRegExp() ) )
          return true;
      else
          return false;
  }

  if ( !d->mimeTypes.isEmpty() ) {
    bool found = false;
    foreach ( const QString &mimeType, index.data( AgentTypeModel::MimeTypesRole ).toStringList() ) {
      if ( d->mimeTypes.contains( mimeType ) ) {
        found = true;
      } else {
        KMimeType::Ptr mimeTypePtr = KMimeType::mimeType( mimeType, KMimeType::ResolveAliases );
        if ( !mimeTypePtr.isNull() ) {
          foreach ( const QString &type, d->mimeTypes ) {
            if ( mimeTypePtr->is( type ) ) {
              found = true;
              break;
            }
          }
        }
      }

      if ( found ) {
        break;
      }
    }

    if ( !found ) {
      return false;
    }
  }

  if ( !d->capabilities.isEmpty() ) {
    bool found = false;
    foreach ( const QString &capability, index.data( AgentTypeModel::CapabilitiesRole ).toStringList() ) {
      if ( d->capabilities.contains( capability ) ) {
        found = true;
        break;
      }
    }

    if ( !found ) {
      return false;
    }

    if ( found && !d->excludeCapabilities.isEmpty() ) {
      foreach ( const QString &capability, index.data( AgentTypeModel::CapabilitiesRole ).toStringList() ) {
        if ( d->excludeCapabilities.contains( capability ) ) {
          found = false;
          break;
        }
      }

      if ( !found ) {
        return false;
      }
    }
  }

  return true;
}