/*

  * JBoss, Home of Professional Open Source.

  * Copyright 2006, Red Hat Middleware LLC, and individual contributors

  * as indicated by the @author tags. See the copyright.txt file in the

  * distribution for a full listing of individual contributors.

  *

  * This is free software; you can redistribute it and/or modify it

  * under the terms of the GNU Lesser General Public License as

  * published by the Free Software Foundation; either version 2.1 of

  * the License, or (at your option) any later version.

  *

  * This software is distributed in the hope that it will be useful,

  * but WITHOUT ANY WARRANTY; without even the implied warranty of

  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU

  * Lesser General Public License for more details.

  *

  * You should have received a copy of the GNU Lesser General Public

  * License along with this software; if not, write to the Free

  * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA

  * 02110-1301 USA, or see the FSF site: http://www.fsf.org.

  */

 package org.apache.ws.scout.transport;

 import org.apache.commons.logging.Log;

 import org.apache.commons.logging.LogFactory;

 import org.apache.ws.scout.registry.RegistryException;

 import org.apache.ws.scout.transport.Transport;

 import org.apache.ws.scout.util.XMLUtils;

 import org.w3c.dom.Document;

 import org.w3c.dom.Element;

 import org.w3c.dom.NamedNodeMap;

 import org.w3c.dom.Node;

 import org.w3c.dom.NodeList;

 import org.xml.sax.InputSource;

 import javax.xml.parsers.DocumentBuilderFactory;

 import javax.xml.soap.MessageFactory;

 import javax.xml.soap.Name;

 import javax.xml.soap.SOAPBody;

 import javax.xml.soap.SOAPBodyElement;

 import javax.xml.soap.SOAPConnection;

 import javax.xml.soap.SOAPConnectionFactory;

 import javax.xml.soap.SOAPElement;

 import javax.xml.soap.SOAPException;

 import javax.xml.soap.SOAPFactory;

 import javax.xml.soap.SOAPFault;

 import javax.xml.soap.SOAPMessage;

 import javax.xml.soap.SOAPPart;

 import java.io.StringReader;

 import java.net.URI;

 /**

  * Transport based on SAAJ

  *

  * @author Anil Saldhana (anil@apache.org)

  * @author Richard Opalka (richard.opalka@redhat.com)

  */

 public class SaajTransport implements Transport {

     public static final String UDDI_V2_NAMESPACE = "urn:uddi-org:api_v2";

     private static Log log = LogFactory.getLog(SaajTransport.class);

     public Element send(Element request, URI endpointURL) throws TransportException {

         if (log.isDebugEnabled()) {

             String requestMessage = XMLUtils.convertNodeToXMLString(request);

             log.debug("Request message: %s\n%s" + endpointURL + ":" + requestMessage);

         }

         Element response = null;

         try {

             SOAPMessage message = this.createSOAPMessage(request);

             //Make the SAAJ Call now

             SOAPConnectionFactory soapConnectionFactory = SOAPConnectionFactory.newInstance();

             SOAPConnection connection = soapConnectionFactory.createConnection();

             SOAPMessage soapResponse = connection.call(message, endpointURL.toURL());

             SOAPBody soapBody = soapResponse.getSOAPBody();

             boolean hasFault = soapBody.hasFault();

             if (hasFault) {

                 SOAPFault soapFault = soapBody.getFault();

                 String faultStr = soapFault.getFaultCode() + "::" + soapFault.getFaultString();

                 throw new RegistryException(faultStr);

             }

             response = getFirstChildElement(soapBody);

         } catch (Exception ex) {

             log.error("Exception::" + ex.getMessage(), ex);

             throw new TransportException(ex);

         }

         if (log.isDebugEnabled()) {

             String responseMessage = XMLUtils.convertNodeToXMLString(response);

             log.debug("Response message: %s" + responseMessage);

         }

         return response;

     }

     public String send(String request, URI endpointURL) throws TransportException {

         Element reqEl = getElement(request);

         Element respEl = this.send(reqEl, endpointURL);

         return XMLUtils.convertNodeToXMLString(respEl);

     }

     private SOAPMessage createSOAPMessage(Element elem) throws Exception {

         String prefix = "";

         MessageFactory msgFactory = MessageFactory.newInstance();

         SOAPFactory factory = SOAPFactory.newInstance();

         SOAPMessage message = msgFactory.createMessage();

         message.getSOAPHeader().detachNode();

         SOAPPart soapPart = message.getSOAPPart();

         SOAPBody soapBody = soapPart.getEnvelope().getBody();

         //Create the outer body element

         Name bodyName = factory.createName(elem.getNodeName(), prefix, UDDI_V2_NAMESPACE);

         SOAPBodyElement bodyElement = soapBody.addBodyElement(bodyName);

         bodyElement.addNamespaceDeclaration(prefix, UDDI_V2_NAMESPACE);

         appendAttributes(bodyElement, elem.getAttributes(), factory);

         appendElements(bodyElement, elem.getChildNodes(), factory);

         return message;

     }

     private void appendAttributes(SOAPElement bodyElement, NamedNodeMap nnm, SOAPFactory factory) throws SOAPException {

         int len = nnm != null ? nnm.getLength() : 0;

         for (int i = 0; i < len; i++) {

             Node n = nnm.item(i);

             String nodename = n.getNodeName();

             String nodevalue = n.getNodeValue();

             if ("xmlns".equals(nodename))

                 continue;

             if (nodename.startsWith("xmlns:"))

                 continue;

             //Special case:  xml:lang

             if ("xml:lang".equals(nodename)) {

                 Name xmlLang = factory.createName("lang", "xml", "");

                 bodyElement.addAttribute(xmlLang, nodevalue);

             } else

                 bodyElement.addAttribute(factory.createName(nodename), nodevalue);

         }

     }

     private void appendElements(SOAPElement bodyElement, NodeList nlist, SOAPFactory factory) throws SOAPException {

         String prefix = "";

         int len = nlist != null ? nlist.getLength() : 0;

         for (int i = 0; i < len; i++) {

             Node node = nlist.item(i);

             short nodeType = node != null ? node.getNodeType() : -100;

             if (Node.ELEMENT_NODE == nodeType) {

                 Element el = (Element) node;

                 Name name = factory.createName(el.getNodeName(), prefix, UDDI_V2_NAMESPACE);

                 SOAPElement attachedEl = bodyElement.addChildElement(name);

                 appendAttributes(attachedEl, el.getAttributes(), factory);

                 appendElements(attachedEl, el.getChildNodes(), factory);

             } else if (nodeType == Node.TEXT_NODE) {

                 bodyElement.addTextNode(node.getNodeValue());

             }

         }

     }

     private static Element getElement(String xmlFrag) {

         Document doc = null;

         Element reqElement = null;

         try {

             DocumentBuilderFactory factory = DocumentBuilderFactory.newInstance();

             doc = factory.newDocumentBuilder().parse(new InputSource(new StringReader(xmlFrag)));

             reqElement = doc.getDocumentElement();

         } catch (Exception ex) {

             log.error("Exception:" + ex.getMessage(), ex);

         }

         return reqElement;

     }

     private Element getFirstChildElement(Element el) {

         return getFirstChildElement(el, null);

     }

     private Element getFirstChildElement(Element el, String tagName) {

         Element childEl = null;

         NodeList nlist = el != null ? el.getChildNodes() : null;

         int len = nlist != null ? nlist.getLength() : 0;

         for (int i = 0; childEl == null && i < len; i++) {

             Node node = nlist.item(i);

             if (node.getNodeType() == Node.ELEMENT_NODE) {

                 if (tagName == null || tagName.equals(node.getLocalName()))

                     childEl = (Element) node;

             }

         }

         if (log.isDebugEnabled()) {

             String responseObtained = XMLUtils.convertNodeToXMLString(childEl);

             log.debug("Response obtained: %s" + responseObtained);

         }

         return childEl;

     }

 }