/*

 * Copyright (c) 1997, 2014, Oracle and/or its affiliates. All rights reserved.

 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.

 *

 * This code is free software; you can redistribute it and/or modify it

 * under the terms of the GNU General Public License version 2 only, as

 * published by the Free Software Foundation.  Oracle designates this

 * particular file as subject to the "Classpath" exception as provided

 * by Oracle in the LICENSE file that accompanied this code.

 *

 * This code is distributed in the hope that it will be useful, but WITHOUT

 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or

 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License

 * version 2 for more details (a copy is included in the LICENSE file that

 * accompanied this code).

 *

 * You should have received a copy of the GNU General Public License version

 * 2 along with this work; if not, write to the Free Software Foundation,

 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.

 *

 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA

 * or visit www.oracle.com if you need additional information or have any

 * questions.

 */

package com.sun.xml.internal.messaging.saaj.soap;

import com.sun.xml.internal.messaging.saaj.SOAPExceptionImpl;

import com.sun.xml.internal.messaging.saaj.packaging.mime.util.ASCIIUtility;

import com.sun.xml.internal.messaging.saaj.packaging.mime.Header;

import com.sun.xml.internal.messaging.saaj.packaging.mime.internet.MimePartDataSource;

import com.sun.xml.internal.messaging.saaj.packaging.mime.internet.InternetHeaders;

import com.sun.xml.internal.messaging.saaj.packaging.mime.internet.MimeBodyPart;

import com.sun.xml.internal.messaging.saaj.packaging.mime.internet.MimeUtility;

import com.sun.xml.internal.messaging.saaj.util.ByteOutputStream;

import com.sun.xml.internal.messaging.saaj.util.LogDomainConstants;

import java.io.IOException;

import java.io.InputStream;

import java.io.ByteArrayOutputStream;

import java.io.ByteArrayInputStream;

import java.io.OutputStream;

import java.util.Iterator;

import java.util.List;

import java.util.logging.Level;

import java.util.logging.Logger;

import javax.activation.*;

import javax.xml.soap.*;

import com.sun.xml.internal.org.jvnet.mimepull.MIMEPart;

/**

 * Implementation of attachments.

 *

 * @author Anil Vijendran (akv@eng.sun.com)

 */

public class AttachmentPartImpl extends AttachmentPart {

    protected static final Logger log =

        Logger.getLogger(LogDomainConstants.SOAP_DOMAIN,

                         "com.sun.xml.internal.messaging.saaj.soap.LocalStrings");

    private final MimeHeaders headers;

    private MimeBodyPart rawContent = null;

    private DataHandler dataHandler = null;

    //alternate impl that uses a MIMEPart

    private MIMEPart mimePart = null;

    public AttachmentPartImpl() {

        headers = new MimeHeaders();

    }

    public AttachmentPartImpl(MIMEPart part) {

        headers = new MimeHeaders();

        mimePart = part;

        List<? extends com.sun.xml.internal.org.jvnet.mimepull.Header> hdrs = part.getAllHeaders();

        for (com.sun.xml.internal.org.jvnet.mimepull.Header hd : hdrs) {

            headers.addHeader(hd.getName(), hd.getValue());

        }

    }

    public int getSize() throws SOAPException {

        if (mimePart != null) {

            try {

                return mimePart.read().available();

            } catch (IOException e) {

                return -1;

            }

        }

        if ((rawContent == null) && (dataHandler == null))

            return 0;

        if (rawContent != null) {

            try {

                return rawContent.getSize();

            } catch (Exception ex) {

                log.log(

                    Level.SEVERE,

                    "SAAJ0573.soap.attachment.getrawbytes.ioexception",

                    new String[] { ex.getLocalizedMessage()});

                throw new SOAPExceptionImpl("Raw InputStream Error: " + ex);

            }

        } else {

            ByteOutputStream bout = new ByteOutputStream();

            try {

                dataHandler.writeTo(bout);

            } catch (IOException ex) {

                log.log(

                    Level.SEVERE,

                    "SAAJ0501.soap.data.handler.err",

                    new String[] { ex.getLocalizedMessage()});

                throw new SOAPExceptionImpl("Data handler error: " + ex);

            }

            return bout.size();

        }

    }

