From 51497d87e01c5a7296a3b805e890a3e9e7b46dfd Mon Sep 17 00:00:00 2001
From: ronoaer <ronoaer@gmail.com>
Date: Mon, 25 Sep 2023 07:22:11 +0800
Subject: [PATCH] redirected to pagesetupwizardeasy when none containers
 installed

---
 client/ui/models/containers_model.cpp        | 14 ++++++++++++++
 client/ui/models/containers_model.h          |  2 +-
 client/ui/qml/Components/ConnectButton.qml   |  8 ++++++++
 client/ui/qml/Pages2/PageSetupWizardEasy.qml |  2 ++
 4 files changed, 25 insertions(+), 1 deletion(-)

diff --git a/client/ui/models/containers_model.cpp b/client/ui/models/containers_model.cpp
index afff44b6..6cf855a6 100644
--- a/client/ui/models/containers_model.cpp
+++ b/client/ui/models/containers_model.cpp
@@ -214,6 +214,20 @@ bool ContainersModel::isAmneziaDnsContainerInstalled(const int serverIndex)
     return containers.contains(DockerContainer::Dns);
 }
 
+bool ContainersModel::isAnyContainerInstalled()
+{
+    for (int row=0; row < rowCount(); row++) {
+        QModelIndex idx = this->index(row, 0);
+
+        if (this->data(idx, IsInstalledRole).toBool() &&
+            this->data(idx, ServiceTypeRole).toInt() == ServiceType::Vpn) {
+            return true;
+        }
+    }
+
+    return false;
+}
+
 QHash<int, QByteArray> ContainersModel::roleNames() const
 {
     QHash<int, QByteArray> roles;
diff --git a/client/ui/models/containers_model.h b/client/ui/models/containers_model.h
index 741a0620..2cc41cbf 100644
--- a/client/ui/models/containers_model.h
+++ b/client/ui/models/containers_model.h
@@ -63,7 +63,7 @@ public slots:
     bool isAmneziaDnsContainerInstalled();
     bool isAmneziaDnsContainerInstalled(const int serverIndex);
 
-    //    bool isOnlyServicesInstalled(const int serverIndex);
+    bool isAnyContainerInstalled();
 
 protected:
     QHash<int, QByteArray> roleNames() const override;
diff --git a/client/ui/qml/Components/ConnectButton.qml b/client/ui/qml/Components/ConnectButton.qml
index ab28d0d0..b7484c73 100644
--- a/client/ui/qml/Components/ConnectButton.qml
+++ b/client/ui/qml/Components/ConnectButton.qml
@@ -5,6 +5,7 @@ import QtQuick.Shapes
 import Qt5Compat.GraphicalEffects
 
 import ConnectionState 1.0
+import PageEnum 1.0
 
 Button {
     id: root
@@ -137,6 +138,13 @@ Button {
     }
 
     onClicked: {
+        if (!ContainersModel.isAnyContainerInstalled()) {
+            ServersModel.currentlyProcessedIndex = ServersModel.getDefaultServerIndex()
+            PageController.goToPage(PageEnum.PageSetupWizardEasy)
+
+            return
+        }
+
         if (ConnectionController.isConnectionInProgress) {
             ConnectionController.closeConnection()
         } else if (ConnectionController.isConnected) {
diff --git a/client/ui/qml/Pages2/PageSetupWizardEasy.qml b/client/ui/qml/Pages2/PageSetupWizardEasy.qml
index b228a7a3..4f94e985 100644
--- a/client/ui/qml/Pages2/PageSetupWizardEasy.qml
+++ b/client/ui/qml/Pages2/PageSetupWizardEasy.qml
@@ -183,6 +183,8 @@ PageType {
                 textColor: "#D7D8DB"
                 borderWidth: 1
 
+                visible: ContainersModel.isAnyContainerInstalled()
+
                 text: qsTr("Set up later")
 
                 onClicked: function() {