/* 
 * @(#)FunBooks.java
 *
 * Copyright 2002 Sun Microsystems, Inc. All rights reserved.
 * SUN PROPRIETARY/CONFIDENTIAL. Use is subject to license terms
 * 
 */

package com.sun.bookprovider;

import javax.xml.messaging.*;
import java.util.*;
import javax.xml.soap.*;
import javax.servlet.*;
import javax.servlet.http.*;
import javax.xml.transform.*;
import javax.xml.transform.stream.*;
import javax.xml.transform.dom.*;
import org.w3c.dom.*;
import java.sql.*;
import java.io.*;
import java.net.*;
import javax.xml.parsers.*;
import javax.activation.*;
import javax.naming.*;

/**
 * FunBooks.java follows JAXM Synchronous invocation model.
 * BookClient of EmployeePortal uses call method of JAXM to send synchronous
 * message to this provider.
 * This class  does the following 
 *     1) gets the request
 *     2) extract the query criteria and query value from the request.
 *     3) checks the inventory for the books matching the criteria
 *     4) sends the response as a SOAP attachment.
 *     5) also sends any promotion information as another attachment.
 * @author Bhakti Mehta and Ramesh Mandava
 * 
 */


public class FunBooks extends JAXMServlet implements ReqRespListener {

    static MessageFactory fac = null;
    FileOutputStream fos ;
    Hashtable bookHash = new Hashtable();
    Vector bookInfoVector = new Vector();

    static {
        try {
            fac = MessageFactory.newInstance();
        } catch (Exception ex) {
            ex.printStackTrace();
        }
    };

    public void init(ServletConfig servletConfig) throws ServletException {
        super.init(servletConfig);
    }


    public void executeQuery( String queryType, String queryVal ) {
        try {

            /*
             * This method looks for books matching the query type
             * and query value by parsing the funbooksinventory.xml
             */

	    DocumentBuilderFactory dbf = DocumentBuilderFactory.newInstance();
            DocumentBuilder db = dbf.newDocumentBuilder();
            Document doc = db.parse(new File("funbooksinventory.xml"));

            org.w3c.dom.Node parentNode = null;
            NodeList childList=null;
            Vector parentNodeVector = new Vector();
            Book bookInfo;

            NodeList nl = doc.getElementsByTagName(queryType );

            /*
             * This  means that there are no matching values
             */ 
            if (nl.getLength()==0){
               System.out.println("No matching query results");
               DocumentBuilderFactory docBFac=DocumentBuilderFactory.newInstance();
               DocumentBuilder docBuilder = docBFac.newDocumentBuilder();
               Document document = docBuilder.newDocument();
               Element books = document.createElement("books");
               document.appendChild(books);

               DOMSource domSource= new DOMSource(document);
               TransformerFactory tfactory = TransformerFactory.newInstance();

               Transformer transformer = tfactory.newTransformer();
               transformer.setOutputProperty(OutputKeys.INDENT,"yes");

               /* 
                * The file to be sent will be an empty file
                */
               FileOutputStream fileOutputStream = new FileOutputStream
                           ("../webapps/ROOT/funbooks.xml");
               transformer.transform(domSource, new StreamResult(fos));
               fileOutputStream.close();

            }else {
	          bookInfoVector = new Vector();
                  for (int i = 0;i<nl.getLength();i++) {
                      /*
                       * We can now check for values matching our criteria
                       */
                      NodeList queryList = (nl.item(i)).getChildNodes();
                      if ( queryList.item(0).getNodeValue().trim()
                            .indexOf(queryVal )>=0 ){
                         System.out.println("Query Matched"+
                                queryList.item(0).getNodeValue());
                         parentNode = nl.item(i).getParentNode();

                         /*
                          * There is match for the query
                          * Now this gets all other values from the parent
                          * such as bookName,bookAuthor,bookPrice etc.
                          *
                          */
		         String bookISBN="";
		         String bookAuthor="";
		         String bookName="";
		         String bookPrice="";
		         String bookQuantity="";
		         String bookDescription=""; 
		    
                         childList = parentNode.getChildNodes();

                         for (int j = 0;j<childList.getLength();j++) {
                              org.w3c.dom.Node childNode = childList.item(j);

                              if (childNode.getNodeName().equals("isbn")){
                                  System.out.println("ISBN"+ childNode
                                       .getChildNodes().item(0).getNodeValue());

                                  bookISBN = childNode.getChildNodes().item(0).
                                       getNodeValue().trim();

                              } else if (childNode.getNodeName()
                                        .equals("name")){

                	     	 System.out.println("name"+ childNode
                                    .getChildNodes().item(0).getNodeValue());
			         bookName =childNode.getChildNodes().item(0)
                                    .getNodeValue().trim();

                              } else if (childNode.getNodeName()
                                          .equals("author1")){
                                 System.out.println("author"+ childList.item(j)
                                     .getChildNodes().item(0).getNodeValue());
			         bookAuthor = childNode.getChildNodes().item(0)
                                     .getNodeValue().trim();
                              } else if (childList.item(j).getNodeName()
                                          .equals("price")){
                                 System.out.println("price"+ childList.item(j)
                                      .getChildNodes().item(0).getNodeValue());
			         bookPrice= childNode.getChildNodes().item(0)
                                      .getNodeValue().trim();
                              } else if (childList.item(j).getNodeName()
                                      .equals("quantity")){
                                 System.out.println("quantity"+
                                        childList.item(j).getChildNodes()
                                      .item(0).getNodeValue());
			         bookQuantity= childNode.getChildNodes()
                                      .item(0).getNodeValue().trim();
                              } else if (childList.item(j).getNodeName()
                                      .equals("description")){
                                 System.out.println("description"+
                                      childList.item(j).getChildNodes().
                                      item(0).getNodeValue());
			         bookDescription= childNode.getChildNodes()
                                     .item(0).getNodeValue().trim();
                             }

                         }
                         /* Here all the book information is stored in a 
                          * class called Book which will later be used 
                          * by processResults method
                          */
		         bookInfo = new Book ( bookISBN, bookName, bookAuthor,
                         bookPrice, bookQuantity, bookDescription );
		         bookInfoVector.addElement( bookInfo);

                    }//end if

                }//end for
	        processResults();
	    }
        } catch ( Exception e ) {
	    e.printStackTrace();
        }
    }


