login/logout/ctrl+g working correctly

src/framework/net/connection.cpp

@@ -34,6 +34,7 @@ Connection::Connection() :
         m_socket(g_ioService)
 {
     m_connected = false;
+    m_connecting = false;
 }
 
 void Connection::poll()
@@ -50,6 +51,7 @@ void Connection::terminate()
 void Connection::connect(const std::string& host, uint16 port, const SimpleCallback& connectCallback)
 {
     m_connected = false;
+    m_connecting = true;
     m_connectCallback = connectCallback;
 
     asio::ip::tcp::resolver::query query(host, Fw::unsafeCast<std::string>(port));
@@ -61,22 +63,23 @@ void Connection::connect(const std::string& host, uint16 port, const SimpleCallb
 
 void Connection::close()
 {
-    if(!m_connected)
+    if(!m_connected && !m_connecting)
         return;
 
+    m_connecting = false;
     m_connected = false;
+    m_connectCallback = nullptr;
+    m_errorCallback = nullptr;
+    m_recvCallback = nullptr;
 
     m_readTimer.cancel();
     m_writeTimer.cancel();
 
     if(m_socket.is_open()) {
-        m_socket.shutdown(boost::asio::ip::tcp::socket::shutdown_both);
+        boost::system::error_code ec;
+        m_socket.shutdown(boost::asio::ip::tcp::socket::shutdown_both, ec);
         m_socket.close();
     }
-
-    m_connectCallback = nullptr;
-    m_errorCallback = nullptr;
-    m_recvCallback = nullptr;
 }
 
 void Connection::write(uint8* buffer, uint16 size)
@@ -129,9 +132,10 @@ void Connection::onConnect(const boost::system::error_code& error)
 {
     m_readTimer.cancel();
 
-    m_connected = true;
+    m_connecting = false;
 
     if(!error) {
+        m_connected = true;
         if(m_connectCallback)
             g_dispatcher.addEvent(m_connectCallback);
     } else
@@ -174,7 +178,7 @@ void Connection::handleError(const boost::system::error_code& error)
     if(error != asio::error::operation_aborted) {
         if(m_errorCallback)
             g_dispatcher.addEvent(std::bind(m_errorCallback, error));
-        if(m_connected)
+        if(m_connected || m_connecting)
             close();
     }
 }

src/framework/net/connection.h

@@ -51,6 +51,7 @@ public:
 
     void setErrorCallback(const ErrorCallback& errorCallback) { m_errorCallback = errorCallback; }
 
+    bool isConnecting() const { return m_connecting; }
     bool isConnected() const { return m_connected; }
 
 private:
@@ -73,6 +74,7 @@ private:
     uint8 m_recvBuffer[65538];
     uint16 m_recvSize;
     bool m_connected;
+    bool m_connecting;
 };
 
 #endif

src/framework/net/protocol.cpp

@@ -44,6 +44,20 @@ void Protocol::disconnect()
     }
 }
 
+bool Protocol::isConnected()
+{
+    if(m_connection && m_connection->isConnected())
+        return true;
+    return false;
+}
+
+bool Protocol::isConnecting()
+{
+    if(m_connection && m_connection->isConnecting())
+        return true;
+    return false;
+}
+
 void Protocol::send(OutputMessage& outputMessage)
 {
     // encrypt

src/framework/net/protocol.h

@@ -37,6 +37,9 @@ public:
     void connect(const std::string& host, uint16 port);
     void disconnect();
 
+    bool isConnected();
+    bool isConnecting();
+
     void send(OutputMessage& outputMessage);
     void recv();
 

src/framework/ui/uiwidget.cpp

@@ -374,8 +374,7 @@ void UIWidget::insertChild(int index, const UIWidgetPtr& child)
 void UIWidget::removeChild(const UIWidgetPtr& child)
 {
     // remove from children list
-    auto it = std::find(m_children.begin(), m_children.end(), child);
-    if(it != m_children.end()) {
+    if(hasChild(child)) {
         // defocus if needed
         bool focusAnother = false;
         if(m_focusedChild == child) {
@@ -383,9 +382,10 @@ void UIWidget::removeChild(const UIWidgetPtr& child)
             focusAnother = true;
         }
 
-        // unlock child if it was locked
-        unlockChild(child);
+        if(isChildLocked(child))
+            unlockChild(child);
 
+        auto it = std::find(m_children.begin(), m_children.end(), child);
         m_children.erase(it);
 
         // reset child parent
@@ -397,7 +397,7 @@ void UIWidget::removeChild(const UIWidgetPtr& child)
         // update child states
         child->updateStates();
 
-        if(focusAnother)
+        if(focusAnother && !m_focusedChild)
             focusPreviousChild(Fw::ActiveFocusReason);
     } else
         logError("attempt to remove an unknown child from a UIWidget");
@@ -474,7 +474,8 @@ void UIWidget::lockChild(const UIWidgetPtr& child)
     assert(hasChild(child));
 
     // prevent double locks
-    unlockChild(child);
+    if(isChildLocked(child))
+        unlockChild(child);
 
     // disable all other children
     for(const UIWidgetPtr& otherChild : m_children) {
@@ -488,7 +489,7 @@ void UIWidget::lockChild(const UIWidgetPtr& child)
 
     // lock child focus
     if(child->isFocusable())
-      focusChild(child, Fw::ActiveFocusReason);
+        focusChild(child, Fw::ActiveFocusReason);
 
     moveChildToTop(child);
 }
@@ -506,10 +507,12 @@ void UIWidget::unlockChild(const UIWidgetPtr& child)
 
     m_lockedChildren.erase(it);
 
-    // find new chick to lock
+    // find new child to lock
     UIWidgetPtr lockedChild;
-    if(m_lockedChildren.size() > 0)
+    if(m_lockedChildren.size() > 0) {
         lockedChild = m_lockedChildren.front();
+        assert(hasChild(lockedChild));
+    }
 
     for(const UIWidgetPtr& otherChild : m_children) {
         // lock new child
@@ -523,6 +526,19 @@ void UIWidget::unlockChild(const UIWidgetPtr& child)
         else
             otherChild->setEnabled(true);
     }
+
+    if(lockedChild) {
+        if(lockedChild->isFocusable())
+            focusChild(lockedChild, Fw::ActiveFocusReason);
+
+        moveChildToTop(lockedChild);
+    }
+}
+
+bool UIWidget::isChildLocked(const UIWidgetPtr& child)
+{
+    auto it = std::find(m_lockedChildren.begin(), m_lockedChildren.end(), child);
+    return it != m_lockedChildren.end();
 }
 
 void UIWidget::updateParentLayout()

src/framework/ui/uiwidget.h

@@ -135,6 +135,7 @@ public:
     void moveChildToTop(const UIWidgetPtr& child);
     void lockChild(const UIWidgetPtr& child);
     void unlockChild(const UIWidgetPtr& child);
+    bool isChildLocked(const UIWidgetPtr& child);
 
     void updateParentLayout();
     void updateLayout();