/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements. See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership. The ASF licenses this file
 * to you 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.apache.axis2.description.java2wsdl;

import org.apache.axis2.AxisFault;
import org.apache.axis2.deployment.util.Utils;
import org.apache.axis2.description.AxisMessage;
import org.apache.axis2.description.AxisOperation;
import org.apache.axis2.description.AxisService;
import org.apache.axis2.description.WSDL2Constants;
import org.apache.axis2.wsdl.WSDLConstants;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.apache.ws.commons.schema.XmlSchema;
import org.apache.ws.commons.schema.XmlSchemaComplexType;
import org.apache.ws.commons.schema.XmlSchemaElement;
import org.apache.ws.commons.schema.XmlSchemaSequence;
import org.apache.ws.commons.schema.utils.NamespaceMap;
import org.w3c.dom.Document;

import javax.xml.namespace.QName;
import java.lang.annotation.Annotation;
import java.lang.reflect.Method;
import java.lang.reflect.Modifier;
import java.lang.reflect.Type;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collection;
import java.util.HashMap;
import java.util.LinkedHashMap;
import java.util.Map;

public class DocLitBareSchemaGenerator extends DefaultSchemaGenerator {

    private static final Log log = LogFactory.getLog(DocLitBareSchemaGenerator.class);
    private HashMap<String,Method> processedParameters = new LinkedHashMap<String,Method>();

    public DocLitBareSchemaGenerator(ClassLoader loader,
                                     String className,
                                     String schematargetNamespace,
                                     String schematargetNamespacePrefix,
                                     AxisService service) throws Exception {
        super(loader, className, schematargetNamespace,
                schematargetNamespacePrefix, service);
    }

