Simplify message extraction logic, fix #2776
This commit is contained in:
parent
e266547bfc
commit
d9ee8b5cf0
3 changed files with 60 additions and 102 deletions
|
@ -1,6 +1,7 @@
|
||||||
package com.fsck.k9.mailstore;
|
package com.fsck.k9.mailstore;
|
||||||
|
|
||||||
|
|
||||||
|
import java.util.Collections;
|
||||||
import java.util.List;
|
import java.util.List;
|
||||||
|
|
||||||
import com.fsck.k9.mail.Message;
|
import com.fsck.k9.mail.Message;
|
||||||
|
@ -36,24 +37,24 @@ public class MessageViewInfo {
|
||||||
this.extraAttachments = extraAttachments;
|
this.extraAttachments = extraAttachments;
|
||||||
}
|
}
|
||||||
|
|
||||||
public static MessageViewInfo createWithExtractedContent(
|
static MessageViewInfo createWithExtractedContent(Message message, boolean isMessageIncomplete,
|
||||||
Message message, boolean isMessageIncomplete, Part rootPart,
|
String text, List<AttachmentViewInfo> attachments, AttachmentResolver attachmentResolver) {
|
||||||
String text, List<AttachmentViewInfo> attachments,
|
|
||||||
CryptoResultAnnotation cryptoResultAnnotation,
|
|
||||||
AttachmentResolver attachmentResolver,
|
|
||||||
String extraText, List<AttachmentViewInfo> extraAttachments
|
|
||||||
) {
|
|
||||||
return new MessageViewInfo(
|
return new MessageViewInfo(
|
||||||
message, isMessageIncomplete, rootPart,
|
message, isMessageIncomplete, message, text, attachments, null, attachmentResolver, null,
|
||||||
text, attachments,
|
Collections.<AttachmentViewInfo>emptyList());
|
||||||
cryptoResultAnnotation,
|
|
||||||
attachmentResolver,
|
|
||||||
extraText, extraAttachments
|
|
||||||
);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
public static MessageViewInfo createWithErrorState(Message message, boolean isMessageIncomplete) {
|
public static MessageViewInfo createWithErrorState(Message message, boolean isMessageIncomplete) {
|
||||||
return new MessageViewInfo(message, isMessageIncomplete, null, null, null, null, null, null, null);
|
return new MessageViewInfo(message, isMessageIncomplete, null, null, null, null, null, null, null);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
MessageViewInfo withCryptoData(CryptoResultAnnotation rootPartAnnotation, String extraViewableText,
|
||||||
|
List<AttachmentViewInfo> extraAttachmentInfos) {
|
||||||
|
return new MessageViewInfo(
|
||||||
|
message, isMessageIncomplete, rootPart, text, attachments,
|
||||||
|
rootPartAnnotation,
|
||||||
|
attachmentResolver,
|
||||||
|
extraViewableText, extraAttachmentInfos
|
||||||
|
);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -12,22 +12,24 @@ import android.support.annotation.VisibleForTesting;
|
||||||
import android.support.annotation.WorkerThread;
|
import android.support.annotation.WorkerThread;
|
||||||
|
|
||||||
import com.fsck.k9.Globals;
|
import com.fsck.k9.Globals;
|
||||||
|
import com.fsck.k9.K9;
|
||||||
import com.fsck.k9.R;
|
import com.fsck.k9.R;
|
||||||
|
import com.fsck.k9.crypto.MessageCryptoStructureDetector;
|
||||||
import com.fsck.k9.mail.Address;
|
import com.fsck.k9.mail.Address;
|
||||||
import com.fsck.k9.mail.Flag;
|
import com.fsck.k9.mail.Flag;
|
||||||
import com.fsck.k9.mail.Message;
|
import com.fsck.k9.mail.Message;
|
||||||
import com.fsck.k9.mail.MessagingException;
|
import com.fsck.k9.mail.MessagingException;
|
||||||
import com.fsck.k9.mail.Part;
|
import com.fsck.k9.mail.Part;
|
||||||
import com.fsck.k9.mail.internet.MessageExtractor;
|
import com.fsck.k9.mail.internet.MessageExtractor;
|
||||||
|
import com.fsck.k9.mail.internet.MimeUtility;
|
||||||
import com.fsck.k9.mail.internet.Viewable;
|
import com.fsck.k9.mail.internet.Viewable;
|
||||||
import com.fsck.k9.mail.internet.Viewable.Flowed;
|
import com.fsck.k9.mail.internet.Viewable.Flowed;
|
||||||
|
import com.fsck.k9.mailstore.CryptoResultAnnotation.CryptoError;
|
||||||
import com.fsck.k9.mailstore.util.FlowedMessageUtils;
|
import com.fsck.k9.mailstore.util.FlowedMessageUtils;
|
||||||
import com.fsck.k9.message.extractors.AttachmentInfoExtractor;
|
import com.fsck.k9.message.extractors.AttachmentInfoExtractor;
|
||||||
import com.fsck.k9.message.html.HtmlConverter;
|
import com.fsck.k9.message.html.HtmlConverter;
|
||||||
import com.fsck.k9.message.html.HtmlProcessor;
|
import com.fsck.k9.message.html.HtmlProcessor;
|
||||||
import com.fsck.k9.ui.crypto.MessageCryptoAnnotations;
|
import com.fsck.k9.ui.crypto.MessageCryptoAnnotations;
|
||||||
import com.fsck.k9.ui.crypto.MessageCryptoSplitter;
|
|
||||||
import com.fsck.k9.ui.crypto.MessageCryptoSplitter.CryptoMessageParts;
|
|
||||||
import org.openintents.openpgp.util.OpenPgpUtils;
|
import org.openintents.openpgp.util.OpenPgpUtils;
|
||||||
import timber.log.Timber;
|
import timber.log.Timber;
|
||||||
|
|
||||||
|
@ -71,43 +73,58 @@ public class MessageViewInfoExtractor {
|
||||||
@WorkerThread
|
@WorkerThread
|
||||||
public MessageViewInfo extractMessageForView(Message message, @Nullable MessageCryptoAnnotations annotations)
|
public MessageViewInfo extractMessageForView(Message message, @Nullable MessageCryptoAnnotations annotations)
|
||||||
throws MessagingException {
|
throws MessagingException {
|
||||||
Part rootPart;
|
ArrayList<Part> extraParts = new ArrayList<>();
|
||||||
CryptoResultAnnotation cryptoResultAnnotation;
|
Part cryptoContentPart = MessageCryptoStructureDetector.findPrimaryEncryptedOrSignedPart(message, extraParts);
|
||||||
List<Part> extraParts;
|
|
||||||
|
|
||||||
CryptoMessageParts cryptoMessageParts = MessageCryptoSplitter.split(message, annotations);
|
if (cryptoContentPart == null) {
|
||||||
if (cryptoMessageParts != null) {
|
|
||||||
rootPart = cryptoMessageParts.contentPart;
|
|
||||||
cryptoResultAnnotation = cryptoMessageParts.contentCryptoAnnotation;
|
|
||||||
extraParts = cryptoMessageParts.extraParts;
|
|
||||||
} else {
|
|
||||||
if (annotations != null && !annotations.isEmpty()) {
|
if (annotations != null && !annotations.isEmpty()) {
|
||||||
Timber.e("Got message annotations but no crypto root part!");
|
Timber.e("Got crypto message annotations but no crypto root part!");
|
||||||
}
|
}
|
||||||
rootPart = message;
|
return extractSimpleMessageForView(message, message);
|
||||||
cryptoResultAnnotation = null;
|
|
||||||
extraParts = null;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
List<AttachmentViewInfo> attachmentInfos = new ArrayList<>();
|
boolean isOpenPgpEncrypted =
|
||||||
ViewableExtractedText viewable = extractViewableAndAttachments(
|
MimeUtility.isSameMimeType(cryptoContentPart.getMimeType(), "multipart/encrypted") &&
|
||||||
Collections.singletonList(rootPart), attachmentInfos);
|
MimeUtility.isSameMimeType(getHeaderParameter(cryptoContentPart.getContentType(), "protocol"),
|
||||||
|
"application/pgp-encrypted");
|
||||||
|
if (!K9.isOpenPgpProviderConfigured() && isOpenPgpEncrypted) {
|
||||||
|
CryptoResultAnnotation noProviderAnnotation = CryptoResultAnnotation.createErrorAnnotation(
|
||||||
|
CryptoError.OPENPGP_ENCRYPTED_NO_PROVIDER, null);
|
||||||
|
return MessageViewInfo.createWithErrorState(message, false).withCryptoData(noProviderAnnotation, null, null);
|
||||||
|
}
|
||||||
|
|
||||||
|
CryptoResultAnnotation cryptoContentPartAnnotation = annotations != null ? annotations.get(cryptoContentPart) : null;
|
||||||
|
if (cryptoContentPartAnnotation != null) {
|
||||||
|
return extractCryptoMessageForView(message, extraParts, cryptoContentPart, cryptoContentPartAnnotation);
|
||||||
|
}
|
||||||
|
|
||||||
|
return extractSimpleMessageForView(message, message);
|
||||||
|
}
|
||||||
|
|
||||||
|
private MessageViewInfo extractCryptoMessageForView(Message message,
|
||||||
|
ArrayList<Part> extraParts, Part cryptoContentPart, CryptoResultAnnotation cryptoContentPartAnnotation)
|
||||||
|
throws MessagingException {
|
||||||
|
if (cryptoContentPartAnnotation != null && cryptoContentPartAnnotation.hasReplacementData()) {
|
||||||
|
cryptoContentPart = cryptoContentPartAnnotation.getReplacementData();
|
||||||
|
}
|
||||||
|
|
||||||
List<AttachmentViewInfo> extraAttachmentInfos = new ArrayList<>();
|
List<AttachmentViewInfo> extraAttachmentInfos = new ArrayList<>();
|
||||||
String extraViewableText = null;
|
ViewableExtractedText extraViewable = extractViewableAndAttachments(extraParts, extraAttachmentInfos);
|
||||||
if (extraParts != null) {
|
|
||||||
ViewableExtractedText extraViewable =
|
|
||||||
extractViewableAndAttachments(extraParts, extraAttachmentInfos);
|
|
||||||
extraViewableText = extraViewable.text;
|
|
||||||
}
|
|
||||||
|
|
||||||
AttachmentResolver attachmentResolver = AttachmentResolver.createFromPart(rootPart);
|
MessageViewInfo messageViewInfo = extractSimpleMessageForView(message, cryptoContentPart);
|
||||||
|
return messageViewInfo.withCryptoData(cryptoContentPartAnnotation, extraViewable.text, extraAttachmentInfos);
|
||||||
|
}
|
||||||
|
|
||||||
boolean isMessageIncomplete = !message.isSet(Flag.X_DOWNLOADED_FULL) ||
|
private MessageViewInfo extractSimpleMessageForView(Message message, Part contentPart) throws MessagingException {
|
||||||
MessageExtractor.hasMissingParts(message);
|
List<AttachmentViewInfo> attachmentInfos = new ArrayList<>();
|
||||||
|
ViewableExtractedText viewable = extractViewableAndAttachments(
|
||||||
|
Collections.singletonList(contentPart), attachmentInfos);
|
||||||
|
AttachmentResolver attachmentResolver = AttachmentResolver.createFromPart(contentPart);
|
||||||
|
boolean isMessageIncomplete =
|
||||||
|
!message.isSet(Flag.X_DOWNLOADED_FULL) || MessageExtractor.hasMissingParts(message);
|
||||||
|
|
||||||
return MessageViewInfo.createWithExtractedContent(message, isMessageIncomplete, rootPart, viewable.html,
|
return MessageViewInfo.createWithExtractedContent(
|
||||||
attachmentInfos, cryptoResultAnnotation, attachmentResolver, extraViewableText, extraAttachmentInfos);
|
message, isMessageIncomplete, viewable.html, attachmentInfos, attachmentResolver);
|
||||||
}
|
}
|
||||||
|
|
||||||
private ViewableExtractedText extractViewableAndAttachments(List<Part> parts,
|
private ViewableExtractedText extractViewableAndAttachments(List<Part> parts,
|
||||||
|
|
|
@ -1,60 +0,0 @@
|
||||||
package com.fsck.k9.ui.crypto;
|
|
||||||
|
|
||||||
|
|
||||||
import java.util.ArrayList;
|
|
||||||
import java.util.Collections;
|
|
||||||
import java.util.List;
|
|
||||||
|
|
||||||
import android.support.annotation.NonNull;
|
|
||||||
import android.support.annotation.Nullable;
|
|
||||||
|
|
||||||
import com.fsck.k9.crypto.MessageCryptoStructureDetector;
|
|
||||||
import com.fsck.k9.mail.Message;
|
|
||||||
import com.fsck.k9.mail.Part;
|
|
||||||
import com.fsck.k9.mailstore.CryptoResultAnnotation;
|
|
||||||
import com.fsck.k9.mailstore.CryptoResultAnnotation.CryptoError;
|
|
||||||
|
|
||||||
|
|
||||||
public class MessageCryptoSplitter {
|
|
||||||
private MessageCryptoSplitter() { }
|
|
||||||
|
|
||||||
@Nullable
|
|
||||||
public static CryptoMessageParts split(@NonNull Message message, @Nullable MessageCryptoAnnotations annotations) {
|
|
||||||
ArrayList<Part> extraParts = new ArrayList<>();
|
|
||||||
Part primaryPart = MessageCryptoStructureDetector.findPrimaryEncryptedOrSignedPart(message, extraParts);
|
|
||||||
if (primaryPart == null) {
|
|
||||||
return null;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (annotations == null) {
|
|
||||||
CryptoResultAnnotation rootPartAnnotation =
|
|
||||||
CryptoResultAnnotation.createErrorAnnotation(CryptoError.OPENPGP_ENCRYPTED_NO_PROVIDER, null);
|
|
||||||
return new CryptoMessageParts(primaryPart, rootPartAnnotation, extraParts);
|
|
||||||
}
|
|
||||||
|
|
||||||
CryptoResultAnnotation rootPartAnnotation = annotations.get(primaryPart);
|
|
||||||
Part rootPart;
|
|
||||||
if (rootPartAnnotation != null && rootPartAnnotation.hasReplacementData()) {
|
|
||||||
rootPart = rootPartAnnotation.getReplacementData();
|
|
||||||
} else {
|
|
||||||
rootPart = primaryPart;
|
|
||||||
}
|
|
||||||
|
|
||||||
return new CryptoMessageParts(rootPart, rootPartAnnotation, extraParts);
|
|
||||||
}
|
|
||||||
|
|
||||||
public static class CryptoMessageParts {
|
|
||||||
public final Part contentPart;
|
|
||||||
@Nullable
|
|
||||||
public final CryptoResultAnnotation contentCryptoAnnotation;
|
|
||||||
|
|
||||||
public final List<Part> extraParts;
|
|
||||||
|
|
||||||
CryptoMessageParts(Part contentPart, @Nullable CryptoResultAnnotation contentCryptoAnnotation, List<Part> extraParts) {
|
|
||||||
this.contentPart = contentPart;
|
|
||||||
this.contentCryptoAnnotation = contentCryptoAnnotation;
|
|
||||||
this.extraParts = Collections.unmodifiableList(extraParts);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
Loading…
Reference in a new issue