com.github.hemanthsridhar.builder.CustomAnnotations Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of custom-page-factory Show documentation
Custom Page Factory to read web elements or mobile elements from JSON
The newest version!
package com.github.hemanthsridhar.builder;

import com.github.hemanthsridhar.pagefactory.AbstractCustomFindByBuilder;
import com.github.hemanthsridhar.support.FilePath;
import com.github.hemanthsridhar.support.SearchAll;
import com.github.hemanthsridhar.support.SearchBy;
import com.github.hemanthsridhar.support.SearchBys;
import org.openqa.selenium.By;
import org.openqa.selenium.support.CacheLookup;
import org.openqa.selenium.support.pagefactory.AbstractAnnotations;
import org.openqa.selenium.support.pagefactory.Annotations;

import java.lang.annotation.Annotation;
import java.lang.reflect.Field;
import java.net.URL;

/**
 * @author hemanthsridhar
 */

public class CustomAnnotations extends AbstractAnnotations {
    private final Field field;
    private final Object page;
    private String filePath;
    private static final String CLASSPATH_PROTOCOL = "classpath:";

    public CustomAnnotations(Field field, Object page) {
        this.field = field;
        this.page = page;
        setFilePath(page);
    }

    private String getFilePath() {
        return filePath;
    }

    private void setFilePath(Object page) {
        String filePath;
        Class clazz = page.getClass();
        if (!clazz.isAnnotationPresent(FilePath.class)) {
            filePath = null;
        } else {
           String fileCompletePath = clazz.getAnnotation(FilePath.class).value();
            if (fileCompletePath.startsWith(CLASSPATH_PROTOCOL)) {
                String path = fileCompletePath.substring(CLASSPATH_PROTOCOL.length());
                URL url = clazz.getClassLoader().getResource(path);
                    if (url != null) {
                        filePath = url.getPath();
                    }
                    else{
                        throw new IllegalArgumentException("Unable to find by classpath with path "+path);
                    }
            }
            else{
                filePath = clazz.getAnnotation(FilePath.class).value();
            }
        }
        this.filePath = filePath;
    }

    @Override
    public By buildBy() {
        SearchBy searchBy = field.getAnnotation(SearchBy.class);
        SearchAll searchAll = field.getAnnotation(SearchAll.class);
        SearchBys searchBys = field.getAnnotation(SearchBys.class);

        if (searchBy != null || searchAll != null || searchBys != null) {
            for (Annotation annotation : field.getDeclaredAnnotations()) {
                AbstractCustomFindByBuilder builder = null;
                try {
                    builder = annotation.annotationType()
                            .getAnnotation(CustomPageFactoryFinder.class).value()
                            .newInstance();

                } catch (InstantiationException | IllegalAccessException e) {
                    e.printStackTrace();
                }
                if (builder != null) {
                    return builder.buildIt(annotation, field, getFilePath());
                }
            }
            return new Annotations(field).buildBy();
        } else {
            return new Annotations(field).buildBy();
        }
    }

    @Override
    public boolean isLookupCached() {
        return (field.getAnnotation(CacheLookup.class) != null);
    }
}