org.apache.tika.config.TikaConfigSerializer Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of tika-core Show documentation
Show all versions of tika-core Show documentation
This is the core Apache Tika™ toolkit library from which all other modules inherit functionality. It also
includes the core facades for the Tika API.
/*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You under the Apache License, Version 2.0
* (the "License"); you may not use this file except in compliance with
* the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.apache.tika.config;
import javax.xml.parsers.DocumentBuilder;
import javax.xml.transform.OutputKeys;
import javax.xml.transform.Transformer;
import javax.xml.transform.TransformerFactory;
import javax.xml.transform.dom.DOMSource;
import javax.xml.transform.stream.StreamResult;
import java.io.Writer;
import java.nio.charset.Charset;
import java.util.Collections;
import java.util.List;
import java.util.Set;
import java.util.TreeSet;
import org.apache.tika.detect.CompositeDetector;
import org.apache.tika.detect.DefaultDetector;
import org.apache.tika.detect.Detector;
import org.apache.tika.language.translate.DefaultTranslator;
import org.apache.tika.language.translate.Translator;
import org.apache.tika.mime.MediaType;
import org.apache.tika.parser.CompositeParser;
import org.apache.tika.parser.DefaultParser;
import org.apache.tika.parser.ParseContext;
import org.apache.tika.parser.Parser;
import org.apache.tika.parser.ParserDecorator;
import org.w3c.dom.Document;
import org.w3c.dom.Element;
import org.w3c.dom.Node;
public class TikaConfigSerializer {
public enum Mode {
MINIMAL, CURRENT, STATIC;
}
/**
*
* @param config config to serialize
* @param mode serialization mode
* @param writer writer
* @param charset charset
* @throws Exception
*/
public static void serialize(TikaConfig config, Mode mode, Writer writer, Charset charset)
throws Exception {
DocumentBuilder docBuilder = new ParseContext().getDocumentBuilder();
// root elements
Document doc = docBuilder.newDocument();
Element rootElement = doc.createElement("properties");
doc.appendChild(rootElement);
addMimeComment(mode, rootElement, doc);
addServiceLoader(mode, rootElement, doc, config);
addExecutorService(mode, rootElement, doc, config);
addTranslator(mode, rootElement, doc, config);
addDetectors(mode, rootElement, doc, config);
addParsers(mode, rootElement, doc, config);
// TODO Service Loader section
// now write
TransformerFactory transformerFactory = TransformerFactory.newInstance();
Transformer transformer = transformerFactory.newTransformer();
transformer.setOutputProperty(OutputKeys.INDENT, "yes");
transformer.setOutputProperty("{http://xml.apache.org/xslt}indent-amount", "2");
transformer.setOutputProperty(OutputKeys.ENCODING, charset.name());
DOMSource source = new DOMSource(doc);
StreamResult result = new StreamResult(writer);
transformer.transform(source, result);
}
private static void addExecutorService(Mode mode, Element rootElement, Document doc, TikaConfig config) {
//TODO
}
private static void addServiceLoader(Mode mode, Element rootElement, Document doc, TikaConfig config) {
ServiceLoader loader = config.getServiceLoader();
if (mode == Mode.MINIMAL) {
// Is this the default?
if (loader.isDynamic() && loader.getLoadErrorHandler() == LoadErrorHandler.IGNORE) {
// Default config, no need to output anything
return;
}
}
Element dslEl = doc.createElement("service-loader");
dslEl.setAttribute("dynamic", Boolean.toString(loader.isDynamic()));
dslEl.setAttribute("loadErrorHandler", loader.getLoadErrorHandler().toString());
rootElement.appendChild(dslEl);
}
private static void addTranslator(Mode mode, Element rootElement, Document doc, TikaConfig config) {
// Unlike the other entries, TikaConfig only wants one of
// these, and no outer list
Translator translator = config.getTranslator();
if (mode == Mode.MINIMAL && translator instanceof DefaultTranslator) {
Node mimeComment = doc.createComment(
"for example: ");
rootElement.appendChild(mimeComment);
} else {
if (translator instanceof DefaultTranslator && mode == Mode.STATIC) {
translator = ((DefaultTranslator)translator).getTranslator();
}
if (translator != null) {
Element translatorElement = doc.createElement("translator");
translatorElement.setAttribute("class", translator.getClass().getCanonicalName());
rootElement.appendChild(translatorElement);
} else {
rootElement.appendChild(doc.createComment("No translators available"));
}
}
}
private static void addMimeComment(Mode mode, Element rootElement, Document doc) {
Node mimeComment = doc.createComment(
"for example: ");
rootElement.appendChild(mimeComment);
}
private static void addDetectors(Mode mode, Element rootElement, Document doc, TikaConfig config) throws Exception {
Detector detector = config.getDetector();
if (mode == Mode.MINIMAL && detector instanceof DefaultDetector) {
// Don't output anything, all using defaults
Node detComment = doc.createComment(
"for example: children = ((CompositeDetector)detector).getDetectors();
for (Detector d : children) {
Element detectorElement = doc.createElement("detector");
detectorElement.setAttribute("class", d.getClass().getCanonicalName());
detectorsElement.appendChild(detectorElement);
}
}
rootElement.appendChild(detectorsElement);
}
private static void addParsers(Mode mode, Element rootElement, Document doc, TikaConfig config) throws Exception {
Parser parser = config.getParser();
if (mode == Mode.MINIMAL && parser instanceof DefaultParser) {
// Don't output anything, all using defaults
return;
} else if (mode == Mode.MINIMAL) {
mode = Mode.CURRENT;
}
Element parsersElement = doc.createElement("parsers");
rootElement.appendChild(parsersElement);
addParser(mode, parsersElement, doc, parser);
}
private static void addParser(Mode mode, Element rootElement, Document doc, Parser parser) throws Exception {
// If the parser is decorated, is it a kind where we output the parser inside?
ParserDecorator decoration = null;
if (parser instanceof ParserDecorator) {
if (parser.getClass().getName().startsWith(ParserDecorator.class.getName()+"$")) {
decoration = ((ParserDecorator)parser);
parser = decoration.getWrappedParser();
}
}
boolean outputParser = true;
List children = Collections.emptyList();
if (mode == Mode.CURRENT && parser instanceof DefaultParser) {
// Only output the parser, not the children
} else if (parser instanceof CompositeParser) {
children = ((CompositeParser)parser).getAllComponentParsers();
// Special case for a naked composite
if (parser.getClass().equals(CompositeParser.class)) {
outputParser = false;
}
// Special case for making Default to static
if (mode == Mode.STATIC && parser instanceof DefaultParser) {
outputParser = false;
}
}
if (outputParser) {
rootElement = addParser(rootElement, doc, parser, decoration);
}
for (Parser childParser : children) {
addParser(mode, rootElement, doc, childParser);
}
// TODO Parser Exclusions
}
private static Element addParser(Element rootElement, Document doc, Parser parser, ParserDecorator decorator) throws Exception {
ParseContext context = new ParseContext();
Set addedTypes = new TreeSet<>();
Set excludedTypes = new TreeSet<>();
if (decorator != null) {
Set types = new TreeSet<>();
types.addAll(decorator.getSupportedTypes(context));
addedTypes.addAll(types);
for (MediaType type : parser.getSupportedTypes(context)) {
if (! types.contains(type)) {
excludedTypes.add(type);
}
addedTypes.remove(type);
}
}
String className = parser.getClass().getCanonicalName();
Element parserElement = doc.createElement("parser");
parserElement.setAttribute("class", className);
rootElement.appendChild(parserElement);
for (MediaType type : addedTypes) {
Element mimeElement = doc.createElement("mime");
mimeElement.appendChild(doc.createTextNode(type.toString()));
parserElement.appendChild(mimeElement);
}
for (MediaType type : excludedTypes) {
Element mimeElement = doc.createElement("mime-exclude");
mimeElement.appendChild(doc.createTextNode(type.toString()));
parserElement.appendChild(mimeElement);
}
return parserElement;
}
}