/* 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
 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

#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 "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(Message& 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(Message& aMessage) { return aMessage.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(Message& aMessage) {
    return mFolderIds.Contains(aMessage.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(Message& message) {
    // TODO: This is incomplete. We haven't matched the message against the
    // search terms.
    return mSearchFolderIds.Contains(message.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(Message& message) {
    nsMsgKey threadId;
    message.GetThreadId(&threadId);
    return threadId == mConversationId;
  }

 protected:
  uint64_t mConversationId;
};

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));
  }

 protected:
  nsAutoCString mTag;
  bool mWanted;
};

}  // namespace mozilla::mailnews

#endif  // COMM_MAILNEWS_DB_PANORAMA_SRC_LIVEVIEWFILTERS_H_