/*
* 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.rest.panels.request.inspectors.representations;
import java.awt.event.ItemEvent;
import java.awt.event.ItemListener;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import javax.swing.JCheckBox;
import org.apache.xmlbeans.XmlCursor;
import com.eviware.soapui.impl.rest.RestRepresentation;
import com.eviware.soapui.impl.rest.RestRequestInterface;
import com.eviware.soapui.impl.settings.XmlBeansSettingsImpl;
import com.eviware.soapui.impl.support.HttpUtils;
import com.eviware.soapui.impl.wsdl.submit.transports.http.HttpResponse;
import com.eviware.soapui.model.iface.Submit;
import com.eviware.soapui.model.iface.SubmitContext;
import com.eviware.soapui.model.iface.SubmitListener;
import com.eviware.soapui.support.UISupport;
import com.eviware.soapui.support.components.JXToolBar;
import com.eviware.soapui.support.xml.XmlUtils;
public class RestResponseRepresentationsInspector extends AbstractRestRepresentationsInspector implements
SubmitListener {
private JCheckBox enableRecordingCheckBox;
public static final String RECORD_RESPONSE_REPRESENTATIONS = "RecordResponseRepresentations";
private RestRequestInterface request;
protected RestResponseRepresentationsInspector(RestRequestInterface request) {
super(request.getRestMethod(), "Representations", "Response Representations", new RestRepresentation.Type[]{
RestRepresentation.Type.RESPONSE, RestRepresentation.Type.FAULT});
request.addSubmitListener(this);
this.request = request;
}
@Override
protected void addToToolbar(JXToolBar toolbar) {
enableRecordingCheckBox = new JCheckBox("Auto-Create");
enableRecordingCheckBox.setToolTipText("Automatically create Representations from received Responses");
enableRecordingCheckBox.setOpaque(false);
UISupport.setFixedSize(enableRecordingCheckBox, 150, 20);
toolbar.addFixed(enableRecordingCheckBox);
XmlBeansSettingsImpl settings = getMethod().getSettings();
if (settings.isSet(RECORD_RESPONSE_REPRESENTATIONS)) {
enableRecordingCheckBox.setSelected(settings.getBoolean(RECORD_RESPONSE_REPRESENTATIONS));
} else {
enableRecordingCheckBox.setSelected(getMethod().getResource() == null
|| getMethod().getResource().getService().isGenerated());
}
enableRecordingCheckBox.addItemListener(new ItemListener() {
public void itemStateChanged(ItemEvent e) {
getMethod().getSettings()
.setBoolean(RECORD_RESPONSE_REPRESENTATIONS, enableRecordingCheckBox.isSelected());
}
});
}
@Override
public boolean beforeSubmit(Submit submit, SubmitContext context) {
return true;
}
public void afterSubmit(Submit submit, SubmitContext context) {
HttpResponse response = (HttpResponse) submit.getResponse();
if (response != null && enableRecordingCheckBox.isSelected()) {
if (HttpUtils.isErrorStatus(response.getStatusCode())) {
extractRepresentation(response, RestRepresentation.Type.FAULT);
} else {
extractRepresentation(response, RestRepresentation.Type.RESPONSE);
}
}
}
@SuppressWarnings("unchecked")
protected void extractRepresentation(HttpResponse response, RestRepresentation.Type type) {
RestRepresentation[] representations = getMethod().getRepresentations(type, null);
int c = 0;
for (; c < representations.length; c++) {
if (representations[c].getMediaType() != null
&& representations[c].getMediaType().equals(response.getContentType())) {
@SuppressWarnings("rawtypes")
List status = representations[c].getStatus();
if (status == null || !status.contains(response.getStatusCode())) {
status = status == null ? new ArrayList<Integer>() : new ArrayList<Integer>(status);
status.add(response.getStatusCode());
representations[c].setStatus(status);
}
break;
}
}
if (c == representations.length) {
RestRepresentation representation = getMethod().addNewRepresentation(type);
representation.setMediaType(response.getContentType());
representation.setStatus(Arrays.asList(response.getStatusCode()));
String xmlContent = response.getContentAsXml();
if (xmlContent != null && !xmlContent.equals("<xml/>")) {
// if(response.getContentType().equals("text/xml") ||
// response.getContentType().equals("application/xml")) {
try {
// XmlCursor cursor = XmlObject.Factory.parse( xmlContent
// ).newCursor();
XmlCursor cursor = XmlUtils.createXmlObject(xmlContent).newCursor();
cursor.toFirstChild();
representation.setElement(cursor.getName());
} catch (Exception e) {
}
}
}
}
@Override
public void release() {
super.release();
request.removeSubmitListener(this);
}
}