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
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
#include "EwsCopyMoveTransaction.h"
#include "nsIMessenger.h"
#include "nsIMsgFolder.h"
namespace {
class UpdateHeaderSetListener : public IEwsFolderOperationListener {
public:
NS_DECL_IEWSFOLDEROPERATIONLISTENER
NS_DECL_ISUPPORTS
explicit UpdateHeaderSetListener(RefPtr<EwsCopyMoveTransaction> transaction)
: mTransaction(std::move(transaction)) {}
protected:
virtual ~UpdateHeaderSetListener() = default;
private:
RefPtr<EwsCopyMoveTransaction> mTransaction;
};
NS_IMPL_ISUPPORTS(UpdateHeaderSetListener, IEwsFolderOperationListener);
NS_IMETHODIMP UpdateHeaderSetListener::OnComplete(
const nsTArray<RefPtr<nsIMsgDBHdr>>& headers) {
mTransaction->UpdateHeaderSet(headers);
return NS_OK;
}
} // namespace
RefPtr<EwsCopyMoveTransaction> EwsCopyMoveTransaction::ForCopy(
nsCOMPtr<IEwsFolder> originalSourceFolder,
nsCOMPtr<IEwsFolder> originalDestinationFolder,
nsCOMPtr<nsIMsgWindow> window,
nsTArray<RefPtr<nsIMsgDBHdr>> originalHeaders,
nsTArray<RefPtr<nsIMsgDBHdr>> newHeaders) {
return new EwsCopyMoveTransaction(
std::move(originalSourceFolder), std::move(originalDestinationFolder),
std::move(window), false, std::move(originalHeaders),
std::move(newHeaders));
}
RefPtr<EwsCopyMoveTransaction> EwsCopyMoveTransaction::ForMove(
nsCOMPtr<IEwsFolder> originalSourceFolder,
nsCOMPtr<IEwsFolder> originalDestinationFolder,
nsCOMPtr<nsIMsgWindow> window, nsTArray<RefPtr<nsIMsgDBHdr>> newHeaders) {
// The move case has no need to hold on to a reference for the original
// headers.
return new EwsCopyMoveTransaction(
std::move(originalSourceFolder), std::move(originalDestinationFolder),
std::move(window), true, nsTArray<RefPtr<nsIMsgDBHdr>>(),
std::move(newHeaders));
}
EwsCopyMoveTransaction::EwsCopyMoveTransaction(
nsCOMPtr<IEwsFolder> originalSourceFolder,
nsCOMPtr<IEwsFolder> originalDestinationFolder,
nsCOMPtr<nsIMsgWindow> window, bool isMove,
nsTArray<RefPtr<nsIMsgDBHdr>>&& originalHeaders,
nsTArray<RefPtr<nsIMsgDBHdr>>&& newHeaders)
: mOriginalSourceFolder(std::move(originalSourceFolder)),
mOriginalDestinationFolder(std::move(originalDestinationFolder)),
mWindow(std::move(window)),
mIsMove(isMove),
mOriginalHeaderSet(std::move(originalHeaders)),
mCurrentHeaderSet(std::move(newHeaders)) {}
EwsCopyMoveTransaction::~EwsCopyMoveTransaction() = default;
NS_IMETHODIMP EwsCopyMoveTransaction::UndoTransaction() {
if (mIsMove) {
return PerformOperation(mOriginalDestinationFolder, mOriginalSourceFolder);
}
nsresult rv;
nsCOMPtr<nsIMsgFolder> msgFolder =
do_QueryInterface(mOriginalDestinationFolder, &rv);
NS_ENSURE_SUCCESS(rv, rv);
rv = msgFolder->DeleteMessages(mCurrentHeaderSet, mWindow, true, false,
nullptr, false);
NS_ENSURE_SUCCESS(rv, rv);
mCurrentHeaderSet.ClearAndRetainStorage();
return NS_OK;
}
NS_IMETHODIMP EwsCopyMoveTransaction::RedoTransaction() {
return PerformOperation(mOriginalSourceFolder, mOriginalDestinationFolder);
}
void EwsCopyMoveTransaction::UpdateHeaderSet(
const nsTArray<RefPtr<nsIMsgDBHdr>>& headers) {
mCurrentHeaderSet.ClearAndRetainStorage();
for (auto&& header : headers) {
mCurrentHeaderSet.AppendElement(header);
}
}
nsresult EwsCopyMoveTransaction::PerformOperation(IEwsFolder* fromFolder,
IEwsFolder* toFolder) {
RefPtr<UpdateHeaderSetListener> listener = new UpdateHeaderSetListener(this);
const auto& transactionHeaders =
mIsMove ? mCurrentHeaderSet : mOriginalHeaderSet;
// We can pass `eUnknown` as the undo operation type because we're passing
// another argument to disallow undo for this operation.
nsresult rv = toFolder->CopyItemsOnSameServer(
fromFolder, transactionHeaders, mIsMove, mWindow, nullptr, false,
nsIMessenger::eUnknown, listener);
NS_ENSURE_SUCCESS(rv, rv);
return NS_OK;
}