    public void clearContent() {

        if (mimePart != null) {

            mimePart.close();

            mimePart = null;

        }

        dataHandler = null;

        rawContent = null;

    }

    public Object getContent() throws SOAPException {

        try {

            if (mimePart != null) {

                //return an inputstream

                return mimePart.read();

            }

            if (dataHandler != null) {

                return getDataHandler().getContent();

            } else if (rawContent != null) {

                return rawContent.getContent();

            } else {

                log.severe("SAAJ0572.soap.no.content.for.attachment");

                throw new SOAPExceptionImpl("No data handler/content associated with this attachment");

            }

        } catch (Exception ex) {

            log.log(Level.SEVERE, "SAAJ0575.soap.attachment.getcontent.exception", ex);

            throw new SOAPExceptionImpl(ex.getLocalizedMessage());

        }

    }

    public void setContent(Object object, String contentType)

        throws IllegalArgumentException {

        if (mimePart != null) {

            mimePart.close();

            mimePart = null;

        }

        DataHandler dh = new DataHandler(object, contentType);

        setDataHandler(dh);

    }

    public DataHandler getDataHandler() throws SOAPException {

        if (mimePart != null) {

            //return an inputstream

            return new DataHandler(new DataSource() {

                public InputStream getInputStream() throws IOException {

                    return mimePart.read();

                }

                public OutputStream getOutputStream() throws IOException {

                    throw new UnsupportedOperationException("getOutputStream cannot be supported : You have enabled LazyAttachments Option");

                }

                public String getContentType() {

                    return mimePart.getContentType();

                }

                public String getName() {

                    return "MIMEPart Wrapper DataSource";

                }

            });

        }

        if (dataHandler == null) {

            if (rawContent != null) {

                return new DataHandler(new MimePartDataSource(rawContent));

            }

            log.severe("SAAJ0502.soap.no.handler.for.attachment");

            throw new SOAPExceptionImpl("No data handler associated with this attachment");

        }

        return dataHandler;

    }

    public void setDataHandler(DataHandler dataHandler)

        throws IllegalArgumentException {

        if (mimePart != null) {

            mimePart.close();

            mimePart = null;

        }

        if (dataHandler == null) {

            log.severe("SAAJ0503.soap.no.null.to.dataHandler");

            throw new IllegalArgumentException("Null dataHandler argument to setDataHandler");

        }

        this.dataHandler = dataHandler;

        rawContent = null;

        if (log.isLoggable(Level.FINE))

            log.log(Level.FINE, "SAAJ0580.soap.set.Content-Type",

                    new String[] { dataHandler.getContentType() });

        setMimeHeader("Content-Type", dataHandler.getContentType());

    }

    public void removeAllMimeHeaders() {

        headers.removeAllHeaders();

    }

    public void removeMimeHeader(String header) {

        headers.removeHeader(header);

    }

    public String[] getMimeHeader(String name) {

        return headers.getHeader(name);

    }

    public void setMimeHeader(String name, String value) {

        headers.setHeader(name, value);

    }

    public void addMimeHeader(String name, String value) {

        headers.addHeader(name, value);

    }

    public Iterator getAllMimeHeaders() {

        return headers.getAllHeaders();

    }

    public Iterator getMatchingMimeHeaders(String[] names) {

        return headers.getMatchingHeaders(names);

    }

    public Iterator getNonMatchingMimeHeaders(String[] names) {

        return headers.getNonMatchingHeaders(names);

    }

    boolean hasAllHeaders(MimeHeaders hdrs) {

        if (hdrs != null) {

            Iterator i = hdrs.getAllHeaders();

            while (i.hasNext()) {

                MimeHeader hdr = (MimeHeader) i.next();

                String[] values = headers.getHeader(hdr.getName());

                boolean found = false;

                if (values != null) {

                    for (int j = 0; j < values.length; j++)

                        if (hdr.getValue().equalsIgnoreCase(values[j])) {

                            found = true;

                            break;

                        }

                }

                if (!found) {

                    return false;

                }

            }

        }

        return true;

    }

    MimeBodyPart getMimePart() throws SOAPException {

        try {

            if (this.mimePart != null) {

                return new MimeBodyPart(mimePart);

            }

            if (rawContent != null) {

                copyMimeHeaders(headers, rawContent);

                return rawContent;

            }

            MimeBodyPart envelope = new MimeBodyPart();

            envelope.setDataHandler(dataHandler);

            copyMimeHeaders(headers, envelope);

            return envelope;

        } catch (Exception ex) {

            log.severe("SAAJ0504.soap.cannot.externalize.attachment");

            throw new SOAPExceptionImpl("Unable to externalize attachment", ex);

        }

    }

