Login working, segfaults sometimes

1 files changed, 11 insertions, 1 deletions

diff --git a/src/login/LoginController.cpp b/src/login/LoginController.cpp
index 64c8221..5a81a55 100644
--- a/src/login/LoginController.cpp
+++ b/src/login/LoginController.cpp
@@ -6,8 +6,9 @@
  */
 
 #include "LoginController.h"
+#include "client/User.h"
 
-LoginController::LoginController() {
+LoginController::LoginController(ControllerCallback * callback) {
 	// TODO Auto-generated constructor stub
 	loginView = new LoginView();
 	loginModel = new LoginModel();
@@ -15,6 +16,8 @@ LoginController::LoginController() {
 	loginView->AttachController(this);
 	loginModel->AddObserver(loginView);
 
+	this->callback = callback;
+
 }
 
 void LoginController::Login(string username, string password)
@@ -22,6 +25,11 @@ void LoginController::Login(string username, string password)
 	loginModel->Login(username, password);
 }
 
+User LoginController::GetUser()
+{
+	return loginModel->GetUser();
+}
+
 void LoginController::Exit()
 {
 	if(ui::Engine::Ref().GetWindow() == loginView)
@@ -29,6 +37,8 @@ void LoginController::Exit()
 		ui::Engine::Ref().CloseWindow();
 		loginView = NULL;
 	}
+	if(callback)
+		callback->ControllerExit();
 }
 
 LoginController::~LoginController() {