mirror of https://github.com/M66B/FairEmail.git
M66B
6 years ago
parent
17c894b3d1
commit
9f5f34815a
@ -0,0 +1,210 @@
|
|||||||
|
package eu.faircode.email;
|
||||||
|
|
||||||
|
import android.content.Context;
|
||||||
|
import android.text.TextUtils;
|
||||||
|
import android.util.Log;
|
||||||
|
import android.util.SparseArray;
|
||||||
|
|
||||||
|
import com.sun.mail.imap.IMAPFolder;
|
||||||
|
import com.sun.mail.imap.IMAPStore;
|
||||||
|
|
||||||
|
import java.io.UnsupportedEncodingException;
|
||||||
|
import java.util.ArrayList;
|
||||||
|
import java.util.List;
|
||||||
|
import java.util.Properties;
|
||||||
|
|
||||||
|
import javax.mail.FetchProfile;
|
||||||
|
import javax.mail.Folder;
|
||||||
|
import javax.mail.Message;
|
||||||
|
import javax.mail.MessagingException;
|
||||||
|
import javax.mail.Session;
|
||||||
|
import javax.mail.UIDFolder;
|
||||||
|
import javax.mail.internet.MimeMessage;
|
||||||
|
import javax.mail.search.BodyTerm;
|
||||||
|
import javax.mail.search.FromStringTerm;
|
||||||
|
import javax.mail.search.OrTerm;
|
||||||
|
import javax.mail.search.SubjectTerm;
|
||||||
|
|
||||||
|
import androidx.lifecycle.Lifecycle;
|
||||||
|
import androidx.lifecycle.LifecycleObserver;
|
||||||
|
import androidx.lifecycle.LifecycleOwner;
|
||||||
|
import androidx.lifecycle.OnLifecycleEvent;
|
||||||
|
import androidx.paging.PositionalDataSource;
|
||||||
|
|
||||||
|
public class SearchDataSource extends PositionalDataSource<TupleMessageEx> implements LifecycleObserver {
|
||||||
|
private Context context;
|
||||||
|
private LifecycleOwner owner;
|
||||||
|
private long fid;
|
||||||
|
private String search;
|
||||||
|
|
||||||
|
private IMAPStore istore = null;
|
||||||
|
private IMAPFolder ifolder;
|
||||||
|
private Message[] imessages;
|
||||||
|
|
||||||
|
private SparseArray<TupleMessageEx> cache = new SparseArray<>();
|
||||||
|
|
||||||
|
SearchDataSource(Context context, LifecycleOwner owner, long folder, String search) {
|
||||||
|
Log.i(Helper.TAG, "SDS create");
|
||||||
|
|
||||||
|
this.context = context;
|
||||||
|
this.owner = owner;
|
||||||
|
this.fid = folder;
|
||||||
|
this.search = search;
|
||||||
|
|
||||||
|
owner.getLifecycle().addObserver(this);
|
||||||
|
}
|
||||||
|
|
||||||
|
@OnLifecycleEvent(Lifecycle.Event.ON_DESTROY)
|
||||||
|
public void onDestroyed() {
|
||||||
|
Log.i(Helper.TAG, "SDS destroy");
|
||||||
|
|
||||||
|
new Thread(new Runnable() {
|
||||||
|
@Override
|
||||||
|
public void run() {
|
||||||
|
try {
|
||||||
|
if (istore != null)
|
||||||
|
istore.close();
|
||||||
|
} catch (MessagingException ex) {
|
||||||
|
Log.e(Helper.TAG, ex + "\n" + Log.getStackTraceString(ex));
|
||||||
|
} finally {
|
||||||
|
istore = null;
|
||||||
|
ifolder = null;
|
||||||
|
imessages = null;
|
||||||
|
cache.clear();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}).start();
|
||||||
|
|
||||||
|
owner.getLifecycle().removeObserver(this);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void loadInitial(LoadInitialParams params, LoadInitialCallback<TupleMessageEx> callback) {
|
||||||
|
Log.i(Helper.TAG, "SDS load initial");
|
||||||
|
try {
|
||||||
|
SearchResult result = search(search, params.requestedStartPosition, params.requestedLoadSize);
|
||||||
|
callback.onResult(result.messages, params.requestedStartPosition, result.total);
|
||||||
|
} catch (Throwable ex) {
|
||||||
|
Log.e(Helper.TAG, ex + "\n" + Log.getStackTraceString(ex));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void loadRange(LoadRangeParams params, LoadRangeCallback<TupleMessageEx> callback) {
|
||||||
|
try {
|
||||||
|
Log.i(Helper.TAG, "SDS load range");
|
||||||
|
SearchResult result = search(search, params.startPosition, params.loadSize);
|
||||||
|
callback.onResult(result.messages);
|
||||||
|
} catch (Throwable ex) {
|
||||||
|
Log.e(Helper.TAG, ex + "\n" + Log.getStackTraceString(ex));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private SearchResult search(String term, int from, int count) throws MessagingException, UnsupportedEncodingException {
|
||||||
|
Log.i(Helper.TAG, "SDS search from=" + from + " count=" + count);
|
||||||
|
|
||||||
|
DB db = DB.getInstance(context);
|
||||||
|
EntityFolder folder = db.folder().getFolder(fid);
|
||||||
|
EntityAccount account = db.account().getAccount(folder.account);
|
||||||
|
|
||||||
|
if (istore == null) {
|
||||||
|
Properties props = MessageHelper.getSessionProperties();
|
||||||
|
Session isession = Session.getInstance(props, null);
|
||||||
|
|
||||||
|
Log.i(Helper.TAG, "SDS connecting account=" + account.name);
|
||||||
|
istore = (IMAPStore) isession.getStore("imaps");
|
||||||
|
istore.connect(account.host, account.port, account.user, account.password);
|
||||||
|
|
||||||
|
Log.i(Helper.TAG, "SDS opening folder=" + folder.name);
|
||||||
|
ifolder = (IMAPFolder) istore.getFolder(folder.name);
|
||||||
|
ifolder.open(Folder.READ_WRITE);
|
||||||
|
|
||||||
|
Log.i(Helper.TAG, "SDS searching term=" + term);
|
||||||
|
imessages = ifolder.search(
|
||||||
|
new OrTerm(
|
||||||
|
new FromStringTerm(term),
|
||||||
|
new OrTerm(
|
||||||
|
new SubjectTerm(term),
|
||||||
|
new BodyTerm(term))));
|
||||||
|
Log.i(Helper.TAG, "SDS found messages=" + imessages.length);
|
||||||
|
}
|
||||||
|
|
||||||
|
SearchResult result = new SearchResult();
|
||||||
|
result.total = imessages.length;
|
||||||
|
result.messages = new ArrayList<>();
|
||||||
|
|
||||||
|
List<Message> selected = new ArrayList<>();
|
||||||
|
int base = imessages.length - 1 - from;
|
||||||
|
for (int i = base; i >= 0 && i >= base - count + 1; i--)
|
||||||
|
selected.add(imessages[i]);
|
||||||
|
Log.i(Helper.TAG, "SDS selected messages=" + selected.size());
|
||||||
|
|
||||||
|
FetchProfile fp = new FetchProfile();
|
||||||
|
fp.add(UIDFolder.FetchProfileItem.UID);
|
||||||
|
fp.add(IMAPFolder.FetchProfileItem.FLAGS);
|
||||||
|
fp.add(FetchProfile.Item.ENVELOPE);
|
||||||
|
fp.add(FetchProfile.Item.CONTENT_INFO);
|
||||||
|
fp.add(IMAPFolder.FetchProfileItem.HEADERS);
|
||||||
|
fp.add(IMAPFolder.FetchProfileItem.MESSAGE);
|
||||||
|
ifolder.fetch(selected.toArray(new Message[0]), fp);
|
||||||
|
|
||||||
|
for (int s = 0; s < selected.size(); s++) {
|
||||||
|
int pos = from + s;
|
||||||
|
if (cache.get(pos) != null) {
|
||||||
|
Log.i(Helper.TAG, "SDS from cache pos=" + pos);
|
||||||
|
result.messages.add(cache.get(pos));
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
Message imessage = selected.get(s);
|
||||||
|
|
||||||
|
long uid = ifolder.getUID(imessage);
|
||||||
|
|
||||||
|
MessageHelper helper = new MessageHelper((MimeMessage) imessage);
|
||||||
|
boolean seen = helper.getSeen();
|
||||||
|
|
||||||
|
TupleMessageEx message = new TupleMessageEx();
|
||||||
|
message.id = uid;
|
||||||
|
message.account = folder.account;
|
||||||
|
message.folder = folder.id;
|
||||||
|
message.uid = uid;
|
||||||
|
message.msgid = helper.getMessageID();
|
||||||
|
message.references = TextUtils.join(" ", helper.getReferences());
|
||||||
|
message.inreplyto = helper.getInReplyTo();
|
||||||
|
message.thread = helper.getThreadId(uid);
|
||||||
|
message.from = helper.getFrom();
|
||||||
|
message.to = helper.getTo();
|
||||||
|
message.cc = helper.getCc();
|
||||||
|
message.bcc = helper.getBcc();
|
||||||
|
message.reply = helper.getReply();
|
||||||
|
message.subject = imessage.getSubject();
|
||||||
|
message.received = imessage.getReceivedDate().getTime();
|
||||||
|
message.sent = (imessage.getSentDate() == null ? null : imessage.getSentDate().getTime());
|
||||||
|
message.seen = seen;
|
||||||
|
message.ui_seen = seen;
|
||||||
|
message.ui_hide = false;
|
||||||
|
|
||||||
|
message.accountName = account.name;
|
||||||
|
message.folderName = folder.name;
|
||||||
|
message.folderType = folder.type;
|
||||||
|
message.count = 1;
|
||||||
|
message.unseen = (seen ? 0 : 1);
|
||||||
|
message.attachments = 0;
|
||||||
|
|
||||||
|
message.body = helper.getHtml();
|
||||||
|
message.virtual = true;
|
||||||
|
|
||||||
|
result.messages.add(message);
|
||||||
|
|
||||||
|
cache.put(pos, message);
|
||||||
|
}
|
||||||
|
|
||||||
|
Log.i(Helper.TAG, "SDS result=" + result.messages.size());
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
|
||||||
|
private class SearchResult {
|
||||||
|
int total;
|
||||||
|
List<TupleMessageEx> messages;
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Reference in new issue