gecko-dev/mailnews/local/build/nsMsgLocalFactory.cpp

/* -*- Mode: C++; tab-width: 2; 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) 1998 Netscape Communications Corporation.  All Rights
 * Reserved.
 */

#include "msgCore.h" // for pre-compiled headers...
#include "nsIServiceManager.h"
#include "nsIFactory.h"
#include "nsISupports.h"
#include "nsMsgLocalCID.h"
#include "pratom.h"

// include files for components this factory creates...
#include "nsMailboxUrl.h"
#include "nsPop3URL.h"
#include "nsMailboxService.h"
#include "nsLocalMailFolder.h"
#include "nsParseMailbox.h"
#include "nsPop3Service.h"
#include "nsPop3IncomingServer.h"
#include "nsCOMPtr.h"
#include "nsLocalMessage.h"

static NS_DEFINE_CID(kComponentManagerCID, NS_COMPONENTMANAGER_CID);
static NS_DEFINE_CID(kMailboxUrlCID, NS_MAILBOXURL_CID);
static NS_DEFINE_CID(kMailboxParserCID, NS_MAILBOXPARSER_CID);
static NS_DEFINE_CID(kMailboxServiceCID, NS_MAILBOXSERVICE_CID);
static NS_DEFINE_CID(kLocalMailFolderResourceCID, NS_LOCALMAILFOLDERRESOURCE_CID);
static NS_DEFINE_CID(kMailboxMessageResourceCID, NS_MAILBOXMESSAGERESOURCE_CID);
static NS_DEFINE_CID(kPop3ServiceCID, NS_POP3SERVICE_CID);
static NS_DEFINE_CID(kPop3UrlCID, NS_POP3URL_CID);
static NS_DEFINE_CID(kPop3IncomingServerCID, NS_POP3INCOMINGSERVER_CID);
static NS_DEFINE_CID(kParseMailMsgStateCID, NS_PARSEMAILMSGSTATE_CID);

////////////////////////////////////////////////////////////
//
////////////////////////////////////////////////////////////
static PRInt32 g_InstanceCount = 0;
static PRInt32 g_LockCount = 0;

class nsMsgLocalFactory : public nsIFactory
{   
public:
	// nsISupports methods
	NS_DECL_ISUPPORTS 

  nsMsgLocalFactory(const nsCID &aClass, const char* aClassName, const char* aProgID); 

  // nsIFactory methods   
  NS_IMETHOD CreateInstance(nsISupports *aOuter, const nsIID &aIID, void **aResult);   
  NS_IMETHOD LockFactory(PRBool aLock);   

protected:
  virtual ~nsMsgLocalFactory();   

  nsCID mClassID;
  char* mClassName;
  char* mProgID;
};   

nsMsgLocalFactory::nsMsgLocalFactory(const nsCID &aClass, const char* aClassName, const char* aProgID)
  : mClassID(aClass), mClassName(nsCRT::strdup(aClassName)), mProgID(nsCRT::strdup(aProgID))
{   
	NS_INIT_REFCNT();
}   

nsMsgLocalFactory::~nsMsgLocalFactory()   
{
	NS_ASSERTION(mRefCnt == 0, "non-zero refcnt at destruction");   
	PL_strfree(mClassName);
	PL_strfree(mProgID);
}   

nsresult nsMsgLocalFactory::QueryInterface(const nsIID &aIID, void **aResult)   
{   
  if (aResult == nsnull)  
    return NS_ERROR_NULL_POINTER;  

  // Always NULL result, in case of failure   
  *aResult = nsnull;   

  // we support two interfaces....nsISupports and nsFactory.....
  if (aIID.Equals(nsCOMTypeInfo<nsISupports>::GetIID()))    
    *aResult = (void *)(nsISupports*)this;   
  else if (aIID.Equals(nsIFactory::GetIID()))   
    *aResult = (void *)(nsIFactory*)this;   

  if (*aResult == nsnull)
    return NS_NOINTERFACE;

  AddRef(); // Increase reference count for caller   
  return NS_OK;   
}   

NS_IMPL_ADDREF(nsMsgLocalFactory)
NS_IMPL_RELEASE(nsMsgLocalFactory)

