1 files changed, 152 insertions, 0 deletions
diff --git a/sdo-java/branches/sdo-1.1-incubating/impl/src/main/java/org/apache/tuscany/sdo/helper/XMLStreamHelperImpl.java b/sdo-java/branches/sdo-1.1-incubating/impl/src/main/java/org/apache/tuscany/sdo/helper/XMLStreamHelperImpl.java
new file mode 100644
index 0000000000..7b0c2fc597
--- /dev/null
+++ b/sdo-java/branches/sdo-1.1-incubating/impl/src/main/java/org/apache/tuscany/sdo/helper/XMLStreamHelperImpl.java
@@ -0,0 +1,152 @@
+/**
+ *
+ *  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.tuscany.sdo.helper;
+
+import java.io.IOException;
+import java.util.Iterator;
+import java.util.Map;
+import java.util.Set;
+import javax.xml.namespace.QName;
+import javax.xml.stream.XMLStreamConstants;
+import javax.xml.stream.XMLStreamException;
+import javax.xml.stream.XMLStreamReader;
+import javax.xml.stream.XMLStreamWriter;
+
+import org.apache.tuscany.sdo.util.resource.DataObjectXMLStreamReader;
+import org.apache.tuscany.sdo.util.resource.XMLDocumentStreamReader;
+import org.apache.tuscany.sdo.util.resource.XMLStreamSerializer;
+//import org.apache.tuscany.sdo.api.XMLStreamHelper;
+import org.eclipse.emf.ecore.resource.Resource;
+
+import commonj.sdo.DataObject;
+import commonj.sdo.Property;
+import commonj.sdo.helper.HelperContext;
+import commonj.sdo.helper.XMLDocument;
+
+/**
+ * @version $Rev$ $Date$
+ */
+public class XMLStreamHelperImpl implements XMLStreamHelper {
+    protected HelperContext helperContext;
+
+    public XMLStreamHelperImpl(HelperContext hc) {
+        this.helperContext = hc;
+    }
+
+    private Map checkSetOptions(Map options) {
+        if (helperContext != null) {
+            return ((HelperContextImpl)helperContext).getMergedOption(options);
+        } else {// null is acceptable as it will be ignored
+            return options;
+        }
+    }
+
+    public XMLDocument load(XMLStreamReader reader) throws XMLStreamException, IllegalStateException {
+        if (reader.getEventType() != XMLStreamConstants.START_DOCUMENT)
+            throw new IllegalStateException();
+
+        return loadDocument(reader, null);
+    }
+
+    public void save(XMLDocument document, XMLStreamWriter writer) throws XMLStreamException {
+        XMLStreamReader reader = createXMLStreamReader(document);
+        new XMLStreamSerializer().serialize(reader, writer);
+    }
+
+    public void save(XMLDocument document, XMLStreamWriter writer, Map options) throws XMLStreamException {
+        XMLStreamReader reader = createXMLStreamReader(document);
+        options = checkSetOptions(options);
+        new XMLStreamSerializer().serialize(reader, writer, options);
+    }
+
+    public XMLStreamReader createXMLStreamReader(XMLDocument document) throws XMLStreamException {
+        XMLStreamReader reader =
+            new DataObjectXMLStreamReader(document.getRootObject(), document.getRootElementURI(), document
+                .getRootElementName(), helperContext.getTypeHelper());
+        // Wrap the reader so that its position will be START_ELEMENT
+        return new XMLDocumentStreamReader(reader);
+
+    }
+
+    public final DataObject loadObject(XMLStreamReader reader, Map options) throws XMLStreamException,
+        IllegalStateException {
+        if (reader.getEventType() != XMLStreamConstants.START_ELEMENT)
+            throw new IllegalStateException();
+
+        // StAX2SAXAdapter won't produce START_DOCUMENT if the reader is
+        // posisitioned at START_ELEMENT and the EMF loader will fail
+        // Wrap the reader so it represents a document
+        reader = new XMLDocumentStreamReader(reader);
+
+        return loadDocument(reader, options).getRootObject();
+    }
+
+    public DataObject loadObject(XMLStreamReader reader) throws XMLStreamException, IllegalStateException {
+        return loadObject(reader, null);
+    }
+
+    public void saveObject(DataObject sdo, XMLStreamWriter writer) throws XMLStreamException {
+        XMLStreamReader reader = createXMLStreamReader(sdo);
+        new XMLStreamSerializer().serialize(new XMLDocumentStreamReader(reader), writer);
+    }
+
+    public void saveObject(DataObject sdo, XMLStreamWriter writer, Map options) throws XMLStreamException {
+        XMLStreamReader reader = createXMLStreamReader(sdo);
+        options = checkSetOptions(options);
+        new XMLStreamSerializer().serialize(new XMLDocumentStreamReader(reader), writer, options);
+    }
+
+    public XMLStreamReader createXMLStreamReader(DataObject dataObject) {
+        String rootElementURI;
+        String rootElementName;
+
+        Property property = dataObject.getContainmentProperty();
+        if (property != null) {
+            rootElementName = property.getName();
+            rootElementURI = property.getType().getURI();
+        } else {
+            rootElementName = dataObject.getType().getName();
+            rootElementURI = dataObject.getType().getURI();
+        }
+
+        return new DataObjectXMLStreamReader(dataObject, rootElementURI, rootElementName, helperContext.getTypeHelper());
+    }
+
+    protected XMLDocument loadDocument(XMLStreamReader reader, Map options) throws XMLStreamException {
+        try {
+            XMLDocumentImpl document = new XMLDocumentImpl(((HelperContextImpl)helperContext).extendedMetaData, null);
+            options = checkSetOptions(options);
+            document.load(reader, options);
+            return document;
+        } catch (Exception e) {
+            if (e instanceof Resource.IOWrappedException) {
+                Resource.IOWrappedException ioe = (Resource.IOWrappedException)e;
+                if (ioe.getWrappedException() instanceof XMLStreamException) {
+                    throw (XMLStreamException)ioe.getWrappedException();
+                }
+            }
+            throw new RuntimeException(e); // ????
+        }
+    }
+
+    public HelperContext getHelperContext() {
+        return helperContext;
+    }
+}