public class PasswordUIController implements Initializable {

private PasswordList passList;
/**
 * Initializes the controller class.
 */
@FXML
TreeTableView passwordInfo;
@FXML
TreeTableColumn<String,String> acctColumn;
@FXML
Parent root;
@FXML
private void backButtonAction(ActionEvent event) throws Exception {
    Stage stage = (Stage) root.getScene().getWindow();
    FXMLLoader loader = new FXMLLoader(getClass().getResource("MainMenuUI.fxml"));
    Pane mainMenuUI = (Pane)loader.load();
    MainMenuController controller = loader.<MainMenuController>getController();
    controller.setPassList(passList);

    Scene scene = new Scene(mainMenuUI);
    stage.setTitle("Main Menu");
    stage.setScene(scene);
    stage.show();
}
@FXML
private void addButtonAction(ActionEvent event) throws Exception {
    Stage stage = (Stage) root.getScene().getWindow();
    FXMLLoader loader = new FXMLLoader(getClass().getResource("CreatePasswordUI.fxml"));
    Pane createPassUI = (Pane)loader.load();
    CreatePasswordUIController controller = loader.<CreatePasswordUIController>getController();
    controller.setPassList(passList);

    Scene scene = new Scene(createPassUI);
    stage.setTitle("Create Password");
    stage.setScene(scene);
    stage.show();
}
@Override
public void initialize(URL url, ResourceBundle rb) {
    PasswordList pList = passList;
    if (pList != null) {
        ArrayList<Password> thePasswords = pList.getPasswordList();
        TreeItem<String> passwords = new TreeItem<>("Passwords");
        for (int i = 0; i < thePasswords.size(); i++) {
            Password thePass = thePasswords.get(i);
            //Creating tree items
            TreeItem<String> username = new TreeItem<>("Username: " + thePass.getUsername());
            TreeItem<String> password = new TreeItem<>("Password: " + thePass.getPassword());
            TreeItem<String> comment = new TreeItem<>("Comment: " + thePass.getComment());

            //Creating the root element
            TreeItem<String> account = new TreeItem<>(thePass.getAccount());
            account.setExpanded(true);

            //Adding tree items to the root
            account.getChildren().setAll(username, password, comment);
            passwords.getChildren().add(account);
        }
        //Creating a column
        acctColumn = new TreeTableColumn<>("Account");

        //Defining cell content
        acctColumn.setCellValueFactory(new Callback<TreeTableColumn.CellDataFeatures<String, String>, ObservableValue<String>>() {
            @Override
            public ObservableValue<String> call(TreeTableColumn.CellDataFeatures<String, String> param){
                return new SimpleStringProperty(param.getValue().getValue());
            }
        });

        //Creating a tree table view
        passwordInfo = new TreeTableView<>();
        passwordInfo.setRoot(passwords);
        passwordInfo.getColumns().add(acctColumn);
    } else {

    }
}

/**
 * @return the passList
 */
public PasswordList getPassList() {
    return passList;
}

/**
 * @param passList the passList to set
 */
public void setPassList(PasswordList passList) {
    this.passList = passList;
}


    }

public class CreatePasswordUIController implements Initializable {

private PasswordList passList;
/**
 * Initializes the controller class.
 */
@FXML
Parent root;
@FXML
TextField accountTxt;
@FXML
TextField usernameTxt;
@FXML
TextField passwordTxt;
@FXML
TextField passwordTxt2;
@FXML
TextArea commentArea;
@FXML
TextField passwordGeneratorTxt;
@FXML
Label rating;
@FXML
Label matching;
@FXML
Label incorrect;
@FXML
private void backButtonAction(ActionEvent event) throws Exception {
    Stage stage = (Stage) root.getScene().getWindow();
    FXMLLoader loader = new FXMLLoader(getClass().getResource("MainMenuUI.fxml"));
    Pane mainMenuUI = (Pane)loader.load();
    MainMenuController controller = loader.<MainMenuController>getController();
    controller.setPassList(passList);

    Scene scene = new Scene(mainMenuUI);
    stage.setTitle("Main Menu");
    stage.setScene(scene);
    stage.show();
}
@FXML
 private void submitButtonAction(ActionEvent event) throws Exception {
    String account = accountTxt.getText();
    String username = usernameTxt.getText();
    String password = passwordTxt.getText();
    String password2 = passwordTxt2.getText();
    String comment = commentArea.getText();
    if (password.equals(password2) && username.length()>2) {
        PasswordList pList = passList;
        Password thePass = new Password(account, username, password, comment);
        pList.addPassword(thePass);
        Stage stage = (Stage) root.getScene().getWindow();
        FXMLLoader loader = new FXMLLoader(getClass().getResource("PasswordUI.fxml"));
        Pane passwordUI = (Pane)loader.load();
        PasswordUIController controller = loader.<PasswordUIController>getController();
        controller.setPassList(pList);

        Scene scene = new Scene(passwordUI);
        stage.setTitle("Password");
        stage.setScene(scene);
        stage.show();
    } else {
        incorrect.setVisible(true);
    }
}
@FXML
private void generateButtonAction(ActionEvent event) {
    PasswordGenerator passGen = new PasswordGenerator();
    String password = passGen.generatePassword(true);
    passwordGeneratorTxt.setText(password);
}
@Override
public void initialize(URL url, ResourceBundle rb) {
    PasswordRater rater = new PasswordRater();
    passwordTxt.textProperty().addListener(new ChangeListener() {
        @Override
        public void changed(ObservableValue observable, Object oldVal,Object newVal) {
           String password = passwordTxt.getText();
           rater.setPassword(password);
           String ratingStr = rater.displayLevel();
           rating.setText(ratingStr);
           rating.setVisible(true);
           if (passwordTxt.getText().equals(passwordTxt2.getText()) && !password.equals("")) {
                matching.setText("Good!");
           }
           else
           {
               matching.setText("Passwords do not match.");
           }
        }
    });
    passwordTxt2.textProperty().addListener(new ChangeListener() {
        @Override
        public void changed(ObservableValue observable, Object oldVal,Object newVal) {
           String password = passwordTxt.getText();
           String password2 = passwordTxt2.getText();
            if (password.equals(password2) && !password.equals("")) {
                matching.setText("Good!");
            } else {
                matching.setText("Passwords do not match.");
            }
            matching.setVisible(true);
            }
    });
}

/**
 * @return the passList
 */
public PasswordList getPassList() {
    return passList;
}

/**
 * @param passList the passList to set
 */
public void setPassList(PasswordList passList) {
    this.passList = passList;
}



    }