org.jsoftware.javamail.AbstractFileTransport Maven / Gradle / Ivy

Go to download
package org.jsoftware.javamail;

import javax.mail.*;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStream;

/**
 * {@link javax.mail.Transport} that saves messages to file using different format writers
 * @author szalik
 */
public abstract class AbstractFileTransport extends AbstractDevTransport {
	private final File dir;

	public AbstractFileTransport(Session session, URLName urlname) {
		super(session, urlname);
		String s = session.getProperties().getProperty("mail.files.path", ".");
		dir = new File(s);
		if (! dir.exists()) {
			if (! dir.mkdirs()) {
				throw new IllegalArgumentException("Unable to create output directory " + dir.getAbsolutePath());
			}
		}
	}


	@Override
	public void sendMessage(Message message, Address[] addresses) throws MessagingException {
		validate(message, addresses);
        File file = null;
        int suffix = -1;
        String base = Long.toString(System.currentTimeMillis(), 32);
        try {
            synchronized (this) {
                do {
                    suffix++;
                    StringBuilder sb = new StringBuilder(base);
                    if (suffix > 0) {
                        sb.append('-').append(suffix);
                    }
                    sb.append('.').append(getFilenameExtension());
                    file = new File(dir, sb.toString());
                } while (file.exists());
                if (! file.createNewFile()) {
                    throw new IOException("Cannot create file " + file.getAbsolutePath());
                }
            }
            FileOutputStream fw = null;
			try {
                fw = new FileOutputStream(file);
				writeMessage(message, addresses, fw);
			} finally {
                if (fw != null) {
				    fw.close();
                }
			}
		} catch (IOException e) {
			throw new MessagingException("Failed to write file " + (file != null ? file.getAbsolutePath() : ""), e);
		}
	}


    /**
     * Write message
     * @param message message
     * @param os where to write
     * @throws IOException
     * @throws MessagingException
     */
    protected abstract void writeMessage(Message message, Address[] addresses, OutputStream os) throws IOException, MessagingException;


    /**
     * @return filename extension
     */
    protected abstract String getFilenameExtension();

}