Qt wifi cleanup (#2594)

* cleanup wifi

* spacing

* cleanup

* little more

* wifi manager

* typo
old-commit-hash: 45eccc842a

selfdrive/ui/qt/offroad/wifi.cc

@@ -1,24 +1,12 @@
 #include <QDebug>
-#include <QListWidget>
-#include <QVBoxLayout>
 #include <QHBoxLayout>
 #include <QLabel>
 #include <QPixmap>
 #include <QPushButton>
-#include <QInputDialog>
 #include <QLineEdit>
-#include <QCoreApplication>
-#include <QButtonGroup>
-#include <QStackedWidget>
 
 #include "wifi.hpp"
-#include "wifiManager.hpp"
-#include "input_field.hpp"
 
-CustomConnectButton::CustomConnectButton(QString text, int iid){
-    setText(text);
-    id=iid;
-}
 
 void clearLayout(QLayout* layout) {
   while (QLayoutItem* item = layout->takeAt(0)) {
@@ -78,13 +66,19 @@ void WifiUI::refresh(){
   wifi->refreshNetworks();
 
   clearLayout(vlayout);
-  int i=0;
 
-  QButtonGroup* connectButtons=new QButtonGroup(this);
+  connectButtons = new QButtonGroup(this);
   QObject::connect(connectButtons, SIGNAL(buttonClicked(QAbstractButton*)), this, SLOT(handleButton(QAbstractButton*)));
+
+  int i = 0;
   for (Network &network : wifi->seen_networks){
     QHBoxLayout *hlayout = new QHBoxLayout;
+
+    // SSID
+    hlayout->addSpacing(50);
     hlayout->addWidget(new QLabel(QString::fromUtf8(network.ssid)));
+
+    // strength indicator
     unsigned int strength_scale = std::round(network.strength / 25.0) * 25;
     QPixmap pix("../assets/offroad/indicator_wifi_" + QString::number(strength_scale) + ".png");
     QLabel *icon = new QLabel();
@@ -93,21 +87,22 @@ void WifiUI::refresh(){
     hlayout->addWidget(icon);
     hlayout->addSpacing(20);
 
-    CustomConnectButton* m_button = new CustomConnectButton(network.connected ? "Connected" : "Connect",i);
+    // connect button
-    m_button->setFixedWidth(300);
+    QPushButton* btn = new QPushButton(network.connected ? "Connected" : "Connect");
-    m_button->setDisabled(network.connected || network.security_type == SecurityType::UNSUPPORTED);
+    btn->setFixedWidth(300);
-    connectButtons->addButton(m_button,i);
+    btn->setDisabled(network.connected || network.security_type == SecurityType::UNSUPPORTED);
-
+    hlayout->addWidget(btn);
-    hlayout->addWidget(m_button);
     hlayout->addSpacing(20);
 
+    connectButtons->addButton(btn, i++);
+
     QWidget * w = new QWidget;
     w->setLayout(hlayout);
     vlayout->addWidget(w);
 
     w->setStyleSheet(R"(
       QLabel {
-        font-size: 40px 
+        font-size: 40px;
       }
       QPushButton:enabled {
         background-color: #114265;
@@ -119,37 +114,33 @@ void WifiUI::refresh(){
         background-color: #114265;
       }
     )");
-    i+=1;
   }
 }
 
 void WifiUI::handleButton(QAbstractButton* button) {
-  CustomConnectButton* m_button = static_cast<CustomConnectButton*>(button);
+  QPushButton* btn = static_cast<QPushButton*>(button);
-  int id = m_button->id;
+  qDebug() << connectButtons->id(btn);
-  qDebug()<<id;
+  Network n = wifi->seen_networks[connectButtons->id(btn)];
-  Network n = wifi->seen_networks[id];
+
-  a->label->setText("Password for "+n.ssid);
+  a->label->setText("Enter password for \"" + n.ssid  + "\"");
+
   if(n.security_type == SecurityType::OPEN){
     wifi->connect(n);
   } else if (n.security_type == SecurityType::WPA){
-
     QString password = getStringFromUser();
-    if(password != ""){
+
+    if(password.size()){
       wifi->connect(n, password);
     }
   } else {
-    qDebug() << "Cannot determine a network's security type";
+    qDebug() << "Cannot determine network's security type";
   }
-
 }
 
 QString WifiUI::getStringFromUser(){
   swidget->setCurrentIndex(1);
-
   loop.exec();
-
   swidget->setCurrentIndex(0);
-
   return text;
 }
 

selfdrive/ui/qt/offroad/wifi.hpp

@@ -1,22 +1,14 @@
 #pragma once
-#include "wifiManager.hpp"
+
-#include "input_field.hpp"
 #include <QWidget>
-#include <QtDBus>
-#include <QPushButton>
 #include <QButtonGroup>
 #include <QVBoxLayout>
-#include <QStackedLayout>
 #include <QStackedWidget>
 #include <QTimer>
 
+#include "wifiManager.hpp"
+#include "input_field.hpp"
 
-class CustomConnectButton : public QPushButton{
-
-public:
-  explicit CustomConnectButton(QString text, int iid);
-  int id;
-};
 
 class WifiUI : public QWidget {
   Q_OBJECT
@@ -32,6 +24,8 @@ private:
   QEventLoop loop;
   QTimer * timer;
   QString text;
+  QButtonGroup *connectButtons;
+
   QString getStringFromUser();
 
 public:
@@ -41,6 +35,4 @@ private slots:
   void handleButton(QAbstractButton* m_button);
   void refresh();
   void receiveText(QString text);
-signals:
-  void gotText();
 };

selfdrive/ui/qt/offroad/wifiManager.cc

@@ -2,8 +2,7 @@
 #include <set>
 
 #include "wifiManager.hpp"
-#include "wifi.hpp"
+
-typedef QMap<QString, QMap<QString, QVariant> > Connection;
 
 QString nm_path                = "/org/freedesktop/NetworkManager";
 QString nm_settings_path       = "/org/freedesktop/NetworkManager/Settings";
@@ -91,6 +90,7 @@ SecurityType WifiManager::getSecurityType(QString path){
   int wpaflag = get_property(path, "WpaFlags").toInt();
   int rsnflag = get_property(path, "RsnFlags").toInt();
   int wpa_props = wpaflag | rsnflag;
+
   if(sflag == 0){
     return SecurityType::OPEN;
   } else if((sflag & 0x1) && (wpa_props & (0x333) && !(wpa_props & 0x200))) {
@@ -103,9 +103,11 @@ SecurityType WifiManager::getSecurityType(QString path){
     return SecurityType::UNSUPPORTED;
   }
 }
+
 void WifiManager::connect(Network n){
   return connect(n, "", "");
 }
+
 void WifiManager::connect(Network n, QString password){
   return connect(n, "", password);
 }
@@ -119,6 +121,7 @@ void WifiManager::connect(Network n, QString username, QString password){
   qDebug() << "Connecting to"<< n.ssid << "with username, password =" << username << "," <<password;
   connect(n.ssid, username, password, n.security_type);
 }
+
 void WifiManager::connect(QByteArray ssid, QString username, QString password, SecurityType security_type){
   Connection connection;
   connection["connection"]["type"] = "802-11-wireless";
@@ -145,7 +148,6 @@ void WifiManager::connect(QByteArray ssid, QString username, QString password, S
   } else {
     qDebug() << result.value().path();
   }
-
 }
 
 void WifiManager::print_active_connections(){
@@ -164,6 +166,7 @@ void WifiManager::print_active_connections(){
   }
   step4.endArray();
 }
+
 void WifiManager::clear_connections(QString ssid){
   QDBusInterface nm(nm_service, nm_settings_path, nm_settings_iface, bus);
   QDBusMessage response = nm.call("ListConnections");
@@ -194,6 +197,7 @@ void WifiManager::clear_connections(QString ssid){
     }
   }
 }
+
 void WifiManager::request_scan(){
   if (!has_adapter) return;
 
@@ -207,12 +211,14 @@ uint WifiManager::get_wifi_device_state(){
   uint resp = get_response<uint>(response);
   return resp;
 }
+
 QString WifiManager::get_active_ap(){
   QDBusInterface device_props(nm_service, adapter, props_iface, bus);
   QDBusMessage response = device_props.call("Get", wireless_device_iface, "ActiveAccessPoint");
   QDBusObjectPath r = get_response<QDBusObjectPath>(response);
   return r.path();
 }
+
 QByteArray WifiManager::get_property(QString network_path ,QString property){
   QDBusInterface device_props(nm_service, network_path, props_iface, bus);
   QDBusMessage response = device_props.call("Get", ap_iface, property);

selfdrive/ui/qt/offroad/wifiManager.hpp

@@ -1,18 +1,37 @@
 #pragma once
+
 #include <QWidget>
 #include <QtDBus>
-enum class SecurityType{OPEN, WPA, UNSUPPORTED};
+
+enum class SecurityType {
+  OPEN,
+  WPA,
+  UNSUPPORTED
+};
+
+typedef QMap<QString, QMap<QString, QVariant>> Connection;
 
 struct Network {
   QString path;
   QByteArray ssid;
   unsigned int strength;
   bool connected;
-
   SecurityType security_type;
 };
 
 class WifiManager{
+public:
+  explicit WifiManager();
+
+  bool has_adapter;
+  void request_scan();
+  QVector<Network> seen_networks;
+
+  void refreshNetworks();
+  void connect(Network ssid);
+  void connect(Network ssid, QString password);
+  void connect(Network ssid, QString username, QString password);
+
 private:
   QVector<QByteArray> seen_ssids;
   QString adapter;//Path to network manager wifi-device
@@ -25,19 +44,7 @@ class WifiManager{
   void clear_connections(QString ssid);
   void print_active_connections();
   uint get_wifi_device_state();
-    QByteArray get_ap_ssid(QString network_path);
   QByteArray get_property(QString network_path, QString property);
   unsigned int get_ap_strength(QString network_path);
   SecurityType getSecurityType(QString ssid);
-
-  public:
-    bool has_adapter;
-    void request_scan();
-    QVector<Network> seen_networks;
-
-    explicit WifiManager();
-    void refreshNetworks();
-    void connect(Network ssid);
-    void connect(Network ssid, QString password);
-    void connect(Network ssid, QString username, QString password);
 };