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
|
/* -*- Mode: C++; tab-width: 2; 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 "nsMsgComposeContentHandler.h"
#include "nsMsgComposeService.h"
#include "nsIChannel.h"
#include "nsIURI.h"
#include "plstr.h"
#include "nsServiceManagerUtils.h"
#include "nsCOMPtr.h"
#include "nsPIDOMWindow.h"
#include "mozIDOMWindow.h"
#include "mozilla/dom/Document.h"
#include "nsIInterfaceRequestor.h"
#include "nsIInterfaceRequestorUtils.h"
#include "nsIMsgMailNewsUrl.h"
#include "nsNetUtil.h"
#include "nsIMsgFolder.h"
#include "nsIMsgIncomingServer.h"
#include "nsIMsgAccountManager.h"
#define NS_MSGCOMPOSESERVICE_CID \
{ /* 588595FE-1ADA-11d3-A715-0060B0EB39B5 */ \
0x588595fe, 0x1ada, 0x11d3, { \
0xa7, 0x15, 0x0, 0x60, 0xb0, 0xeb, 0x39, 0xb5 \
} \
}
static NS_DEFINE_CID(kMsgComposeServiceCID, NS_MSGCOMPOSESERVICE_CID);
nsMsgComposeContentHandler::nsMsgComposeContentHandler() {}
// The following macro actually implement addref, release and query interface
// for our component.
NS_IMPL_ISUPPORTS(nsMsgComposeContentHandler, nsIContentHandler)
nsMsgComposeContentHandler::~nsMsgComposeContentHandler() {}
// Try to get an appropriate nsIMsgIdentity by going through the window, getting
// the document's URI, then the corresponding nsIMsgDBHdr. Then find the server
// associated with that header and get the first identity for it.
nsresult nsMsgComposeContentHandler::GetBestIdentity(
nsIInterfaceRequestor* aWindowContext, nsIMsgIdentity** aIdentity) {
nsresult rv;
nsCOMPtr<mozIDOMWindowProxy> domWindow = do_GetInterface(aWindowContext);
NS_ENSURE_TRUE(domWindow, NS_ERROR_FAILURE);
nsCOMPtr<nsPIDOMWindowOuter> window = nsPIDOMWindowOuter::From(domWindow);
nsAutoString documentURIString;
rv = window->GetDoc()->GetDocumentURI(documentURIString);
NS_ENSURE_SUCCESS(rv, rv);
nsCOMPtr<nsIURI> documentURI;
rv = NS_NewURI(getter_AddRefs(documentURI), documentURIString);
NS_ENSURE_SUCCESS(rv, rv);
nsCOMPtr<nsIMsgMessageUrl> msgURI = do_QueryInterface(documentURI);
if (!msgURI) return NS_ERROR_FAILURE;
nsCOMPtr<nsIMsgDBHdr> msgHdr;
rv = msgURI->GetMessageHeader(getter_AddRefs(msgHdr));
NS_ENSURE_SUCCESS(rv, rv);
nsCOMPtr<nsIMsgFolder> folder;
rv = msgHdr->GetFolder(getter_AddRefs(folder));
NS_ENSURE_SUCCESS(rv, rv);
// nsIMsgDBHdrs from .eml messages have a null folder, so bail out if that's
// the case.
if (!folder) return NS_ERROR_FAILURE;
nsCOMPtr<nsIMsgIncomingServer> server;
rv = folder->GetServer(getter_AddRefs(server));
NS_ENSURE_SUCCESS(rv, rv);
nsCOMPtr<nsIMsgAccountManager> accountManager =
do_GetService("@mozilla.org/messenger/account-manager;1", &rv);
NS_ENSURE_SUCCESS(rv, rv);
rv = accountManager->GetFirstIdentityForServer(server, aIdentity);
NS_ENSURE_SUCCESS(rv, rv);
return rv;
}
NS_IMETHODIMP nsMsgComposeContentHandler::HandleContent(
const char* aContentType, nsIInterfaceRequestor* aWindowContext,
nsIRequest* request) {
nsresult rv = NS_OK;
if (!request) return NS_ERROR_NULL_POINTER;
// First of all, get the content type and make sure it is a content type we
// know how to handle!
if (PL_strcasecmp(aContentType, "application/x-mailto") == 0) {
nsCOMPtr<nsIMsgIdentity> identity;
if (aWindowContext)
GetBestIdentity(aWindowContext, getter_AddRefs(identity));
nsCOMPtr<nsIURI> aUri;
nsCOMPtr<nsIChannel> aChannel = do_QueryInterface(request);
if (!aChannel) return NS_ERROR_FAILURE;
rv = aChannel->GetURI(getter_AddRefs(aUri));
if (aUri) {
nsCOMPtr<nsIMsgComposeService> composeService =
do_GetService(kMsgComposeServiceCID, &rv);
if (NS_SUCCEEDED(rv))
rv = composeService->OpenComposeWindowWithURI(nullptr, aUri, identity);
}
} else {
// The content-type was not application/x-mailto...
return NS_ERROR_WONT_HANDLE_CONTENT;
}
return rv;
}
|
