package com.iluwatar; import java.util.ArrayList; import java.util.List; /** * * Collection class. * */ public class TreasureChest { private List items; public TreasureChest() { items = new ArrayList<>(); items.add(new Item(ItemType.POTION, "Potion of courage")); items.add(new Item(ItemType.RING, "Ring of shadows")); items.add(new Item(ItemType.POTION, "Potion of wisdom")); items.add(new Item(ItemType.POTION, "Potion of blood")); items.add(new Item(ItemType.WEAPON, "Sword of silver +1")); items.add(new Item(ItemType.POTION, "Potion of rust")); items.add(new Item(ItemType.POTION, "Potion of healing")); items.add(new Item(ItemType.RING, "Ring of armor")); items.add(new Item(ItemType.WEAPON, "Steel halberd")); items.add(new Item(ItemType.WEAPON, "Dagger of poison")); } ItemIterator Iterator(ItemType type) { return new TreasureChestItemIterator(this, type); } public List getItems() { ArrayList list = new ArrayList<>(); list.addAll(items); return list; } }