bundles/org.eclipse.jst.ws.jaxws.core/src/org/eclipse/jst/ws/internal/jaxws/core/annotations/validation/SOAPBindingMethodUseRule.java

/*******************************************************************************
 * Copyright (c) 2009 Progress Software Inc.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *******************************************************************************/
package org.eclipse.jst.ws.internal.jaxws.core.annotations.validation;

import static org.eclipse.jst.ws.internal.jaxws.core.utils.JAXWSUtils.PARAMETER_STYLE;
import static org.eclipse.jst.ws.internal.jaxws.core.utils.JAXWSUtils.USE;

import java.util.Collection;
import java.util.Map;
import java.util.Set;

import javax.jws.soap.SOAPBinding;

import org.eclipse.jst.ws.annotations.core.processor.AbstractAnnotationProcessor;
import org.eclipse.jst.ws.internal.jaxws.core.JAXWSCoreMessages;

import com.sun.mirror.declaration.AnnotationMirror;
import com.sun.mirror.declaration.AnnotationTypeDeclaration;
import com.sun.mirror.declaration.AnnotationTypeElementDeclaration;
import com.sun.mirror.declaration.AnnotationValue;
import com.sun.mirror.declaration.Declaration;
import com.sun.mirror.declaration.EnumConstantDeclaration;
import com.sun.mirror.declaration.MethodDeclaration;

/**
 * 
 * @author ohurley
 *
 */
public class SOAPBindingMethodUseRule extends AbstractAnnotationProcessor {

    @Override
    public void process() {
        AnnotationTypeDeclaration annotationDeclaration = (AnnotationTypeDeclaration) environment
                .getTypeDeclaration(SOAPBinding.class.getName());

        Collection<Declaration> annotatedTypes = environment
                .getDeclarationsAnnotatedWith(annotationDeclaration);
        
        boolean methodUseEncoded = false;
        boolean methodParameterStylePresent = false;
        
        for (Declaration declaration : annotatedTypes) {
            if (declaration instanceof MethodDeclaration) {

                Collection<AnnotationMirror> annotationMirrors = declaration.getAnnotationMirrors();
    
                for (AnnotationMirror mirror : annotationMirrors) {
                    Map<AnnotationTypeElementDeclaration, AnnotationValue> valueMap = mirror
                            .getElementValues();
                    Set<Map.Entry<AnnotationTypeElementDeclaration, AnnotationValue>> valueSet = valueMap
                            .entrySet();
                    for (Map.Entry<AnnotationTypeElementDeclaration, AnnotationValue> annotationKeyValue : 
                            valueSet) {

                        if (annotationKeyValue.getKey().getSimpleName().equals(USE)) {
                            if (annotationKeyValue.getValue() != null) {
                                AnnotationValue annotationValue = annotationKeyValue.getValue();
                                EnumConstantDeclaration enumConstantDeclaration = 
                                    (EnumConstantDeclaration) annotationValue.getValue();
                                if (enumConstantDeclaration.getSimpleName().equals(
                                        javax.jws.soap.SOAPBinding.Use.ENCODED.name())) {
                                	methodUseEncoded = true;
                                }
                            }
                        } else {
                        	if (annotationKeyValue.getKey().getSimpleName().equals(PARAMETER_STYLE)) {
                        		methodParameterStylePresent = true;
                        	}
                        }
                        
                        if (methodParameterStylePresent && methodUseEncoded) {
                            printError(mirror.getPosition(),
                                    JAXWSCoreMessages.SOAPBINDING_NO_PARAMETERSTYLE_WHEN_ENCODED);
                        	return;
                        }
                    }
                }
            }
        }
    }
}