/**

  * Alt2XML

  * Copyright © 2014 František Kučera (frantovo.cz)

  *

  * This program 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

  * (at your option) any later version.

  *

  * This program 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 this program. If not, see <http://www.gnu.org/licenses/>.

  */

 package cz.frantovo.alt2xml;

 import java.util.Deque;

 import java.util.HashMap;

 import java.util.LinkedList;

 import java.util.Map;

 import java.util.Map.Entry;

 import java.util.ServiceLoader;

 import javax.xml.parsers.FactoryConfigurationError;

 import javax.xml.parsers.ParserConfigurationException;

 import javax.xml.parsers.SAXParser;

 import javax.xml.parsers.SAXParserFactory;

 import org.xml.sax.SAXException;

 import org.xml.sax.SAXNotRecognizedException;

 import org.xml.sax.SAXNotSupportedException;

 import org.xml.sax.XMLReader;

 /**

  *

  * @author Ing. František Kučera (frantovo.cz)

  */

 public class ParserFactory extends SAXParserFactory implements ReaderFinder {

 	private final Deque<Alt2XmlReaderFactory> readerFactories = new LinkedList();

 	/**

 	 * @see #DEFAULT_FACTORY_PROPERTY

 	 */

 	private SAXParserFactory fallbackFactory;

 	/**

 	 * System property which contains SAXParserFactory class name for XML.

 	 * Will be used as fallback if no alternative factory matches systemId to be parsed.

 	 */

 	public static final String DEFAULT_FACTORY_PROPERTY = "cz.frantovo.alt2xml.fallback.javax.xml.parsers.SAXParserFactory";

 	private final Map<String, Boolean> features = new HashMap<>();

 	public ParserFactory() {

 		super();

 		for (Alt2XmlReaderFactory f : ServiceLoader.load(Alt2XmlReaderFactory.class)) {

 			readerFactories.add(f);

 		}

 		readerFactories.add(new FallbackReaderFactory());

 	}

 	/**

 	 * @return factory to be used for XML documents (default/fallback)

 	 * @throws FactoryConfigurationError if fallback factory is the same one as this – avoid

 	 * infinite recursion.

 	 */

 	public SAXParserFactory getFallbackFactory() {

 		if (fallbackFactory == null) {

 			String className = System.getProperty(DEFAULT_FACTORY_PROPERTY);

 			if (className == null) {

 				fallbackFactory = SAXParserFactory.newInstance();

 			} else {

 				fallbackFactory = SAXParserFactory.newInstance(className, null);

 			}

 		}

 		if (fallbackFactory.getClass().equals(getClass())) {

 			throw new FactoryConfigurationError("Fallback factory is the same class as this one – avoid infinite recursion: " + getClass());

 		} else {

 			return fallbackFactory;

 		}

 	}

 	public void setFallbackFactory(SAXParserFactory fallbackFactory) {

 		this.fallbackFactory = fallbackFactory;

 	}

 	private class FallbackReaderFactory implements Alt2XmlReaderFactory {

 		@Override

 		public boolean canRead(AltInputSource inputSource) {

 			return true;

 		}

 		@Override

 		public XMLReader getReader() throws SAXException {

 			try {

 				return getFallbackFactory().newSAXParser().getXMLReader();

 			} catch (ParserConfigurationException e) {

 				throw new SAXException("Unable to instantiate the fallback factory.", e);

 			}

 		}

 	}

 	@Override

 	public XMLReader findReader(AltInputSource inputSource) throws SAXException {

 		for (Alt2XmlReaderFactory f : readerFactories) {

 			if (f.canRead(inputSource)) {

 				return f.getReader();

 			}

 		}

 		throw new SAXException("Iterated over " + readerFactories.size() + " and was unable to find XMLReader for SystemId: " + inputSource);

 	}

 	@Override

 	public SAXParser newSAXParser() throws ParserConfigurationException, SAXException {

 		SuperReader r = new SuperReader(this);

 		for (Entry<String, Boolean> f : features.entrySet()) {

 			r.setFeature(f.getKey(), f.getValue());

 		}

 		return new AltSAXParser(r);

 	}

 	@Override

 	public void setFeature(String name, boolean value) throws ParserConfigurationException, SAXNotRecognizedException, SAXNotSupportedException {

 		features.put(name, value);

 	}

 	@Override

 	public boolean getFeature(String name) throws ParserConfigurationException, SAXNotRecognizedException, SAXNotSupportedException {

 		return features.get(name);

 	}

 }