diff --git a/src/main/java/edu/group5/app/control/MainController.java b/src/main/java/edu/group5/app/control/MainController.java index d488a20..8106975 100644 --- a/src/main/java/edu/group5/app/control/MainController.java +++ b/src/main/java/edu/group5/app/control/MainController.java @@ -1,6 +1,7 @@ package edu.group5.app.control; import edu.group5.app.control.donationpage.DonationPageController; +import edu.group5.app.model.user.User; import edu.group5.app.view.MainView; import edu.group5.app.view.donationpage.DonationPageView; @@ -12,6 +13,7 @@ public class MainController { private final BrowseCardController browseCardController; private final OrganizationPageController organizationPageController; private final DonationPageController donationPageController; + private User currentUser; public MainController() { this.view = new MainView(this); @@ -23,6 +25,19 @@ public MainController() { this.donationPageController = new DonationPageController(this); } + public void setCurrentUser(User user) { + this.currentUser = user; + } + + public User getCurrentUser() { + return this.currentUser; + } + + public void logout() { + currentUser = null; + showLoginPage(); + } + public MainView getMainView() { return view; }