    public static void copyMimeHeaders(MimeHeaders headers, MimeBodyPart mbp)

        throws SOAPException {

        Iterator i = headers.getAllHeaders();

        while (i.hasNext())

            try {

                MimeHeader mh = (MimeHeader) i.next();

                mbp.setHeader(mh.getName(), mh.getValue());

            } catch (Exception ex) {

                log.severe("SAAJ0505.soap.cannot.copy.mime.hdr");

                throw new SOAPExceptionImpl("Unable to copy MIME header", ex);

            }

    }

    public static void copyMimeHeaders(MimeBodyPart mbp, AttachmentPartImpl ap)

        throws SOAPException {

        try {

            List hdr = mbp.getAllHeaders();

            int sz = hdr.size();

            for( int i=0; i<sz; i++ ) {

                Header h = (Header)hdr.get(i);

                if(h.getName().equalsIgnoreCase("Content-Type"))

                    continue;   // skip

                ap.addMimeHeader(h.getName(), h.getValue());

            }

        } catch (Exception ex) {

            log.severe("SAAJ0506.soap.cannot.copy.mime.hdrs.into.attachment");

            throw new SOAPExceptionImpl(

                "Unable to copy MIME headers into attachment",

                ex);

        }

    }

    public  void setBase64Content(InputStream content, String contentType)

        throws SOAPException {

        if (mimePart != null) {

            mimePart.close();

            mimePart = null;

        }

        dataHandler = null;

        InputStream decoded = null;

        ByteOutputStream bos = null;

        try {

            decoded = MimeUtility.decode(content, "base64");

            InternetHeaders hdrs = new InternetHeaders();

            hdrs.setHeader("Content-Type", contentType);

            //TODO: reading the entire attachment here is ineffcient. Somehow the MimeBodyPart

            // Ctor with inputStream causes problems based on the InputStream

            // has markSupported()==true

            bos = new ByteOutputStream();

            bos.write(decoded);

            rawContent = new MimeBodyPart(hdrs, bos.getBytes(), bos.getCount());

            setMimeHeader("Content-Type", contentType);

        } catch (Exception e) {

            log.log(Level.SEVERE, "SAAJ0578.soap.attachment.setbase64content.exception", e);

            throw new SOAPExceptionImpl(e.getLocalizedMessage());

        } finally {

            if (bos != null)

                bos.close();

            try {

                if (decoded != null)

                    decoded.close();

            } catch (IOException ex) {

                throw new SOAPException(ex);

            }

        }

    }

    public  InputStream getBase64Content() throws SOAPException {

        InputStream stream;

        if (mimePart != null) {

            stream = mimePart.read();

        } else if (rawContent != null) {

            try {

                 stream = rawContent.getInputStream();

            } catch (Exception e) {

                log.log(Level.SEVERE,"SAAJ0579.soap.attachment.getbase64content.exception", e);

                throw new SOAPExceptionImpl(e.getLocalizedMessage());

            }

        } else if (dataHandler != null) {

            try {

                stream = dataHandler.getInputStream();

            } catch (IOException e) {

                log.severe("SAAJ0574.soap.attachment.datahandler.ioexception");

                throw new SOAPExceptionImpl("DataHandler error" + e);

            }

        } else {

            log.severe("SAAJ0572.soap.no.content.for.attachment");

            throw new SOAPExceptionImpl("No data handler/content associated with this attachment");

        }

        //TODO: Write a BASE64EncoderInputStream instead,

        // this code below is inefficient

        // where we are trying to read the whole attachment first

        int len;

        int size = 1024;

        byte [] buf;

        if (stream != null) {

            try {

                ByteArrayOutputStream bos = new ByteArrayOutputStream(size);

                //TODO: try and optimize this on the same lines as

                // ByteOutputStream : to eliminate the temp buffer here

                OutputStream ret = MimeUtility.encode(bos, "base64");

                buf = new byte[size];

                while ((len = stream.read(buf, 0, size)) != -1) {

                    ret.write(buf, 0, len);

                }

                ret.flush();

                buf = bos.toByteArray();

                return new ByteArrayInputStream(buf);

            } catch (Exception e) {

                // throw new SOAPException

                log.log(Level.SEVERE,"SAAJ0579.soap.attachment.getbase64content.exception", e);

                throw new SOAPExceptionImpl(e.getLocalizedMessage());

            } finally {

                try {

                    stream.close();

                } catch (IOException ex) {

                  //close the stream

                }

            }

        } else {

          //throw  new SOAPException

          log.log(Level.SEVERE,"SAAJ0572.soap.no.content.for.attachment");

          throw new SOAPExceptionImpl("No data handler/content associated with this attachment");

        }

    }

