/*
 * 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.support.http.HttpRequestInterface;
import com.eviware.soapui.impl.wsdl.submit.transports.http.support.attachments.WsdlRequestMimeMessageRequestEntity.DummyOutputStream;
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.IOException;
import java.io.InputStream;
import java.io.OutputStream;

/**
 * MimeMessage request class
 *
 * @author ole.matzura
 */

public class HttpRequestMimeMessageRequestEntity extends AbstractHttpEntity {
    private final MimeMessage message;
    private final HttpRequestInterface<?> restRequest;

    public HttpRequestMimeMessageRequestEntity(MimeMessage message, HttpRequestInterface<?> restRequest) {
        this.message = message;
        this.restRequest = restRequest;
    }

    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 {
            String header = message.getHeader("Content-Type")[0];
            int ix = header.indexOf("boundary");

            return new BasicHeader("Content-Type", restRequest.getMediaType() + "; " + header.substring(ix));
        } catch (MessagingException e) {
            SoapUI.logError(e);
        }

        return new BasicHeader("Content-Type", restRequest.getMediaType());
    }

    public boolean isRepeatable() {
        return true;
    }

    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, IllegalStateException {
        try {
            return message.getInputStream();
        } catch (MessagingException e) {
            throw new IOException(e);
        }
    }

    @Override
    public boolean isStreaming() {
        return false;
    }
}