SummaryGatheringVisitor.java

/**
 *    Copyright 2006-2016 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.mybatis.generator.eclipse.tests.harness.summary.support;

import java.util.ArrayList;
import java.util.List;

import org.eclipse.jdt.core.dom.ASTVisitor;
import org.eclipse.jdt.core.dom.AnnotationTypeDeclaration;
import org.eclipse.jdt.core.dom.AnnotationTypeMemberDeclaration;
import org.eclipse.jdt.core.dom.EnumDeclaration;
import org.eclipse.jdt.core.dom.MethodDeclaration;
import org.eclipse.jdt.core.dom.TypeDeclaration;
import org.eclipse.jdt.core.dom.VariableDeclarationFragment;
import org.mybatis.generator.eclipse.core.merge.visitors.MethodSignatureStringifier;
import org.mybatis.generator.eclipse.tests.harness.summary.AnnotationSummary;
import org.mybatis.generator.eclipse.tests.harness.summary.ClassSummary;
import org.mybatis.generator.eclipse.tests.harness.summary.EnumSummary;
import org.mybatis.generator.eclipse.tests.harness.summary.FieldSummary;
import org.mybatis.generator.eclipse.tests.harness.summary.InterfaceSummary;

public class SummaryGatheringVisitor extends ASTVisitor {
    private List<String> methodSignatures = new ArrayList<String>();
    private List<FieldSummary> fieldSummaries = new ArrayList<FieldSummary>();
    private List<AnnotationSummary> annotationSummaries = new ArrayList<AnnotationSummary>();
    private List<ClassSummary> classSummaries = new ArrayList<ClassSummary>();
    private List<EnumSummary> enumSummaries = new ArrayList<EnumSummary>();
    private List<InterfaceSummary> interfaceSummaries = new ArrayList<InterfaceSummary>();
    private List<String> annotationMembers = new ArrayList<String>();

    @Override
    public boolean visit(AnnotationTypeMemberDeclaration node) {
        annotationMembers.add(node.getName().getFullyQualifiedName());
        return false;
    }

    @Override
    public boolean visit(MethodDeclaration node) {
        MethodSignatureStringifier visitor = new MethodSignatureStringifier();
        node.accept(visitor);
        methodSignatures.add(visitor.toString());
        return false;
    }

    @Override
    public boolean visit(VariableDeclarationFragment node) {
        fieldSummaries.add(FieldSummary.from(node));
        return false;
    }

    @Override
    public boolean visit(TypeDeclaration node) {
        if (node.isInterface()) {
            interfaceSummaries.add(InterfaceSummary.from(node));
        } else {
            classSummaries.add(ClassSummary.from(node));
        }
        return false;
    }

    @Override
    public boolean visit(AnnotationTypeDeclaration node) {
        annotationSummaries.add(AnnotationSummary.from(node));
        return false;
    }

    @Override
    public boolean visit(EnumDeclaration node) {
        enumSummaries.add(EnumSummary.from(node));
        return false;
    }

    public List<String> getAnnotationMembers() {
        return annotationMembers;
    }
    
    public List<String> getMethodSignatures() {
        return methodSignatures;
    }

    public List<FieldSummary> getFieldSummaries() {
        return fieldSummaries;
    }

    public List<AnnotationSummary> getAnnotationSummaries() {
        return annotationSummaries;
    }

    public List<ClassSummary> getClassSummaries() {
        return classSummaries;
    }

    public List<EnumSummary> getEnumSummaries() {
        return enumSummaries;
    }

    public List<InterfaceSummary> getInterfaceSummaries() {
        return interfaceSummaries;
    }
}