com.moviejukebox.tools.DOMHelper Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of yamj Show documentation
Show all versions of yamj Show documentation
Static analysis of MovieJukebox project
The newest version!
/*
* Copyright (c) 2004-2013 YAMJ Members
* http://code.google.com/p/moviejukebox/people/list
*
* This file is part of the Yet Another Movie Jukebox (YAMJ).
*
* The YAMJ is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* any later version.
*
* YAMJ 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 General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with the YAMJ. If not, see childAttributes) {
Element child = doc.createElement(elementName);
Text text = doc.createTextNode(elementValue);
child.appendChild(text);
if (childAttributes != null && !childAttributes.isEmpty()) {
for (Map.Entry attrib : childAttributes.entrySet()) {
child.setAttribute(attrib.getKey(), attrib.getValue());
}
}
parentElement.appendChild(child);
}
/**
* Append a child element to a parent element with a single attribute/value pair
*
* @param doc
* @param parentElement
* @param elementName
* @param elementValue
* @param attribName
* @param attribValue
*/
public static void appendChild(Document doc, Element parentElement, String elementName, String elementValue, String attribName, String attribValue) {
Element child = doc.createElement(elementName);
Text text = doc.createTextNode(elementValue);
child.appendChild(text);
child.setAttribute(attribName, attribValue);
parentElement.appendChild(child);
}
/**
* Convert a DOM document to a string
*
* @param doc
* @return
* @throws TransformerException
*/
public static String convertDocToString(Document doc) throws TransformerException {
//set up a transformer
TransformerFactory transfac = TransformerFactory.newInstance();
Transformer trans = transfac.newTransformer();
trans.setOutputProperty(OutputKeys.OMIT_XML_DECLARATION, YES);
trans.setOutputProperty(OutputKeys.INDENT, YES);
//create string from xml tree
StringWriter sw = new StringWriter();
StreamResult result = new StreamResult(sw);
DOMSource source = new DOMSource(doc);
trans.transform(source, result);
return sw.toString();
}
/**
* Create a blank Document
*
* @return a Document
* @throws ParserConfigurationException
*/
public static Document createDocument() throws ParserConfigurationException {
DocumentBuilderFactory dbfac = DocumentBuilderFactory.newInstance();
DocumentBuilder docBuilder = dbfac.newDocumentBuilder();
return docBuilder.newDocument();
}
/**
* Get a DOM document from the supplied string
*
* @param docString
* @return
*/
public static Document getDocFromString(String docString) throws ParserConfigurationException, SAXException, IOException {
DocumentBuilderFactory dbf = DocumentBuilderFactory.newInstance();
InputSource is = new InputSource(new StringReader(docString));
DocumentBuilder db;
db = dbf.newDocumentBuilder();
Document doc = db.parse(is);
return doc;
}
/**
* Get a DOM document from the supplied file
*
* @param xmlFile
* @return
* @throws IOException
* @throws ParserConfigurationException
* @throws SAXException
*/
public static Document getDocFromFile(File xmlFile) throws ParserConfigurationException, SAXException, IOException {
URL url = xmlFile.toURI().toURL();
InputStream in = url.openStream();
DocumentBuilderFactory dbf = DocumentBuilderFactory.newInstance();
DocumentBuilder db = dbf.newDocumentBuilder();
Document doc;
// Custom error handler
db.setErrorHandler(new SaxErrorHandler());
try {
doc = db.parse(in);
} catch (SAXParseException ex) {
if (FilenameUtils.isExtension(xmlFile.getName().toLowerCase(), "xml")) {
throw new SAXParseException("Failed to process file as XML", null, ex);
} else {
// Try processing the file a different way
doc = null;
}
} finally {
// close the stream
in.close();
}
if (doc == null) {
// try wrapping the file in a root
StringReader sr = new StringReader(wrapInXml(FileTools.readFileToString(xmlFile)));
try {
doc = db.parse(new InputSource(sr));
} finally {
sr.close();
}
}
doc.getDocumentElement().normalize();
return doc;
}
/**
* Gets the string value of the tag element name passed
*
* @param element
* @param tagName
* @return
*/
public static String getValueFromElement(Element element, String tagName) {
String returnValue = DEFAULT_RETURN;
try {
NodeList nlElement = element.getElementsByTagName(tagName);
Element tagElement = (Element) nlElement.item(0);
NodeList tagNodeList = tagElement.getChildNodes();
returnValue = ((Node) tagNodeList.item(0)).getNodeValue();
} catch (Exception ignore) {
return returnValue;
}
return returnValue;
}
/**
* Get an element from a parent element node
*
* @param eParent
* @param elementName
* @return
*/
public static Element getElementByName(Element eParent, String elementName) {
NodeList nlParent = eParent.getElementsByTagName(elementName);
for (int looper = 0; looper < nlParent.getLength(); looper++) {
if (nlParent.item(looper).getNodeType() == Node.ELEMENT_NODE) {
return (Element) nlParent.item(looper);
}
}
return null;
}
/**
* Write the Document out to a file using nice formatting
*
* @param doc The document to save
* @param localFile The file to write to
* @return
*/
public static boolean writeDocumentToFile(Document doc, String localFilename) {
return writeDocumentToFile(doc, new File(localFilename));
}
/**
* Write the Document out to a file using nice formatting
*
* @param doc The document to save
* @param localFile The file to write to
* @return
*/
public static boolean writeDocumentToFile(Document doc, File localFile) {
try {
Transformer trans = TransformerFactory.newInstance().newTransformer();
// Define the output properties
trans.setOutputProperty(OutputKeys.OMIT_XML_DECLARATION, "no");
trans.setOutputProperty(OutputKeys.ENCODING, "UTF-8");
trans.setOutputProperty(OutputKeys.INDENT, YES);
trans.setOutputProperty("{http://xml.apache.org/xslt}indent-amount", "4");
doc.setXmlStandalone(true);
trans.transform(new DOMSource(doc), new StreamResult(localFile));
return true;
} catch (Exception error) {
LOG.error(LOG_MESSAGE + "Error writing the document to " + localFile);
LOG.error(LOG_MESSAGE + "Message: " + error.getMessage());
return false;
}
}
/**
* Override the standard Sax ErrorHandler with this one, to minimise noise about failed parsing errors
*/
public static class SaxErrorHandler implements ErrorHandler {
@Override
public void warning(SAXParseException exception) throws SAXException {
// logger.warn(buildMessage(exception, "Warning"));
throw new SAXParseException(null, null, exception);
}
@Override
public void error(SAXParseException exception) throws SAXException {
// logger.warn(buildMessage(exception, "Error"));
throw new SAXParseException(null, null, exception);
}
@Override
public void fatalError(SAXParseException exception) throws SAXException {
// logger.warn(buildMessage(exception, "Fatal Error"));
throw new SAXParseException(null, null, exception);
}
public String buildMessage(SAXParseException exception) {
return buildMessage(exception, null);
}
public String buildMessage(SAXParseException exception, String level) {
StringBuilder message = new StringBuilder("Parsing ");
if (StringUtils.isNotBlank(level)) {
message.append(level);
} else {
message.append("Error");
}
message.append(" - ");
message.append(" Line: ").append(exception.getLineNumber());
message.append(" Column: ").append(exception.getColumnNumber());
message.append(" Message: ").append(exception.getMessage());
return message.toString();
}
}
/**
* Take a file and wrap it in a new root element
*
* @param fileString
* @return
*/
public static String wrapInXml(String fileString) {
StringBuilder newOutput = new StringBuilder(fileString);
int posMovie = fileString.indexOf("<" + MovieNFOReader.TYPE_MOVIE);
int posTvShow = fileString.indexOf("<" + MovieNFOReader.TYPE_TVSHOW);
int posEpisode = fileString.indexOf("<" + MovieNFOReader.TYPE_EPISODE);
boolean posValid = Boolean.FALSE;
if (posMovie == -1) {
posMovie = fileString.length();
} else {
posValid = Boolean.TRUE;
}
if (posTvShow == -1) {
posTvShow = fileString.length();
} else {
posValid = Boolean.TRUE;
}
if (posEpisode == -1) {
posEpisode = fileString.length();
} else {
posValid = Boolean.TRUE;
}
if (posValid) {
int pos = Math.min(posMovie, Math.min(posTvShow, posEpisode));
newOutput.insert(pos, "<" + TYPE_ROOT + ">");
newOutput.append("");
}
return newOutput.toString();
}
}