USER

import java.util.Scanner;

public class User {
    private String name;
    private String pin;
    private BankAccount account;

    public User(String name, String pin, BankAccount account) {
        this.name = name;
        this.pin = pin;
        this.account = account;
    }

    public String getName() {
        return name;
    }

    public String getPin() {
        return pin;
    }

    public BankAccount getAccount() {
        return account;
    }
}
public class BankAccount {
    private double balance;

    public BankAccount(double balance) {
        this.balance = balance;
    }

    public double getBalance() {
        return balance;
    }

    public void setBalance(double balance) {
        this.balance = balance;
    }

    public void deposit(double amount) {
        balance += amount;
    }

    public boolean withdraw(double amount) {
        if (balance >= amount) {
            balance -= amount;
            return true;
        } else {
            return false;
        }
    }
}

public class ATM {
    private User user;

    public ATM(User user) {
        this.user = user;
    }

    public void start() {
        Scanner scanner = new Scanner(System.in);
        while (true) {
            System.out.println("Please choose an option:");
            System.out.println("1. Check balance");
            System.out.println("2. Withdraw");
            System.out.println("3. Deposit");
            System.out.println("4. Quit");

            int option = scanner.nextInt();
            scanner.nextLine(); // Consume newline not consumed by nextInt()

            switch (option) {
                case 1:
                    checkBalance();
                    break;
                case 2:
                withdraw(scanner);
                    break;
                case 3:
                    deposit(scanner);
                    break;
                case 4:
                    quit();
                    break;
                default:
                System.out.println("Invalid option.");
            }
        }
    }
    private void checkBalance(){
        System.out.printf("Your current balance is $%.2f\n", user.getBalance());
    }
    private void withdraw(Scanner scanner){
        if (!user.hasAccount()) {
            System.out.println("You do not have a checking account!");
            return;
        }
        double amount = getAmountFromUser(scanner, "Withdrawal Amount");
        if (amount > user.getCheckingBalance()) {
            System.out.println("Insufficient funds for that transaction.");
            return;
        }
        user.withdraw(amount);
        System.out.printf("$%.2f has been withdrawn from your account.\n", amount);
    }
    private void deposit(Scanner scanner){
        if (!user.hasAccount()) {
            System.out.println("You do not have a checking account!");
            return;
        }
        double amount = getAmountFromUser(scanner, "Deposit Amount");
        user.deposit(amount);
        System.out.printf("$%.2f has been deposited into your account.\n", amount);
    }
    public static double getAmountFromUser(Scanner scanner, String prompt){
        boolean validInput = false;
        while(!validInput){
            try{
                System.out.print(prompt + ": ");
                double input = scanner.nextDouble();
                if (input < 0) throw new Exception();
                validInput = true;
                return input;
            } catch (Exception e){
                System.out.println("That was an invalid entry! Please enter a positive number.");
            }
        }
        return -1; //This should never be reached
    }
    }