package org.apache.torque.task;

 import java.io.File;

 import java.io.FileWriter;

 import java.io.IOException;

 import java.io.StringWriter;

 import javax.xml.parsers.DocumentBuilder;

 import javax.xml.parsers.DocumentBuilderFactory;

 import javax.xml.parsers.ParserConfigurationException;

 import javax.xml.transform.OutputKeys;

 import javax.xml.transform.Result;

 import javax.xml.transform.Transformer;

 import javax.xml.transform.TransformerConfigurationException;

 import javax.xml.transform.TransformerException;

 import javax.xml.transform.TransformerFactory;

 import javax.xml.transform.dom.DOMSource;

 import javax.xml.transform.stream.StreamResult;

 import org.apache.tools.ant.BuildException;

 import org.apache.tools.ant.Task;

 import org.w3c.dom.DOMImplementation;

 import org.w3c.dom.Document;

 import org.w3c.dom.Element;

 import org.w3c.dom.Node;

 import org.w3c.dom.NodeList;

 import org.xml.sax.SAXException;

 /**

  * This class is task used to generate xml file with database information and its description.

  * 

  * @author Kuali Rice Team (kuali-rice@googlegroups.com)

  */

 public class TorqueXMLWithDesc extends Task {

     File outputFile;

     File inputFile;

     String inputFileString;

     String outputFileString;

     public void setOutputFileString(String outputFileString) {

         this.outputFileString = outputFileString;

         outputFile = new File(outputFileString);

     }

     public void setInputFileString(String inputFileString) {

         this.inputFileString = inputFileString;

         inputFile = new File(inputFileString);

     }

     /**

      * Returns a document object with table and column name from the input file and a blank description

      * 

      * @return Document document object with table/column names and blank description added

      * @throws ParserConfigurationException

      * @throws SAXException

      * @throws IOException

      */

     public Document createXMLWithDescription() throws ParserConfigurationException, SAXException, IOException {

         DocumentBuilderFactory documentBuilderFactory = DocumentBuilderFactory.newInstance();

         DocumentBuilder documentBuilder = documentBuilderFactory.newDocumentBuilder();

         Document inputDocument = documentBuilder.parse(inputFile);

         DOMImplementation domImplementation = documentBuilder.getDOMImplementation();

         inputDocument.getDocumentElement().normalize();

         Document outputDocument = domImplementation.createDocument(null, null, null);

         Element outputRootElement = outputDocument.createElement("database");

         outputDocument.appendChild(outputRootElement);

         NodeList listOfTableElementsInInput = inputDocument.getElementsByTagName("table");

         for (int i = 0; i < listOfTableElementsInInput.getLength(); i++) {

             Node currentTableNode = listOfTableElementsInInput.item(i);

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

                 Element currentInputTableElement = (Element) currentTableNode;

                 Element currentOutputTableElement = outputDocument.createElement("table");

                 System.out.println(currentTableNode.getAttributes().toString());

                 currentOutputTableElement.setAttribute("name", currentInputTableElement.getAttribute("name"));

                 currentOutputTableElement.setAttribute("description", "");

                 currentOutputTableElement.setAttribute("javaName", "");

                 NodeList listOfChildNodesInCurrentTableNode = currentTableNode.getChildNodes();

                 for (int j = 0; j < listOfChildNodesInCurrentTableNode.getLength(); j++) {

                     Node currentChildNode = listOfChildNodesInCurrentTableNode.item(j);

                     if (currentChildNode.getNodeName().equals("column")) {

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

                             Element currentInputColumnElement = (Element) currentChildNode;

                             Element currentOutputColumnElement = outputDocument.createElement("column");

                             currentOutputColumnElement.setAttribute("name",

                                     currentInputColumnElement.getAttribute("name"));

                             currentOutputColumnElement.setAttribute("description", "");

                             currentOutputColumnElement.setAttribute("javaName", "");

                             currentOutputTableElement.appendChild(currentOutputColumnElement);

                         }

                     }

                 }

                 outputRootElement.appendChild(currentOutputTableElement);

             }

         }

         return outputDocument;

     }

     /**

      * Writes the XMLDocument to an output file.

      * 

      * @param newXMLDocument

      */

     public void writeXMLToFile(Document newXMLDocument) {

         TransformerFactory tFactory = TransformerFactory.newInstance();

         try {

             Transformer transformer = tFactory.newTransformer();

             transformer.setOutputProperty(OutputKeys.DOCTYPE_SYSTEM, "database.dtd");

             DOMSource domSource = new DOMSource(newXMLDocument);

             StringWriter writer = new StringWriter();

             Result result = new StreamResult(writer);

             transformer.transform(domSource, result);

             FileWriter fileWriter = new FileWriter(outputFile);

             if (outputFile.exists()) {

                 StringBuffer bufferedWriter = new StringBuffer(writer.toString());

                 fileWriter.write(bufferedWriter.toString());

                 fileWriter.close();

                 System.out.println("The data has been written");

             } else {

                 System.out.println("This file is not exist");

             }

         } catch (TransformerConfigurationException e) {

             // TODO Auto-generated catch block

             e.printStackTrace();

         } catch (TransformerException e) {

             // TODO Auto-generated catch block

             e.printStackTrace();

         } catch (IOException e) {

             // TODO Auto-generated catch block

             e.printStackTrace();

         }

     }

     /**

      * This is the exceute method of the Task called by the ant script

      */

     @Override

     public void execute() throws BuildException {

         try {

             writeXMLToFile(createXMLWithDescription());

         } catch (ParserConfigurationException e) {

             // TODO Auto-generated catch block

             e.printStackTrace();

         } catch (SAXException e) {

             // TODO Auto-generated catch block

             e.printStackTrace();

         } catch (IOException e) {

             // TODO Auto-generated catch block

             e.printStackTrace();

         }

     }

 }