package org.codehaus.xfire.service.binding;

import java.lang.reflect.Array;

import javax.xml.stream.XMLStreamException;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.codehaus.xfire.MessageContext;
import org.codehaus.xfire.XFireRuntimeException;
import org.codehaus.xfire.exchange.MessageExchange;
import org.codehaus.xfire.exchange.OutMessage;
import org.codehaus.xfire.fault.XFireFault;
import org.codehaus.xfire.handler.AbstractHandler;
import org.codehaus.xfire.service.OperationInfo;
import org.codehaus.xfire.util.DepthXMLStreamReader;

public abstract class AbstractBinding
    extends AbstractHandler
    implements ObjectBinding
{
    private static final Log logger = LogFactory.getLog(AbstractBinding.class.getName());

    public static final String OPERATION_KEY = "xfire.operation";
    
    public static final String RESPONSE_VALUE = "xfire.java.response";

    public static final String RESPONSE_PIPE = "xfire.java.responsePipe";

    private String style;
    private String use;
    private Invoker invoker;
    private BindingProvider bindingProvider;
    private boolean clientModeOn = false;
    
    public void setOperation(OperationInfo operation, MessageContext context)
    {
        MessageExchange exchange = context.createMessageExchange(operation);
        
        context.setExchange(exchange);
    }

    public void invoke(final MessageContext context)
        throws Exception
    {
        try
        {
            // Read in the parameters...

            // Don't read the operation in until after reading. Otherwise
            // it won't work for document style services.
            final OperationInfo operation = context.getExchange().getOperation();
            final Class[] parameterTypes = operation.getMethod().getParameterTypes();
            
            final Invoker invoker = getInvoker();
            
            // invoke the service method...
            if (!operation.isAsync())
            {
                final Object value = invoker.invoke(operation.getMethod(), castToRightParameterTypes(parameterTypes,context.getInMessage().getBody()), context);

                OutMessage outMsg = context.getOutMessage();
                if (outMsg != null)
                {
                    outMsg.setBody(new Object[] {value});
                    context.setOutMessage(outMsg);
                }
            }
            else
            {
                Runnable runnable = new Runnable() 
                {
                    public void run() 
                    {
                        try
                        {
                            final Object value = invoker.invoke(operation.getMethod(), castToRightParameterTypes(parameterTypes,context.getInMessage().getBody()), context);

                            OutMessage outMsg = context.getOutMessage();
                            if (outMsg != null)
                            {
                                outMsg.setBody(new Object[] {value});
                                context.setOutMessage(outMsg);
                            }
                        }
                        catch (XFireFault e)
                        {
                            context.getExchange().handleFault(e);
                        }
                    }
                };
                
                Thread opthread = new Thread(runnable);
                opthread.start();
            }
        }
        catch (XFireRuntimeException e)
        {
            logger.warn("Error invoking service.", e);
            throw new XFireFault("Error invoking service.", e, XFireFault.SENDER);
        }
    }
    
    private Object makeArray(Class arrayType,Object[] objects){
        if(String.class.equals(arrayType)){
            Object array = Array.newInstance(String.class,objects.length);
            for(int i=0,n=objects.length;i<n;i++){
                Array.set(array,i,objects[i]);
            }
            return array;
        }
        else if(Integer.TYPE.equals(arrayType)){
            Object array = Array.newInstance(Integer.TYPE,objects.length);
            for(int i=0,n=objects.length;i<n;i++){
                Array.set(array,i,objects[i]);
            }
            return array;
        }
        else if(Long.TYPE.equals(arrayType)){
            Object array = Array.newInstance(Long.TYPE,objects.length);
            for(int i=0,n=objects.length;i<n;i++){
                Array.set(array,i,objects[i]);
            }
            return array;
        } 
        else if(Short.TYPE.equals(arrayType)){
            Object array = Array.newInstance(Short.TYPE,objects.length);
            for(int i=0,n=objects.length;i<n;i++){
                Array.set(array,i,objects[i]);
            }
            return array;
        } 
        else if(Double.TYPE.equals(arrayType)){
            Object array = Array.newInstance(Double.TYPE,objects.length);
            for(int i=0,n=objects.length;i<n;i++){
                Array.set(array,i,objects[i]);
            }
            return array;
        } 
        else if(Float.TYPE.equals(arrayType)){
            Object array = Array.newInstance(Float.TYPE,objects.length);
            for(int i=0,n=objects.length;i<n;i++){
                Array.set(array,i,objects[i]);
            }
            return array;
        }
        else if(Byte.TYPE.equals(arrayType)){
            Object array = Array.newInstance(Byte.TYPE,objects.length);
            for(int i=0,n=objects.length;i<n;i++){
                Array.set(array,i,objects[i]);
            }
            return array;
        }
        else if(Boolean.TYPE.equals(arrayType)){
            Object array = Array.newInstance(Boolean.TYPE,objects.length);
            for(int i=0,n=objects.length;i<n;i++){
                Array.set(array,i,objects[i]);
            }
            return array;
        }
        return null;
    }
    
    private Object[] castToRightParameterTypes(Class[] parameterTypes,Object obj){
        Object[] parameter = (Object[]) obj;
        Object[] result = new Object[parameterTypes.length];
        for(int i=0,n=parameterTypes.length;i<n;i++) {
            Class arrayType = parameterTypes[i].getComponentType();
            if(arrayType!=null){
                Object[] objects = (Object[]) parameter[i];
                Object array = makeArray(arrayType,objects);
                if(array!=null){
                    result[i] = array;
                }
                else {
                    result[i] = parameter[i];
                }
            }
            else{
                result[i] = parameter[i];
            }
        }
        return result;
    }

    protected void nextEvent(DepthXMLStreamReader dr)
    {
        try
        {
            dr.next();
        }
        catch (XMLStreamException e)
        {
            throw new XFireRuntimeException("Couldn't parse stream.", e);
        }
    }

    public Invoker getInvoker()
    {
        return invoker;
    }
 
    public void setInvoker(Invoker invoker)
    {
        this.invoker = invoker;
    }
    
    public String getStyle()
    {
        return style;
    }
    
    protected void setStyle(String style)
    {
        this.style = style;
    }
    
    public String getUse()
    {
        return use;
    }
    
    protected void setUse(String use)
    {
        this.use = use;
    }

    public BindingProvider getBindingProvider()
    {
        return bindingProvider;
    }

    public void setBindingProvider(BindingProvider bindingProvider)
    {
        this.bindingProvider = bindingProvider;
    }

    public boolean isClientModeOn()
    {
        return clientModeOn;
    }

    public void setClientModeOn(boolean clientModeOn)
    {
        this.clientModeOn = clientModeOn;
    }
}