    /*
     * This method will create the xml file based on all
     * the book information obtained from the funbooksinventory.xml
     * This xml file will be sent as an attachment
     */
    public void processResults() {
        try {

            DocumentBuilderFactory dbf = DocumentBuilderFactory.newInstance();
            DocumentBuilder db = dbf.newDocumentBuilder();
            Document doc = db.newDocument();

            Element books = doc.createElement("books");
            doc.appendChild(books);
            Element book = null;
            Element isbn = null;
            Element name = null;
            Element author = null;
            Element price = null;
            Element quantity = null;
            Element description = null;
           
            /* 
             * This iterates through the bookInfoVector
             * The bookInfoVector contains all the book
             * information which will be used to populate the 
             * nodes
             */
            for ( int k=0; k < bookInfoVector.size(); k++ ) {

	       Book bookInfo = (Book)bookInfoVector.elementAt( k);
	       book = doc.createElement("book");
	       isbn = doc.createElement("ISBN");
	       org.w3c.dom.Text t = doc.createTextNode(bookInfo.getIsbn() );
	       isbn.appendChild(t);
	       book.appendChild(isbn);

	       name = doc.createElement("name");
	       org.w3c.dom.Text t1 = doc.createTextNode(bookInfo.getName());
	       name.appendChild(t1);
               book.appendChild(name);

	       author = doc.createElement("author");
	       org.w3c.dom.Text t2 = doc.createTextNode( bookInfo.getAuthor() );
	       author.appendChild(t2);
	       book.appendChild(author);

	       description = doc.createElement("description");
	       org.w3c.dom.Text t3 = doc.createTextNode
                     ( bookInfo.getDescription() );
	       description.appendChild(t3);
	       book.appendChild(description);

	       price = doc.createElement("price");
	       org.w3c.dom.Text t4 = doc.createTextNode( bookInfo.getPrice() );
	       price.appendChild(t4);
	       book.appendChild(price);

	       quantity = doc.createElement("quantity");
	       org.w3c.dom.Text t5 = doc.createTextNode( 
                     bookInfo.getQuantity() );
	       quantity.appendChild(t5);
	       book.appendChild(quantity);
               
               books.appendChild(book);
            }

            /*
             * The document is then written to an xml file
             * This file will be sent as an attachment
             */

            DOMSource domSource= new DOMSource(doc);
            TransformerFactory tfactory = TransformerFactory.newInstance();

            Transformer transformer = tfactory.newTransformer();
            transformer.setOutputProperty(OutputKeys.INDENT,"yes");

            fos = new FileOutputStream("../webapps/ROOT/funbooks.xml");
            transformer.transform(domSource, new StreamResult(fos));
            fos.close();
            

        }catch (Exception e) {
            e.printStackTrace();

        }

    }


