mirror of
https://github.com/moparisthebest/k-9
synced 2024-11-27 11:42:16 -05:00
Fix setUsing7bitTransport() functionality for BinaryTempFileBody
This commit is contained in:
parent
9f4f0cf6a8
commit
f7d3eaa006
@ -2,7 +2,10 @@ package com.fsck.k9.mail.internet;
|
|||||||
|
|
||||||
import com.fsck.k9.mail.Body;
|
import com.fsck.k9.mail.Body;
|
||||||
import com.fsck.k9.mail.MessagingException;
|
import com.fsck.k9.mail.MessagingException;
|
||||||
|
import com.fsck.k9.mail.filter.Base64OutputStream;
|
||||||
import org.apache.commons.io.IOUtils;
|
import org.apache.commons.io.IOUtils;
|
||||||
|
import org.apache.james.mime4j.codec.QuotedPrintableOutputStream;
|
||||||
|
import org.apache.james.mime4j.util.MimeUtil;
|
||||||
|
|
||||||
import java.io.*;
|
import java.io.*;
|
||||||
|
|
||||||
@ -24,13 +27,50 @@ public class BinaryTempFileBody implements Body {
|
|||||||
}
|
}
|
||||||
|
|
||||||
public void setEncoding(String encoding) throws MessagingException {
|
public void setEncoding(String encoding) throws MessagingException {
|
||||||
mEncoding = encoding;
|
if (mEncoding != null && mEncoding.equalsIgnoreCase(encoding)) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
// The encoding changed, so we need to convert the message
|
||||||
|
if (!MimeUtil.ENC_8BIT.equalsIgnoreCase(mEncoding)) {
|
||||||
|
throw new RuntimeException("Can't convert from encoding: " + mEncoding);
|
||||||
|
}
|
||||||
|
|
||||||
|
try {
|
||||||
|
File newFile = File.createTempFile("body", null, mTempDirectory);
|
||||||
|
OutputStream out = new FileOutputStream(newFile);
|
||||||
|
try {
|
||||||
|
if (MimeUtil.ENC_QUOTED_PRINTABLE.equals(encoding)) {
|
||||||
|
out = new QuotedPrintableOutputStream(out, false);
|
||||||
|
} else if (MimeUtil.ENC_BASE64.equals(encoding)) {
|
||||||
|
out = new Base64OutputStream(out);
|
||||||
|
} else {
|
||||||
|
throw new RuntimeException("Target encoding not supported: " + encoding);
|
||||||
|
}
|
||||||
|
|
||||||
|
InputStream in = getInputStream();
|
||||||
|
try {
|
||||||
|
IOUtils.copy(in, out);
|
||||||
|
} finally {
|
||||||
|
in.close();
|
||||||
|
}
|
||||||
|
} finally {
|
||||||
|
out.close();
|
||||||
|
}
|
||||||
|
|
||||||
|
mFile = newFile;
|
||||||
|
mEncoding = encoding;
|
||||||
|
} catch (IOException e) {
|
||||||
|
throw new MessagingException("Unable to convert body", e);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
public BinaryTempFileBody() {
|
public BinaryTempFileBody(String encoding) {
|
||||||
if (mTempDirectory == null) {
|
if (mTempDirectory == null) {
|
||||||
throw new RuntimeException("setTempDirectory has not been called on BinaryTempFileBody!");
|
throw new RuntimeException("setTempDirectory has not been called on BinaryTempFileBody!");
|
||||||
}
|
}
|
||||||
|
|
||||||
|
mEncoding = encoding;
|
||||||
}
|
}
|
||||||
|
|
||||||
public OutputStream getOutputStream() throws IOException {
|
public OutputStream getOutputStream() throws IOException {
|
||||||
|
@ -18,6 +18,10 @@ import com.fsck.k9.mail.MessagingException;
|
|||||||
*/
|
*/
|
||||||
public class BinaryTempFileMessageBody extends BinaryTempFileBody implements CompositeBody {
|
public class BinaryTempFileMessageBody extends BinaryTempFileBody implements CompositeBody {
|
||||||
|
|
||||||
|
public BinaryTempFileMessageBody(String encoding) {
|
||||||
|
super(encoding);
|
||||||
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void setEncoding(String encoding) throws MessagingException {
|
public void setEncoding(String encoding) throws MessagingException {
|
||||||
if (!MimeUtil.ENC_7BIT.equalsIgnoreCase(encoding)
|
if (!MimeUtil.ENC_7BIT.equalsIgnoreCase(encoding)
|
||||||
|
@ -1136,11 +1136,10 @@ public class MimeUtility {
|
|||||||
|
|
||||||
BinaryTempFileBody tempBody;
|
BinaryTempFileBody tempBody;
|
||||||
if (MimeUtil.isMessage(contentType)) {
|
if (MimeUtil.isMessage(contentType)) {
|
||||||
tempBody = new BinaryTempFileMessageBody();
|
tempBody = new BinaryTempFileMessageBody(contentTransferEncoding);
|
||||||
} else {
|
} else {
|
||||||
tempBody = new BinaryTempFileBody();
|
tempBody = new BinaryTempFileBody(contentTransferEncoding);
|
||||||
}
|
}
|
||||||
tempBody.setEncoding(contentTransferEncoding);
|
|
||||||
|
|
||||||
OutputStream out = tempBody.getOutputStream();
|
OutputStream out = tempBody.getOutputStream();
|
||||||
try {
|
try {
|
||||||
|
@ -291,8 +291,7 @@ public class MessageTest extends AndroidTestCase {
|
|||||||
|
|
||||||
private MimeMessage nestedMessage(MimeMessage subMessage)
|
private MimeMessage nestedMessage(MimeMessage subMessage)
|
||||||
throws MessagingException, IOException {
|
throws MessagingException, IOException {
|
||||||
BinaryTempFileMessageBody tempMessageBody = new BinaryTempFileMessageBody();
|
BinaryTempFileMessageBody tempMessageBody = new BinaryTempFileMessageBody(MimeUtil.ENC_8BIT);
|
||||||
tempMessageBody.setEncoding(MimeUtil.ENC_8BIT);
|
|
||||||
|
|
||||||
OutputStream out = tempMessageBody.getOutputStream();
|
OutputStream out = tempMessageBody.getOutputStream();
|
||||||
try {
|
try {
|
||||||
@ -334,7 +333,7 @@ public class MessageTest extends AndroidTestCase {
|
|||||||
String encodedTestString = "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
|
String encodedTestString = "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
|
||||||
+ "abcdefghijklmnopqrstuvwxyz0123456789+/\r\n";
|
+ "abcdefghijklmnopqrstuvwxyz0123456789+/\r\n";
|
||||||
|
|
||||||
BinaryTempFileBody tempFileBody = new BinaryTempFileBody();
|
BinaryTempFileBody tempFileBody = new BinaryTempFileBody(MimeUtil.ENC_BASE64);
|
||||||
|
|
||||||
InputStream in = new ByteArrayInputStream(
|
InputStream in = new ByteArrayInputStream(
|
||||||
encodedTestString.getBytes("UTF-8"));
|
encodedTestString.getBytes("UTF-8"));
|
||||||
|
Loading…
Reference in New Issue
Block a user