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
 	 */
 	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();
 		}
 	}
 }