Source code
Revision control
Copy as Markdown
Other Tools
/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
#ifndef COMM_MAILNEWS_DB_PANORAMA_SRC_LIVEVIEWFILTERS_H_
#define COMM_MAILNEWS_DB_PANORAMA_SRC_LIVEVIEWFILTERS_H_
#include "FolderDatabase.h"
#include "Message.h"
#include "mozilla/Components.h"
#include "mozilla/RefPtr.h"
#include "mozilla/storage/Variant.h"
#include "mozIStorageStatement.h"
#include "nsCOMPtr.h"
#include "nsIMsgHdr.h"
#include "nsIVariant.h"
#include "nsMsgMessageFlags.h"
#include "nsString.h"
#include "nsTArray.h"
#include "nsTString.h"
#include "VirtualFolderWrapper.h"
namespace mozilla::mailnews {
class LiveViewFilter {
friend class LiveView;
public:
LiveViewFilter() {}
virtual ~LiveViewFilter() {}
virtual bool Matches(nsIMsgDBHdr* aMessage) { return false; }
virtual void Refresh() {}
protected:
nsAutoCString mSQLClause;
nsTArray<nsCOMPtr<nsIVariant>> mSQLParams;
};
class SingleFolderFilter final : public LiveViewFilter {
public:
explicit SingleFolderFilter(uint64_t folderId) : mFolderId(folderId) {
mSQLClause.Assign("folderId = ");
mSQLClause.AppendInt(mFolderId);
}
bool Matches(nsIMsgDBHdr* message) {
nsCOMPtr<nsIMsgFolder> folder;
message->GetFolder(getter_AddRefs(folder));
uint64_t folderId;
folder->GetId(&folderId);
return folderId == mFolderId;
}
protected:
uint64_t mFolderId;
};
class MultiFolderFilter final : public LiveViewFilter {
public:
explicit MultiFolderFilter(nsTArray<uint64_t> const& folderIds) {
mFolderIds.ClearAndRetainStorage();
mSQLClause.Assign("folderId IN (");
for (size_t i = 0; i < folderIds.Length(); i++) {
if (i > 0) {
mSQLClause.Append(", ");
}
mSQLClause.AppendInt(folderIds[i]);
mFolderIds.AppendElement(folderIds[i]);
}
mSQLClause.Append(")");
}
bool Matches(nsIMsgDBHdr* message) {
nsCOMPtr<nsIMsgFolder> folder;
message->GetFolder(getter_AddRefs(folder));
uint64_t folderId;
folder->GetId(&folderId);
return mFolderIds.Contains(folderId);
}
protected:
nsTArray<uint64_t> mFolderIds;
};
class VirtualFolderFilter final : public LiveViewFilter {
public:
explicit VirtualFolderFilter(uint64_t folderId) : mVirtualFolderId(folderId) {
mWrapper = new VirtualFolderWrapper(folderId);
Refresh();
}
void Refresh();
bool Matches(nsIMsgDBHdr* message) {
nsCOMPtr<nsIMsgFolder> folder;
message->GetFolder(getter_AddRefs(folder));
uint64_t folderId;
folder->GetId(&folderId);
// TODO: This is incomplete. We haven't matched the message against the
// search terms.
return mSearchFolderIds.Contains(folderId);
}
protected:
uint64_t mVirtualFolderId;
nsTArray<uint64_t> mSearchFolderIds;
RefPtr<VirtualFolderWrapper> mWrapper;
};
class ConversationFilter final : public LiveViewFilter {
public:
explicit ConversationFilter(uint64_t conversationId)
: mConversationId(conversationId) {
mSQLClause.Assign("threadId = ");
mSQLClause.AppendInt(conversationId);
}
bool Matches(nsIMsgDBHdr* message) {
nsMsgKey threadId;
message->GetThreadId(&threadId);
return threadId == mConversationId;
}
protected:
uint64_t mConversationId;
};
class MessageFlagsFilter : public LiveViewFilter {
public:
MessageFlagsFilter(uint64_t aFlag, uint64_t aWanted)
: mFlag(aFlag), mWanted(aWanted) {
mSQLClause.Assign("flags & ");
mSQLClause.AppendInt(aFlag);
mSQLClause.Append(" = ");
mSQLClause.AppendInt(aWanted);
}
bool Matches(nsIMsgDBHdr* message) {
uint32_t flags;
message->GetFlags(&flags);
return (flags & mFlag) == mWanted;
}
protected:
uint64_t mFlag;
uint64_t mWanted;
};
class UnreadMessagesFilter final : public MessageFlagsFilter {
public:
UnreadMessagesFilter() : MessageFlagsFilter(nsMsgMessageFlags::Read, 0) {}
};
class FlaggedMessagesFilter final : public MessageFlagsFilter {
public:
FlaggedMessagesFilter()
: MessageFlagsFilter(nsMsgMessageFlags::Marked,
nsMsgMessageFlags::Marked) {}
};
class TaggedMessagesFilter final : public LiveViewFilter {
public:
explicit TaggedMessagesFilter(const nsACString& aTag, bool aWanted)
: mTag(aTag), mWanted(aWanted) {
mSQLClause.Assign(aWanted ? "TAGS_INCLUDE(tags, ?)"
: "TAGS_EXCLUDE(tags, ?)");
mSQLParams.AppendElement(new mozilla::storage::UTF8TextVariant(mTag));
}
bool Matches(nsIMsgDBHdr* message);
protected:
nsAutoCString mTag;
bool mWanted;
};
class DateRangeFilter final : public LiveViewFilter {
public:
explicit DateRangeFilter(PRTime aStartDate, PRTime aEndDate)
: mStartDate(aStartDate), mEndDate(aEndDate) {
mSQLClause.Assign("date BETWEEN ? AND ?");
mSQLParams.AppendElement(new mozilla::storage::IntegerVariant(mStartDate));
mSQLParams.AppendElement(new mozilla::storage::IntegerVariant(mEndDate));
}
bool Matches(nsIMsgDBHdr* message) {
PRTime date;
message->GetDate(&date);
return date >= mStartDate && date <= mEndDate;
}
protected:
PRTime mStartDate;
PRTime mEndDate;
};
} // namespace mozilla::mailnews
#endif // COMM_MAILNEWS_DB_PANORAMA_SRC_LIVEVIEWFILTERS_H_