org.eclipse.skalli.testutil/src/main/java/org/eclipse/skalli/testutil/ConverterWrapper.java

/*******************************************************************************
 * Copyright (c) 2010-2014 SAP AG and others.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *     SAP AG - initial API and implementation
 *******************************************************************************/
package org.eclipse.skalli.testutil;

import java.io.IOException;

import org.apache.commons.lang.StringUtils;
import org.eclipse.skalli.commons.XMLUtils;
import org.eclipse.skalli.model.Derived;
import org.eclipse.skalli.model.ExtensionEntityBase;
import org.eclipse.skalli.services.extension.rest.RestConverter;
import org.eclipse.skalli.services.extension.rest.RestException;
import org.eclipse.skalli.services.rest.RestReader;
import org.eclipse.skalli.services.rest.RestWriter;

import com.thoughtworks.xstream.converters.MarshallingContext;
import com.thoughtworks.xstream.converters.UnmarshallingContext;
import com.thoughtworks.xstream.io.HierarchicalStreamReader;
import com.thoughtworks.xstream.io.HierarchicalStreamWriter;

@SuppressWarnings("rawtypes")
public class ConverterWrapper implements RestConverter {

    private static final String XSI_INSTANCE_NS = "http://www.w3.org/2001/XMLSchema-instance"; //$NON-NLS-1$
    private static final String URL_SCHEMAS = "/schemas/"; //$NON-NLS-1$

    private RestConverter<?> converter;
    private String nodeName;
    private boolean isInherited;
    private boolean omitInheritedAttribute;
    private String host;

    public ConverterWrapper(RestConverter<?> converter) {
        this.converter = converter;
    }

    public ConverterWrapper(String host, RestConverter<?> converter, String nodeName) {
        this.host = host;
        this.converter = converter;
        this.nodeName = nodeName;
        this.omitInheritedAttribute = true;
    }

    public ConverterWrapper(String host, RestConverter<?> converter, String nodeName, boolean isInherited) {
        this.converter = converter;
        this.nodeName = nodeName;
        this.isInherited = isInherited;
        this.omitInheritedAttribute = false;
    }

    @Override
    public void marshal(Object ext, RestWriter writer) throws IOException {
        writer.object(nodeName);
        namespaces(writer);
        commonAttributes((ExtensionEntityBase)ext, writer);
        converter.marshal(ext, writer);
        writer.end();
    }

    @Override
    public Object unmarshal(RestReader reader) throws RestException, IOException {
        return converter.unmarshal(reader);
    }

    @Override
    public void marshal(Object source, HierarchicalStreamWriter writer, MarshallingContext context) {
        writer.startNode(nodeName);
        marshalNSAttributes(writer);
        marshalCommonAttributes((ExtensionEntityBase) source, writer);
        converter.marshal(source, writer, context);
        writer.endNode();
    }

    private void marshalNSAttributes(HierarchicalStreamWriter writer) {
        writer.addAttribute(XMLUtils.XMLNS, getNamespace());
        writer.addAttribute(XMLUtils.XMLNS_XSI, XSI_INSTANCE_NS);
        writer.addAttribute(XMLUtils.XSI_SCHEMA_LOCATION, getSchemaLocation());
    }

    @SuppressWarnings("nls")
    private void marshalCommonAttributes(ExtensionEntityBase ext, HierarchicalStreamWriter writer) {
        if (!omitInheritedAttribute) {
            writer.addAttribute("inherited", Boolean.toString(isInherited)); //$NON-NLS-1$
        }
        writer.addAttribute("derived", Boolean.toString(ext.getClass().isAnnotationPresent(Derived.class)));
        writer.addAttribute("apiVersion", getApiVersion());
        String lastModified = ext.getLastModified();
        if (StringUtils.isNotBlank(lastModified)) {
            writer.addAttribute("lastModified", lastModified);
        }
        String modifiedBy = ext.getLastModifiedBy();
        if (StringUtils.isNotBlank(lastModified)) {
            writer.addAttribute("modifiedBy", modifiedBy);
        }
    }

    protected void namespaces(RestWriter writer) throws IOException {
        writer.namespace(XMLUtils.XMLNS, getNamespace());
        writer.namespace(XMLUtils.XMLNS_XSI, XMLUtils.XSI_INSTANCE_NS);
        writer.namespace(XMLUtils.XSI_SCHEMA_LOCATION, getSchemaLocation());
    }

    @SuppressWarnings("nls")
    protected void commonAttributes(ExtensionEntityBase ext, RestWriter writer) throws IOException {
        if (!omitInheritedAttribute) {
            writer.attribute("inherited", Boolean.toString(isInherited)); //$NON-NLS-1$
        }
        writer.attribute("derived", Boolean.toString(ext.getClass().isAnnotationPresent(Derived.class)));
        writer.attribute("apiVersion", getApiVersion());
        long lastModifiedMillis = ext.getLastModifiedMillis();
        if (lastModifiedMillis > 0) {
            writer.attribute("lastModifiedMillis", lastModifiedMillis);
        }
        String lastModified = ext.getLastModified();
        if (StringUtils.isNotBlank(lastModified)) {
            writer.attribute("lastModified", lastModified);
        }
        String modifiedBy = ext.getLastModifiedBy();
        if (StringUtils.isNotBlank(modifiedBy)) {
            writer.attribute("modifiedBy", modifiedBy);
        }
    }

    @Override
    public Object unmarshal(HierarchicalStreamReader arg0, UnmarshallingContext arg1) {
        return converter.unmarshal(arg0, arg1);
    }

    @Override
    public boolean canConvert(Class arg0) {
        return converter.canConvert(arg0);
    }

    @Override
    public String getAlias() {
        return converter.getAlias();
    }

    @Override
    public Class<?> getConversionClass() {
        return converter.getConversionClass();
    }

    @Override
    public String getApiVersion() {
        return converter.getApiVersion();
    }

    @Override
    public String getNamespace() {
        return converter.getNamespace();
    }

    @Override
    public String getXsdFileName() {
        return converter.getXsdFileName();
    }

    private String getSchemaLocation() {
        return converter.getNamespace() + " " + host + URL_SCHEMAS + converter.getXsdFileName();
    }

}