/****************************************************************
* 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.james.mime4j.message;
import java.io.IOException;
import java.io.InputStream;
import java.util.Stack;
import java.nio.charset.Charset;
import org.apache.james.mime4j.MimeException;
import org.apache.james.mime4j.codec.Base64InputStream;
import org.apache.james.mime4j.codec.QuotedPrintableInputStream;
import org.apache.james.mime4j.descriptor.BodyDescriptor;
import org.apache.james.mime4j.field.AbstractField;
import org.apache.james.mime4j.parser.ContentHandler;
import org.apache.james.mime4j.parser.Field;
import org.apache.james.mime4j.parser.MimeStreamParser;
import org.apache.james.mime4j.storage.StorageProvider;
import org.apache.james.mime4j.util.ByteArrayBuffer;
import org.apache.james.mime4j.util.ByteSequence;
import org.apache.james.mime4j.util.MimeUtil;
import org.apache.james.mime4j.util.CharsetUtil;
/**
* A <code>ContentHandler</code> for building an <code>Entity</code> to be
* used in conjunction with a {@link MimeStreamParser}.
*/
public class MessageBuilder implements ContentHandler {
private final Entity entity;
private final BodyFactory bodyFactory;
private Stack<Object> stack = new Stack<Object>();
private String defaultCharset = "ISO8859-1";
public MessageBuilder(Entity entity) {
this.entity = entity;
this.bodyFactory = new BodyFactory();
}
public MessageBuilder(Entity entity, StorageProvider storageProvider,
String defaultCharset) {
this.entity = entity;
this.bodyFactory = new BodyFactory(storageProvider);
this.defaultCharset = defaultCharset;
}
private void expect(Class<?> c) {
if (!c.isInstance(stack.peek())) {
throw new IllegalStateException("Internal stack error: "
+ "Expected '" + c.getName() + "' found '"
+ stack.peek().getClass().getName() + "'");
}
}
/**
* @see org.apache.james.mime4j.parser.ContentHandler#startMessage()
*/
public void startMessage() throws MimeException {
if (stack.isEmpty()) {
stack.push(this.entity);
} else {
expect(Entity.class);
Message m = new Message();
((Entity) stack.peek()).setBody(m);
stack.push(m);
}
}
/**
* @see org.apache.james.mime4j.parser.ContentHandler#endMessage()
*/
public void endMessage() throws MimeException {
expect(Message.class);
stack.pop();
}
/**
* @see org.apache.james.mime4j.parser.ContentHandler#startHeader()
*/
public void startHeader() throws MimeException {
stack.push(new Header());
}
/**
* @see org.apache.james.mime4j.parser.ContentHandler#field(Field)
*/
public void field(Field field) throws MimeException {
expect(Header.class);
Field parsedField;
ByteSequence raw = field.getRaw();
if (CharsetUtil.isASCII(raw.toByteArray()))
parsedField = AbstractField.parse(raw);
else
parsedField = AbstractField.parse(raw, Charset.forName(defaultCharset));
((Header) stack.peek()).addField(parsedField);
}
/**
* @see org.apache.james.mime4j.parser.ContentHandler#endHeader()
*/
public void endHeader() throws MimeException {
expect(Header.class);
Header h = (Header) stack.pop();
expect(Entity.class);
((Entity) stack.peek()).setHeader(h);
}
/**
* @see org.apache.james.mime4j.parser.ContentHandler#startMultipart(org.apache.james.mime4j.descriptor.BodyDescriptor)
*/
public void startMultipart(final BodyDescriptor bd) throws MimeException {
expect(Entity.class);
final Entity e = (Entity) stack.peek();
final String subType = bd.getSubType();
final Multipart multiPart = new Multipart(subType);
e.setBody(multiPart);
stack.push(multiPart);
}
/**
* @see org.apache.james.mime4j.parser.ContentHandler#body(org.apache.james.mime4j.descriptor.BodyDescriptor, java.io.InputStream)
*/
public void body(BodyDescriptor bd, final InputStream is) throws MimeException, IOException {
expect(Entity.class);
final String enc = bd.getTransferEncoding();
final Body body;
final InputStream decodedStream;
if (MimeUtil.ENC_BASE64.equals(enc)) {
decodedStream = new Base64InputStream(is);
} else if (MimeUtil.ENC_QUOTED_PRINTABLE.equals(enc)) {
decodedStream = new QuotedPrintableInputStream(is);
} else {
decodedStream = is;
}
if (bd.getMimeType().startsWith("text/")) {
String charset = bd.getCharset();
if (charset == null)
charset = defaultCharset;
body = bodyFactory.textBody(decodedStream, charset);
} else {
body = bodyFactory.binaryBody(decodedStream);
}
Entity entity = ((Entity) stack.peek());
entity.setBody(body);
}
/**
* @see org.apache.james.mime4j.parser.ContentHandler#endMultipart()
*/
public void endMultipart() throws MimeException {
stack.pop();
}
/**
* @see org.apache.james.mime4j.parser.ContentHandler#startBodyPart()
*/
public void startBodyPart() throws MimeException {
expect(Multipart.class);
BodyPart bodyPart = new BodyPart();
((Multipart) stack.peek()).addBodyPart(bodyPart);
stack.push(bodyPart);
}
/**
* @see org.apache.james.mime4j.parser.ContentHandler#endBodyPart()
*/
public void endBodyPart() throws MimeException {
expect(BodyPart.class);
stack.pop();
}
/**
* @see org.apache.james.mime4j.parser.ContentHandler#epilogue(java.io.InputStream)
*/
public void epilogue(InputStream is) throws MimeException, IOException {
expect(Multipart.class);
ByteSequence bytes = loadStream(is);
((Multipart) stack.peek()).setEpilogueRaw(bytes);
}
/**
* @see org.apache.james.mime4j.parser.ContentHandler#preamble(java.io.InputStream)
*/
public void preamble(InputStream is) throws MimeException, IOException {
expect(Multipart.class);
ByteSequence bytes = loadStream(is);
((Multipart) stack.peek()).setPreambleRaw(bytes);
}
/**
* Unsupported.
* @see org.apache.james.mime4j.parser.ContentHandler#raw(java.io.InputStream)
*/
public void raw(InputStream is) throws MimeException, IOException {
throw new UnsupportedOperationException("Not supported");
}
private static ByteSequence loadStream(InputStream in) throws IOException {
ByteArrayBuffer bab = new ByteArrayBuffer(64);
int b;
while ((b = in.read()) != -1) {
bab.append(b);
}
return bab;
}
}