1 files changed, 28 insertions, 11 deletions

diff --git a/src/lobby.cpp b/src/lobby.cpp
index 424f634..29101c6 100644
--- a/src/lobby.cpp
+++ b/src/lobby.cpp
@@ -471,19 +471,21 @@ void Lobby::list_servers()
     QTreeWidgetItem *treeItem = new QTreeWidgetItem(ui_serverlist_tree);
     treeItem->setData(0, Qt::DisplayRole, i);
 
-    if (i_server.protocol == "tcp")
+    QIcon insecureIcon(":/data/icons/noencryption.svg");
+    QIcon secureIcon(":/data/icons/https.svg");
+    QColor secureColor(80, 120, 80);
+    treeItem->setText(1, i_server.name);
+    if (i_server.scheme == "wss")
     {
-      treeItem->setText(1, "(Legacy) " + i_server.name);
-      treeItem->setBackground(0, Qt::darkRed);
-      treeItem->setBackground(1, Qt::darkRed);
-
-      QString tooltip = tr("Unable to connect to server. Server is missing WebSocket support.");
-      treeItem->setToolTip(0, tooltip);
-      treeItem->setToolTip(1, tooltip);
+      treeItem->setIcon(1, secureIcon);
+      treeItem->setToolTip(1, "Connection is secure.");
+      treeItem->setBackground(0, secureColor);
+      treeItem->setBackground(1, secureColor);
     }
-    else
+    else if (i_server.scheme == "ws")
     {
-      treeItem->setText(1, i_server.name);
+      treeItem->setIcon(1, insecureIcon);
+      treeItem->setToolTip(1, "Insecure. Traffic can be intercepted or modified.");
     }
 
     i++;
@@ -504,7 +506,7 @@ void Lobby::list_favorites()
     QTreeWidgetItem *treeItem = new QTreeWidgetItem(ui_favorites_tree);
     treeItem->setData(0, Qt::DisplayRole, i);
 
-    if (i_server.protocol == "tcp")
+    if (i_server.legacy)
     {
       treeItem->setText(1, "(Legacy) " + i_server.name);
       treeItem->setBackground(0, Qt::darkRed);
@@ -516,7 +518,22 @@ void Lobby::list_favorites()
     }
     else
     {
+      QIcon insecureIcon(":/data/icons/noencryption.svg");
+      QIcon secureIcon(":/data/icons/https.svg");
+      QColor secureColor(80, 120, 80);
       treeItem->setText(1, i_server.name);
+      if (i_server.scheme == "wss")
+      {
+        treeItem->setIcon(1, secureIcon);
+        treeItem->setToolTip(1, "Connection is secure.");
+        treeItem->setBackground(0, secureColor);
+        treeItem->setBackground(1, secureColor);
+      }
+      else if (i_server.scheme == "ws")
+      {
+        treeItem->setIcon(1, insecureIcon);
+        treeItem->setToolTip(1, "Insecure. Traffic can be intercepted or modified.");
+      }
     }
 
     i++;