private void saveBudget() {
if (!(currentBudget.getName().equals("No budget loaded"))) {
Savior.writeBudgetToDisk(currentBudget.getName(), currentBudget);
} else {
errorMessage = Printer.getPrintout("noBudgetLoaded");
getOption(this);
}
}
private void reportToConsole() {
if (!(currentBudget.getName().equals("No budget loaded"))) {
System.out.println(currentBudget.toString());
} else {
errorMessage = Printer.getPrintout("noBudgetLoaded");
getOption(this);
}
}
private void startEditor() {
if (!(currentBudget.getName().equals("No budget loaded"))) {
EditorMenu editor = new EditorMenu(currentBudget);
while (editor.stillEditingBudget()) editor.getOption(editor);
} else {
errorMessage = Printer.getPrintout("noBudgetLoaded");
getOption(this);
}
}
public class MainMenu implements MasterMenu {
private Budget currentBudget;
private String errorMessage = "";
public MainMenu(Budget currentBudget) {
this.currentBudget = currentBudget;
}
public void setErrorMessage(String newMessage) {
errorMessage = newMessage;
}
public int getNumberOfOptions() {
return menuOptions.length;
}
private MasterOption[] menuOptions =
new MasterOption[] {
new MasterOption() {
String optionPrintout = Printer.getPrintout("reportToConsoleOption");
public void optionMethod() {
reportToConsole();
}
public String printOption() {
return optionPrintout;
}
},
new MasterOption() {
String optionPrintout = Printer.getPrintout("reportToDiskOption");
public void optionMethod() {
reportToDisk();
}
public String printOption() {
return optionPrintout;
}
},
new MasterOption() {
String optionPrintout = Printer.getPrintout("startEditorOption");
public void optionMethod() {
startEditor();
}
public String printOption() {
return optionPrintout;
}
},
new MasterOption() {
String optionPrintout = Printer.getPrintout("saveBudgetOption");
public void optionMethod() {
saveBudget();
}
public String printOption() {
return optionPrintout;
}
},
new MasterOption() {
String optionPrintout = Printer.getPrintout("openBudgetOption");
public void optionMethod() {
openBudget();
}
public String printOption() {
return optionPrintout;
}
},
new MasterOption() {
String optionPrintout = Printer.getPrintout("createBudgetOption");
public void optionMethod() {
createBudget();
}
public String printOption() {
return optionPrintout;
}
},
new MasterOption() {
String optionPrintout = Printer.getPrintout("exitOption");
public void optionMethod() {
exitProgram();
}
public String printOption() {
return optionPrintout;
}
}
};
public void displayMenu() {
AnsiConsole.out.println(ansi().eraseScreen());
Printer.print("mainMenuHeader");
System.out.println("Budget in use: " + currentBudget.getName() + "\n");
Printer.printMenuOptions(menuOptions);
}
public void getOption(MasterMenu menu) {
displayMenu();
System.out.println(errorMessage);
String userInput = Listener.getInput();
while (InputValidator.menuChoiceIsInvalid(userInput, menu)) {
displayMenu();
System.out.println(errorMessage);
userInput = Listener.getInput();
}
if (!(userInput.equalsIgnoreCase("exit"))) {
int optionChose = Integer.valueOf(userInput);
chooseOption(optionChose);
}
}
private void reportToConsole() {
if (!(currentBudget.getName().equals("No budget loaded"))) {
System.out.println(currentBudget.toString());
} else {
errorMessage = Printer.getPrintout("noBudgetLoaded");
getOption(this);
}
}
private void reportToDisk() {
if (!(currentBudget.getName().equals("No budget loaded"))) {
Savior.generateBudgetReport(currentBudget.getName(), currentBudget);
} else {
errorMessage = Printer.getPrintout("noBudgetLoaded");
getOption(this);
}
}
private void startEditor() {
if (!(currentBudget.getName().equals("No budget loaded"))) {
EditorMenu editor = new EditorMenu(currentBudget);
while (editor.stillEditingBudget()) editor.getOption(editor);
} else {
errorMessage = Printer.getPrintout("noBudgetLoaded");
getOption(this);
}
}
private void saveBudget() {
if (!(currentBudget.getName().equals("No budget loaded"))) {
Savior.writeBudgetToDisk(currentBudget.getName(), currentBudget);
} else {
errorMessage = Printer.getPrintout("noBudgetLoaded");
getOption(this);
}
}
private void openBudget() {
BudgetBoss.loadSavedBudget();
BudgetBoss.endNeedNewBudget();
}
private void createBudget() {
BudgetBoss.endLoadSavedBudget();
BudgetBoss.needNewBudget();
}
private void exitProgram() {
BudgetBoss.doneUsingBudgetBoss();
BudgetBoss.endLoadSavedBudget();
BudgetBoss.endNeedNewBudget();
}
public void chooseOption(int optionChose) {
int index = (optionChose - 1);
menuOptions[index].optionMethod();
}
}