nsresult nsMsgLocalFactory::CreateInstance(nsISupports * /* aOuter */, const nsIID &aIID, void **aResult)  
{  
	nsresult rv = NS_OK;

	if (aResult == nsnull)  
		return NS_ERROR_NULL_POINTER;  

	*aResult = nsnull;  
  
	// ClassID check happens here
	// Whenever you add a new class that supports an interface, plug it in here!!!
	
	// do they want a local datasource ?
	if (mClassID.Equals(kMailboxUrlCID)) 
	{
		nsMailboxUrl * url = new nsMailboxUrl();
		if (url)
			rv = url->QueryInterface(aIID, aResult);
		else
			rv = NS_ERROR_OUT_OF_MEMORY;
		
		if (NS_FAILED(rv) && url)
			delete url;
	}
	else if (mClassID.Equals(kPop3UrlCID))
	{
		rv = NS_NewPopUrl(nsIPop3URL::GetIID(), aResult);
	}
	else if (mClassID.Equals(kMailboxParserCID)) 
	{
		nsMsgMailboxParser * parser = new nsMsgMailboxParser();
		if (parser)
			rv =  parser->QueryInterface(aIID, aResult);
		else
			rv = NS_ERROR_OUT_OF_MEMORY;
		
		if (NS_FAILED(rv) && parser)
			delete parser;
	}
	else if (mClassID.Equals(kMailboxServiceCID)) 
	{
		nsMailboxService * mboxService = new nsMailboxService();
		if (mboxService)
			rv = mboxService->QueryInterface(aIID, aResult);
		else
			rv = NS_ERROR_OUT_OF_MEMORY;

		if (NS_FAILED(rv) && mboxService)
			delete mboxService;
	}
	else if (mClassID.Equals(kPop3ServiceCID))
	{
		nsPop3Service * popService = new nsPop3Service();
		if (popService)
			rv = popService->QueryInterface(aIID, aResult);
		else
			rv = NS_ERROR_OUT_OF_MEMORY;

		if (NS_FAILED(rv) && popService)
			delete popService;
	}
	else if (mClassID.Equals(kLocalMailFolderResourceCID)) 
	{
		nsMsgLocalMailFolder * localFolder = new nsMsgLocalMailFolder();
		if (localFolder)
			rv = localFolder->QueryInterface(aIID, aResult);
		else
			rv = NS_ERROR_OUT_OF_MEMORY;

		if (NS_FAILED(rv) && localFolder)
			delete localFolder;
	}
	else if (mClassID.Equals(kMailboxMessageResourceCID)) 
	{
		nsLocalMessage * localMessage = new nsLocalMessage();
		if (localMessage)
			rv = localMessage->QueryInterface(aIID, aResult);
		else
			rv = NS_ERROR_OUT_OF_MEMORY;

		if (NS_FAILED(rv) && localMessage)
			delete localMessage;
 	}
	else if (mClassID.Equals(kParseMailMsgStateCID))
		rv = NS_NewParseMailMessageState(aIID, aResult);
  else if (mClassID.Equals(kPop3IncomingServerCID))
    rv = NS_NewPop3IncomingServer(nsCOMTypeInfo<nsISupports>::GetIID(), aResult);
	
  else
		rv = NS_NOINTERFACE;
  
	return rv;
}  

nsresult nsMsgLocalFactory::LockFactory(PRBool aLock)  
{  
	if (aLock) { 
		PR_AtomicIncrement(&g_LockCount); 
	} else { 
		PR_AtomicDecrement(&g_LockCount); 
	} 

  return NS_OK;
}  

// return the proper factory to the caller. 
extern "C" NS_EXPORT nsresult NSGetFactory(nsISupports* /* aServMgr */,
                                           const nsCID &aClass,
                                           const char *aClassName,
                                           const char *aProgID,
                                           nsIFactory **aFactory)
{
	if (nsnull == aFactory)
		return NS_ERROR_NULL_POINTER;

	// If we decide to implement multiple factories in the msg.dll, then we need to check the class
	// type here and create the appropriate factory instead of always creating a nsMsgFactory...
	*aFactory = new nsMsgLocalFactory(aClass, aClassName, aProgID);

	if (aFactory)
		return (*aFactory)->QueryInterface(nsIFactory::GetIID(), (void**)aFactory); // they want a Factory Interface so give it to them
	else
		return NS_ERROR_OUT_OF_MEMORY;
}

extern "C" NS_EXPORT PRBool NSCanUnload(nsISupports* /* aServMgr */) 
{
    return PRBool(g_InstanceCount == 0 && g_LockCount == 0);
}

