/*
* SoapUI, Copyright (C) 2004-2016 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.wsdl.submit.transports.http.support.attachments;
import com.eviware.soapui.SoapUI;
import com.eviware.soapui.impl.wsdl.WsdlRequest;
import com.eviware.soapui.impl.wsdl.support.soap.SoapVersion;
import org.apache.http.Header;
import org.apache.http.entity.AbstractHttpEntity;
import org.apache.http.message.BasicHeader;
import javax.mail.MessagingException;
import javax.mail.internet.MimeMessage;
import javax.mail.internet.MimeMultipart;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
/**
* MimeMessage request class
*
* @author ole.matzura
*/
public class WsdlRequestMimeMessageRequestEntity extends AbstractHttpEntity {
private final MimeMessage message;
private final boolean isXOP;
private final WsdlRequest wsdlRequest;
public WsdlRequestMimeMessageRequestEntity(MimeMessage message, boolean isXOP, WsdlRequest wsdlRequest) {
this.message = message;
this.isXOP = isXOP;
this.wsdlRequest = wsdlRequest;
}
public long getContentLength() {
try {
DummyOutputStream out = new DummyOutputStream();
writeTo(out);
return out.getSize();
} catch (Exception e) {
SoapUI.logError(e);
return -1;
}
}
public Header getContentType() {
try {
SoapVersion soapVersion = wsdlRequest.getOperation().getInterface().getSoapVersion();
if (isXOP) {
String header = message.getHeader("Content-Type")[0];
return new BasicHeader("Content-Type", AttachmentUtils.buildMTOMContentType(header,
wsdlRequest.getAction(), soapVersion));
} else {
String header = message.getHeader("Content-Type")[0];
int ix = header.indexOf("boundary");
return new BasicHeader("Content-Type", "multipart/related; type=\"" + soapVersion.getContentType()
+ "\"; " + "start=\"" + AttachmentUtils.ROOTPART_SOAPUI_ORG + "\"; " + header.substring(ix));
}
} catch (MessagingException e) {
SoapUI.logError(e);
}
return null;
}
@Override
public boolean isRepeatable() {
return true;
}
@Override
public void writeTo(OutputStream arg0) throws IOException {
try {
arg0.write("\r\n".getBytes());
((MimeMultipart) message.getContent()).writeTo(arg0);
} catch (Exception e) {
SoapUI.logError(e);
}
}
@Override
public InputStream getContent() throws IOException {
try {
return message.getInputStream();
} catch (MessagingException e) {
throw new IOException(e);
}
}
@Override
public boolean isStreaming() {
return false;
}
public static class DummyOutputStream extends OutputStream {
private int intLength;
private long size = 0;
public DummyOutputStream() {
ByteArrayOutputStream tempOut = new ByteArrayOutputStream();
tempOut.write(1);
intLength = tempOut.toByteArray().length;
}
@Override
public void write(int b) throws IOException {
size += intLength;
}
public long getSize() {
return size;
}
}
}