diff --git a/src/main/java/controllers/MainController.java b/src/main/java/controllers/MainController.java index 3030be4..2616aeb 100644 --- a/src/main/java/controllers/MainController.java +++ b/src/main/java/controllers/MainController.java @@ -1,12 +1,13 @@ package controllers; +import javafx.collections.FXCollections; +import javafx.collections.ObservableList; import javafx.fxml.FXML; import javafx.fxml.Initializable; -import javafx.scene.control.Alert; -import javafx.scene.control.Control; -import javafx.scene.control.PasswordField; -import javafx.scene.control.TextField; +import javafx.scene.control.*; import model.SceneCode; +import model.Session; +import model.User; import java.net.URL; import java.util.ResourceBundle; @@ -20,13 +21,18 @@ public class MainController extends Controller{ TextField username; @FXML PasswordField password; + @FXML + private ComboBox accountType; public void register() throws Exception{ changeScene(SceneCode.REGISTER); } public boolean validateLogin(String user, String pass){ - if (user.equals(pass)){ + if (Session.session.getDataManager().getUsers().get(user).checkPassword(pass) && User.AccountType.getValueOf(accountType.getValue()) == User.AccountType.PASSENGER){ + return true; + } + if (Session.session.getDataManager().getDrivers().get(user).checkPassword(pass) && User.AccountType.getValueOf(accountType.getValue()) == User.AccountType.DRIVER){ return true; } return false; @@ -44,6 +50,9 @@ public class MainController extends Controller{ @Override public void initialize(URL location, ResourceBundle resources) { - + ObservableList accountTypes = FXCollections.observableArrayList(); + accountTypes.addAll(User.AccountType.DRIVER.name, User.AccountType.PASSENGER.name); + accountType.setItems(accountTypes); + accountType.getSelectionModel().select(0); } } diff --git a/src/main/java/model/User.java b/src/main/java/model/User.java index b7b2479..58dac00 100644 --- a/src/main/java/model/User.java +++ b/src/main/java/model/User.java @@ -52,6 +52,10 @@ public class User { } + public boolean checkPassword(String p){ + return p.equals(password); + } + public void serialise(){ try { BufferedImage bi = SwingFXUtils.fromFXImage(photo, null); diff --git a/src/main/resources/scenes/main.fxml b/src/main/resources/scenes/main.fxml index ba20a99..f74bd72 100644 --- a/src/main/resources/scenes/main.fxml +++ b/src/main/resources/scenes/main.fxml @@ -23,8 +23,7 @@ - - + @@ -47,6 +46,7 @@ +