mirror of
https://github.com/mozilla/gecko-dev.git
synced 2024-11-05 00:25:27 +00:00
232 lines
6.7 KiB
C++
232 lines
6.7 KiB
C++
/* -*- 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) 1999 Netscape Communications Corporation. All Rights
|
|
* Reserved.
|
|
*/
|
|
|
|
#include "msgCore.h" // for pre-compiled headers...
|
|
#include "nsIServiceManager.h"
|
|
#include "nsIFactory.h"
|
|
#include "nsISupports.h"
|
|
#include "nsMsgDBCID.h"
|
|
#include "pratom.h"
|
|
|
|
// include files for components this factory creates...
|
|
#include "nsMailDatabase.h"
|
|
#include "nsNewsDatabase.h"
|
|
#include "nsImapMailDatabase.h"
|
|
#include "nsCOMPtr.h"
|
|
|
|
static NS_DEFINE_CID(kComponentManagerCID, NS_COMPONENTMANAGER_CID);
|
|
static NS_DEFINE_CID(kCMailDB, NS_MAILDB_CID);
|
|
static NS_DEFINE_CID(kCNewsDB, NS_NEWSDB_CID);
|
|
static NS_DEFINE_CID(kCImapDB, NS_IMAPDB_CID);
|
|
////////////////////////////////////////////////////////////
|
|
//
|
|
////////////////////////////////////////////////////////////
|
|
static PRInt32 g_InstanceCount = 0;
|
|
static PRInt32 g_LockCount = 0;
|
|
|
|
class nsMsgDBFactory : public nsIFactory
|
|
{
|
|
public:
|
|
// nsISupports methods
|
|
NS_DECL_ISUPPORTS
|
|
|
|
nsMsgDBFactory(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 ~nsMsgDBFactory();
|
|
|
|
nsCID mClassID;
|
|
char* mClassName;
|
|
char* mProgID;
|
|
};
|
|
|
|
nsMsgDBFactory::nsMsgDBFactory(const nsCID &aClass, const char* aClassName, const char* aProgID)
|
|
: mClassID(aClass), mClassName(nsCRT::strdup(aClassName)), mProgID(nsCRT::strdup(aProgID))
|
|
{
|
|
NS_INIT_REFCNT();
|
|
}
|
|
|
|
nsMsgDBFactory::~nsMsgDBFactory()
|
|
{
|
|
NS_ASSERTION(mRefCnt == 0, "non-zero refcnt at destruction");
|
|
PL_strfree(mClassName);
|
|
PL_strfree(mProgID);
|
|
}
|
|
|
|
nsresult nsMsgDBFactory::QueryInterface(const nsIID &aIID, void **aResult)
|
|
{
|
|
if (aResult == NULL)
|
|
return NS_ERROR_NULL_POINTER;
|
|
|
|
// Always NULL result, in case of failure
|
|
*aResult = NULL;
|
|
|
|
// we support two interfaces....nsISupports and nsFactory.....
|
|
if (aIID.Equals(::nsISupports::GetIID()))
|
|
*aResult = (void *)(nsISupports*)this;
|
|
else if (aIID.Equals(nsIFactory::GetIID()))
|
|
*aResult = (void *)(nsIFactory*)this;
|
|
|
|
if (*aResult == NULL)
|
|
return NS_NOINTERFACE;
|
|
|
|
AddRef(); // Increase reference count for caller
|
|
return NS_OK;
|
|
}
|
|
|
|
NS_IMPL_ADDREF(nsMsgDBFactory)
|
|
NS_IMPL_RELEASE(nsMsgDBFactory)
|
|
|
|
nsresult nsMsgDBFactory::CreateInstance(nsISupports *aOuter, const nsIID &aIID, void **aResult)
|
|
{
|
|
nsresult rv = NS_OK;
|
|
|
|
if (aResult == NULL)
|
|
return NS_ERROR_NULL_POINTER;
|
|
|
|
*aResult = NULL;
|
|
|
|
nsISupports *inst = nsnull;
|
|
|
|
// ClassID check happens here
|
|
// Whenever you add a new class that supports an interface, plug it in here!!!
|
|
|
|
// do they want an nsMailDatabase ?
|
|
if (mClassID.Equals(kCMailDB))
|
|
{
|
|
inst = NS_STATIC_CAST(nsIMsgDatabase*, new nsMailDatabase());
|
|
}
|
|
// do they want an nsNewsDatabase ?
|
|
else if (mClassID.Equals(kCNewsDB))
|
|
{
|
|
inst = NS_STATIC_CAST(nsIMsgDatabase*, new nsNewsDatabase());
|
|
}
|
|
// do they want an nsImapDatabase?
|
|
else if (mClassID.Equals(kCImapDB))
|
|
{
|
|
inst = new nsImapMailDatabase();
|
|
}
|
|
|
|
if (inst == nsnull)
|
|
return NS_ERROR_OUT_OF_MEMORY;
|
|
|
|
rv = inst->QueryInterface(aIID, aResult);
|
|
if (NS_FAILED(rv))
|
|
delete inst;
|
|
return rv;
|
|
}
|
|
|
|
nsresult nsMsgDBFactory::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 nsMsgDBFactory(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;
|
|
|
|
nsCOMPtr<nsIServiceManager> servMgr(do_QueryInterface(aServMgr, &rv));
|
|
if (NS_FAILED(rv)) return rv;
|
|
|
|
nsIComponentManager* compMgr;
|
|
rv = servMgr->GetService(kComponentManagerCID,
|
|
nsIComponentManager::GetIID(),
|
|
(nsISupports**)&compMgr);
|
|
if (NS_FAILED(rv)) return rv;
|
|
|
|
rv = compMgr->RegisterComponent(kCMailDB, nsnull, nsnull,
|
|
path, PR_TRUE, PR_TRUE);
|
|
if (NS_FAILED(rv)) goto done;
|
|
|
|
rv = compMgr->RegisterComponent(kCNewsDB, nsnull, nsnull,
|
|
path, PR_TRUE, PR_TRUE);
|
|
if (NS_FAILED(rv)) goto done;
|
|
|
|
rv = compMgr->RegisterComponent(kCImapDB, nsnull, nsnull,
|
|
path, PR_TRUE, PR_TRUE);
|
|
if (NS_FAILED(rv)) goto done;
|
|
|
|
done:
|
|
(void)servMgr->ReleaseService(kComponentManagerCID, compMgr);
|
|
return rv;
|
|
}
|
|
|
|
extern "C" NS_EXPORT nsresult
|
|
NSUnregisterSelf(nsISupports* aServMgr, const char* path)
|
|
{
|
|
nsresult rv;
|
|
|
|
nsCOMPtr<nsIServiceManager> servMgr(do_QueryInterface(aServMgr, &rv));
|
|
if (NS_FAILED(rv)) return rv;
|
|
|
|
nsIComponentManager* compMgr;
|
|
rv = servMgr->GetService(kComponentManagerCID,
|
|
nsIComponentManager::GetIID(),
|
|
(nsISupports**)&compMgr);
|
|
if (NS_FAILED(rv)) return rv;
|
|
|
|
rv = compMgr->UnregisterFactory(kCMailDB, path);
|
|
if (NS_FAILED(rv)) goto done;
|
|
|
|
rv = compMgr->UnregisterFactory(kCImapDB, path);
|
|
if (NS_FAILED(rv)) goto done;
|
|
|
|
rv = compMgr->UnregisterFactory(kCNewsDB, path);
|
|
if (NS_FAILED(rv)) goto done;
|
|
|
|
done:
|
|
(void)servMgr->ReleaseService(kComponentManagerCID, compMgr);
|
|
return rv;
|
|
}
|