mmarquee.automation.pattern.BasePattern Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of ui-automation Show documentation
A Java library that wraps the MS UIAutomation library.
There is a newer version: 0.7.0
/*
 * Copyright 2016-18 [email protected]
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package mmarquee.automation.pattern;

import com.sun.jna.platform.win32.COM.COMUtils;
import com.sun.jna.platform.win32.COM.Unknown;
import com.sun.jna.platform.win32.Guid;
import com.sun.jna.platform.win32.WinNT;
import com.sun.jna.ptr.PointerByReference;
import mmarquee.automation.*;
import mmarquee.uiautomation.IUIAutomationElement;
import mmarquee.uiautomation.IUIAutomationElementArray;
import mmarquee.uiautomation.IUIAutomationElementArrayConverter;
import mmarquee.uiautomation.IUIAutomationElementConverter;
import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;

import java.util.function.Function;

/**
 * The base pattern functionality, shared for all patterns.
 *
 * @author Mark Humphreys
 * Date 29/02/2016
 */
public abstract class BasePattern
        extends BaseAutomation
        implements Pattern {
    /**
     * The logger.
     */
    final Logger logger = LogManager.getLogger(BasePattern.class.getName());

    /**
     * The guid of the pattern.
     */
    protected Guid.IID IID;

    /**
     * The associated automation element.
     */
    final protected Element element;

    /**
     * The associated pattern id.
     */
    protected PatternID patternID;

    /**
     * The associated property id.
     */
    protected PropertyID availabilityPropertyID;

    /**
     * Constructor for BasePattern.
     *
     * @param element The Element
     * @throws AutomationException Error from automation
     */
    public BasePattern(Element element) throws AutomationException {
        super();
        this.element = element;
    }

    /**
     * Gets the pattern id.
     *
     * @return The pattern id
     */
    public PatternID getPatternID() {
    	return patternID;
    }

    /**
     * Is this pattern available.
     * @return True if available.
     */
    public boolean isAvailable() {
        try {
            final Object propertyValue = this.element.getPropertyValue(availabilityPropertyID.getValue());
            return BaseAutomation.isPropertyValueTrue(propertyValue);
        } catch (AutomationException ex) {
            return false;
        }
    }

    /**
     * Gets the raw pointer to the pattern.
     * @param pbr The raw pointer
     * @return Result of the call from the COM library
     * @throws PatternNotFoundException When the pattern was not found
     */
    public WinNT.HRESULT getRawPatternPointer(
            final PointerByReference pbr) throws PatternNotFoundException {
    	PointerByReference unknown;
		try {
			unknown = this.element.getPattern(patternID.getValue());
		} catch (AutomationException e) {
        	throw new PatternNotFoundException(e);
		}

//    	if (unknown == null) {
//        	logger.warn("Failed to find pattern");
//        	throw new PatternNotFoundException("Failed to find pattern");
//    	}

        Unknown uElement = makeUnknown(unknown.getValue());
        return uElement.QueryInterface(new Guid.REFIID(this.IID), pbr);
    }

    /**
     * Returns the Class of the unmocked pattern class.
     * @return the class without mocking
     */
	@SuppressWarnings("unchecked")
	public final Class getPatternClass() {
		Class patternClass = this.getClass();
        while (patternClass.getSimpleName().contains("Mockito")) { // Support for Pattern mocks
            patternClass = (Class) patternClass.getSuperclass();
        }
		return patternClass;
	}

	/**
	 * Gets a pattern from the raw pattern pointer, or returns the override pattern.
	 *
	 * @param   The class of the pattern to return
	 * @param overridePattern the pattern to use if to
	 * @param convertPointerToInterface the method to convert the result pointer
	 * @return the pattern interface
	 * @throws AutomationException if something goes wrong
	 */
    protected  T getPattern(final T overridePattern,
        Function convertPointerToInterface)
    		throws AutomationException {
    	if (overridePattern != null) {
            return overridePattern;
        } else {
            PointerByReference pbr = new PointerByReference();

            WinNT.HRESULT result0 = null;
			result0 = this.getRawPatternPointer(pbr);

            if (COMUtils.SUCCEEDED(result0)) {
                return (T) convertPointerToInterface.apply(pbr);
            } else {
                throw new AutomationException(result0.intValue());
            }
        }
    }
	 /**
     * Converts the unknown value to a IUIAutomationElement.
     * @param pUnknownA The Unknown pointer
     * @return The pattern
     */
    public IUIAutomationElement convertPointerToElementInterface(
            final PointerByReference pUnknownA) {
        return IUIAutomationElementConverter.pointerToInterface(pUnknownA);
    }

    /**
     * Converts the unknown value to a IUIAutomationElementArray.
     * @param pUnknownA The Unknown pointer
     * @return The pattern
     */
    public IUIAutomationElementArray convertPointerToElementArrayInterface(
            final PointerByReference pUnknownA) {
        return IUIAutomationElementArrayConverter.pointerToInterface(pUnknownA);
    }
}