/*******************************************************************************
* Copyright 2013 SAP AG
*
* Licensed 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 com.sap.core.odata.core.ep.producer;
import java.io.Writer;
import java.util.List;
import javax.xml.stream.FactoryConfigurationError;
import javax.xml.stream.XMLOutputFactory;
import javax.xml.stream.XMLStreamException;
import javax.xml.stream.XMLStreamWriter;
import com.sap.core.odata.api.edm.Edm;
import com.sap.core.odata.api.edm.EdmEntitySetInfo;
import com.sap.core.odata.api.edm.EdmServiceMetadata;
import com.sap.core.odata.api.ep.EntityProviderException;
import com.sap.core.odata.api.exception.ODataException;
import com.sap.core.odata.core.commons.ContentType;
import com.sap.core.odata.core.ep.util.FormatXml;
/**
* Writes the OData service document in XML.
* @author SAP AG
*/
public class AtomServiceDocumentProducer {
private static final String DEFAULT_CHARSET = ContentType.CHARSET_UTF_8;
private static final String XML_VERSION = "1.0";
private final Edm edm;
private final String serviceRoot;
public AtomServiceDocumentProducer(final Edm edm, final String serviceRoot) {
this.edm = edm;
this.serviceRoot = serviceRoot;
}
public void writeServiceDocument(final Writer writer) throws EntityProviderException {
EdmServiceMetadata serviceMetadata = edm.getServiceMetadata();
try {
XMLStreamWriter xmlStreamWriter = XMLOutputFactory.newInstance().createXMLStreamWriter(writer);
xmlStreamWriter.writeStartDocument(DEFAULT_CHARSET, XML_VERSION);
xmlStreamWriter.setPrefix(Edm.PREFIX_XML, Edm.NAMESPACE_XML_1998);
xmlStreamWriter.setPrefix(Edm.PREFIX_ATOM, Edm.NAMESPACE_ATOM_2005);
xmlStreamWriter.setDefaultNamespace(Edm.NAMESPACE_APP_2007);
xmlStreamWriter.writeStartElement(FormatXml.APP_SERVICE);
xmlStreamWriter.writeAttribute(Edm.PREFIX_XML, Edm.NAMESPACE_XML_1998, FormatXml.XML_BASE, serviceRoot);
xmlStreamWriter.writeDefaultNamespace(Edm.NAMESPACE_APP_2007);
xmlStreamWriter.writeNamespace(Edm.PREFIX_ATOM, Edm.NAMESPACE_ATOM_2005);
xmlStreamWriter.writeStartElement(FormatXml.APP_WORKSPACE);
xmlStreamWriter.writeStartElement(Edm.NAMESPACE_ATOM_2005, FormatXml.ATOM_TITLE);
xmlStreamWriter.writeCharacters(FormatXml.ATOM_TITLE_DEFAULT);
xmlStreamWriter.writeEndElement();
List<EdmEntitySetInfo> entitySetInfos = serviceMetadata.getEntitySetInfos();
for (EdmEntitySetInfo info : entitySetInfos) {
xmlStreamWriter.writeStartElement(FormatXml.APP_COLLECTION);
xmlStreamWriter.writeAttribute(FormatXml.ATOM_HREF, info.getEntitySetUri().toASCIIString());
xmlStreamWriter.writeStartElement(Edm.NAMESPACE_ATOM_2005, FormatXml.ATOM_TITLE);
xmlStreamWriter.writeCharacters(info.getEntitySetName());
xmlStreamWriter.writeEndElement();
xmlStreamWriter.writeEndElement();
}
// Collection<Schema> schemas = edmProvider.getSchemas();
// if (schemas != null) {
// for (Schema schema : schemas) {
// Collection<EntityContainer> entityContainers = schema.getEntityContainers();
// if (entityContainers != null) {
// for (EntityContainer entityContainer : entityContainers) {
// Collection<EntitySet> entitySets = entityContainer.getEntitySets();
// for (EntitySet entitySet : entitySets) {
// xmlStreamWriter.writeStartElement(FormatXml.APP_COLLECTION);
// if (entityContainer.isDefaultEntityContainer()) {
// xmlStreamWriter.writeAttribute(FormatXml.ATOM_HREF, entitySet.getName());
// } else {
// xmlStreamWriter.writeAttribute(FormatXml.ATOM_HREF, entityContainer.getName() + Edm.DELIMITER + entitySet.getName());
// }
// xmlStreamWriter.writeStartElement(Edm.NAMESPACE_ATOM_2005, FormatXml.ATOM_TITLE);
// xmlStreamWriter.writeCharacters(entitySet.getName());
// xmlStreamWriter.writeEndElement();
// xmlStreamWriter.writeEndElement();
// }
// }
// }
// }
// }
xmlStreamWriter.writeEndElement();
xmlStreamWriter.writeEndElement();
xmlStreamWriter.writeEndDocument();
xmlStreamWriter.flush();
} catch (FactoryConfigurationError e) {
throw new EntityProviderException(EntityProviderException.COMMON, e);
} catch (XMLStreamException e) {
throw new EntityProviderException(EntityProviderException.COMMON, e);
} catch (ODataException e) {
throw new EntityProviderException(EntityProviderException.COMMON, e);
}
}
}