001    /**
002     *
003     * Copyright 2005 The Apache Software Foundation
004     *
005     *  Licensed under the Apache License, Version 2.0 (the "License");
006     *  you may not use this file except in compliance with the License.
007     *  You may obtain a copy of the License at
008     *
009     *     http://www.apache.org/licenses/LICENSE-2.0
010     *
011     *  Unless required by applicable law or agreed to in writing, software
012     *  distributed under the License is distributed on an "AS IS" BASIS,
013     *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
014     *  See the License for the specific language governing permissions and
015     *  limitations under the License.
016     */
017    package org.apache.geronimo.activation.handlers;
018    
019    import java.awt.datatransfer.DataFlavor;
020    import java.awt.datatransfer.UnsupportedFlavorException;
021    import java.io.IOException;
022    import java.io.OutputStream;
023    import javax.activation.ActivationDataFlavor;
024    import javax.activation.DataContentHandler;
025    import javax.activation.DataSource;
026    import javax.mail.MessagingException;
027    import javax.mail.internet.MimeMultipart;
028    
029    /**
030     * @version $Rev: 356022 $ $Date: 2005-12-11 12:58:34 -0800 (Sun, 11 Dec 2005) $
031     */
032    public class MultipartHandler implements DataContentHandler {
033        private final DataFlavor flavour;
034    
035        public MultipartHandler() {
036            flavour = new ActivationDataFlavor(MimeMultipart.class, "multipart/mixed", "Multipart MIME");
037        }
038    
039        public DataFlavor[] getTransferDataFlavors() {
040            return new DataFlavor[]{flavour};
041        }
042    
043        public Object getTransferData(DataFlavor df, DataSource ds) throws UnsupportedFlavorException, IOException {
044            return flavour.equals(df) ? getContent(ds) : null;
045        }
046    
047        public Object getContent(DataSource ds) throws IOException {
048            try {
049                return new MimeMultipart(ds);
050            } catch (MessagingException e) {
051                throw (IOException) new IOException(e.getMessage()).initCause(e);
052            }
053        }
054    
055        public void writeTo(Object obj, String mimeType, OutputStream os) throws IOException {
056            if (obj instanceof MimeMultipart) {
057                MimeMultipart mp = (MimeMultipart) obj;
058                try {
059                    mp.writeTo(os);
060                } catch (MessagingException e) {
061                    throw (IOException) new IOException(e.getMessage()).initCause(e);
062                }
063            }
064        }
065    }