address collecter, not part of build

mailnews/addrbook/public/nsIAbAddressCollecter.idl

@@ -0,0 +1,28 @@
+/* -*- Mode: IDL; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*-
+ *
+ * The contents of this file are subject to the Netscape Public License
+ * Version 1.0 (the "NPL"); you may not use this file except in
+ * compliance with the NPL.  You may obtain a copy of the NPL at
+ * http://www.mozilla.org/NPL/
+ *
+ * Software distributed under the NPL is distributed on an "AS IS" basis,
+ * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the NPL
+ * for the specific language governing rights and limitations under the
+ * NPL.
+ *
+ * The Initial Developer of this code under the NPL is Netscape
+ * Communications Corporation.  Portions created by Netscape are
+ * Copyright (C) 1999 Netscape Communications Corporation.  All Rights
+ * Reserved.
+ */
+
+#include "nsISupports.idl"
+
+
+[scriptable, uuid(fe04c8e6-501e-11d3-a527-0060b0fc04b7)]
+
+interface nsIAbAddressCollecter : nsISupports {
+
+  void CollectAddress(in string address);
+};
+

mailnews/addrbook/src/nsAbAddressCollecter.cpp

@@ -0,0 +1,142 @@
+/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*-
+ *
+ * The contents of this file are subject to the Netscape Public License
+ * Version 1.0 (the "NPL"); you may not use this file except in
+ * compliance with the NPL.  You may obtain a copy of the NPL at
+ * http://www.mozilla.org/NPL/
+ *
+ * Software distributed under the NPL is distributed on an "AS IS" basis,
+ * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the NPL
+ * for the specific language governing rights and limitations under the
+ * NPL.
+ *
+ * The Initial Developer of this code under the NPL is Netscape
+ * Communications Corporation.  Portions created by Netscape are
+ * Copyright (C) 1999 Netscape Communications Corporation.  All Rights
+ * Reserved.
+ */
+
+#include "msgCore.h"  // for pre-compiled headers
+
+#include "nsIServiceManager.h"
+
+#include "nsIAbCard.h"
+#include "nsAbBaseCID.h"
+#include "nsAbAddressCollecter.h"
+#include "nsIPref.h"
+#include "nsIAddrBookSession.h"
+#include "nsIMsgHeaderParser.h"
+  
+  // For the new pref API's
+static NS_DEFINE_CID(kPrefCID, NS_PREF_CID);
+static NS_DEFINE_CID(kAbCardPropertyCID, NS_ABCARDPROPERTY_CID);
+static NS_DEFINE_CID(kAddrBookSessionCID, NS_ADDRBOOKSESSION_CID);
+static NS_DEFINE_CID(kAddressBookDBCID, NS_ADDRESSBOOKDB_CID);
+static NS_DEFINE_CID(kMsgHeaderParserCID,		NS_MSGHEADERPARSER_CID); 
+
+NS_IMPL_ISUPPORTS(nsAbAddressCollecter, nsCOMTypeInfo<nsIAbAddressCollecter>::GetIID());
+
+nsAbAddressCollecter::nsAbAddressCollecter()
+{
+	NS_INIT_REFCNT();
+}
+
+nsAbAddressCollecter::~nsAbAddressCollecter()
+{
+	if (m_historyAB)
+	{
+		m_historyAB->Commit(kSessionCommit);
+		m_historyAB->Close(PR_FALSE);
+	}
+}
+
+NS_IMETHODIMP nsAbAddressCollecter::CollectAddress(const char *address)
+{
+	nsresult rv;
+	PRBool collectAddresses;
+
+    NS_WITH_SERVICE(nsIPref, pPref, kPrefCID, &rv); 
+    if (NS_FAILED(rv) || !pPref) 
+		return NS_ERROR_FAILURE;
+
+    rv = pPref->GetBoolPref("mail.collect_email_address", &collectAddresses);
+	if (!NS_SUCCEEDED(rv) || !collectAddresses)
+		return rv;
+
+	if (!m_historyAB)
+	{
+		rv = OpenHistoryAB(getter_AddRefs(m_historyAB));
+		if (!NS_SUCCEEDED(rv) || !m_historyAB)
+			return rv;
+	}
+	// note that we're now setting the whole recipient list,
+	// not just the pretty name of the first recipient.
+	PRUint32 numAddresses;
+	char	*names;
+	char	*addresses;
+
+	nsCOMPtr<nsIMsgHeaderParser>  pHeader;
+
+	nsresult res = nsComponentManager::CreateInstance(kMsgHeaderParserCID, 
+                                                    NULL, nsIMsgHeaderParser::GetIID(), 
+                                                    (void **) getter_AddRefs(pHeader)); 
+
+	nsresult ret = pHeader->ParseHeaderAddresses (nsnull, address, &names, &addresses, &numAddresses);
+	if (ret == NS_OK)
+	{
+		char *curName = names;
+		char *curAddress = addresses;
+		char *excludeDomainList = nsnull;
+		PRUint32 numAddresses;
+
+		rv = pPref->CopyCharPref("mail.address_collection_ignore_domain_list", &excludeDomainList);
+		if (NS_SUCCEEDED(rv)) 
+		{
+			for (PRUint32 i = 0; i < numAddresses; i++)
+			{
+				nsCOMPtr<nsIAbCard> senderCard;
+				rv = nsComponentManager::CreateInstance(kAbCardPropertyCID, nsnull, nsCOMTypeInfo<nsIAbCard>::GetIID(), getter_AddRefs(senderCard));
+				if (NS_SUCCEEDED(rv) && senderCard)
+				{
+					senderCard->SetDisplayName(curName);
+					senderCard->SetPrimaryEmail(curAddress);
+					senderCard->AddCardToDatabase("abdirectory://history.mab");
+				}
+				curName += strlen(curName) + 1;
+				curAddress += strlen(curAddress) + 1;
+			}
+			PR_FREEIF(addresses);
+			PR_FREEIF(names);
+			PR_FREEIF(excludeDomainList);
+		}
+	}
+
+// pref("mail.address_collection_ignore_domain_list","");
+
+	return NS_OK;
+}
+
+
+nsresult nsAbAddressCollecter::OpenHistoryAB(nsIAddrDatabase **aDatabase)
+{
+	if (!aDatabase)
+		return NS_ERROR_NULL_POINTER;
+
+	nsresult rv = NS_OK;
+	nsFileSpec* dbPath = nsnull;
+
+	NS_WITH_SERVICE(nsIAddrBookSession, abSession, kAddrBookSessionCID, &rv); 
+	if(NS_SUCCEEDED(rv))
+		abSession->GetUserProfileDirectory(&dbPath);
+	
+	if (dbPath)
+	{
+		(*dbPath) += "Collected Addresses.mab";
+
+		NS_WITH_SERVICE(nsIAddrDatabase, addrDBFactory, kAddressBookDBCID, &rv);
+
+		if (NS_SUCCEEDED(rv) && addrDBFactory)
+			rv = addrDBFactory->Open(dbPath, PR_FALSE, aDatabase, PR_TRUE);
+	}
+	return rv;
+}