    @Override
    protected Method[] processMethods(Method[] declaredMethods) throws Exception {
        ArrayList<Method> list = new ArrayList<Method>();
        //short the elements in the array
        Arrays.sort(declaredMethods, new MathodComparator());

        // since we do not support overload
        HashMap<String, Method> uniqueMethods = new LinkedHashMap<String, Method>();
        XmlSchemaComplexType methodSchemaType;
        XmlSchemaSequence sequence;

        for (Method jMethod : declaredMethods) {
            if (jMethod.isBridge() || jMethod.getDeclaringClass().getName().equals(Object.class.getName())) {
                continue;
            }
            String methodName = jMethod.getName();
            // no need to think abt this method , since that is system
            // config method
            if (excludeMethods.contains(methodName)) {
                continue;
            }

            if (uniqueMethods.get(methodName) != null) {
                log.warn("We don't support method overloading. Ignoring [" +
                         methodName + "]");
                continue;
            }

            if (!Modifier.isPublic(jMethod.getModifiers())) {
                // no need to generate Schema for non public methods
                continue;
            }

            boolean addToService = false;
            AxisOperation axisOperation = service.getOperation(new QName(methodName));
            if (axisOperation == null) {
                axisOperation = Utils.getAxisOperationForJmethod(jMethod);
                if (WSDL2Constants.MEP_URI_ROBUST_IN_ONLY.equals(
                        axisOperation.getMessageExchangePattern())) {
                    AxisMessage outMessage = axisOperation.getMessage(
                            WSDLConstants.MESSAGE_LABEL_OUT_VALUE);
                    if (outMessage != null) {
                        outMessage.setName(methodName + RESULT);
                    }
                }
                addToService = true;
            }

            // Maintain a list of methods we actually work with
            list.add(jMethod);
            processException(jMethod, axisOperation);
            uniqueMethods.put(methodName, jMethod);
            //create the schema type for the method wrapper

            uniqueMethods.put(methodName, jMethod);
            Class<?>[] paras = jMethod.getParameterTypes();
            Type[] genericParameterTypes = jMethod.getGenericParameterTypes();
            String parameterNames[] = methodTable.getParameterNames(methodName);
            AxisMessage inMessage = axisOperation.getMessage(WSDLConstants.MESSAGE_LABEL_IN_VALUE);
            if (inMessage != null) {
                inMessage.setName(methodName + "RequestMessage");
            }
            Annotation[][] parameterAnnotation = jMethod.getParameterAnnotations();
            if (paras.length > 1) {
                sequence = new XmlSchemaSequence();
                methodSchemaType = createSchemaTypeForMethodPart(methodName);
                methodSchemaType.setParticle(sequence);
                inMessage.setElementQName(typeTable.getQNamefortheType(methodName));
                service.addMessageElementQNameToOperationMapping(methodSchemaType.getQName(),
                                                                 axisOperation);
                inMessage.setPartName(methodName);
                for (int j = 0; j < paras.length; j++) {
                    Class<?> methodParameter = paras[j];
                    String parameterName = getParameterName(parameterAnnotation, j, parameterNames);
                    if (generateRequestSchema(methodParameter, parameterName, jMethod, sequence, genericParameterTypes[j])) {
                        break;
                    }
                }
            } else if (paras.length == 1) {
                if (paras[0].isArray()) {
                    sequence = new XmlSchemaSequence();

                    methodSchemaType = createSchemaTypeForMethodPart(methodName);
                    methodSchemaType.setParticle(sequence);
                    Class<?> methodParameter = paras[0];
                    inMessage.setElementQName(typeTable.getQNamefortheType(methodName));
                    service.addMessageElementQNameToOperationMapping(methodSchemaType.getQName(),
                                                                     axisOperation);
                    inMessage.setPartName(methodName);
                    String parameterName = getParameterName(parameterAnnotation, 0, parameterNames);
                    if (generateRequestSchema(methodParameter, parameterName, jMethod, sequence, genericParameterTypes[0])) {
                        break;
                    }
                } else {
                    String parameterName = getParameterName(parameterAnnotation, 0, parameterNames);
                    Class<?> methodParameter = paras[0];
                    Method processMethod = processedParameters.get(parameterName);
                    if (processMethod != null) {
                        throw new AxisFault("Inavalid Java class," +
                                            " there are two methods [" + processMethod.getName() + " and " +
                                            jMethod.getName() + " ]which have the same parameter names");
                    } else {
                        processedParameters.put(parameterName, jMethod);
                        if (methodParameter != null && Map.class.isAssignableFrom(methodParameter)) {
                            generateBareSchemaTypeForMap(parameterName, genericParameterTypes[0], null);

                        } else if (methodParameter != null
                                   && Collection.class
                                .isAssignableFrom(methodParameter)) {

                            sequence = new XmlSchemaSequence();
                            methodSchemaType = createSchemaTypeForMethodPart(methodName);
                            methodSchemaType.setParticle(sequence);
                            generateBareSchemaTypeForCollection(sequence,
                                                                genericParameterTypes[0], parameterName,
                                                                methodName);
                            parameterName = methodName;

                        } else if (methodParameter != null && Document.class.isAssignableFrom(methodParameter)) {
                            generateBareSchemaTypeForDocument(null, parameterName);
                        } else {
                            generateSchemaForType(null, methodParameter, parameterName);
                        }
                        inMessage.setElementQName(typeTable.getQNamefortheType(parameterName));
                        inMessage.setPartName(parameterName);
                        inMessage.setWrapped(false);
                        service.addMessageElementQNameToOperationMapping(typeTable.getQNamefortheType(parameterName),
                                                                         axisOperation);
                    }
                }
            }

            // for its return type
            Class<?> returnType = jMethod.getReturnType();
            Type genericReturnType = jMethod.getGenericReturnType();
            String methodTypeName = jMethod.getName() + RESULT;
            String returnName = "return";

            if (!"void".equals(jMethod.getReturnType().getName())) {
                AxisMessage outMessage = axisOperation.getMessage(
                        WSDLConstants.MESSAGE_LABEL_OUT_VALUE);
                if (returnType.isArray()) {
                    methodSchemaType =
                            createSchemaTypeForMethodPart(methodTypeName);
                    sequence = new XmlSchemaSequence();
                    methodSchemaType.setParticle(sequence);
                    if (nonRpcMethods.contains(methodName)) {
                        generateSchemaForType(sequence, null, returnName);
                    } else {
                        generateSchemaForType(sequence, returnType, returnName);
                    }
                } else {
                    if (returnType != null && Document.class.isAssignableFrom(returnType)) {
                        generateBareSchemaTypeForDocument(null, methodTypeName);

                    } else if (returnType != null && Map.class.isAssignableFrom(returnType)) {
                        generateBareSchemaTypeForMap(methodTypeName, genericReturnType, null);

                    } else if (returnType != null
                               && Collection.class.isAssignableFrom(returnType)) {
                        sequence = new XmlSchemaSequence();
                        methodSchemaType = createSchemaTypeForMethodPart(methodTypeName);
                        methodSchemaType.setParticle(sequence);
                        generateBareSchemaTypeForCollection(sequence,
                                                            genericReturnType, returnName, methodName);

                    } else {
                        generateSchemaForType(null, returnType, methodTypeName);
                    }
                    outMessage.setWrapped(false);
                }
                outMessage.setElementQName(typeTable.getQNamefortheType(methodTypeName));
                outMessage.setName(methodName + "ResponseMessage");
                outMessage.setPartName(methodTypeName);
                service.addMessageElementQNameToOperationMapping(
                        typeTable.getQNamefortheType(methodTypeName),
                        axisOperation);
            }
            if (addToService) {
                service.addOperation(axisOperation);
            }
        }
        return list.toArray(new Method[list.size()]);
    }