    public void setRawContent(InputStream content, String contentType)

        throws SOAPException {

        if (mimePart != null) {

            mimePart.close();

            mimePart = null;

        }

        dataHandler = null;

        ByteOutputStream bos = null;

        try {

            InternetHeaders hdrs = new InternetHeaders();

            hdrs.setHeader("Content-Type", contentType);

            //TODO: reading the entire attachment here is ineffcient. Somehow the MimeBodyPart

            // Ctor with inputStream causes problems based on whether the InputStream has

            // markSupported()==true or false

            bos = new ByteOutputStream();

            bos.write(content);

            rawContent = new MimeBodyPart(hdrs, bos.getBytes(), bos.getCount());

            setMimeHeader("Content-Type", contentType);

        } catch (Exception e) {

            log.log(Level.SEVERE, "SAAJ0576.soap.attachment.setrawcontent.exception", e);

            throw new SOAPExceptionImpl(e.getLocalizedMessage());

        } finally {

            if (bos != null)

                bos.close();

            try {

                content.close();

            } catch (IOException ex) {

                throw new SOAPException(ex);

            }

        }

    }

   /*

    public void setRawContentBytes(byte[] content, String contentType)

        throws SOAPException {

        if (content == null) {

            throw new SOAPExceptionImpl("Null content passed to setRawContentBytes");

        }

        dataHandler = null;

        try {

            InternetHeaders hdrs = new InternetHeaders();

            hdrs.setHeader("Content-Type", contentType);

            rawContent = new MimeBodyPart(hdrs, content, content.length);

            setMimeHeader("Content-Type", contentType);

        } catch (Exception e) {

            log.log(Level.SEVERE, "SAAJ0576.soap.attachment.setrawcontent.exception", e);

            throw new SOAPExceptionImpl(e.getLocalizedMessage());

        }

    } */

    public void setRawContentBytes(

        byte[] content, int off, int len, String contentType)

        throws SOAPException {

        if (mimePart != null) {

            mimePart.close();

            mimePart = null;

        }

        if (content == null) {

            throw new SOAPExceptionImpl("Null content passed to setRawContentBytes");

        }

        dataHandler = null;

        try {

            InternetHeaders hdrs = new InternetHeaders();

            hdrs.setHeader("Content-Type", contentType);

            rawContent = new MimeBodyPart(hdrs, content, off, len);

            setMimeHeader("Content-Type", contentType);

        } catch (Exception e) {

            log.log(Level.SEVERE,

                "SAAJ0576.soap.attachment.setrawcontent.exception", e);

            throw new SOAPExceptionImpl(e.getLocalizedMessage());

        }

    }

    public  InputStream getRawContent() throws SOAPException {

        if (mimePart != null) {

            return mimePart.read();

        }

        if (rawContent != null) {

            try {

                return rawContent.getInputStream();

            } catch (Exception e) {

                log.log(Level.SEVERE,"SAAJ0577.soap.attachment.getrawcontent.exception", e);

                throw new SOAPExceptionImpl(e.getLocalizedMessage());

            }

        } else if (dataHandler != null) {

            try {

                return dataHandler.getInputStream();

            } catch (IOException e) {

                log.severe("SAAJ0574.soap.attachment.datahandler.ioexception");

                throw new SOAPExceptionImpl("DataHandler error" + e);

            }

        } else {

            log.severe("SAAJ0572.soap.no.content.for.attachment");

            throw new SOAPExceptionImpl("No data handler/content associated with this attachment");

        }

    }

    public  byte[] getRawContentBytes() throws SOAPException {

        InputStream ret;