mailnews/addrbook/src/nsAbAddressCollecter.h

@@ -0,0 +1,44 @@
+/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*-
+ *
+ * The contents of this file are subject to the Netscape Public License
+ * Version 1.0 (the "NPL"); you may not use this file except in
+ * compliance with the NPL.  You may obtain a copy of the NPL at
+ * http://www.mozilla.org/NPL/
+ *
+ * Software distributed under the NPL is distributed on an "AS IS" basis,
+ * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the NPL
+ * for the specific language governing rights and limitations under the
+ * NPL.
+ *
+ * The Initial Developer of this code under the NPL is Netscape
+ * Communications Corporation.  Portions created by Netscape are
+ * Copyright (C) 1999 Netscape Communications Corporation.  All Rights
+ * Reserved.
+ */
+
+#ifndef _nsAbAddressCollecter_H_
+#define _nsAbAddressCollecter_H_
+
+#include "nsIAbAddressCollecter.h"
+#include "nsCOMPtr.h"
+#include "nsIAbAddressCollecter.h"
+#include "nsIAddrDatabase.h"
+#include "nsAddrDatabase.h"
+
+class nsAbAddressCollecter : public nsIAbAddressCollecter
+{
+public:
+	nsAbAddressCollecter();
+	virtual ~nsAbAddressCollecter();
+
+	NS_DECL_ISUPPORTS
+	NS_IMETHOD	CollectAddress(const char *address);
+
+	nsresult OpenHistoryAB(nsIAddrDatabase **aDatabase);
+protected:
+	nsCOMPtr <nsIAddrDatabase> m_historyAB;
+
+};
+
+#endif  // _nsAbAddressCollecter_H_
+