    /* This is method that will get invoked when the call method of Connection
     * is called
     * The message received contains the searchCriteria and searchValue
     * which the user supplied.
     */
    public SOAPMessage onMessage(SOAPMessage message) {
        try {
            System.out.println("Here's the message: ");
            message.writeTo(System.out);

            SOAPEnvelope menv = message.getSOAPPart().getEnvelope();
            SOAPBody sb = menv.getBody();
            Iterator sbes = sb.getChildElements(menv.createName
                     ("GetBookDetails","bp","http://www.bookprovider.com"));

            String searchCriteria="";
            String searchValue="";

            while ( sbes.hasNext() ) {
                SOAPBodyElement sbe = (SOAPBodyElement)sbes.next();

                Iterator scIterator =sbe.getChildElements(
                      menv.createName("searchCriteria", "bp",
                      "http://www.bookprovider.com" ));
		if ( scIterator.hasNext() ) {
		 
                     SOAPElement child = (SOAPElement)scIterator.next();

                     searchCriteria=child.getValue();
		} else {
	 	    System.err.println("ERROR : Received Message doesn't "+
                     " contain searchCriteria. Not processing the message."+
                     " Returning null" );
		    return null;
		}

                Iterator svIterator =sbe.getChildElements(
                     menv.createName("searchValue", "bp",
                     "http://www.bookprovider.com" ));
		if ( svIterator.hasNext() ) {
		 
                    SOAPElement searchValueElement = (SOAPElement)svIterator
                             .next();
                    searchValue=searchValueElement.getValue();
		} else {
	 	    System.err.println("ERROR : Received Message doesn't "+
                      " contain searchValue. Not processing the message."+
                      " Returning null" );
		    return null;
		}
		 
		
            }//end while hasNext

            /* Now the searchCriteria and searchValue will be used to check if
             * there are any books matching the criteria
             */
            System.out.println("searchCriteria"+searchCriteria+" "+searchValue);
            executeQuery(searchCriteria,searchValue);

            /*
             * Construct the reply message with the book information
             * and the promotion attachment
             */
            SOAPMessage msg = fac.createMessage();
            SOAPEnvelope env = msg.getSOAPPart().getEnvelope();

	    SOAPHeader header = env.getHeader();
	    header.addHeaderElement( env.createName("organization", "bp",
			"http://www.bookprovider.com" ) ).
			addTextNode("FunBooks" );

           env.getBody() .addChildElement(env.createName("Response"))
                .addTextNode("This is a response"+searchCriteria+searchValue);


           /*
            * Attaching the xml file tot he soap message
            */
           URL url = new URL("http://localhost:8080/funbooks.xml");

           AttachmentPart ap = msg.createAttachmentPart(
                                        new DataHandler(url));
           ap.setContentType("text/xml");

           msg.addAttachmentPart(ap);
           msg.saveChanges();

                
           System.out.println("now adding promotion attachment");
           url = new URL ("http://localhost:8080/funbooks/promotion.html");
           ap = msg.createAttachmentPart(new DataHandler (url));  
           ap.setContentType("text/html");
           msg.addAttachmentPart(ap);
           System.out.println("No of attachments " + msg.countAttachments());
           msg.saveChanges();


           FileOutputStream sentFile = new FileOutputStream("sent.msg");
           msg.writeTo(sentFile);
           sentFile.close();


           return msg;
        } catch(Exception e) {
            e.printStackTrace();
            return null;
        }
    }


}