/**
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You under the Apache License, Version 2.0
* (the "License"); you may not use this file except in compliance with
* the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.apache.camel.component.mail;
import java.io.IOException;
import java.util.Map;
import javax.activation.DataHandler;
import javax.mail.Message;
import javax.mail.MessagingException;
import javax.mail.Multipart;
import javax.mail.Part;
import org.apache.camel.RuntimeCamelException;
import org.apache.camel.impl.DefaultMessage;
import org.apache.camel.util.CollectionHelper;
/**
* Represents a {@link org.apache.camel.Message} for working with Mail
*
* @version $Revision:520964 $
*/
public class MailMessage extends DefaultMessage {
private Message mailMessage;
public MailMessage() {
}
public MailMessage(Message message) {
this.mailMessage = message;
}
@Override
public String toString() {
if (mailMessage != null) {
return "MailMessage: " + MailUtils.dumpMessage(mailMessage);
} else {
return "MailMessage: " + getBody();
}
}
@Override
public MailExchange getExchange() {
return (MailExchange)super.getExchange();
}
public MailMessage copy() {
MailMessage answer = (MailMessage)super.copy();
answer.mailMessage = mailMessage;
return answer;
}
/**
* Returns the underlying Mail message
*/
public Message getMessage() {
return mailMessage;
}
public void setMessage(Message mailMessage) {
this.mailMessage = mailMessage;
}
@Override
public Object getHeader(String name) {
Object answer = super.getHeader(name);
// mimic case insensitive search of mail message getHeader
if (answer == null) {
answer = super.getHeader(name.toLowerCase());
}
return answer;
}
@Override
public MailMessage newInstance() {
return new MailMessage();
}
@Override
protected Object createBody() {
if (mailMessage != null) {
return getExchange().getBinding().extractBodyFromMail(getExchange(), mailMessage);
}
return null;
}
@Override
protected void populateInitialHeaders(Map<String, Object> map) {
if (mailMessage != null) {
try {
map.putAll(getExchange().getBinding().extractHeadersFromMail(mailMessage));
} catch (MessagingException e) {
throw new RuntimeCamelException("Error accessing headers due to: " + e.getMessage(), e);
}
}
}
@Override
protected void populateInitialAttachments(Map<String, DataHandler> map) {
if (mailMessage != null) {
try {
extractAttachments(mailMessage, map);
} catch (Exception e) {
throw new RuntimeCamelException("Error populating the initial mail message attachments", e);
}
}
}
public void copyFrom(org.apache.camel.Message that) {
super.copyFrom(that);
if (that instanceof MailMessage) {
MailMessage mailMessage = (MailMessage) that;
this.mailMessage = mailMessage.mailMessage;
}
}
/**
* Parses the attachments of the given mail message and adds them to the map
*
* @param message the mail message with attachments
* @param map the map to add found attachments (attachmentFilename is the key)
*/
protected static void extractAttachments(Message message, Map<String, DataHandler> map)
throws javax.mail.MessagingException, IOException {
Object content = message.getContent();
if (content instanceof Multipart) {
// mail with attachment
Multipart mp = (Multipart)content;
for (int i = 0; i < mp.getCount(); i++) {
Part part = mp.getBodyPart(i);
String disposition = part.getDisposition();
if (disposition != null) {
if (disposition.equalsIgnoreCase(Part.ATTACHMENT) || disposition.equalsIgnoreCase(Part.INLINE)) {
// only add named attachments
if (part.getFileName() != null) {
// Parts marked with a disposition of Part.ATTACHMENT are clearly attachments
CollectionHelper.appendValue(map, part.getFileName(), part.getDataHandler());
}
}
}
}
}
}
}