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