/*
 * Copyright 2009-2017 the original author or authors.
 *
 * 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 org.codehaus.groovy.eclipse.codebrowsing.elements;

import org.codehaus.groovy.ast.ASTNode;
import org.codehaus.groovy.ast.ClassHelper;
import org.codehaus.groovy.ast.ClassNode;
import org.codehaus.groovy.ast.MethodNode;
import org.codehaus.groovy.ast.Variable;
import org.eclipse.jdt.core.JavaModelException;
import org.eclipse.jdt.internal.core.JavaElement;
import org.eclipse.jdt.internal.core.ResolvedBinaryType;

/**
 * A resolved IType/IMember suitable for hovers. May include extra Javadoc
 * information to appear in the hover.
 */
public class GroovyResolvedBinaryType extends ResolvedBinaryType implements IGroovyResolvedElement {

    private final String extraDoc;
    private ASTNode inferredElement;
    private Boolean isAnnotationCollector;

    public GroovyResolvedBinaryType(JavaElement parent, String name, String uniqueKey, String extraDoc, ASTNode inferredElement) {
        super(parent, name, uniqueKey);
        this.extraDoc = extraDoc;
        this.inferredElement = inferredElement;
    }

    public String getExtraDoc() {
        return extraDoc;
    }

    public int getFlags() throws JavaModelException {
        // a compiled collector (aka BinaryType) is actually a final class; adjust flags so it appears as an annotation
        return super.getFlags() ^ (isAnnotationCollector() ? 0x00002010 /*aka Modifier.ANNOTATION and Modifier.FINAL*/ : 0);
    }

    public ASTNode getInferredElement() {
        return inferredElement;
    }

    public String getInferredElementName() {
        if (inferredElement instanceof Variable) {
            return ((Variable) inferredElement).getName();
        } else if (inferredElement instanceof MethodNode) {
            return ((MethodNode) inferredElement).getName();
        } else if (inferredElement instanceof ClassNode) {
            return ((ClassNode) inferredElement).getName();
        } else {
            return inferredElement.getText();
        }
    }

    protected boolean isAnnotationCollector() {
        if (isAnnotationCollector == null) {
            isAnnotationCollector = (inferredElement instanceof ClassNode && !((ClassNode) inferredElement)
                .redirect().getAnnotations(ClassHelper.make("groovy.transform.AnnotationCollector")).isEmpty());
        }
        return isAnnotationCollector;
    }
}