org.noos.xing.mydoggy.itest.impl.RobotComponentLookuper Maven / Gradle / Ivy

Go to download
package org.noos.xing.mydoggy.itest.impl;

import org.noos.xing.mydoggy.itest.ComponentAdapter;
import org.noos.xing.mydoggy.itest.ComponentFilter;
import org.noos.xing.mydoggy.itest.ComponentLookuper;

import java.awt.*;
import java.util.ArrayList;
import java.util.Vector;

/**
 * @author Angelo De Caro ([email protected])
 */
public class RobotComponentLookuper implements ComponentLookuper {
    protected java.util.List roots;
    protected Robot robot;

    public RobotComponentLookuper(Robot robot, Container rootContainer) {
        this.roots = new ArrayList();
        this.roots.add(rootContainer);
        this.robot = robot;
    }

    public ComponentAdapter lookup() {
        return new RobotComponentAdapter(robot, roots.get(0), 1000);
    }

    public ComponentAdapter lookup(String componentName) {
        return lookup(new NamedComponentFilter(componentName));
    }

    public ComponentAdapter lookup(ComponentFilter componentFilter) {
        ComponentAdapter result = lookupInternal(componentFilter);
        if (result == null) {
            for (Window window : getTopContainers()) {
                if (!roots.contains(window))
                    roots.add(window);
            }

            result = lookupInternal(componentFilter);
        }
        if (result != null)
            return result;
        throw new IllegalStateException("Cannot find any components using the passed filter [filter : " + componentFilter + "]");
    }



    protected ComponentAdapter lookupInternal(ComponentFilter componentFilter) {
        if (componentFilter == null)
            return createComponentAdapter(roots.get(0));
        for (Container root : roots) {
            Component filteredComponent = findComponentByName(root, componentFilter);
            if (filteredComponent != null)
                return createComponentAdapter(filteredComponent);
        }
        return null; 
    }
    
    protected Component findComponentByName(Container root, ComponentFilter componentFilter) {
        if (root == null || componentFilter == null)
            return null;

        if (componentFilter.filter(root))
            return root;

        for (Component component : root.getComponents()) {
//            System.out.println(component.getName());
            if (componentFilter.filter(root))
                return component;

            if (component instanceof Container) {
                Component result = findComponentByName((Container) component, componentFilter);
                if (result != null)
                    return result;
            }
        }

        return null;
    }

    protected ComponentAdapter createComponentAdapter(Component component) {
        return new RobotComponentAdapter(robot, component, 1000);
    }

    protected java.util.List getTopContainers() {
        Vector containers = new Vector();

        Frame frames[] = Frame.getFrames();
        for (Frame frame : frames) {
            Window[] windows = frame.getOwnedWindows();

            for (Window window : windows) {
                if (window.getName() != null)
                    containers.add(window);
            }

            if (!containers.contains(frame)) {
                containers.add(frame);
            }
        }
        return containers;
    }

}