package com.peterfranza;

import java.awt.Component;
import java.awt.Container;
import java.util.ArrayList;
import java.util.List;

import javax.swing.AbstractButton;
import javax.swing.JComponent;

public class UiTestUtils {
	
	private UiTestUtils() {}
	
	@SuppressWarnings("unchecked")
	public static <T> List<T> findAllInstances(Class<T> type, Container container) {
		
		final List<T> list = new ArrayList<T>();

		if(type.isInstance(container)) {
			list.add((T)container);
		}
		
		for(final Component comp: container.getComponents()) {
			if(comp instanceof Container) {
				list.addAll( findAllInstances(type, (Container) comp) );
			} else {
				if(type.isInstance(comp)) {
					list.add((T)comp);
				}
			}
		}
		
		return list;
	}
	
	public static <T extends AbstractButton> T findButtonByText(String text, Class<T> type, Container container) {
		List<T> items = findAllInstances(type, container);
		for (T item : items) {
			if(item.getText().equals(text)) {
				return item;
			}
		}
		return null;
	}
	
	public static <T extends JComponent> T findComponentByToolTipText(String text, Class<T> type, Container container) {
		List<T> items = findAllInstances(type, container);
		for (T item : items) {
			if(item.getToolTipText() != null && item.getToolTipText().equals(text)) {
				return item;
			}
		}
		return null;
	}	
	
}