package test.wsdl.jaxrpchandlereval;

import javax.xml.namespace.QName;
import javax.xml.rpc.handler.Handler;
import javax.xml.rpc.handler.HandlerInfo;
import javax.xml.rpc.handler.MessageContext;
import javax.xml.rpc.handler.soap.SOAPMessageContext;
import javax.xml.soap.Name;
import javax.xml.soap.SOAPEnvelope;
import javax.xml.soap.SOAPHeader;
import javax.xml.soap.SOAPBody;
import javax.xml.soap.SOAPHeaderElement;
import javax.xml.soap.SOAPMessage;
import javax.xml.soap.SOAPPart;
import java.util.Iterator;
import javax.xml.soap.*;
import javax.xml.rpc.handler.soap.SOAPMessageContext;
import javax.xml.rpc.soap.*;
import javax.xml.rpc.*;


/**
 */
public class ClientHandler implements Handler {

private final static String _actorURI = "myActorURI";
/**
 * Constructor for ClientHandler.
 */
public ClientHandler() {
    System.out.println("ClientHandler:Constructor");
}

/**
 * @see javax.xml.rpc.handler.Handler#handleRequest(MessageContext)
 */
public boolean handleRequest(MessageContext context) {
    System.out.println("ClientHandler:handleRequest");
	HandlerTracker.addClientHandler("clienthandler1.handleRequest");
    if (context instanceof SOAPMessageContext) {
        try {
            SOAPMessageContext soapMsgCtx = (SOAPMessageContext)context;
            SOAPMessage soapMsg = soapMsgCtx.getMessage();
            SOAPPart sp = soapMsg.getSOAPPart();
            SOAPEnvelope se = sp.getEnvelope();
            SOAPBody sb = se.getBody();
            SOAPHeader sh = se.getHeader();

            Name name = se.createName("ClientHandler-handleRequest", "", "");
            SOAPHeaderElement hdr = sh.addHeaderElement(name);
            hdr.addTextNode("Processed");

            String instruction = sb.toString();
            if (instruction.indexOf("client-return-false") >= 0) {
                soapMsgCtx.setProperty("fault", "Returning false from ClientHandler.handleRequest");
                return false;
            }
        } catch (SOAPException ex) {
            ex.printStackTrace();
            return false;
        }
    }
    return true;
}

public SOAPMessage prepareError(SOAPMessageContext soapMsgCtx) throws Exception{
        MessageFactory messageFactory = MessageFactory.newInstance();
        SOAPMessage soapMsg = messageFactory.createMessage();
        soapMsgCtx.setMessage(soapMsg);

        String fault = (String)soapMsgCtx.getProperty("fault");
        if (fault != null) {
            SOAPFault soapFault =
                soapMsg.getSOAPPart().getEnvelope().getBody().addFault();
            soapFault.setFaultString(fault);
            soapFault.setFaultCode("client.Exception");
        }
        return soapMsg;
}

	private void trackServerHandler(SOAPHeader sh) {
		Iterator i = sh.getChildElements();
		while (i.hasNext()) {
			Node n = (Node) i.next();
			if (n instanceof Text) {
				continue;
			}
			SOAPElement e = (SOAPElement) n;
			String local = e.getLocalName();
			if (local.equals("ServiceHandler2-handleFault")) {
				HandlerTracker.addServerHandler("serverhandler2.handleFault");
			} else if (local.equals("ServiceHandler1-handleFault")) {
				HandlerTracker.addServerHandler("serverhandler1.handleFault");
			} else if (local.equals("ServiceHandler2-handleResponse")) {
				HandlerTracker.addServerHandler("serverhandler2.handleResponse");
			} else if (local.equals("ServiceHandler1-handleResponse")) {
				HandlerTracker.addServerHandler("serverhandler1.handleResponse");
			}
		}
	}


/**
 * @see javax.xml.rpc.handler.Handler#handleResponse(MessageContext)
 */
public boolean handleResponse(MessageContext context) {
    System.out.println("ClientHandler:handleResponse");
	HandlerTracker.addClientHandler("clienthandler1.handleResponse");

    if (context instanceof SOAPMessageContext) {
        try {
            SOAPMessageContext soapMsgCtx = (SOAPMessageContext)context;
            SOAPMessage soapMsg = soapMsgCtx.getMessage();
            if (soapMsg == null) {
                soapMsg = prepareError(soapMsgCtx);
            }
            SOAPPart sp = soapMsg.getSOAPPart();
            SOAPEnvelope se = sp.getEnvelope();
            SOAPBody sb = se.getBody();
            SOAPHeader sh = se.getHeader();

			trackServerHandler(sh);

            Name name = se.createName("ClientHandler-handleResponse", "", "");
            SOAPHeaderElement hdr = sh.addHeaderElement(name);
            hdr.addTextNode("Processed");
        } catch (SOAPException ex) {
            ex.printStackTrace();
            return false;
        } catch (Exception ex) {
            throw new JAXRPCException (ex);
        }
    }
    return true;
}

/**
 * @see javax.xml.rpc.handler.Handler#handleFault(MessageContext)
 */
public boolean handleFault(MessageContext context) {
    System.out.println("ClientHandler:handleFault");
	HandlerTracker.addClientHandler("clienthandler1.handleFault");
    if (context instanceof SOAPMessageContext) {
        try {
            SOAPMessageContext soapMsgCtx = (SOAPMessageContext)context;
            SOAPMessage soapMsg = soapMsgCtx.getMessage();
            SOAPPart sp = soapMsg.getSOAPPart();
            SOAPEnvelope se = sp.getEnvelope();
            SOAPBody sb = se.getBody();
            SOAPHeader sh = se.getHeader();

            Name name = se.createName("ClientHandler-handleFault", "", "");
            SOAPHeaderElement hdr = sh.addHeaderElement(name);
            hdr.addTextNode("Processed");
        } catch (SOAPException ex) {
            ex.printStackTrace();
            return false;
        }
    }
    return false;
}

/**
 * @see javax.xml.rpc.handler.Handler#init(HandlerInfo)
 */
public void init(HandlerInfo config) {
    System.out.println("ClientHandler.init");

}

/**
 * @see javax.xml.rpc.handler.Handler#destroy()
 */
public void destroy() {
    System.out.println("ClientHandler.destroy");
}

/**
 * @see javax.xml.rpc.handler.Handler#getHeaders()
 */
public QName[] getHeaders() {
    System.out.println("ClientHandler.getheaders");

    return null;
}

}