package prison;

import java.util.InputMismatchException;
import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        Main main = new Main();
        Scanner scanner = new Scanner(System.in);
        Prison prison = new Prison();
        boolean close = false;
        int menuChoice = 0;

        while (!close) {
            menuChoice = main.menuScreen(scanner);
            if (menuChoice == 1) {
                prison.printPrisoners();
            } else if (menuChoice == 2) {
            } else if (menuChoice == 3) {
                prison.removePrisoner(scanner);
            } else if (menuChoice == 4) {

            } else if (menuChoice == 5) {
                prison.loadFile(scanner);
            } else if (menuChoice == 6) {
                System.out.println("Exiting program");
                close = true;
            }
        }
    }

    public int menuScreen(Scanner scanner) {
        boolean inputWrong = false;

        while (!inputWrong) {
            try {
                System.out.println("*******************");
                System.out.println("1) Print prisoners ");
                System.out.println("2) Add prisoner ");
                System.out.println("3) Remove prisoner ");
                System.out.println("4) Save prisoners ");
                System.out.println("5) Load prisoners ");
                System.out.println("6) Close program");
                System.out.println();
                System.out.println("Please choose an option");

                int choice = scanner.nextInt();
                if (choice > 0 && choice < 7) {
                    return choice;
                } else {
                    System.out.println("Wrong input, please try again");
                    inputWrong = true;
                }
            } catch (InputMismatchException e) {
                scanner.next();
                System.out.println("Please try again");
            }
        }
        return 0;
    }
}