com.sleepcamel.bsoneer.processor.domain.ElementResolver.java Source code

Introduction

Here is the source code for com.sleepcamel.bsoneer.processor.domain.ElementResolver.java

Source

/*
 * Copyright (C) 2015 Sleepcamel.
 *
 * 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 com.sleepcamel.bsoneer.processor.domain;

import javax.lang.model.element.TypeElement;
import javax.lang.model.type.DeclaredType;
import javax.lang.model.type.TypeKind;
import javax.lang.model.type.TypeMirror;
import javax.lang.model.type.TypeVariable;
import javax.lang.model.type.WildcardType;
import javax.lang.model.util.SimpleTypeVisitor7;

import com.google.common.base.Optional;

public class ElementResolver extends SimpleTypeVisitor7<Optional<TypeElement>, Void> {

    protected ElementResolver() {
        super(Optional.<TypeElement>fromNullable(null));
    }

    @Override
    public Optional<TypeElement> visitDeclared(DeclaredType t, Void p) {
        TypeElement asElement = (TypeElement) t.asElement();
        if (asElement.getSuperclass().getKind().equals(TypeKind.NONE)) {
            asElement = null;
        }
        return Optional.fromNullable(asElement);
    }

    @Override
    public Optional<TypeElement> visitTypeVariable(TypeVariable t, Void p) {
        TypeVariable tv = t;
        TypeMirror lb = tv.getLowerBound();
        TypeKind lbk = lb.getKind();
        if (TypeKind.NONE.equals(lbk) == false && TypeKind.NULL.equals(lbk) == false) {
            return lb.accept(this, null);
        } else {
            return tv.getUpperBound().accept(this, null);
        }

    }

    @Override
    public Optional<TypeElement> visitWildcard(WildcardType t, Void p) {
        if (t.getSuperBound() != null) {
            return t.getSuperBound().accept(this, null);
        } else {
            return t.getExtendsBound().accept(this, null);
        }
    }
}