ch.powerunit.extensions.matchers.provideprocessor.fields.FieldDescriptionProvider Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of powerunit-extensions-matchers-factory Show documentation
This is a test framework for the JDK 1.8 - Extension to provide matchers based on annotation. - Factory Support
There is a newer version: 0.1.6
Show newest version
/**
 * Powerunit - A JDK1.8 test framework
 * Copyright (C) 2014 Mathieu Boretti.
 *
 * This file is part of Powerunit
 *
 * Powerunit is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * Powerunit 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 for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with Powerunit. If not, see .
 */
package ch.powerunit.extensions.matchers.provideprocessor.fields;

import javax.annotation.processing.ProcessingEnvironment;
import javax.lang.model.element.Element;
import javax.lang.model.element.ExecutableElement;
import javax.lang.model.type.ArrayType;
import javax.lang.model.type.DeclaredType;
import javax.lang.model.type.TypeMirror;
import javax.lang.model.type.TypeVariable;
import javax.lang.model.util.TypeKindVisitor8;
import javax.tools.Diagnostic.Kind;

import ch.powerunit.extensions.matchers.IgnoreInMatcher;
import ch.powerunit.extensions.matchers.provideprocessor.ProvidesMatchersAnnotatedElementData;

public final class FieldDescriptionProvider {

	private FieldDescriptionProvider() {
	}

	public static enum Type {
		NA, ARRAY, COLLECTION, LIST, SET, OPTIONAL, COMPARABLE, STRING, SUPPLIER
	}

	public static final class ExtracTypeVisitor extends TypeKindVisitor8 {

		@Override
		protected Type defaultAction(TypeMirror t, ProcessingEnvironment processingEnv) {
			return Type.NA;
		}

		@Override
		public Type visitArray(ArrayType t, ProcessingEnvironment processingEnv) {
			return Type.ARRAY;
		}

		@Override
		public Type visitDeclared(DeclaredType t, ProcessingEnvironment processingEnv) {
			if (processingEnv.getTypeUtils().isAssignable(t, processingEnv.getTypeUtils()
					.erasure(processingEnv.getElementUtils().getTypeElement("java.util.Optional").asType()))) {
				return Type.OPTIONAL;
			} else if (processingEnv.getTypeUtils().isAssignable(t, processingEnv.getTypeUtils()
					.erasure(processingEnv.getElementUtils().getTypeElement("java.util.Set").asType()))) {
				return Type.SET;
			} else if (processingEnv.getTypeUtils().isAssignable(t, processingEnv.getTypeUtils()
					.erasure(processingEnv.getElementUtils().getTypeElement("java.util.List").asType()))) {
				return Type.LIST;
			} else if (processingEnv.getTypeUtils().isAssignable(t, processingEnv.getTypeUtils()
					.erasure(processingEnv.getElementUtils().getTypeElement("java.util.Collection").asType()))) {
				return Type.COLLECTION;
			} else if (processingEnv.getTypeUtils().isAssignable(t, processingEnv.getTypeUtils()
					.erasure(processingEnv.getElementUtils().getTypeElement("java.lang.String").asType()))) {
				return Type.STRING;
			} else if (processingEnv.getTypeUtils().isAssignable(t, processingEnv.getTypeUtils()
					.erasure(processingEnv.getElementUtils().getTypeElement("java.lang.Comparable").asType()))) {
				return Type.COMPARABLE;
			} else if (processingEnv.getTypeUtils().isAssignable(t, processingEnv.getTypeUtils()
					.erasure(processingEnv.getElementUtils().getTypeElement("java.util.function.Supplier").asType()))) {
				return Type.SUPPLIER;
			}
			return Type.NA;
		}

		@Override
		public Type visitTypeVariable(TypeVariable t, ProcessingEnvironment processingEnv) {
			return Type.NA;
		}

		@Override
		public Type visitUnknown(TypeMirror t, ProcessingEnvironment processingEnv) {
			processingEnv.getMessager().printMessage(Kind.MANDATORY_WARNING, "Unsupported type element");
			return Type.NA;
		}
	}

	public static AbstractFieldDescription of(ProvidesMatchersAnnotatedElementData containingElementMirror,
			FieldDescriptionMirror mirror) {
		Element te = mirror.getFieldElement();
		ProcessingEnvironment processingEnv = containingElementMirror.getRoundMirror().getProcessingEnv();
		Type type = new ExtracTypeVisitor().visit(
				(te instanceof ExecutableElement) ? ((ExecutableElement) te).getReturnType() : te.asType(),
				processingEnv);
		if (te.getAnnotation(IgnoreInMatcher.class) != null) {
			return new IgoreFieldDescription(containingElementMirror, mirror);
		}

		switch (type) {
		case ARRAY:
			return new ArrayFieldDescription(containingElementMirror, mirror);
		case COLLECTION:
		case SET:
		case LIST:
			return new CollectionFieldDescription(containingElementMirror, mirror);
		case COMPARABLE:
			return new ComparableFieldDescription(containingElementMirror, mirror);
		case OPTIONAL:
			return new OptionalFieldDescription(containingElementMirror, mirror);
		case STRING:
			return new StringFieldDescription(containingElementMirror, mirror);
		case SUPPLIER:
			return new SupplierFieldDescription(containingElementMirror, mirror);
		default:
			return new DefaultFieldDescription(containingElementMirror, mirror);
		}
	}

}