src/dolphinmodel.cpp

   1 /**
   2   * This file is part of the KDE project
   3   * Copyright (C) 2007 Rafael Fernández López <ereslibre@kde.org>
   4   *
   5   * This library is free software; you can redistribute it and/or
   6   * modify it under the terms of the GNU Library General Public
   7   * License as published by the Free Software Foundation; either
   8   * version 2 of the License, or (at your option) any later version.
   9   *
  10   * This library is distributed in the hope that it will be useful,
  11   * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12   * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13   * Library General Public License for more details.
  14   *
  15   * You should have received a copy of the GNU Library General Public License
  16   * along with this library; see the file COPYING.LIB.  If not, write to
  17   * the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
  18   * Boston, MA 02110-1301, USA.
  19   */
  20
  21 #include "dolphinmodel.h"
  22
  23 #include "dolphinsortfilterproxymodel.h"
  24
  25 #include "kcategorizedview.h"
  26
  27 #include <config-nepomuk.h>
  28 #ifdef HAVE_NEPOMUK
  29 #include <nepomuk/global.h>
  30 #include <nepomuk/resource.h>
  31 #include <nepomuk/tag.h>
  32 #include <Soprano/Vocabulary/Xesam>
  33 #endif
  34
  35 #include <kdatetime.h>
  36 #include <kdirmodel.h>
  37 #include <kfileitem.h>
  38 #include <kiconloader.h>
  39 #include <klocale.h>
  40 #include <kurl.h>
  41 #include <kuser.h>
  42 #include <kmimetype.h>
  43 #include <kstandarddirs.h>
  44
  45 #include <QList>
  46 #include <QSortFilterProxyModel>
  47 #include <QPainter>
  48 #include <QPersistentModelIndex>
  49 #include <QDir>
  50 #include <QFileInfo>
  51
  52 const char* DolphinModel::m_others = I18N_NOOP2("@title:group Name", "Others");
  53
  54 DolphinModel::DolphinModel(QObject* parent) :
  55     KDirModel(parent),
  56     m_hasRevisionData(false),
  57     m_revisionHash()
  58 {
  59 }
  60
  61 DolphinModel::~DolphinModel()
  62 {
  63 }
  64
  65 bool DolphinModel::setData(const QModelIndex& index, const QVariant& value, int role)
  66 {
  67     if ((index.column() == DolphinModel::Revision) && (role == Qt::DecorationRole)) {
  68         // TODO: remove data again when items are deleted...
  69
  70         const QPersistentModelIndex key = index;
  71         const RevisionControlPlugin::RevisionState state = static_cast<RevisionControlPlugin::RevisionState>(value.toInt());
  72         if (m_revisionHash.value(key, RevisionControlPlugin::UnversionedRevision) != state) {
  73             if (!m_hasRevisionData) {
  74                 connect(this, SIGNAL(rowsRemoved (const QModelIndex&, int, int)),
  75                         this, SLOT(slotRowsRemoved(const QModelIndex&, int, int)));
  76                 m_hasRevisionData = true;
  77             }
  78
  79             m_revisionHash.insert(key, state);
  80             emit dataChanged(index, index);
  81             return true;
  82         }
  83     }
  84
  85     return KDirModel::setData(index, value, role);
  86 }
  87
  88 QVariant DolphinModel::data(const QModelIndex& index, int role) const
  89 {
  90     switch (role) {
  91     case KCategorizedSortFilterProxyModel::CategoryDisplayRole:
  92         return displayRoleData(index);
  93
  94     case KCategorizedSortFilterProxyModel::CategorySortRole:
  95         return sortRoleData(index);
  96
  97     case Qt::DecorationRole:
  98         if (index.column() == DolphinModel::Revision) {
  99             return m_revisionHash.value(index, RevisionControlPlugin::UnversionedRevision);
 100         }
 101         break;
 102
 103     case Qt::DisplayRole:
 104         if (index.column() == DolphinModel::Revision) {
 105             switch (m_revisionHash.value(index, RevisionControlPlugin::UnversionedRevision)) {
 106             case RevisionControlPlugin::NormalRevision:
 107                 return i18nc("@item::intable", "Normal");
 108             case RevisionControlPlugin::LocallyModifiedRevision:
 109                 return i18nc("@item::intable", "Locally modified");
 110             case RevisionControlPlugin::UpdateRequiredRevision:
 111                 return i18nc("@item::intable", "Update required");
 112             case RevisionControlPlugin::UnversionedRevision:
 113             default:
 114                 return i18nc("@item::intable", "Unversioned");
 115             }
 116         }
 117         break;
 118
 119     default:
 120         break;
 121     }
 122
 123     return KDirModel::data(index, role);
 124 }
 125
 126 QVariant DolphinModel::headerData(int section, Qt::Orientation orientation, int role) const
 127 {
 128     if ((orientation == Qt::Horizontal) && (role == Qt::DisplayRole)) {
 129         if (section < KDirModel::ColumnCount) {
 130             return KDirModel::headerData(section, orientation, role);
 131         }
 132
 133         Q_ASSERT(section == DolphinModel::Revision);
 134         return i18nc("@title::column", "Revision");
 135     }
 136     return QVariant();
 137 }
 138
 139 int DolphinModel::columnCount(const QModelIndex& parent) const
 140 {
 141     return KDirModel::columnCount(parent) + (ExtraColumnCount - ColumnCount);
 142 }
 143
 144 void DolphinModel::clearRevisionData()
 145 {
 146     m_revisionHash.clear();
 147     m_hasRevisionData = false;
 148 }
 149
 150 bool DolphinModel::hasRevisionData() const
 151 {
 152     return m_hasRevisionData;
 153 }
 154
 155 void DolphinModel::slotRowsRemoved(const QModelIndex& parent, int start, int end)
 156 {
 157     if (m_hasRevisionData) {
 158         const int column = parent.column();
 159         for (int row = start; row <= end; ++row) {
 160             m_revisionHash.remove(parent.child(row, column));
 161         }
 162     }
 163 }
 164
 165 QVariant DolphinModel::displayRoleData(const QModelIndex& index) const
 166 {
 167     QString retString;
 168
 169     if (!index.isValid()) {
 170         return retString;
 171     }
 172
 173     const KDirModel *dirModel = qobject_cast<const KDirModel*>(index.model());
 174     KFileItem item = dirModel->itemForIndex(index);
 175
 176     switch (index.column()) {
 177     case KDirModel::Name: {
 178         // KDirModel checks columns to know to which role are
 179         // we talking about
 180         const QModelIndex nameIndex = index.model()->index(index.row(), KDirModel::Name, index.parent());
 181         if (!nameIndex.isValid()) {
 182             return retString;
 183         }
 184         const QVariant data = nameIndex.model()->data(nameIndex, Qt::DisplayRole);
 185         const QString name = data.toString();
 186         if (!name.isEmpty()) {
 187             if (!item.isHidden() && name.at(0).isLetter())
 188                 retString = name.at(0).toUpper();
 189             else if (item.isHidden()) {
 190                 if (name.at(0) == '.') {
 191                     if (name.size() > 1 && name.at(1).isLetter()) {
 192                         retString = name.at(1).toUpper();
 193                     } else {
 194                         retString = i18nc("@title:group Name", m_others);
 195                     }
 196                 } else {
 197                     retString = name.at(0).toUpper();
 198                 }
 199             } else {
 200                 bool validCategory = false;
 201
 202                 const QString str(name.toUpper());
 203                 const QChar* currA = str.unicode();
 204                 while (!currA->isNull() && !validCategory) {
 205                     if (currA->isLetter()) {
 206                         validCategory = true;
 207                     } else if (currA->isDigit()) {
 208                         return i18nc("@title:group Name", m_others);
 209                     } else {
 210                         ++currA;
 211                     }
 212                 }
 213
 214                 if (!validCategory) {
 215                     retString = validCategory ? *currA : i18nc("@title:group Name", m_others);
 216                 } else {
 217                     retString = *currA;
 218                 }
 219             }
 220         }
 221         break;
 222     }
 223
 224     case KDirModel::Size: {
 225         const KIO::filesize_t fileSize = !item.isNull() ? item.size() : ~0U;
 226         if (!item.isNull() && item.isDir()) {
 227             retString = i18nc("@title:group Size", "Folders");
 228         } else if (fileSize < 5242880) {
 229             retString = i18nc("@title:group Size", "Small");
 230         } else if (fileSize < 10485760) {
 231             retString = i18nc("@title:group Size", "Medium");
 232         } else {
 233             retString = i18nc("@title:group Size", "Big");
 234         }
 235         break;
 236     }
 237
 238     case KDirModel::ModifiedTime: {
 239         KDateTime modifiedTime = item.time(KFileItem::ModificationTime);
 240         modifiedTime = modifiedTime.toLocalZone();
 241
 242         const QDate currentDate = KDateTime::currentLocalDateTime().date();
 243         const QDate modifiedDate = modifiedTime.date();
 244
 245         const int daysDistance = modifiedDate.daysTo(currentDate);
 246
 247         int yearForCurrentWeek = 0;
 248         int currentWeek = currentDate.weekNumber(&yearForCurrentWeek);
 249         if (yearForCurrentWeek == currentDate.year() + 1) {
 250             currentWeek = 53;
 251         }
 252
 253         int yearForModifiedWeek = 0;
 254         int modifiedWeek = modifiedDate.weekNumber(&yearForModifiedWeek);
 255         if (yearForModifiedWeek == modifiedDate.year() + 1) {
 256             modifiedWeek = 53;
 257         }
 258
 259         if (currentDate.year() == modifiedDate.year() && currentDate.month() == modifiedDate.month()) {
 260             switch (currentWeek - modifiedWeek) {
 261             case 0:
 262                 switch (daysDistance) {
 263                 case 0:  retString = i18nc("@title:group Date", "Today"); break;
 264                 case 1:  retString = i18nc("@title:group Date", "Yesterday"); break;
 265                 default: retString = modifiedTime.toString(i18nc("@title:group The week day name: %A", "%A"));
 266                 }
 267                 break;
 268             case 1:
 269                 retString = i18nc("@title:group Date", "Last Week");
 270                 break;
 271             case 2:
 272                 retString = i18nc("@title:group Date", "Two Weeks Ago");
 273                 break;
 274             case 3:
 275                 retString = i18nc("@title:group Date", "Three Weeks Ago");
 276                 break;
 277             case 4:
 278             case 5:
 279                 retString = i18nc("@title:group Date", "Earlier this Month");
 280                 break;
 281             default:
 282                 Q_ASSERT(false);
 283             }
 284         } else {
 285             const QDate lastMonthDate = currentDate.addMonths(-1);
 286             if  (lastMonthDate.year() == modifiedDate.year() && lastMonthDate.month() == modifiedDate.month()) {
 287                 if (daysDistance == 1) {
 288                     retString = modifiedTime.toString(i18nc("@title:group Date: %B is full month name in current locale, and %Y is full year number", "Yesterday (%B, %Y)"));
 289                 } else if (daysDistance <= 7) {
 290                     retString = modifiedTime.toString(i18nc("@title:group The week day name: %A, %B is full month name in current locale, and %Y is full year number", "%A (%B, %Y)"));
 291                 } else if (daysDistance <= 7 * 2) {
 292                     retString = modifiedTime.toString(i18nc("@title:group Date: %B is full month name in current locale, and %Y is full year number", "Last Week (%B, %Y)"));
 293                 } else if (daysDistance <= 7 * 3) {
 294                     retString = modifiedTime.toString(i18nc("@title:group Date: %B is full month name in current locale, and %Y is full year number", "Two Weeks Ago (%B, %Y)"));
 295                 } else if (daysDistance <= 7 * 4) {
 296                     retString = modifiedTime.toString(i18nc("@title:group Date: %B is full month name in current locale, and %Y is full year number", "Three Weeks Ago (%B, %Y)"));
 297                 } else {
 298                     retString = modifiedTime.toString(i18nc("@title:group Date: %B is full month name in current locale, and %Y is full year number", "Earlier on %B, %Y"));
 299                 }
 300             } else {
 301                 retString = modifiedTime.toString(i18nc("@title:group The month and year: %B is full month name in current locale, and %Y is full year number", "%B, %Y"));
 302             }
 303         }
 304         break;
 305     }
 306
 307     case KDirModel::Permissions: {
 308         QString user;
 309         QString group;
 310         QString others;
 311
 312         QFileInfo info(item.url().pathOrUrl());
 313
 314         // set user string
 315         if (info.permission(QFile::ReadUser)) {
 316             user = i18nc("@item:intext Access permission, concatenated", "Read, ");
 317         }
 318         if (info.permission(QFile::WriteUser)) {
 319             user += i18nc("@item:intext Access permission, concatenated", "Write, ");
 320         }
 321         if (info.permission(QFile::ExeUser)) {
 322             user += i18nc("@item:intext Access permission, concatenated", "Execute, ");
 323         }
 324         user = user.isEmpty() ? i18nc("@item:intext Access permission, concatenated", "Forbidden") : user.mid(0, user.count() - 2);
 325
 326         // set group string
 327         if (info.permission(QFile::ReadGroup)) {
 328             group = i18nc("@item:intext Access permission, concatenated", "Read, ");
 329         }
 330         if (info.permission(QFile::WriteGroup)) {
 331             group += i18nc("@item:intext Access permission, concatenated", "Write, ");
 332         }
 333         if (info.permission(QFile::ExeGroup)) {
 334             group += i18nc("@item:intext Access permission, concatenated", "Execute, ");
 335         }
 336         group = group.isEmpty() ? i18nc("@item:intext Access permission, concatenated", "Forbidden") : group.mid(0, group.count() - 2);
 337
 338         // set permission string
 339         if (info.permission(QFile::ReadOther)) {
 340             others = i18nc("@item:intext Access permission, concatenated", "Read, ");
 341         }
 342         if (info.permission(QFile::WriteOther)) {
 343             others += i18nc("@item:intext Access permission, concatenated", "Write, ");
 344         }
 345         if (info.permission(QFile::ExeOther)) {
 346             others += i18nc("@item:intext Access permission, concatenated", "Execute, ");
 347         }
 348         others = others.isEmpty() ? i18nc("@item:intext Access permission, concatenated", "Forbidden") : others.mid(0, others.count() - 2);
 349
 350         retString = i18nc("@title:group Files and folders by permissions", "(User: %1) (Group: %2) (Others: %3)", user, group, others);
 351         break;
 352     }
 353
 354     case KDirModel::Owner:
 355         retString = item.user();
 356         break;
 357
 358     case KDirModel::Group:
 359         retString = item.group();
 360         break;
 361
 362     case KDirModel::Type:
 363         retString = item.mimeComment();
 364         break;
 365
 366     case DolphinModel::Revision:
 367         retString = "test";
 368         break;
 369     }
 370
 371     return retString;
 372 }
 373
 374 QVariant DolphinModel::sortRoleData(const QModelIndex& index) const
 375 {
 376     QVariant retVariant;
 377
 378     if (!index.isValid()) {
 379         return retVariant;
 380     }
 381
 382     const KDirModel *dirModel = qobject_cast<const KDirModel*>(index.model());
 383     KFileItem item = dirModel->itemForIndex(index);
 384
 385     switch (index.column()) {
 386     case KDirModel::Name: {
 387         retVariant = data(index, KCategorizedSortFilterProxyModel::CategoryDisplayRole);
 388         if (retVariant == i18nc("@title:group Name", m_others)) {
 389             // assure that the "Others" group is always the last categorization
 390             retVariant = QString(QChar(QChar::ReplacementCharacter));
 391         }
 392         break;
 393     }
 394
 395     case KDirModel::Size: {
 396         const KIO::filesize_t fileSize = !item.isNull() ? item.size() : ~0U;
 397         if (item.isDir()) {
 398             retVariant = 0;
 399         } else if (fileSize < 5242880) {
 400             retVariant = 1;
 401         } else if (fileSize < 10485760) {
 402             retVariant = 2;
 403         } else {
 404             retVariant = 3;
 405         }
 406         break;
 407     }
 408
 409     case KDirModel::ModifiedTime: {
 410         KDateTime modifiedTime = item.time(KFileItem::ModificationTime);
 411         modifiedTime = modifiedTime.toLocalZone();
 412
 413         const QDate currentDate = KDateTime::currentLocalDateTime().date();
 414         const QDate modifiedDate = modifiedTime.date();
 415
 416         retVariant = -modifiedDate.daysTo(currentDate);
 417         break;
 418     }
 419
 420     case KDirModel::Permissions: {
 421         QFileInfo info(item.url().pathOrUrl());
 422
 423         retVariant = -KDirSortFilterProxyModel::pointsForPermissions(info);
 424         break;
 425     }
 426
 427     case KDirModel::Owner:
 428         retVariant = item.user();
 429         break;
 430
 431     case KDirModel::Group:
 432         retVariant = item.group();
 433         break;
 434
 435     case KDirModel::Type:
 436         if (item.isDir()) {
 437             // when sorting we want folders to be placed first
 438             retVariant = QString(); // krazy:exclude=nullstrassign
 439         } else {
 440             retVariant = item.mimeComment();
 441         }
 442         break;
 443
 444     default:
 445         break;
 446     }
 447
 448     return retVariant;
 449 }