Revision control
Copy as Markdown
Other Tools
/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* vim: set ts=8 sts=2 et sw=2 tw=80: */
/* 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
#include "mozilla/mailnews/MimeHeaderParser.h"
#include "mozilla/DebugOnly.h"
#include "nsMemory.h"
#include "nsCOMPtr.h"
#include "nsIMsgHeaderParser.h"
#include "mozilla/Components.h"
namespace mozilla {
namespace mailnews {
void detail::DoConversion(const nsTArray<nsString>& aUTF16Array,
nsTArray<nsCString>& aUTF8Array) {
uint32_t count = aUTF16Array.Length();
aUTF8Array.SetLength(count);
for (uint32_t i = 0; i < count; ++i)
CopyUTF16toUTF8(aUTF16Array[i], aUTF8Array[i]);
}
void MakeMimeAddress(const nsACString& aName, const nsACString& aEmail,
nsACString& full) {
nsAutoString utf16Address;
MakeMimeAddress(NS_ConvertUTF8toUTF16(aName), NS_ConvertUTF8toUTF16(aEmail),
utf16Address);
CopyUTF16toUTF8(utf16Address, full);
}
void MakeMimeAddress(const nsAString& aName, const nsAString& aEmail,
nsAString& full) {
nsCOMPtr<nsIMsgHeaderParser> headerParser(
components::HeaderParser::Service());
nsCOMPtr<msgIAddressObject> address;
headerParser->MakeMailboxObject(aName, aEmail, getter_AddRefs(address));
nsTArray<RefPtr<msgIAddressObject>> addresses;
addresses.AppendElement(address);
headerParser->MakeMimeHeader(addresses, full);
}
void MakeDisplayAddress(const nsAString& aName, const nsAString& aEmail,
nsAString& full) {
nsCOMPtr<nsIMsgHeaderParser> headerParser(
components::HeaderParser::Service());
nsCOMPtr<msgIAddressObject> object;
headerParser->MakeMailboxObject(aName, aEmail, getter_AddRefs(object));
object->ToString(full);
}
void RemoveDuplicateAddresses(const nsACString& aHeader,
const nsACString& aOtherEmails,
nsACString& result) {
nsCOMPtr<nsIMsgHeaderParser> headerParser(
components::HeaderParser::Service());
headerParser->RemoveDuplicateAddresses(aHeader, aOtherEmails, result);
}
/////////////////////////////////////////////
// These are the core shim methods we need //
/////////////////////////////////////////////
nsCOMArray<msgIAddressObject> DecodedHeader(const nsAString& aHeader) {
nsCOMArray<msgIAddressObject> retval;
if (aHeader.IsEmpty()) {
return retval;
}
nsCOMPtr<nsIMsgHeaderParser> headerParser(
components::HeaderParser::Service());
NS_ENSURE_TRUE(headerParser, retval);
nsTArray<RefPtr<msgIAddressObject>> addresses;
nsresult rv = headerParser->ParseDecodedHeader(aHeader, false, addresses);
MOZ_ASSERT(NS_SUCCEEDED(rv), "Javascript jsmime returned an error!");
if (NS_SUCCEEDED(rv) && addresses.Length() > 0) {
retval.SetCapacity(addresses.Length());
for (auto& addr : addresses) {
retval.AppendElement(addr);
}
}
return retval;
}
nsCOMArray<msgIAddressObject> EncodedHeader(const nsACString& aHeader,
const char* aCharset) {
nsCOMArray<msgIAddressObject> retval;
if (aHeader.IsEmpty()) {
return retval;
}
nsCOMPtr<nsIMsgHeaderParser> headerParser(
components::HeaderParser::Service());
NS_ENSURE_TRUE(headerParser, retval);
nsTArray<RefPtr<msgIAddressObject>> addresses;
nsresult rv =
headerParser->ParseEncodedHeader(aHeader, aCharset, false, addresses);
MOZ_ASSERT(NS_SUCCEEDED(rv), "This should never fail!");
if (NS_SUCCEEDED(rv) && addresses.Length() > 0) {
retval.SetCapacity(addresses.Length());
for (auto& addr : addresses) {
retval.AppendElement(addr);
}
}
return retval;
}
nsCOMArray<msgIAddressObject> EncodedHeaderW(const nsAString& aHeader) {
nsCOMArray<msgIAddressObject> retval;
if (aHeader.IsEmpty()) {
return retval;
}
nsCOMPtr<nsIMsgHeaderParser> headerParser(
components::HeaderParser::Service());
NS_ENSURE_TRUE(headerParser, retval);
nsTArray<RefPtr<msgIAddressObject>> addresses;
nsresult rv = headerParser->ParseEncodedHeaderW(aHeader, addresses);
MOZ_ASSERT(NS_SUCCEEDED(rv), "This should never fail!");
if (NS_SUCCEEDED(rv) && addresses.Length() > 0) {
retval.SetCapacity(addresses.Length());
for (auto& addr : addresses) {
retval.AppendElement(addr);
}
}
return retval;
}
void ExtractAllAddresses(const nsCOMArray<msgIAddressObject>& aHeader,
nsTArray<nsString>& names,
nsTArray<nsString>& emails) {
uint32_t count = aHeader.Length();
// Prefill arrays before we start
names.SetLength(count);
emails.SetLength(count);
for (uint32_t i = 0; i < count; i++) {
aHeader[i]->GetName(names[i]);
aHeader[i]->GetEmail(emails[i]);
}
if (count == 1 && names[0].IsEmpty() && emails[0].IsEmpty()) {
names.Clear();
emails.Clear();
}
}
void ExtractDisplayAddresses(const nsCOMArray<msgIAddressObject>& aHeader,
nsTArray<nsString>& displayAddrs) {
uint32_t count = aHeader.Length();
displayAddrs.SetLength(count);
for (uint32_t i = 0; i < count; i++) aHeader[i]->ToString(displayAddrs[i]);
if (count == 1 && displayAddrs[0].IsEmpty()) displayAddrs.Clear();
}
/////////////////////////////////////////////////
// All of these are based on the above methods //
/////////////////////////////////////////////////
void ExtractEmails(const nsCOMArray<msgIAddressObject>& aHeader,
nsTArray<nsString>& emails) {
nsTArray<nsString> names;
ExtractAllAddresses(aHeader, names, emails);
}
void ExtractEmail(const nsCOMArray<msgIAddressObject>& aHeader,
nsACString& email) {
AutoTArray<nsString, 1> names;
AutoTArray<nsString, 1> emails;
ExtractAllAddresses(aHeader, names, emails);
if (emails.Length() > 0)
CopyUTF16toUTF8(emails[0], email);
else
email.Truncate();
}
void ExtractFirstAddress(const nsCOMArray<msgIAddressObject>& aHeader,
nsACString& name, nsACString& email) {
AutoTArray<nsString, 1> names, emails;
ExtractAllAddresses(aHeader, names, emails);
if (names.Length() > 0) {
CopyUTF16toUTF8(names[0], name);
CopyUTF16toUTF8(emails[0], email);
} else {
name.Truncate();
email.Truncate();
}
}
void ExtractFirstAddress(const nsCOMArray<msgIAddressObject>& aHeader,
nsAString& name, nsACString& email) {
AutoTArray<nsString, 1> names, emails;
ExtractAllAddresses(aHeader, names, emails);
if (names.Length() > 0) {
name = names[0];
CopyUTF16toUTF8(emails[0], email);
} else {
name.Truncate();
email.Truncate();
}
}
void ExtractName(const nsCOMArray<msgIAddressObject>& aHeader,
nsACString& name) {
nsCString email;
ExtractFirstAddress(aHeader, name, email);
if (name.IsEmpty()) name = email;
}
void ExtractName(const nsCOMArray<msgIAddressObject>& aHeader,
nsAString& name) {
AutoTArray<nsString, 1> names;
AutoTArray<nsString, 1> emails;
ExtractAllAddresses(aHeader, names, emails);
if (names.Length() > 0) {
if (names[0].IsEmpty())
name = emails[0];
else
name = names[0];
} else {
name.Truncate();
}
}
} // namespace mailnews
} // namespace mozilla