akonadiconsole/jobtrackermodel.cpp

   1 /*
   2  This file is part of Akonadi.
   3
   4  Copyright (c) 2009 Till Adam <adam@kde.org>
   5
   6  This program is free software; you can redistribute it and/or modify
   7  it under the terms of the GNU General Public License as published by
   8  the Free Software Foundation; either version 2 of the License, or
   9  (at your option) any later version.
  10
  11  This program is distributed in the hope that it will be useful,
  12  but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  14  GNU General Public License for more details.
  15
  16  You should have received a copy of the GNU General Public License
  17  along with this program; if not, write to the Free Software
  18  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301,
  19  USA.
  20  */
  21
  22 #include "jobtrackermodel.h"
  23 #include "jobtracker.h"
  24
  25 #include <KGlobal>
  26 #include <KLocale>
  27 #include <QtCore/QStringList>
  28 #include <QtCore/QModelIndex>
  29 #include <QtCore/QDebug>
  30 #include <QtCore/QDateTime>
  31 #include <QFont>
  32
  33 #include <cassert>
  34
  35 class JobTrackerModel::Private
  36 {
  37 public:
  38   Private( const char *name, JobTrackerModel* _q )
  39   :tracker( name), q(_q)
  40   {
  41
  42   }
  43   JobTracker tracker;
  44 private:
  45   JobTrackerModel* const q;
  46 };
  47
  48 JobTrackerModel::JobTrackerModel( const char *name, QObject* parent )
  49 :QAbstractItemModel( parent ), d( new Private( name, this ) )
  50 {
  51   connect( &d->tracker, SIGNAL( updated() ),
  52            this, SIGNAL( modelReset() ) );
  53 }
  54
  55 JobTrackerModel::~JobTrackerModel()
  56 {
  57   delete d;
  58 }
  59
  60 QModelIndex JobTrackerModel::index(int row, int column, const QModelIndex & parent) const
  61 {
  62   if ( !parent.isValid() ) // session, at top level
  63   {
  64     if ( row < 0 || row >= d->tracker.sessions().size() )
  65       return QModelIndex();
  66     return createIndex( row, column, d->tracker.idForSession( d->tracker.sessions().at(row) ) );
  67   }
  68   // non-toplevel job
  69   const QList<JobInfo> jobs = d->tracker.jobs( parent.internalId() );
  70   if ( row >= jobs.size() ) return QModelIndex();
  71   return createIndex( row, column, d->tracker.idForJob( jobs.at( row ).id ) );
  72 }
  73
  74 QModelIndex JobTrackerModel::parent(const QModelIndex & idx) const
  75 {
  76   if ( !idx.isValid() ) return QModelIndex();
  77
  78   const int parentid = d->tracker.parentId( idx.internalId() );
  79   if ( parentid == -1 ) return QModelIndex(); // top level session
  80
  81   const int grandparentid = d->tracker.parentId( parentid );
  82   int row = -1;
  83   if ( grandparentid == -1 )
  84   {
  85     row = d->tracker.sessions().indexOf( d->tracker.sessionForId( parentid ) );
  86   }
  87   else
  88   {
  89     // offset of the parent in the list of children of the grandparent
  90     row = d->tracker.jobs( grandparentid ).indexOf( d->tracker.info( parentid ) );
  91   }
  92   assert( row != -1 );
  93   return createIndex( row, 0, parentid );
  94 }
  95
  96 int JobTrackerModel::rowCount(const QModelIndex & parent) const
  97 {
  98   if ( !parent.isValid() )
  99   {
 100     return d->tracker.sessions().size();
 101   }
 102   else
 103   {
 104     return d->tracker.jobs( parent.internalId() ).size();
 105   }
 106 }
 107
 108 int JobTrackerModel::columnCount(const QModelIndex & parent) const
 109 {
 110   Q_UNUSED( parent );
 111   return 4;
 112 }
 113
 114 QVariant JobTrackerModel::data(const QModelIndex & idx, int role) const
 115 {
 116   // top level items are sessions
 117   if ( !idx.parent().isValid() )
 118   {
 119     if ( role == Qt::DisplayRole )
 120     {
 121       if( idx.column() == 0 )
 122       {
 123         assert(d->tracker.sessions().size() > idx.row());
 124         return d->tracker.sessions().at(idx.row());
 125       }
 126     }
 127   }
 128   else // not top level, so a job or subjob
 129   {
 130     const int id = idx.internalId();
 131     const JobInfo info = d->tracker.info( id );
 132     if ( role == Qt::DisplayRole )
 133     {
 134       if ( idx.column() == 0 )
 135         return info.id;
 136       if ( idx.column() == 1 )
 137         return KGlobal::locale()->formatTime( info.timestamp.time(), true )
 138           + QString::fromLatin1( ".%1" ).arg( info.timestamp.time().msec(), 3, 10, QLatin1Char('0') );
 139       if ( idx.column() == 2 )
 140         return info.type;
 141       if ( idx.column() == 3 )
 142         return info.stateAsString();
 143     }
 144     else if ( role == Qt::ForegroundRole ) {
 145       if ( info.state == JobInfo::Failed )
 146         return Qt::red;
 147     }
 148     else if ( role == Qt::FontRole ) {
 149       if ( info.state == JobInfo::Running ) {
 150         QFont f;
 151         f.setBold( true );
 152         return f;
 153       }
 154     }
 155     else if ( role == Qt::ToolTipRole ) {
 156       if ( info.state == JobInfo::Failed )
 157         return info.error;
 158     }
 159   }
 160   return QVariant();
 161 }
 162
 163 QVariant JobTrackerModel::headerData(int section, Qt::Orientation orientation, int role) const
 164 {
 165   if ( role == Qt::DisplayRole )
 166   {
 167     if ( orientation == Qt::Horizontal )
 168     {
 169       if ( section == 0 )
 170         return QLatin1String("Job ID");
 171       if ( section == 1 )
 172         return QLatin1String("Timestamp");
 173       if ( section == 2 )
 174         return QLatin1String("Job Type");
 175       if ( section == 3 )
 176         return QLatin1String("State");
 177
 178     }
 179   }
 180   return QVariant();
 181 }
 182
 183 void JobTrackerModel::resetTracker()
 184 {
 185   d->tracker.reset();
 186 }
 187
 188
 189 bool JobTrackerModel::isEnabled() const
 190 {
 191   return d->tracker.isEnabled();
 192 }
 193
 194 void JobTrackerModel::setEnabled( bool on )
 195 {
 196   d->tracker.setEnabled( on );
 197 }
 198
 199 #include "jobtrackermodel.moc"