WithdrawalTransaction.java

package com.transactions.transactions;

import com.transactions.exceptions.InsufficientFundsException;
import com.transactions.models.BankAccount;
import java.util.Calendar;

public class WithdrawalTransaction extends BaseTransaction {

    private BankAccount appliedAccount;
    private boolean applied;
    private double amountNotWithdrawn;
    private double amountDebited;

    public WithdrawalTransaction(double amount, Calendar date) {
        super(amount, date);
        this.applied = false;
        this.amountNotWithdrawn = 0;
        this.amountDebited = 0;
    }

    @Override
    public void printTransactionDetails() {
        System.out.println("--- Withdrawal Transaction ---");
        System.out.println("Transaction ID: " + getTransactionID());
        System.out.println("Amount: " + getAmount());
        System.out.println("Date: " + getDate().getTime());
        System.out.println("Type: WITHDRAWAL (reversible)");
        System.out.println("Applied: " + applied);
        if (amountNotWithdrawn > 0) {
            System.out.println("Amount not withdrawn: " + amountNotWithdrawn);
        }
    }

    // apply method uses throws to propagate InsufficientFundsException to the caller
    @Override
    public void apply(BankAccount ba) throws InsufficientFundsException {
        if (ba.getBalance() < getAmount()) {
            throw new InsufficientFundsException(getAmount(), ba.getBalance());
        }

        double balanceBefore = ba.getBalance();
        ba.debit(getAmount());
        this.appliedAccount = ba;
        this.applied = true;
        this.amountDebited = getAmount();

        System.out.println("Withdrawal applied to account: " + ba.getAccountHolder());
        System.out.println("Balance before: " + balanceBefore);
        System.out.println("Balance after: " + ba.getBalance());
        printTransactionDetails();
    }

    // overloaded apply - handles partial withdrawal using try catch finally
    public void apply(BankAccount ba, boolean allowPartial) {
        try {
            if (ba.getBalance() <= 0) {
                throw new InsufficientFundsException(getAmount(), ba.getBalance());
            }

            if (allowPartial && ba.getBalance() < getAmount()) {
                // balance is between 0 and the withdrawal amount so withdraw whatever is available
                double available = ba.getBalance();
                this.amountNotWithdrawn = getAmount() - available;
                this.amountDebited = available;

                ba.debit(available);
                this.appliedAccount = ba;
                this.applied = true;

                System.out.println("Partial withdrawal: only " + available + " was available");
                System.out.println("Amount not withdrawn: " + amountNotWithdrawn);
                System.out.println("Balance after: " + ba.getBalance());

            } else {
                if (ba.getBalance() < getAmount()) {
                    throw new InsufficientFundsException(getAmount(), ba.getBalance());
                }
                double balanceBefore = ba.getBalance();
                ba.debit(getAmount());
                this.appliedAccount = ba;
                this.applied = true;
                this.amountDebited = getAmount();

                System.out.println("Withdrawal applied: " + getAmount());
                System.out.println("Balance before: " + balanceBefore);
                System.out.println("Balance after: " + ba.getBalance());
            }

        } catch (InsufficientFundsException e) {
            System.out.println("Error: " + e.getMessage());

        } finally {
            System.out.println("Transaction complete.");
            printTransactionDetails();
        }
    }

    public boolean reverse() {
        if (!applied || appliedAccount == null) {
            System.out.println("Cannot reverse: transaction was not applied.");
            return false;
        }

        appliedAccount.credit(amountDebited);
        this.applied = false;

        System.out.println("Withdrawal reversed for: " + appliedAccount.getAccountHolder());
        System.out.println("Amount restored: " + amountDebited);
        System.out.println("New balance: " + appliedAccount.getBalance());
        return true;
    }

    public double getAmountNotWithdrawn() {
        return amountNotWithdrawn;
    }
}