validation for login window

Author: StackTheCode

desktop-agent/src/main.cpp

@@ -231,30 +231,29 @@ ShowWindow(GetConsoleWindow(), SW_HIDE);
     }
 
 
- if (!loggedIn) {
+if (!loggedIn) {
     LoginWindow loginGui; 
-    if (loginGui.show()) { 
-        std::string enteredUser = loginGui.getUsername();
-        std::string enteredPass = loginGui.getPassword();
-        backendClient.setUsername(enteredUser);
-        backendClient.setPassword(enteredPass);
-
-        if (backendClient.login()) {
-            PasswordManager::storePassword(enteredUser, enteredPass);
-            config.setUsername(enteredUser);
-            config.save("config.json"); 
-            loggedIn = true;
+    while (!loggedIn) {  // Loop until authenticated or exited
+        if (loginGui.show()) { 
+            backendClient.setUsername(loginGui.getUsername());
+            backendClient.setPassword(loginGui.getPassword());
+
+            if (backendClient.login()) {
+                PasswordManager::storePassword(loginGui.getUsername(), loginGui.getPassword());
+                config.setUsername(loginGui.getUsername());
+                config.save("config.json"); 
+                loggedIn = true; // This breaks the while loop
+            } else {
+                loginGui.setErrorMessage("Invalid credentials. Please try again.");
+            } 
         } else {
-            // If the server rejects the login, we stop here.
-            return 1; 
-        } 
-    } else {
-        return 0; 
+            // If the user closes the window manually (X or Cancel)
+            return 0; 
+        }
     }
 }
 
 
-
     // Initialize camera
     CameraCapture camera(config.getCameraIndex());
     if (!camera.initialize()){

desktop-agent/src/ui/LoginWindow.cpp

@@ -127,13 +127,22 @@ bool LoginWindow::show(){
                              ImGuiInputTextFlags_Password);
     ImGui::PopItemWidth();
 
+// Error message if credentials are wrong
+if(!m_errorMessage.empty()){
+ImGui::Dummy(ImVec2(0,5));
+ImGui::PushStyleColor(ImGuiCol_Text,ImVec4(1.0f,0.3f,0.3f,1.0f));
+ImGui::TextWrapped("%s",m_errorMessage.c_str());
+ImGui::PopStyleColor(); 
+}
+
     ImGui::Dummy(ImVec2(0, 25));
 
 
     // Login button
        if(ImGui::Button("Sign In", ImVec2(-1, 55))){
             m_username = m_userBuf;
             m_password = m_passBuf;
+            m_errorMessage ="";
             ImGui::End();
             ImGui::Render();
             ImGui_ImplOpenGL3_RenderDrawData(ImGui::GetDrawData());

desktop-agent/src/ui/LoginWindow.h

@@ -14,7 +14,9 @@ class LoginWindow{
 
     std::string getUsername() const {return m_username; }
     std::string getPassword() const {return m_password;}
-
+    
+   void setErrorMessage(const std::string& message){m_errorMessage = message;}
+    
 
 
     private:
@@ -23,6 +25,7 @@ class LoginWindow{
        char m_passBuf[64] = "";
        std::string m_username;
        std::string m_password;
+       std::string m_errorMessage;
        bool m_loginTriggered = false;