org.apache.torque.task.TorqueXMLWithDesc Maven / Gradle / Ivy

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 ([email protected])
 */
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();
		}
	}
}