SmDirModel: check if file already is in database

Indicate if we already have a file by coloring the filename darkGreen
when browsing the filesystem.

1 files changed, 14 insertions, 0 deletions

diff --git a/smdirmodel.cpp b/smdirmodel.cpp
index 8a6c961..6053890 100644
--- a/smdirmodel.cpp
+++ b/smdirmodel.cpp
@@ -57,6 +57,8 @@ QVariant SmDirModel::data(const QModelIndex &index, int role) const{
             return i->data(Bitrate);
         case FullPathRole:
             return i->data(FullPath);
+        case PresentRole:
+            return i->data(Present);
         case Qt::DecorationRole: {
             if(index.column() == 0){
                 QStringList mime = i->data(Type).toString().split('/');
@@ -67,8 +69,15 @@ QVariant SmDirModel::data(const QModelIndex &index, int role) const{
                     return mIcons.value("other");
                 }
                 return mIcons.value(mime.at(0).toLower());
+            }else{
+                return QVariant();
             }
         }
+        case Qt::ForegroundRole:
+            if(index.column() == 0 && i->data(Present).toInt() > 0){
+                return QVariant(QColor(Qt::darkGreen));
+            }
+            return SmTreeModel::data(index, role);
         case Qt::DisplayRole: {
             if(index.column() == DurSize){
                 QVariant d = i->data(DurSize);
@@ -85,6 +94,7 @@ QVariant SmDirModel::data(const QModelIndex &index, int role) const{
             }
         }
 
+
         default:
             return SmTreeModel::data(index, role);
     }
@@ -187,6 +197,10 @@ void SmDirModel::refresh(){
     setDir(mCurrentDir);
 }
 
+void SmDirModel::setCheckForPresent(bool check){
+    mCollector->setCheckForPresent(check);
+}
+
 void SmDirModel::populate(SmTreeItem *root){
     setRoot(root);
     emit needResize();