package generated.model.de.fhdw.partner;
   
   import de.fhdw.wtf.context.model.AnyType;
   import de.fhdw.wtf.context.model.collections.MutableMap;
   import de.fhdw.wtf.persistence.meta.UserObject;
   
   /**
    * Account locator.
    */
  public class IBAN_Finder extends AnyType {
  	
  	/**
  	 * Maps triples (country, bank identifier, account number) to bank accounts.
  	 */
  	private MutableMap<Country, MutableMap<Bankidentifier, MutableMap<AccountNumber, Account>>> accountTree;
  	
  	/**
  	 * Creates an empty IBAN_Finder.
  	 */
  	public IBAN_Finder() {
  		this.accountTree = new MutableMap<>();
  	}
  	
  	/**
  	 * Loads object from database.
  	 * 
  	 * @param userObject
  	 *            The underlying user object.
  	 */
  	public IBAN_Finder(final UserObject userObject) {
  		super(userObject);
  	}
  	
  	/**
  	 * Maps a triple (country, bank identifier, account number) to an account.
  	 * 
  	 * @param country
  	 *            The country.
  	 * @param blz
  	 *            The bank identifier.
  	 * @param number
  	 *            The account number.
  	 * @return The account or null if no such account exists.
  	 */
  	public Account findAccount(final Country country, final Bankidentifier blz, final AccountNumber number) {
  		final MutableMap<Bankidentifier, MutableMap<AccountNumber, Account>> countryMap = this.accountTree.get(country);
  		if (countryMap != null) {
  			final MutableMap<AccountNumber, Account> bankMap = countryMap.get(blz);
  			if (bankMap != null) {
  				final Account acc = bankMap.get(number);
  				return acc;
  			}
  		}
  		return null;
  	}
  	
  	/**
  	 * Adds a account to the locator map.
  	 * 
  	 * @param country
  	 *            The country.
  	 * @param blz
  	 *            The bank identifier.
  	 * @param number
  	 *            The account number.
  	 * @param account
  	 *            The account.
  	 */
  	public void addAccount(final Country country,
  			final Bankidentifier blz,
  			final AccountNumber number,
  			final Account account) {
  		MutableMap<Bankidentifier, MutableMap<AccountNumber, Account>> countryMap = this.accountTree.get(country);
  		if (countryMap != null) {
  			MutableMap<AccountNumber, Account> bankMap = countryMap.get(blz);
  			if (bankMap != null) {
  				bankMap.put(number, account);
  			} else {
  				bankMap = new MutableMap<>();
  				bankMap.put(number, account);
  				countryMap.put(blz, bankMap);
  			}
  		} else {
  			countryMap = new MutableMap<>();
  			final MutableMap<AccountNumber, Account> bankMap = new MutableMap<>();
  			bankMap.put(number, account);
  			countryMap.put(blz, bankMap);
  			this.accountTree.put(country, countryMap);
  		}
  	}
  }