    private boolean generateRequestSchema(Class<?> methodParameter,
                                          String parameterName,
                                          Method jMethod,
                                          XmlSchemaSequence sequence, Type genericParameterType) throws Exception {
        if (nonRpcMethods.contains(jMethod.getName())) {
            generateSchemaForType(sequence, null, jMethod.getName());
            return true;
        } else if (methodParameter != null && Map.class.isAssignableFrom(methodParameter)){                        	
			generateBareSchemaTypeForMap(parameterName, genericParameterType, sequence);			
        } else if (methodParameter != null
        	&& Collection.class.isAssignableFrom(methodParameter)) {
            generateBareSchemaTypeForCollection(sequence, genericParameterType,
        	    parameterName, jMethod.getName());
            
        } else if (methodParameter != null && Document.class.isAssignableFrom(methodParameter)) {
            generateBareSchemaTypeForDocument(sequence,
                    parameterName);
        } else {
            generateSchemaForType(sequence, methodParameter, parameterName);
        }
        return false;
    }

    private QName generateSchemaForType(XmlSchemaSequence sequence, Class<?> type, String partName)
            throws Exception {

        boolean isArrayType = false;
        if (type != null) {
            isArrayType = type.isArray();
        }
        if (isArrayType) {
            type = type.getComponentType();
        }
        if (AxisFault.class.getName().equals(type)) {
            return null;
        }
        String classTypeName;
        if (type == null) {
            classTypeName = "java.lang.Object";
        } else {
            classTypeName = type.getName();
        }
        if (isArrayType && "byte".equals(classTypeName)) {
            classTypeName = "base64Binary";
            isArrayType = false;
        }
        if (isDataHandler(type)) {
            classTypeName = "base64Binary";
        }
        QName schemaTypeName = typeTable.getSimpleSchemaTypeName(classTypeName);
        if (schemaTypeName == null && type != null) {
            schemaTypeName = generateSchema(type);
            addContentToMethodSchemaType(sequence,
                    schemaTypeName,
                    partName,
                    isArrayType);
            String schemaNamespace = resolveSchemaNamespace(getQualifiedName(type.getPackage()));
            addImport(getXmlSchema(schemaNamespace), schemaTypeName);
            if(sequence==null){
                 generateSchemaForSingleElement(schemaTypeName, partName, isArrayType);
            }
        } else {
            if (sequence == null) {
                generateSchemaForSingleElement(schemaTypeName, partName, isArrayType);
            } else {
                addContentToMethodSchemaType(sequence,
                        schemaTypeName,
                        partName,
                        isArrayType);
            }
        }
        addImport(getXmlSchema(schemaTargetNameSpace), schemaTypeName);
        return schemaTypeName;
    }

    protected void generateSchemaForSingleElement(QName schemaTypeName,
                                                  String paraName,
                                                  boolean isArray) throws Exception {
        XmlSchemaElement elt1 = new XmlSchemaElement(getXmlSchema(schemaTargetNameSpace), false);
        elt1.setName(paraName);
        elt1.setSchemaTypeName(schemaTypeName);
        elt1.setNillable(true);
        QName elementName =
                new QName(schemaTargetNameSpace, paraName, schema_namespace_prefix);
        XmlSchema xmlSchema = getXmlSchema(schemaTargetNameSpace);
        xmlSchema.getElements().put(elementName, elt1);
        xmlSchema.getItems().add(elt1);
        typeTable.addComplexSchema(paraName, elementName);
    }

