com.github.gimmi.AnyXml Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of any Show documentation

An immutable data structure that provide an uniform interface to map, list or scalar data

There is a newer version: 0.4.0

package com.github.gimmi;

import org.xml.sax.Attributes;
import org.xml.sax.InputSource;
import org.xml.sax.SAXException;
import org.xml.sax.helpers.DefaultHandler;

import javax.xml.parsers.ParserConfigurationException;
import javax.xml.parsers.SAXParser;
import javax.xml.parsers.SAXParserFactory;
import java.io.IOException;
import java.io.Reader;
import java.io.StringReader;
import java.util.Stack;

public class AnyXml {
	public static Any fromXml(String xml) {
		return fromXml(new StringReader(xml));
	}

	public static Any fromXml(Reader reader) {
		try {
			SAXParserFactory factory = SAXParserFactory.newInstance();
			factory.setNamespaceAware(true);
			SAXParser saxParser = factory.newSAXParser();
			Handler handler = new Handler();
			saxParser.parse(new InputSource(reader), handler);
			return handler.any;
		} catch (ParserConfigurationException | SAXException | IOException e) {
			throw new RuntimeException("Erro parsing XML", e);
		}
	}

	private static class Handler extends DefaultHandler {
		private final Stack elements = new Stack<>();
		private final StringBuilder sb = new StringBuilder();
		private Any any;

		@Override
		public void startElement(String uri, String localName, String qName, Attributes attributes) throws SAXException {
			AnyMapBuilder builder = new AnyMapBuilder();
			for (int i = 0; i < attributes.getLength(); i++) {
				builder.put(attributes.getLocalName(i), Any.scalar(attributes.getValue(i)));
			}
			elements.push(builder);
		}

		@Override
		public void characters(char[] ch, int start, int length) throws SAXException {
			sb.append(ch, start, length);
		}

		@Override
		public void endElement(String uri, String localName, String qName) throws SAXException {
			AnyMapBuilder builder = elements.pop();
			any = Any.scalar(sb.toString());
			sb.setLength(0);
			if (builder.count() > 0) {
				any = builder.put("text", any).build();
			}

			if (!elements.empty()) {
				elements.peek().put(localName, any);
			}
		}
	}
}