org.jemmy.swt.ToolBarWrap Maven / Gradle / Ivy

Go to download
/*
 * Copyright (c) 2007, 2018, Oracle and/or its affiliates. All rights reserved.
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
 *
 * This code is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License version 2 only, as
 * published by the Free Software Foundation. Oracle designates this
 * particular file as subject to the "Classpath" exception as provided
 * by Oracle in the LICENSE file that accompanied this code.
 *
 * This code is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
 * version 2 for more details (a copy is included in the LICENSE file that
 * accompanied this code).
 *
 * You should have received a copy of the GNU General Public License version
 * 2 along with this work; if not, write to the Free Software Foundation,
 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
 *
 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
 * or visit www.oracle.com if you need additional information or have any
 * questions.
 */
package org.jemmy.swt;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import org.eclipse.swt.widgets.ToolBar;
import org.eclipse.swt.widgets.ToolItem;
import org.jemmy.action.GetAction;
import org.jemmy.control.ControlType;
import org.jemmy.control.Property;
import org.jemmy.env.Environment;
import org.jemmy.interfaces.Parent;
import org.jemmy.interfaces.Selectable;
import org.jemmy.interfaces.Selector;
import org.jemmy.interfaces.TypeControlInterface;
import org.jemmy.lookup.Lookup;

/**
 *
 * @author shura
 * @author erikgreijus
 */
@ControlType(ToolBar.class)
public class ToolBarWrap extends ControlWrap implements Selectable {

    private ItemParent items;
    private ToolTipItemSelector selector;

    public ToolBarWrap(Environment env, T node) {
        super(env, node);
        this.items = null;
        this.selector = null;
    }

    @Override
    public > INTERFACE as(Class interfaceClass, Class type) {
        if (interfaceClass.equals(Parent.class) && ToolItem.class.equals(type)) {
            if (items == null) {
                items = new ItemParent(this, ToolItem.class) {

                    @Override
                    protected List getItems() {
                        return Arrays.asList(new GetAction() {

                            @Override
                            public void run(Object... parameters) {
                                setResult(getControl().getItems());
                            }
                        }.dispatch(getEnvironment()));
                    }
                };
            }
            return (INTERFACE) items;
        }
        return super.as(interfaceClass, type);
    }

    @Override
    public > boolean is(Class interfaceClass, Class type) {
        if (interfaceClass.equals(Parent.class) && ToolItem.class.equals(type)) {
            return true;
        }
        return super.is(interfaceClass, type);
    }

    @Property(Selectable.STATES_PROP_NAME)
    public List getStates() {
        return new GetAction>() {

            @Override
            public void run(Object... parameters) {
                Lookup lookup = as(Parent.class, ToolItem.class).lookup();
                ArrayList res = new ArrayList();
                for (int i = 0; i < lookup.size(); i++) {
                    res.add(lookup.get(i).getToolTipText());
                }
                setResult(res);
            }
        }.dispatch(getEnvironment());
    }

    @Property(Selectable.STATE_PROP_NAME)
    public String getState() {
        return new GetAction() {

            @Override
            public void run(Object... parameters) {
                Lookup lookup = as(Parent.class, ToolItem.class).lookup();
                String selected = "No selection";
                for (int i = 0; i < lookup.size(); i++) {
                    if (lookup.get(i).getSelection()) {
                        selected = lookup.get(i).getToolTipText();
                        break;
                    }
                }
                setResult(selected);
            }
        }.dispatch(getEnvironment());
    }

    public Selector selector() {
        if (selector == null) {
            if (items == null) {
                as(Parent.class, ToolItem.class);
            }
            selector = new ToolTipItemSelector(items);
        }
        return selector;
    }

    public Class getType() {
        return String.class;
    }
}