    /**
     * Generate schema construct for given type
     *
     * @param localPartName
     */
    private XmlSchemaComplexType createSchemaTypeForMethodPart(String localPartName) {
        XmlSchema xmlSchema = getXmlSchema(schemaTargetNameSpace);
        QName elementName =
                new QName(this.schemaTargetNameSpace, localPartName, this.schema_namespace_prefix);

        XmlSchemaComplexType complexType = getComplexTypeForElement(xmlSchema, elementName);
        if (complexType == null) {
            complexType = new XmlSchemaComplexType(xmlSchema, false);

            XmlSchemaElement globalElement = new XmlSchemaElement(xmlSchema, false);
            globalElement.setSchemaType(complexType);
            globalElement.setName(localPartName);
            xmlSchema.getItems().add(globalElement);
            xmlSchema.getElements().put(elementName, globalElement);
        }
        typeTable.addComplexSchema(localPartName, elementName);

        return complexType;
    }

    // TODO: explain why we need to override the method if the implementation is identical!
    @Override
    protected XmlSchema getXmlSchema(String targetNamespace) {
        XmlSchema xmlSchema;

        if ((xmlSchema = schemaMap.get(targetNamespace)) == null) {
            String targetNamespacePrefix;

            if (targetNamespace.equals(schemaTargetNameSpace) &&
                    schema_namespace_prefix != null) {
                targetNamespacePrefix = schema_namespace_prefix;
            } else {
                targetNamespacePrefix = generatePrefix();
            }


            xmlSchema = new XmlSchema(targetNamespace, xmlSchemaCollection);
            xmlSchema.setAttributeFormDefault(getAttrFormDefaultSetting());
            xmlSchema.setElementFormDefault(getElementFormDefaultSetting());


            targetNamespacePrefixMap.put(targetNamespace, targetNamespacePrefix);
            schemaMap.put(targetNamespace, xmlSchema);

            NamespaceMap prefixmap = new NamespaceMap();
            prefixmap.put(DEFAULT_SCHEMA_NAMESPACE_PREFIX, URI_2001_SCHEMA_XSD);
            prefixmap.put(targetNamespacePrefix, targetNamespace);
            xmlSchema.setNamespaceContext(prefixmap);
        }
        return xmlSchema;
    }
    
	/**
	 * Generate bare schema type for map.
	 *
	 * @param paraName the para name
	 * @param genericParameterType the generic parameter type
	 * @param sequence the sequence
	 * @throws Exception the exception
	 */
	private void generateBareSchemaTypeForMap(String paraName,
			Type genericParameterType, XmlSchemaSequence sequence)
			throws Exception {
		QName schemaTypeName = generateSchemaTypeForMap(sequence,
				genericParameterType, paraName, false);
		if (sequence != null) {
			return;
		}
        XmlSchema xmlSchema = getXmlSchema(schemaTargetNameSpace);
		XmlSchemaElement elt1 = new XmlSchemaElement(xmlSchema, false);
		elt1.setSchemaTypeName(schemaTypeName);
		elt1.setName(paraName);
		elt1.setNillable(true);
		QName elementName = new QName(schemaTargetNameSpace, paraName,
				schema_namespace_prefix);
		xmlSchema.getElements().put(elementName, elt1);
		xmlSchema.getItems().add(elt1);
		typeTable.addComplexSchema(paraName, elementName);

	}
	
	/**
	 * Generate bare schema type for collection.
	 *
	 * @param sequence the sequence
	 * @param genericType the generic type
	 * @param partName the part name
	 * @param methodName the method name
	 * @throws Exception the exception
	 */
	private void generateBareSchemaTypeForCollection(
		XmlSchemaSequence sequence, Type genericType, String partName,
		String methodName) throws Exception {
	    QName schemaTypeName = generateSchemaForCollection(sequence,
		    genericType, partName);
	}
	
    /**
     * Generate bare schema type for document.
     * 
     * @param sequence
     *            the sequence
     * @param parameterName
     *            the parameter name
     */
    private void generateBareSchemaTypeForDocument(XmlSchemaSequence sequence,
            String parameterName) {
        QName schemaTypeName = generateSchemaTypeForDocument(sequence,
                parameterName);
        if (sequence != null) {
            return;
        }
        XmlSchema xmlSchema = getXmlSchema(schemaTargetNameSpace);
        XmlSchemaElement elt1 = new XmlSchemaElement(xmlSchema, false);
        elt1.setSchemaTypeName(schemaTypeName);
        elt1.setName(parameterName);
        elt1.setNillable(true);
        QName elementName = new QName(schemaTargetNameSpace, parameterName,
                schema_namespace_prefix);
        xmlSchema.getElements().put(elementName, elt1);
        xmlSchema.getItems().add(elt1);
        typeTable.addComplexSchema(parameterName, elementName);

    }

}