/* -*- 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 "nsIFactory.h" #include "nsISupports.h" #include "msgCore.h" #include "nsMsgBaseCID.h" #include "pratom.h" #include "nsIComponentManager.h" #include "nsIServiceManager.h" #include "nsCRT.h" #include "nsCOMPtr.h" #include "nsNewsFolder.h" #include "nsMsgNewsCID.h" /* Include all of the interfaces our factory can generate components for */ #include "nsNntpUrl.h" #include "nsNntpService.h" #include "nsNntpIncomingServer.h" #include "nsNewsMessage.h" #include "nsNNTPNewsgroup.h" #include "nsNNTPNewsgroupPost.h" #include "nsNNTPNewsgroupList.h" #include "nsNNTPArticleList.h" #include "nsNNTPHost.h" static NS_DEFINE_CID(kComponentManagerCID, NS_COMPONENTMANAGER_CID); static NS_DEFINE_CID(kNntpUrlCID, NS_NNTPURL_CID); static NS_DEFINE_CID(kNntpServiceCID, NS_NNTPSERVICE_CID); static NS_DEFINE_CID(kNewsFolderResourceCID, NS_NEWSFOLDERRESOURCE_CID); static NS_DEFINE_CID(kNntpIncomingServerCID, NS_NNTPINCOMINGSERVER_CID); static NS_DEFINE_CID(kNewsMessageResourceCID, NS_NEWSMESSAGERESOURCE_CID); static NS_DEFINE_CID(kNNTPNewsgroupCID, NS_NNTPNEWSGROUP_CID); static NS_DEFINE_CID(kNNTPNewsgroupPostCID, NS_NNTPNEWSGROUPPOST_CID); static NS_DEFINE_CID(kNNTPNewsgroupListCID, NS_NNTPNEWSGROUPLIST_CID); static NS_DEFINE_CID(kNNTPArticleListCID, NS_NNTPARTICLELIST_CID); static NS_DEFINE_CID(kNNTPHostCID, NS_NNTPHOST_CID); static PRInt32 g_InstanceCount = 0; static PRInt32 g_LockCount = 0; class nsMsgNewsFactory : public nsIFactory { public: // nsISupports methods NS_DECL_ISUPPORTS nsMsgNewsFactory(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 ~nsMsgNewsFactory(); nsCID mClassID; char* mClassName; char* mProgID; }; nsMsgNewsFactory::nsMsgNewsFactory(const nsCID &aClass, const char* aClassName, const char* aProgID) : mClassID(aClass), mClassName(nsCRT::strdup(aClassName)), mProgID(nsCRT::strdup(aProgID)) { NS_INIT_REFCNT(); } nsMsgNewsFactory::~nsMsgNewsFactory() { NS_ASSERTION(mRefCnt == 0, "non-zero refcnt at destruction"); PL_strfree(mClassName); PL_strfree(mProgID); } nsresult nsMsgNewsFactory::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(nsCOMTypeInfo::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(nsMsgNewsFactory) NS_IMPL_RELEASE(nsMsgNewsFactory) nsresult nsMsgNewsFactory::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!!! if (mClassID.Equals(kNntpUrlCID)) { nsNntpUrl *url = new nsNntpUrl(); 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(kNntpServiceCID)) { nsNntpService *service = new nsNntpService(); if (service) rv = service->QueryInterface(aIID, aResult); else rv = NS_ERROR_OUT_OF_MEMORY; if (NS_FAILED(rv) && service) delete service; } else if (mClassID.Equals(kNNTPNewsgroupPostCID)) { nsNNTPNewsgroupPost *newsgroupPost = new nsNNTPNewsgroupPost(); if (newsgroupPost) rv = newsgroupPost->QueryInterface(aIID, aResult); else rv = NS_ERROR_OUT_OF_MEMORY; if (NS_FAILED(rv) && newsgroupPost) delete newsgroupPost; } else if (mClassID.Equals(kNNTPArticleListCID)) { nsNNTPArticleList *articleList = new nsNNTPArticleList(); if (articleList) rv = articleList->QueryInterface(aIID, aResult); else rv = NS_ERROR_OUT_OF_MEMORY; if (NS_FAILED(rv) && articleList) delete articleList; } else if (mClassID.Equals(kNNTPNewsgroupListCID)) { nsNNTPNewsgroupList *newsgroupList = new nsNNTPNewsgroupList(); if (newsgroupList) rv = newsgroupList->QueryInterface(aIID, aResult); else rv = NS_ERROR_OUT_OF_MEMORY; if (NS_FAILED(rv) && newsgroupList) delete newsgroupList; } else if (mClassID.Equals(kNNTPNewsgroupCID)) { nsNNTPNewsgroup *newsgroup = new nsNNTPNewsgroup(); if (newsgroup) rv = newsgroup->QueryInterface(aIID, aResult); else rv = NS_ERROR_OUT_OF_MEMORY; if (NS_FAILED(rv) && newsgroup) delete newsgroup; } else if (mClassID.Equals(kNewsFolderResourceCID)) { nsMsgNewsFolder *folder = new nsMsgNewsFolder(); if (folder) rv = folder->QueryInterface(aIID, aResult); else rv = NS_ERROR_OUT_OF_MEMORY; if (NS_FAILED(rv) && folder) delete folder; } else if (mClassID.Equals(kNNTPHostCID)) { nsNNTPHost *host = new nsNNTPHost(); if (host) rv = host->QueryInterface(aIID, aResult); else rv = NS_ERROR_OUT_OF_MEMORY; if (NS_FAILED(rv) && host) delete host; } else if (mClassID.Equals(kNntpIncomingServerCID)) { nsNntpIncomingServer *server = new nsNntpIncomingServer(); if (server) rv = server->QueryInterface(aIID, aResult); else rv = NS_ERROR_OUT_OF_MEMORY; if (NS_FAILED(rv) && server) delete server; } else if (mClassID.Equals(kNewsMessageResourceCID)) { nsNewsMessage *message = new nsNewsMessage(); if (message) rv = message->QueryInterface(aIID, aResult); else rv = NS_ERROR_OUT_OF_MEMORY; if (NS_FAILED(rv) && message) delete message; } else return NS_NOINTERFACE; return rv; } nsresult nsMsgNewsFactory::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; *aFactory = new nsMsgNewsFactory(aClass, aClassName, aProgID); if (aFactory) return (*aFactory)->QueryInterface(nsIFactory::GetIID(),(void**)aFactory); 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 = NS_OK; nsCOMPtr servMgr(do_QueryInterface(aServMgr, &rv)); if (NS_FAILED(rv)) return rv; NS_WITH_SERVICE1(nsIComponentManager, compMgr, aServMgr, kComponentManagerCID, &rv); if (NS_FAILED(rv)) return rv; rv = compMgr->RegisterComponent(kNntpUrlCID, "NNTP Url", "component://netscape/messenger/nntpurl", path, PR_TRUE, PR_TRUE); if (NS_FAILED(rv)) return rv; rv = compMgr->RegisterComponent(kNntpServiceCID, "NNTP Service", "component://netscape/messenger/nntpservice", path, PR_TRUE, PR_TRUE); if (NS_FAILED(rv)) return rv; rv = compMgr->RegisterComponent(kNntpServiceCID, "NNTP News Service", "component://netscape/messenger/messageservice;type=news", path, PR_TRUE, PR_TRUE); if (NS_FAILED(rv)) return rv; rv = compMgr->RegisterComponent(kNntpServiceCID, "NNTP News Message Service", "component://netscape/messenger/messageservice;type=news_message", path, PR_TRUE, PR_TRUE); if (NS_FAILED(rv)) return rv; rv = compMgr->RegisterComponent(kNewsFolderResourceCID, "News Folder Resource Factory", NS_RDF_RESOURCE_FACTORY_PROGID_PREFIX "news", path, PR_TRUE, PR_TRUE); if (NS_FAILED(rv)) return rv; rv = compMgr->RegisterComponent(kNewsMessageResourceCID, "News Resource Factory", NS_RDF_RESOURCE_FACTORY_PROGID_PREFIX "news_message", path, PR_TRUE, PR_TRUE); if (NS_FAILED(rv)) return rv; rv = compMgr->RegisterComponent(kNntpIncomingServerCID, "Nntp Incoming Server", "component://netscape/messenger/server&type=nntp", path, PR_TRUE, PR_TRUE); if (NS_FAILED(rv)) return rv; rv = compMgr->RegisterComponent(kNNTPNewsgroupCID, "NNTP Newsgroup", "component://netscape/messeneger/nntpnewsgroup", path, PR_TRUE, PR_TRUE); if (NS_FAILED(rv)) return rv; rv = compMgr->RegisterComponent(kNNTPNewsgroupPostCID, "NNTP Newsgroup Post", "component://netscape/messeneger/nntpnewsgrouppost", path, PR_TRUE, PR_TRUE); if (NS_FAILED(rv)) return rv; rv = compMgr->RegisterComponent(kNNTPNewsgroupListCID, "NNTP Newsgroup List", "component://netscape/messeneger/nntpnewsgrouplist", path, PR_TRUE, PR_TRUE); if (NS_FAILED(rv)) return rv; rv = compMgr->RegisterComponent(kNNTPArticleListCID, "NNTP Article List", "component://netscape/messeneger/nntparticlelist", path, PR_TRUE, PR_TRUE); if (NS_FAILED(rv)) return rv; rv = compMgr->RegisterComponent(kNNTPHostCID, "NNTP Host", "component://netscape/messeneger/nntphost", path, PR_TRUE, PR_TRUE); if (NS_FAILED(rv)) return rv; return NS_OK; } extern "C" NS_EXPORT nsresult NSUnregisterSelf(nsISupports* aServMgr, const char* path) { nsresult rv = NS_OK; nsCOMPtr servMgr(do_QueryInterface(aServMgr, &rv)); if (NS_FAILED(rv)) return rv; NS_WITH_SERVICE(nsIComponentManager,compMgr,kComponentManagerCID,&rv); if (NS_FAILED(rv)) return rv; rv = compMgr->UnregisterComponent(kNntpUrlCID, path); if (NS_FAILED(rv)) return rv; rv = compMgr->UnregisterComponent(kNntpServiceCID, path); if (NS_FAILED(rv)) return rv; rv = compMgr->UnregisterComponent(kNewsFolderResourceCID, path); if (NS_FAILED(rv)) return rv; rv = compMgr->UnregisterComponent(kNntpIncomingServerCID, path); if (NS_FAILED(rv)) return rv; rv = compMgr->UnregisterComponent(kNewsMessageResourceCID, path); if (NS_FAILED(rv)) return rv; rv = compMgr->UnregisterComponent(kNNTPNewsgroupCID, path); if (NS_FAILED(rv)) return rv; rv = compMgr->UnregisterComponent(kNNTPNewsgroupPostCID, path); if (NS_FAILED(rv)) return rv; rv = compMgr->UnregisterComponent(kNNTPNewsgroupListCID, path); if (NS_FAILED(rv)) return rv; rv = compMgr->UnregisterComponent(kNNTPArticleListCID, path); if (NS_FAILED(rv)) return rv; rv = compMgr->UnregisterComponent(kNNTPHostCID, path); if (NS_FAILED(rv)) return rv; return NS_OK; } ////////////////////////////////////////////////////////////////////////////////