Package com.eviware.soapui.impl.support.definition.export

Source Code of com.eviware.soapui.impl.support.definition.export.WadlDefinitionExporter

/*
* Copyright 2004-2014 SmartBear Software
*
* Licensed under the EUPL, Version 1.1 or - as soon as they will be approved by the European Commission - subsequent
* versions of the EUPL (the "Licence");
* You may not use this work except in compliance with the Licence.
* You may obtain a copy of the Licence at:
*
* http://ec.europa.eu/idabc/eupl
*
* Unless required by applicable law or agreed to in writing, software distributed under the Licence is
* distributed on an "AS IS" basis, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
* express or implied. See the Licence for the specific language governing permissions and limitations
* under the Licence.
*/

package com.eviware.soapui.impl.support.definition.export;

import java.util.List;

import net.java.dev.wadl.x2009.x02.ApplicationDocument;
import net.java.dev.wadl.x2009.x02.MethodDocument.Method;
import net.java.dev.wadl.x2009.x02.RepresentationDocument.Representation;
import net.java.dev.wadl.x2009.x02.ResourceDocument.Resource;
import net.java.dev.wadl.x2009.x02.ResourcesDocument.Resources;
import net.java.dev.wadl.x2009.x02.ResponseDocument.Response;

import org.apache.xmlbeans.XmlObject;
import org.w3c.dom.Element;

import com.eviware.soapui.impl.rest.RestService;
import com.eviware.soapui.impl.support.definition.InterfaceDefinition;
import com.eviware.soapui.impl.support.definition.InterfaceDefinitionPart;
import com.eviware.soapui.impl.wsdl.support.Constants;

public class WadlDefinitionExporter extends AbstractDefinitionExporter<RestService> {
    public WadlDefinitionExporter(InterfaceDefinition<RestService> definition) {
        super(definition);
    }

    public WadlDefinitionExporter(RestService restService) throws Exception {
        this(restService.getDefinitionContext().getInterfaceDefinition());
    }

    public String export(String folderName) throws Exception {
        setDefinition(getDefinition().getInterface().getWadlContext().regenerateWadl());

        return super.export(folderName);
    }

    protected String[] getLocationXPathsToReplace() {
        return new String[]{
                "declare namespace s='" + getDefinition().getInterface().getWadlVersion()
                        + "' .//s:grammars/s:include/@href",
                "declare namespace s='http://www.w3.org/2001/XMLSchema' .//s:import/@schemaLocation",
                "declare namespace s='http://www.w3.org/2001/XMLSchema' .//s:include/@schemaLocation"};
    }

    @Override
    protected void postProcessing(XmlObject obj, InterfaceDefinitionPart part) {
        if (part.getType().equals(Constants.WADL11_NS)) {
            ApplicationDocument document = (ApplicationDocument) obj;
            for (Resources resources : document.getApplication().getResourcesList()) {
                for (Resource resource : resources.getResourceList()) {
                    for (Method method : resource.getMethodList()) {
                        if (method.getRequest() != null) {
                            fixRepresentations(method.getRequest().getRepresentationList());
                        }

                        for (Response response : method.getResponseList()) {
                            fixRepresentations(response.getRepresentationList());
                        }
                    }
                }
            }
        }
    }

    private void fixRepresentations(List<Representation> representationList) {
        for (Representation representation : representationList) {
            if (!("text/xml".equals(representation.getMediaType()) || "application/xml".equals(representation
                    .getMediaType())) && representation.isSetElement()) {
                String prefix = representation.xgetElement().getDomNode().getNodeValue().split(":")[0];
                representation.unsetElement();
                ((Element) representation.getDomNode()).removeAttribute("xmlns:" + prefix);
            }
        }
    }

}
TOP

Related Classes of com.eviware.soapui.impl.support.definition.export.WadlDefinitionExporter

TOP
Copyright © 2018 www.massapi.com. All rights reserved.
All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.