1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
|
/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* 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 "nsCopyMessageStreamListener.h"
#include "nsIMsgMailNewsUrl.h"
#include "nsIMailboxUrl.h"
#include "nsIMsgHdr.h"
#include "nsIMsgImapMailFolder.h"
#include "nsIMsgMessageService.h"
#include "nsMsgUtils.h"
#include "netCore.h"
NS_IMPL_ISUPPORTS(nsCopyMessageStreamListener, nsIStreamListener,
nsIRequestObserver, nsICopyMessageStreamListener)
static nsresult GetMessage(nsIURI *aURL, nsIMsgDBHdr **message)
{
NS_ENSURE_ARG_POINTER(message);
nsCOMPtr<nsIMsgMessageUrl> uriURL;
nsresult rv;
//Need to get message we are about to copy
uriURL = do_QueryInterface(aURL, &rv);
if(NS_FAILED(rv))
return rv;
// get the uri. first try and use the original message spec
// if that fails, use the spec of nsIURI that we're called with
nsCString uri;
rv = uriURL->GetOriginalSpec(getter_Copies(uri));
if (NS_FAILED(rv) || uri.IsEmpty()) {
rv = uriURL->GetUri(getter_Copies(uri));
NS_ENSURE_SUCCESS(rv,rv);
}
nsCOMPtr <nsIMsgMessageService> msgMessageService;
rv = GetMessageServiceFromURI(uri, getter_AddRefs(msgMessageService));
NS_ENSURE_SUCCESS(rv,rv);
if (!msgMessageService)
return NS_ERROR_FAILURE;
rv = msgMessageService->MessageURIToMsgHdr(uri.get(), message);
return rv;
}
nsCopyMessageStreamListener::nsCopyMessageStreamListener()
{
}
nsCopyMessageStreamListener::~nsCopyMessageStreamListener()
{
//All member variables are nsCOMPtr's.
}
NS_IMETHODIMP nsCopyMessageStreamListener::Init(nsIMsgFolder *srcFolder, nsICopyMessageListener *destination, nsISupports *listenerData)
{
mSrcFolder = srcFolder;
mDestination = destination;
mListenerData = listenerData;
return NS_OK;
}
NS_IMETHODIMP nsCopyMessageStreamListener::StartMessage()
{
if (mDestination)
mDestination->StartMessage();
return NS_OK;
}
NS_IMETHODIMP nsCopyMessageStreamListener::EndMessage(nsMsgKey key)
{
if (mDestination)
mDestination->EndMessage(key);
return NS_OK;
}
NS_IMETHODIMP nsCopyMessageStreamListener::OnDataAvailable(nsIRequest * /* request */, nsISupports *ctxt, nsIInputStream *aIStream, uint64_t sourceOffset, uint32_t aLength)
{
nsresult rv;
rv = mDestination->CopyData(aIStream, aLength);
return rv;
}
NS_IMETHODIMP nsCopyMessageStreamListener::OnStartRequest(nsIRequest * request, nsISupports *ctxt)
{
nsCOMPtr<nsIMsgDBHdr> message;
nsresult rv = NS_OK;
nsCOMPtr<nsIURI> uri = do_QueryInterface(ctxt, &rv);
NS_ASSERTION(NS_SUCCEEDED(rv), "ahah...someone didn't pass in the expected context!!!");
if (NS_SUCCEEDED(rv))
rv = GetMessage(uri, getter_AddRefs(message));
if(NS_SUCCEEDED(rv))
rv = mDestination->BeginCopy(message);
NS_ENSURE_SUCCESS(rv, rv);
return rv;
}
NS_IMETHODIMP nsCopyMessageStreamListener::EndCopy(nsISupports *url, nsresult aStatus)
{
nsresult rv;
nsCOMPtr<nsIURI> uri = do_QueryInterface(url, &rv);
NS_ENSURE_SUCCESS(rv, rv);
bool copySucceeded = (aStatus == NS_BINDING_SUCCEEDED);
rv = mDestination->EndCopy(copySucceeded);
//If this is a move and we finished the copy, delete the old message.
bool moveMessage = false;
nsCOMPtr<nsIMsgMailNewsUrl> mailURL(do_QueryInterface(uri));
if (mailURL)
rv = mailURL->IsUrlType(nsIMsgMailNewsUrl::eMove, &moveMessage);
if (NS_FAILED(rv))
moveMessage = false;
// OK, this is wrong if we're moving to an imap folder, for example. This really says that
// we were able to pull the message from the source, NOT that we were able to
// put it in the destination!
if (moveMessage)
{
// don't do this if we're moving to an imap folder - that's handled elsewhere.
nsCOMPtr<nsIMsgImapMailFolder> destImap = do_QueryInterface(mDestination);
// if the destination is a local folder, it will handle the delete from the source in EndMove
if (!destImap)
rv = mDestination->EndMove(copySucceeded);
}
// Even if the above actions failed we probably still want to return NS_OK.
// There should probably be some error dialog if either the copy or delete failed.
return NS_OK;
}
NS_IMETHODIMP nsCopyMessageStreamListener::OnStopRequest(nsIRequest* request, nsISupports *ctxt, nsresult aStatus)
{
return EndCopy(ctxt, aStatus);
}
|
