Correct the logic for incorrect passwords. Looks like spaghetti. 🍝

src/installerprompt.cpp

@@ -12,7 +12,6 @@
 #include <QScreen>
 #include <QMessageBox>
 #include <QUuid>
-#include <QLineEdit>
 #include <QDBusPendingReply>
 #include "installerprompt.h"
 #include "./ui_installerprompt.h"
@@ -22,6 +21,9 @@ InstallerPrompt::InstallerPrompt(QWidget *parent)
     , ui(new Ui::InstallerPrompt) {
     ui->setupUi(this);
 
+    // Hide the Incorrect Password text
+    ui->incorrectPassword->setVisible(false);
+
     // Set the background image and scale it
     QPixmap bg(":/background");
     if (bg.isNull()) {
@@ -78,6 +80,8 @@ void InstallerPrompt::updateConnectionStatus() {
 
     switch (status) {
         case NetworkManager::Status::Disconnected:
+        case NetworkManager::ConnectedLinkLocal:
+        case NetworkManager::Asleep:
             statusText = tr("Not Connected");
             statusIndicator = "<span style=\"color: red;\">❌</span> " + statusText;
             break;
@@ -95,6 +99,7 @@ void InstallerPrompt::updateConnectionStatus() {
             statusIndicator = "<span style=\"color: yellow;\">🟡</span> " + statusText;
             break;
         default:
+            qDebug() << "Unknown status:" << status;
             statusText = tr("Unknown Status");
             statusIndicator = "<span style=\"color: grey;\">⚪</span> " + statusText;
     }
@@ -120,162 +125,154 @@ void InstallerPrompt::updateConnectionStatus() {
 
 void InstallerPrompt::handleWiFiConnectionChange(NetworkManager::Device::State newstate, NetworkManager::Device::State oldstate, NetworkManager::Device::StateChangeReason reason)
 {
-    if (reason == NetworkManager::Device::NoSecretsReason) {
+    QMutexLocker locker(&wifiChangeMutex);
+    if (reason == NetworkManager::Device::NoSecretsReason && !wifiWrongHandling) {
+        wifiWrongHandling = true;
+        qDebug() << wifiSSID;
+        
         foreach (const NetworkManager::Connection::Ptr &connection, NetworkManager::listConnections()) {
             if (connection->settings()->connectionType() == NetworkManager::ConnectionSettings::Wireless) {
                 auto wirelessSetting = connection->settings()->setting(NetworkManager::Setting::Wireless).dynamicCast<NetworkManager::WirelessSetting>();
-                if (wirelessSetting && wirelessSetting->ssid() == ui->networkComboBox->currentText()) {
-                    qDebug() << "Wiping connection with wrong password: " << ui->networkComboBox->currentText();
+                if (wirelessSetting && wirelessSetting->ssid() == wifiSSID) {
+                    qDebug() << "Wiping connection with wrong password: " << wifiSSID;
+                    // Show the Incorrect Password text
+                    ui->incorrectPassword->setVisible(true);
                     QDBusPendingReply removeReply = connection->remove();
                     removeReply.waitForFinished();
-                    handleWifiConnection(ui->networkComboBox->currentText(), true);
                 }
             }
         }
+        wifiWrongHandling = false;
     }
 }
 
-void InstallerPrompt::handleWifiConnection(const QString &ssid, bool recoverFromWrongPassword) {
-    ui->networkComboBox->setEnabled(false);
-    qDebug() << "Attempting to find connection for SSID:" << ssid;
-    if (!recoverFromWrongPassword) {
-        foreach (const NetworkManager::Connection::Ptr &connection, NetworkManager::listConnections()) {
-            if (connection->settings()->connectionType() == NetworkManager::ConnectionSettings::Wireless) {
-                auto wirelessSetting = connection->settings()->setting(NetworkManager::Setting::Wireless).dynamicCast<NetworkManager::WirelessSetting>();
-                if (wirelessSetting && wirelessSetting->ssid() == ssid) {
-                    qDebug() << "Attempting to use existing connection:" << ssid;
-                    NetworkManager::activateConnection(connection->path(), wifiDevice->uni(), QString());
-                    qDebug() << "Successfully connected:" << ssid;
-                    ui->networkComboBox->setEnabled(true);
-                    return;
-                }
+NetworkManager::Connection::Ptr InstallerPrompt::findConnectionBySsid(const QString &ssid) {
+    foreach (const NetworkManager::Connection::Ptr &connection, NetworkManager::listConnections()) {
+        if (connection->settings()->connectionType() == NetworkManager::ConnectionSettings::Wireless) {
+            auto wirelessSetting = connection->settings()->setting(NetworkManager::Setting::Wireless).dynamicCast<NetworkManager::WirelessSetting>();
+            if (wirelessSetting && wirelessSetting->ssid() == ssid) {
+                return connection;
             }
         }
     }
+    return NetworkManager::Connection::Ptr(); // Return null pointer if not found
+}
 
-    QMap<QString, QVariant> fullSettings = createSettingsBySSID(ssid);
-    NMVariantMapMap nmMap;
+QString InstallerPrompt::promptForWifiPassword(const QString &ssid, bool isWrongPassword) {
+    QDialog passwordDialog(this);
+    passwordDialog.setModal(true);
+    passwordDialog.setWindowTitle(tr("Wi-Fi Password Required"));
+    passwordDialog.setWindowIcon(QIcon::fromTheme("network-wireless"));
+    passwordDialog.setStyleSheet("QLabel { color: black; } ");
+    passwordDialog.setMinimumWidth(250);
+    passwordDialog.setMinimumHeight(120);
+    passwordDialog.setMaximumWidth(5000);
+    passwordDialog.setMaximumHeight(500);
+
+    QVBoxLayout layout;
+    QLabel passwordLabel(tr("Enter password for \"%1\":").arg(ssid), &passwordDialog);
+    QLineEdit passwordLineEdit(&passwordDialog);
+    QPushButton passwordButton(tr("Connect"), &passwordDialog);
+
+    passwordLineEdit.setEchoMode(QLineEdit::Password);
+    layout.addWidget(&passwordLabel);
+    layout.addWidget(&passwordLineEdit);
+    layout.addWidget(&passwordButton);
+
+    passwordDialog.setLayout(&layout);
+
+    // Connect with a lambda function for inline validation
+    connect(&passwordLineEdit, &QLineEdit::textChanged, this, [&passwordLineEdit](const QString &text) {
+        int minLength = 8;
+        int maxLength = 64;
+        bool isValid = text.length() >= minLength && text.length() <= maxLength;
+        passwordLineEdit.setStyleSheet(isValid ? "" : "border: 1px solid red;");
+    });
+
+    connect(&passwordButton, &QPushButton::clicked, &passwordDialog, &QDialog::accept);
 
-    for (const auto &key : fullSettings.keys()) {
-        nmMap[key] = fullSettings[key].toMap();
+    if (passwordDialog.exec() == QDialog::Accepted) {
+        return passwordLineEdit.text();
     }
 
-    NetworkManager::ConnectionSettings::Ptr newConnectionSettings(new NetworkManager::ConnectionSettings(NetworkManager::ConnectionSettings::Wireless));
-    newConnectionSettings->fromMap(nmMap);
-    QVariantMap wirelessSecurity = fullSettings.value("802-11-wireless-security").toMap();
+    return QString();
+}
 
-    //bool isOpenNetwork = wirelessSecurity.isEmpty() || wirelessSecurity.value("key-mgmt").toString().isEmpty();
-    //if (isOpenNetwork && wifiDevice && wifiDevice->isValid() && connection) {
-    //    qDebug() << "Attempting to connect to open network: " << ssid;
-    //    NetworkManager::activateConnection(connection->path(), wifiDevice->uni(), QString());
-    //    return;
-    //}
+void InstallerPrompt::handleWifiConnection(const QString &ssid, bool recoverFromWrongPassword) {
+    ui->incorrectPassword->setVisible(false);
+    ui->networkComboBox->setEnabled(false);
+    wifiSSID = ssid;
+    qDebug() << "Attempting to find connection for SSID:" << ssid;
 
-    // If the network is secured, display the password dialog
-    QDialog passwordDialog(this);
-    QVBoxLayout layout(&passwordDialog);
-    QLabel label(&passwordDialog);
-    if (!recoverFromWrongPassword) {
-        label.setText(tr("Enter Wi-Fi Password for %1:").arg(ssid));
-        label.setStyleSheet("color: black");
-    } else {
-        label.setText(tr("Wrong Wi-Fi password for %1. Try again:").arg(ssid));
-        label.setStyleSheet("color: red");
+    // Check for existing connection
+    NetworkManager::Connection::Ptr connection = findConnectionBySsid(ssid);
+    if (connection && !recoverFromWrongPassword) {
+        qDebug() << "Using existing connection for:" << ssid;
+        NetworkManager::activateConnection(connection->path(), wifiDevice->uni(), QString());
+        ui->networkComboBox->setEnabled(true);
+        return;
     }
-    QLineEdit lineEdit(&passwordDialog);
-    QPushButton button(tr("Connect"), &passwordDialog);
 
-    lineEdit.setEchoMode(QLineEdit::Password);
-    layout.addWidget(&label);
-    layout.addWidget(&lineEdit);
-    layout.addWidget(&button);
+    // Prompt for Wi-Fi password
+    QString password = promptForWifiPassword(ssid);
+    if (password.isEmpty()) {
+        ui->networkComboBox->setEnabled(true);
+        return;
+    }
 
-    connect(&button, &QPushButton::clicked, &passwordDialog, &QDialog::accept);
+    // Create new Wi-Fi connection
+    NMVariantMapMap settings = createSettingsBySSID(ssid);
+    if (settings.isEmpty()) {
+        QMessageBox::warning(this, tr("Error"), tr("Failed to create Wi-Fi settings."));
+        ui->networkComboBox->setEnabled(true);
+        return;
+    }
 
-    for (int attempts = 0; attempts < 3; ++attempts) {
-        if (passwordDialog.exec() == QDialog::Rejected) {
-            ui->networkComboBox->setEnabled(true);
-            return;
-        }
-        QString password = lineEdit.text();
-        if (wifiDevice && wifiDevice->isValid()) {
-            // Update the wireless security settings in the map
-            wirelessSecurity["key-mgmt"] = "wpa-psk";
-            wirelessSecurity["psk"] = password;
-            fullSettings["802-11-wireless-security"] = wirelessSecurity;
-
-            // Convert QMap<QString, QVariant> to NMVariantMapMap
-            NMVariantMapMap nmMap;
-            for (const auto &key : fullSettings.keys()) {
-                nmMap[key] = fullSettings[key].toMap();
-            }
+    // Update the wireless security settings
+    QVariantMap wirelessSecurity;
+    wirelessSecurity["key-mgmt"] = "wpa-psk";
+    wirelessSecurity["psk"] = password;
+    settings["802-11-wireless-security"] = wirelessSecurity;
+
+    // Add the new connection
+    QDBusPendingReply<QDBusObjectPath> reply = NetworkManager::addConnection(settings);
+    reply.waitForFinished();
+    if (reply.isError()) {
+        QMessageBox::warning(this, tr("Error"), tr("Failed to add Wi-Fi connection."));
+        ui->networkComboBox->setEnabled(true);
+        return;
+    }
 
-            // Update the connection settings
-            qDebug() << "Saving the connection...";
-            QDBusObjectPath path;
-            NetworkManager::ConnectionSettings::Ptr newConnectionSettings(new NetworkManager::ConnectionSettings(NetworkManager::ConnectionSettings::Wireless));
-            newConnectionSettings->fromMap(nmMap);
-            QDBusPendingReply<QDBusObjectPath> addreply = NetworkManager::addConnection(nmMap);
-            addreply.waitForFinished();
-            if (addreply.isError()) {
-                qDebug() << nmMap;
-                qDebug() << "Unable to save the connection:" << addreply.error().message();
-            } else {
-                path = addreply.value();
-                qDebug() << "Added connection path:" << path.path();
-            }
+    // Activate the new connection
+    QDBusObjectPath path = reply.value();
+    NetworkManager::activateConnection(path.path(), wifiDevice->uni(), QString());
+    ui->networkComboBox->setEnabled(true);
+}
 
-            NetworkManager::Connection::Ptr connection = NetworkManager::findConnection(path.path());
-            if (!connection) {
-                qDebug() << "Unable to retrieve the connection after saving:" << addreply.error().message();
-            }
+NMVariantMapMap InstallerPrompt::createSettingsBySSID(const QString &ssid) {
+    NMVariantMapMap convertedSettings;
 
-            QDBusPendingReply<QDBusObjectPath> reply = NetworkManager::activateConnection(connection->path(), wifiDevice->uni(), QString());
-            reply.waitForFinished();
-            if (reply.isError()) {
-                qDebug() << "Unable to activate the connection:" << addreply.error().message();
-                QMessageBox::warning(this, tr("Connection Failed"), tr("Unable to connect to the network."));
-                ui->networkComboBox->setEnabled(true);
-                return;
-            } else {
-                NetworkManager::reloadConnections();
-                qDebug() << "Successfully connected:" << ssid;
-                // ui->networkComboBox->setEnabled(true); !!! We don't run this here since we actually *want* the box to remain locked right now
-                return;
-            }
-        }
+    if (!wifiDevice) {
+        qWarning() << "Wi-Fi device not found. Unable to set interface name.";
+        return convertedSettings;
     }
-}
 
-QMap<QString, QVariant> InstallerPrompt::createSettingsBySSID(const QString &ssid) {
-    // Create new connection settings
     NetworkManager::ConnectionSettings::Ptr newConnectionSettings(new NetworkManager::ConnectionSettings(NetworkManager::ConnectionSettings::Wireless));
     newConnectionSettings->setId(ssid);
     newConnectionSettings->setUuid(NetworkManager::ConnectionSettings::createNewUuid());
-
-    // Set interface name from wifiDevice
-    if (wifiDevice) {
-        newConnectionSettings->setInterfaceName(wifiDevice->interfaceName());
-    } else {
-        qWarning() << "Wi-Fi device not found. Unable to set interface name.";
-        return QMap<QString, QVariant>();
-    }
+    newConnectionSettings->setInterfaceName(wifiDevice->interfaceName());
 
     // Configure wireless settings
     QVariantMap wirelessSetting;
     wirelessSetting.insert("ssid", ssid.toUtf8());
+    convertedSettings.insert("802-11-wireless", wirelessSetting);
 
-    // Configure wireless security settings
-    NetworkManager::WirelessSecuritySetting::Ptr wirelessSecuritySetting = newConnectionSettings->setting(NetworkManager::Setting::WirelessSecurity).staticCast<NetworkManager::WirelessSecuritySetting>();
-    wirelessSecuritySetting->setKeyMgmt(NetworkManager::WirelessSecuritySetting::WpaPsk);
-
-    // Convert settings to QVariantMap
-    QMap<QString, QVariant> convertedSettings;
+    // Convert other settings
     const auto settingsMap = newConnectionSettings->toMap();
     for (const auto &key : settingsMap.keys()) {
-        convertedSettings[key] = QVariant::fromValue(settingsMap[key]);
+        QVariant value = settingsMap.value(key);
+        convertedSettings.insert(key, value.toMap());
     }
-    convertedSettings.insert("802-11-wireless", wirelessSetting);
 
     return convertedSettings;
 }
@@ -334,7 +331,6 @@ void InstallerPrompt::refreshNetworkList() {
     }
 
     // Update the main map and combo box only after the new list is ready
-    wifiNetworkMap.swap(tempWifiNetworkMap);
     ui->networkComboBox->clear();
     ui->networkComboBox->addItems(ssidList);
 

src/installerprompt.h

@@ -7,16 +7,12 @@
 #include <QPushButton>
 #include <QLabel>
 #include <QDialog>
+#include <QMutex>
+#include <QLineEdit>
 #include <NetworkManagerQt/Device>
 #include <NetworkManagerQt/WirelessDevice>
 #include <NetworkManagerQt/WirelessNetwork>
 
-namespace NetworkManager {
-    class Device;
-    class WirelessDevice;
-    class WirelessNetwork;
-}
-
 namespace Ui { class InstallerPrompt; }
 
 class InstallerPrompt : public QMainWindow {
@@ -39,14 +35,19 @@ private:
     Ui::InstallerPrompt *ui;
     QProcess *process;
     NetworkManager::WirelessDevice::Ptr wifiDevice;
-    QMap<QString, NetworkManager::WirelessNetwork::Ptr> wifiNetworkMap;
+    QString wifiSSID;
+    QMutex wifiChangeMutex;
+    NetworkManager::Connection::Ptr findConnectionBySsid(const QString &ssid);
+    bool wifiWrongHandling = false;
+    QLineEdit *passwordLineEdit;
 
     void handleWifiConnection(const QString &ssid, bool recoverFromWrongPassword = false);
+    QString promptForWifiPassword(const QString &ssid, bool isWrongPassword = false);
+    void connectToWifi(const QString &ssid, const QString &password, bool recoverFromWrongPassword = false);
     void initLanguageComboBox();
     QStringList getAvailableLanguages() const;
     void showWifiOptions();
-    NetworkManager::Connection::Ptr findConnectionBySsid(const QString &ssid);
-    QMap<QString, QVariant> createSettingsBySSID(const QString &ssid);
+    NMVariantMapMap createSettingsBySSID(const QString &ssid);
 };
 
 #endif // INSTALLERPROMPT_H

src/installerprompt.ui

@@ -59,7 +59,10 @@ QLabel#logoLabel {
     image: url(:/logo);
     background-color: transparent;
 }
-</string>
+
+QLabel#incorrectPassword {
+    color: red;
+}</string>
   </property>
   <widget class="QWidget" name="centralwidget">
    <property name="sizePolicy">
@@ -128,6 +131,7 @@ QLabel#logoLabel {
           <property name="font">
            <font>
             <family>Ubuntu</family>
+            <weight>50</weight>
             <italic>false</italic>
             <bold>false</bold>
            </font>
@@ -200,6 +204,7 @@ QLabel#logoLabel {
           <property name="font">
            <font>
             <pointsize>18</pointsize>
+            <weight>75</weight>
             <bold>true</bold>
            </font>
           </property>
@@ -258,80 +263,61 @@ QLabel#logoLabel {
        </spacer>
       </item>
       <item>
-       <layout class="QHBoxLayout" name="WiFiLayout">
-        <item>
-         <spacer name="networkSpacer1">
-          <property name="orientation">
-           <enum>Qt::Horizontal</enum>
+       <layout class="QGridLayout" name="WiFiLayout">
+        <item row="1" column="3">
+         <widget class="QPushButton" name="connectWiFiButton">
+          <property name="minimumSize">
+           <size>
+            <width>150</width>
+            <height>65</height>
+           </size>
           </property>
-          <property name="sizeHint" stdset="0">
+          <property name="maximumSize">
            <size>
-            <width>40</width>
-            <height>20</height>
+            <width>150</width>
+            <height>65</height>
            </size>
           </property>
-         </spacer>
-        </item>
-        <item>
-         <widget class="QLabel" name="WiFiLabel">
           <property name="font">
            <font>
-            <pointsize>18</pointsize>
+            <weight>75</weight>
             <bold>true</bold>
            </font>
           </property>
           <property name="text">
-           <string>Select a Wi-Fi Network:</string>
+           <string>Connect</string>
           </property>
          </widget>
         </item>
-        <item>
-         <widget class="QComboBox" name="networkComboBox">
-          <property name="minimumSize">
-           <size>
-            <width>352</width>
-            <height>50</height>
-           </size>
-          </property>
-          <property name="maximumSize">
-           <size>
-            <width>400</width>
-            <height>50</height>
-           </size>
-          </property>
+        <item row="2" column="2">
+         <widget class="QLabel" name="WiFiInfoLabel">
           <property name="font">
            <font>
-            <pointsize>16</pointsize>
+            <weight>75</weight>
+            <bold>true</bold>
            </font>
           </property>
+          <property name="text">
+           <string>(For advanced network configuration, select &quot;Try Lubuntu&quot;)</string>
+          </property>
          </widget>
         </item>
-        <item>
-         <widget class="QPushButton" name="connectWiFiButton">
-          <property name="minimumSize">
-           <size>
-            <width>100</width>
-            <height>65</height>
-           </size>
-          </property>
-          <property name="maximumSize">
-           <size>
-            <width>16777215</width>
-            <height>65</height>
-           </size>
-          </property>
+        <item row="1" column="1">
+         <widget class="QLabel" name="WiFiLabel">
           <property name="font">
            <font>
+            <pointsize>18</pointsize>
+            <weight>75</weight>
             <bold>true</bold>
            </font>
           </property>
           <property name="text">
-           <string>Connect</string>
+           <string>Select a Wi-Fi Network:</string>
           </property>
          </widget>
         </item>
-        <item>
-         <spacer name="networkSpacer2">
+        <item row="2" column="3">
+         <spacer name="advancedSpacer_2">
           <property name="orientation">
            <enum>Qt::Horizontal</enum>
           </property>
@@ -343,12 +329,8 @@ QLabel#logoLabel {
           </property>
          </spacer>
         </item>
-       </layout>
-      </item>
-      <item>
-       <layout class="QHBoxLayout" name="WiFiLayout_2">
-        <item>
-         <spacer name="advancedSpacer">
+        <item row="1" column="4">
+         <spacer name="networkSpacer2">
           <property name="orientation">
            <enum>Qt::Horizontal</enum>
           </property>
@@ -360,20 +342,21 @@ QLabel#logoLabel {
           </property>
          </spacer>
         </item>
-        <item>
-         <widget class="QLabel" name="WiFiInfoLabel">
-          <property name="font">
-           <font>
-            <bold>true</bold>
-           </font>
+        <item row="1" column="0">
+         <spacer name="networkSpacer1">
+          <property name="orientation">
+           <enum>Qt::Horizontal</enum>
           </property>
-          <property name="text">
-           <string>(For advanced network configuration, select &quot;Try Lubuntu&quot;)</string>
+          <property name="sizeHint" stdset="0">
+           <size>
+            <width>40</width>
+            <height>20</height>
+           </size>
           </property>
-         </widget>
+         </spacer>
         </item>
-        <item>
-         <spacer name="advancedSpacer_2">
+        <item row="2" column="1">
+         <spacer name="advancedSpacer">
           <property name="orientation">
            <enum>Qt::Horizontal</enum>
           </property>
@@ -385,6 +368,44 @@ QLabel#logoLabel {
           </property>
          </spacer>
         </item>
+        <item row="1" column="2">
+         <widget class="QComboBox" name="networkComboBox">
+          <property name="minimumSize">
+           <size>
+            <width>352</width>
+            <height>50</height>
+           </size>
+          </property>
+          <property name="maximumSize">
+           <size>
+            <width>550</width>
+            <height>50</height>
+           </size>
+          </property>
+          <property name="font">
+           <font>
+            <pointsize>16</pointsize>
+           </font>
+          </property>
+         </widget>
+        </item>
+        <item row="0" column="2">
+         <widget class="QLabel" name="incorrectPassword">
+          <property name="enabled">
+           <bool>true</bool>
+          </property>
+          <property name="font">
+           <font>
+            <pointsize>16</pointsize>
+            <weight>75</weight>
+            <bold>true</bold>
+           </font>
+          </property>
+          <property name="text">
+           <string>You entered an incorrect password. Please try again.</string>
+          </property>
+         </widget>
+        </item>
        </layout>
       </item>
       <item>
@@ -459,6 +480,7 @@ QLabel#logoLabel {
            <font>
             <family>Ubuntu</family>
             <pointsize>24</pointsize>
+            <weight>75</weight>
             <bold>true</bold>
            </font>
           </property>
@@ -535,6 +557,7 @@ QToolTip {
            <font>
             <family>Ubuntu</family>
             <pointsize>24</pointsize>
+            <weight>75</weight>
             <bold>true</bold>
            </font>
           </property>
@@ -677,6 +700,7 @@ QToolTip {
           <property name="font">
            <font>
             <pointsize>24</pointsize>
+            <weight>75</weight>
             <bold>true</bold>
            </font>
           </property>