From ff4f8d7dbf914f2529030701c877053ebacfa0fe Mon Sep 17 00:00:00 2001
From: Martin Kleusberg <mkleusberg@gmail.com>
Date: Mon, 31 Jul 2017 23:11:42 +0200
Subject: [PATCH] dbhub: Use different icon for user's directory

---
 src/RemoteModel.cpp       |  11 ++++++++++-
 src/RemoteModel.h         |   1 +
 src/icons/folder_user.png | Bin 0 -> 730 bytes
 src/icons/icons.qrc       |   1 +
 4 files changed, 12 insertions(+), 1 deletion(-)
 create mode 100644 src/icons/folder_user.png

diff --git a/src/RemoteModel.cpp b/src/RemoteModel.cpp
index de850206..b5311413 100644
--- a/src/RemoteModel.cpp
+++ b/src/RemoteModel.cpp
@@ -110,6 +110,13 @@ RemoteModel::~RemoteModel()
 
 void RemoteModel::setNewRootDir(const QString& url, const QString& cert)
 {
+    // Extract user name
+    QString cn = remoteDatabase.clientCertificates()[cert].subjectInfo(QSslCertificate::CommonName).at(0);
+    QStringList cn_parts = cn.split("@");
+    if(cn_parts.size() < 2)
+        return;
+    currentUserName = cn_parts.first();
+
     // Save settings
     currentRootDirectory = url;
     currentClientCert = cert;
@@ -194,7 +201,9 @@ QVariant RemoteModel::data(const QModelIndex& index, int role) const
     if(role == Qt::DecorationRole && index.column() == 0)
     {
         // Use different icons depending on item type
-        if(type == "folder")
+        if(type == "folder" && index.parent() == QModelIndex() && item->value(RemoteModelColumnName) == currentUserName)
+            return QImage(":/icons/folder_user");
+        else if(type == "folder")
             return QImage(":/icons/folder");
         else if(type == "database")
             return QImage(":/icons/database");
diff --git a/src/RemoteModel.h b/src/RemoteModel.h
index 4d62ec6d..65945f00 100644
--- a/src/RemoteModel.h
+++ b/src/RemoteModel.h
@@ -105,6 +105,7 @@ private:
     // lazy population.
     QString currentRootDirectory;
     QString currentClientCert;
+    QString currentUserName;
 };
 
 #endif
diff --git a/src/icons/folder_user.png b/src/icons/folder_user.png
new file mode 100644
index 0000000000000000000000000000000000000000..f021c3e12a72a17b100089cd10d1ff9f2788c224
GIT binary patch
literal 730
zcmV<00ww*4P)<h;3K|Lk000e1NJLTq000mG000mO1^@s6AM^iV00004XF*Lt006JZ
zHwB960000PbVXQnQ*UN;cVTj606}DLVr3vnZDD6+Qe|Oed2z{QJOBU!Y)M2xR2Uh3
z!AopRVHkkn_dj#a+&Y=6&{AzfHIWbz)hV$Mwv@OmBsNVNVPj)y!4k2v5ZACEMG!;~
zK_peTATF`0rV@9HR<z^HoXh`x+L%y_@jOOVsb=Tu#WS1ha#wSyOu<?~K@-{JjqQhT
zbW!0@#jbT7PGfD&#auF7up(GhA}hupEv?zTacK_~4pp?&YIaJdW_c!GOY6KjG&VIL
zqKF7}nal|)T&hUw$EbeRY-#V>>`eEHOekrZG3#;v?6H6fhbs1c-xE%qmL1FD6Pgsp
z%Q7MO@KoovHHK8U1J?_FIzK;SjKg2b`K_PgZHE@VJXO=*(cIEX$#{4s<ip2d?HyXw
z+sSWc;994)t$Bno29$A%A6RxActY#kc@zo(Pj8*Ym^5B8Pg6%Xr;px=Z@+rKC4Ss<
z!3<n3NZZ;z6Mj964HMwx+1ndYn!1`PowFD$g0=kNDojIpaqeF5=I!fEE|Ebc#D+;!
zl%GQn=+3`__&7YIs^gokqz*pcbIHZ1k+{U@&>+#sCnPFUw_rU#+IR6UWkSw_(UZ&g
zlAKBNq7_M($idCjGh@|$KtL56ha|q=<f~uCs-<KT_%XybqPbxS28f@%D2;giu}h?=
zEfan_2`Ik_DY0b9F(yfj7(5lMM7RW4Akop@G-qYsh)Vz}F&ID<5ReE*BO2Tv)Egf)
zMbs&x43o~Z^A0$(k`czago?p}DyRhwQ3*iGsFqOcqqc~~qp0-(VKP@kA@+xn*aLMy
z6R3J1DO74eBUD2$AviI%TxK#?%faWNBO4k}91Mp4QNY{<(P1in0X<zbFA)!j{r~^~
M07*qoM6N<$g7n%!RsaA1

literal 0
HcmV?d00001

diff --git a/src/icons/icons.qrc b/src/icons/icons.qrc
index 24e7b26d..6cc06d50 100644
--- a/src/icons/icons.qrc
+++ b/src/icons/icons.qrc
@@ -55,5 +55,6 @@
         <file>database.png</file>
         <file>cog_go.png</file>
         <file alias="paste">page_paste.png</file>
+        <file>folder_user.png</file>
     </qresource>
 </RCC>