extern "C" NS_EXPORT nsresult
NSRegisterSelf(nsISupports* aServMgr, const char* path)
{
  nsresult rv;
  nsresult finalResult = NS_OK;

  NS_WITH_SERVICE1(nsIComponentManager, compMgr, aServMgr, kComponentManagerCID, &rv);
  if (NS_FAILED(rv)) return rv;

  rv = compMgr->RegisterComponent(kMailboxUrlCID, nsnull, nsnull,
                                  path, PR_TRUE, PR_TRUE);
  if (NS_FAILED(rv)) finalResult = rv;

  rv = compMgr->RegisterComponent(kMailboxServiceCID, nsnull, 
								  "component://netscape/messenger/mailboxservice", 
                                  path, PR_TRUE, PR_TRUE);
  if (NS_FAILED(rv)) finalResult = rv;

  rv = compMgr->RegisterComponent(kMailboxServiceCID, nsnull, 
								  "component://netscape/messenger/messageservice;type=mailbox", 
                                  path, PR_TRUE, PR_TRUE);
  if (NS_FAILED(rv)) finalResult = rv;

  rv = compMgr->RegisterComponent(kMailboxServiceCID, nsnull, 
								  "component://netscape/messenger/messageservice;type=mailbox_message", 
                                  path, PR_TRUE, PR_TRUE);
  if (NS_FAILED(rv)) finalResult = rv;

  rv = compMgr->RegisterComponent(kMailboxParserCID, nsnull, nsnull,
                                  path, PR_TRUE, PR_TRUE);
  if (NS_FAILED(rv)) finalResult = rv;

  rv = compMgr->RegisterComponent(kPop3UrlCID, nsnull, nsnull,
								  path, PR_TRUE, PR_TRUE);
  if (NS_FAILED(rv)) finalResult = rv;

  rv = compMgr->RegisterComponent(kPop3ServiceCID, nsnull, 
								  "component://netscape/messenger/popservice",
								  path, PR_TRUE, PR_TRUE);
  if (NS_FAILED(rv)) finalResult = rv;

  // register our RDF resource factories:
  rv = compMgr->RegisterComponent(kLocalMailFolderResourceCID,
                                  "Local Mail Folder Resource Factory",
                                  NS_RDF_RESOURCE_FACTORY_PROGID_PREFIX "mailbox",
                                  path, PR_TRUE, PR_TRUE);
  if (NS_FAILED(rv)) finalResult = rv;

   rv = compMgr->RegisterComponent(kMailboxMessageResourceCID,
                                   "Local Message Resource Factory",
                                   NS_RDF_RESOURCE_FACTORY_PROGID_PREFIX "mailbox_message",
                                   path, PR_TRUE, PR_TRUE);
   if (NS_FAILED(rv)) finalResult = rv;

  rv = compMgr->RegisterComponent(kPop3IncomingServerCID,
                                  "Pop3 Incoming Server",
                                  "component://netscape/messenger/server&type=pop3",
                                  path, PR_TRUE, PR_TRUE);
	
  if (NS_FAILED(rv)) finalResult = rv;
   rv = compMgr->RegisterComponent(kParseMailMsgStateCID,
                                  "Parse MailMessage State",
                                  "component://netscape/messenger/messagestateparser",
                                  path, PR_TRUE, PR_TRUE);
  
                                  
  
  if (NS_FAILED(rv)) finalResult = rv;
  return finalResult;
}

extern "C" NS_EXPORT nsresult
NSUnregisterSelf(nsISupports* aServMgr, const char* path)
{
  nsresult rv = NS_OK;
  nsresult finalResult = NS_OK;

  NS_WITH_SERVICE1(nsIComponentManager, compMgr, aServMgr, kComponentManagerCID, &rv);
  if (NS_FAILED(rv)) return rv;

  rv = compMgr->UnregisterComponent(kMailboxUrlCID, path);
  if (NS_FAILED(rv)) finalResult = rv;

  rv = compMgr->UnregisterComponent(kMailboxServiceCID, path);
  if (NS_FAILED(rv)) finalResult = rv;

  rv = compMgr->UnregisterComponent(kPop3UrlCID, path);
  if (NS_FAILED(rv)) finalResult = rv;

  rv = compMgr->UnregisterComponent(kPop3ServiceCID, path);
  if (NS_FAILED(rv)) finalResult = rv;

  rv = compMgr->UnregisterComponent(kMailboxParserCID, path);
  if (NS_FAILED(rv)) finalResult = rv;

  rv = compMgr->UnregisterComponent(kLocalMailFolderResourceCID, path);
  if (NS_FAILED(rv)) finalResult = rv;

  rv = compMgr->UnregisterComponent(kMailboxMessageResourceCID, path);
  if (NS_FAILED(rv)) finalResult = rv;

  rv = compMgr->UnregisterComponent(kPop3IncomingServerCID, path);
  if (NS_FAILED(rv)) finalResult = rv;

  rv = compMgr->UnregisterComponent(kParseMailMsgStateCID, path);
  if (NS_FAILED(rv)) finalResult = rv;

  return finalResult;
}