//////////////////////////////////////////////////////////////////////////////
/*++

Copyright (C) Microsoft Corporation, 1997 - 1999

Module Name:

    NAPMMCDLL.cpp

Abstract:

	Implementation of DLL Exports.

Proxy/Stub Information:

	To build a separate proxy/stub DLL, 
	run nmake -f NAPMMCps.mk in the project directory.

Revision History:
	mmaguire 11/6/97 - created using MMC snap-in wizard


--*/
//////////////////////////////////////////////////////////////////////////////

//////////////////////////////////////////////////////////////////////////////
// BEGIN INCLUDES


//
// standard includes:
//
#include "Precompiled.h"
//
// where we can find declaration for main class in this file:
//

//
// where we can find declarations needed in this file:
//
#include "initguid.h"
#include "NAPMMC_i.c"

// For IComponent, IComponentData, and ISnapinAbout COM classes:

#include "Component.h"
#include "About.h"
#include "LogComp.h"
#include "LogAbout.H"

// For AttributeInfo COM classes:

#include "IASAttributeInfo.h"
#include "IASEnumerableAttributeInfo.h"

// SDO declarations
//TODO: change it when TODD's CLSIDs are ready

//
// Need to include this at least once to compile it in from the common directory:
#include "mmcutility.cpp"

// We are hosting a few extra COM objects in this dll:


// For AttributeEditor COM classes:
#include "IASIPAttributeEditor.h"
#include "IASMultivaluedAttributeEditor.h"
#include "IASVendorSpecificAttributeEditor.h"
#include "IASEnumerableAttributeEditor.h"
#include "IASStringAttributeEditor.h"
#include "IASBooleanAttributeEditor.h"
#include "NTGroups.h"

// For NASVendors COM object:
#include "Vendors.h"

#include <proxyext.h>
#include <proxyres.h>
unsigned int CF_MMC_NodeID = RegisterClipboardFormatW(CCF_NODEID2);


//
// END INCLUDES
//////////////////////////////////////////////////////////////////////////////



CComModule _Module;



BEGIN_OBJECT_MAP(ObjectMap)
	OBJECT_ENTRY(CLSID_NAPSnapin, CComponentData)
	OBJECT_ENTRY(CLSID_NAPSnapinAbout, CSnapinAbout)
	OBJECT_ENTRY(CLSID_LoggingSnapin, CLoggingComponentData)
	OBJECT_ENTRY(CLSID_LoggingSnapinAbout, CLoggingSnapinAbout)
	OBJECT_ENTRY(CLSID_IASAttributeInfo, CAttributeInfo)
	OBJECT_ENTRY(CLSID_IASEnumerableAttributeInfo, CEnumerableAttributeInfo)
	OBJECT_ENTRY(CLSID_IASIPAttributeEditor, CIASIPAttributeEditor)
	OBJECT_ENTRY(CLSID_IASMultivaluedAttributeEditor, CIASMultivaluedAttributeEditor)
	OBJECT_ENTRY(CLSID_IASVendorSpecificAttributeEditor, CIASVendorSpecificAttributeEditor)
	OBJECT_ENTRY(CLSID_IASEnumerableAttributeEditor, CIASEnumerableAttributeEditor)
	OBJECT_ENTRY(CLSID_IASStringAttributeEditor, CIASStringAttributeEditor)
	OBJECT_ENTRY(CLSID_IASBooleanAttributeEditor, CIASBooleanAttributeEditor)
	OBJECT_ENTRY(CLSID_IASGroupsAttributeEditor, CIASGroupsAttributeEditor)
	OBJECT_ENTRY(CLSID_IASNASVendors, CIASNASVendors)
   OBJECT_ENTRY(__uuidof(ProxyExtension), ProxyExtension)
END_OBJECT_MAP()




#if 1  // Use CWinApp for MFC support -- some of the COM objects in this dll use MFC.



class CNAPMMCApp : public CWinApp
{
public:
	virtual BOOL InitInstance();
	virtual int ExitInstance();
};

CNAPMMCApp theApp;

DWORD   g_dwTraceHandle = 0;

//////////////////////////////////////////////////////////////////////////////
/*++

CNAPMMCApp::InitInstance

	MFC's dll entry point.
	
--*/
//////////////////////////////////////////////////////////////////////////////
BOOL CNAPMMCApp::InitInstance()
{
	_Module.Init(ObjectMap, m_hInstance);

    g_dwTraceHandle = TraceRegister(_T("NapMMC"));
	TracePrintf(g_dwTraceHandle, _T("DLL Init"));
	
	// Initialize static class variables of CSnapInItem.
	CSnapInItem::Init();

	// Initialize any other static class variables.
	CMachineNode::InitClipboardFormat();
    CLoggingMachineNode::InitClipboardFormat();



	return CWinApp::InitInstance();
}


//////////////////////////////////////////////////////////////////////////////
/*++

CNAPMMCApp::ExitInstance

	MFC's dll exit point.
	
--*/
//////////////////////////////////////////////////////////////////////////////
int CNAPMMCApp::ExitInstance()
{
	TracePrintf(g_dwTraceHandle, _T("DLL Exit"));
	TraceDeregister(g_dwTraceHandle);

	_Module.Term();

	return CWinApp::ExitInstance();
}



#else // Use CWinApp



//////////////////////////////////////////////////////////////////////////////
/*++

	DllMain


	Remarks
		
		DLL Entry Point
	
--*/
//////////////////////////////////////////////////////////////////////////////
extern "C"
BOOL WINAPI DllMain(HINSTANCE hInstance, DWORD dwReason, LPVOID /*lpReserved*/)
{
	if (dwReason == DLL_PROCESS_ATTACH)
	{
		_Module.Init(ObjectMap, hInstance);
		// Initialize static class variables of CSnapInItem.
		CSnapInItem::Init();

		// Initialize any other static class variables.
		CMachineNode::InitClipboardFormat();

		DisableThreadLibraryCalls(hInstance);
	}
	else if (dwReason == DLL_PROCESS_DETACH)
		_Module.Term();
	return TRUE;    // ok
	}


#endif // Use CWinApp



//////////////////////////////////////////////////////////////////////////////
/*++

	DllCanUnloadNow


	Remarks
		
		Used to determine whether the DLL can be unloaded by OLE

--*/
//////////////////////////////////////////////////////////////////////////////
STDAPI DllCanUnloadNow(void)
{
	return (_Module.GetLockCount()==0) ? S_OK : S_FALSE;
}



//////////////////////////////////////////////////////////////////////////////
/*++

	DllGetClassObject


	Remarks
		
		Returns a class factory to create an object of the requested type

--*/
//////////////////////////////////////////////////////////////////////////////
STDAPI DllGetClassObject(REFCLSID rclsid, REFIID riid, LPVOID* ppv)
{
	return _Module.GetClassObject(rclsid, riid, ppv);
}



//////////////////////////////////////////////////////////////////////////////
/*++

	DllRegisterServer


	Remarks
		
		  Adds entries to the system registry

--*/
//////////////////////////////////////////////////////////////////////////////

STDAPI DllRegisterServer(void)
{
	// Set the protocol.
	TCHAR NapName[IAS_MAX_STRING];
	TCHAR NapName_Indirect[IAS_MAX_STRING];
	TCHAR	ModuleName[MAX_PATH];
	TCHAR LoggingName[IAS_MAX_STRING];
	TCHAR LoggingName_Indirect[IAS_MAX_STRING];

	GetModuleFileNameOnly(_Module.GetModuleInstance(), ModuleName, MAX_PATH);
	
	int iLoadStringResult = LoadString(  _Module.GetResourceInstance(), IDS_SNAPINNAME_NAP, NapName, IAS_MAX_STRING );
	swprintf(NapName_Indirect, L"@%s,-%-d", ModuleName, IDS_SNAPINNAME_NAP);

	iLoadStringResult = LoadString(  _Module.GetResourceInstance(), IDS_SNAPINNAME_LOGGING, LoggingName, IAS_MAX_STRING );
	swprintf(LoggingName_Indirect, L"@%s,-%-d", ModuleName, IDS_SNAPINNAME_LOGGING);

    struct _ATL_REGMAP_ENTRY regMap[] = {
        {OLESTR("NAPSNAPIN"), NapName}, // subsitute %NAPSNAPIN% for registry
        {OLESTR("NAPSNAPIN_INDIRECT"), NapName_Indirect}, // subsitute %IASSNAPIN% for registry
        {OLESTR("LOGGINGSNAPIN"), LoggingName}, // subsitute %LOGGINGSNAPIN% for registry
        {OLESTR("LOGGINGSNAPIN_INDIRECT"), LoggingName_Indirect}, // subsitute %IASSNAPIN% for registry
        {0, 0}
    };
    
   HRESULT hr = _Module.UpdateRegistryFromResource(IDR_NAPSNAPIN, TRUE, regMap);
   if (SUCCEEDED(hr))
   {
      ResourceString proxyName(IDS_PROXY_EXTENSION);
      _ATL_REGMAP_ENTRY entries[] =
      {
         { L"PROXY_EXTENSION", proxyName },
         { NULL, NULL }
      };

      hr = _Module.UpdateRegistryFromResource(
                       IDR_PROXY_REGISTRY,
                       TRUE,
                       entries
                       );
   }

   return hr;
}



//////////////////////////////////////////////////////////////////////////////
/*++

	DllUnregisterServer


	Remarks
		
		Removes entries from the system registry

--*/
//////////////////////////////////////////////////////////////////////////////
STDAPI DllUnregisterServer(void)
{
	// Set the protocol.
	TCHAR NapName[IAS_MAX_STRING];
	TCHAR LoggingName[IAS_MAX_STRING];
	int iLoadStringResult = LoadString(  _Module.GetResourceInstance(), IDS_SNAPINNAME_NAP, NapName, IAS_MAX_STRING );
	iLoadStringResult = LoadString(  _Module.GetResourceInstance(), IDS_SNAPINNAME_LOGGING, LoggingName, IAS_MAX_STRING );

    struct _ATL_REGMAP_ENTRY regMap[] = {
        {OLESTR("NAPSNAPIN"), NapName}, // subsitute %NAPSNAPIN% for registry
        {OLESTR("LOGGINGSNAPIN"), LoggingName}, // subsitute %LOGGINGSNAPIN% for registry
        {0, 0}
    };
    
   _Module.UpdateRegistryFromResource(IDR_NAPSNAPIN, FALSE, regMap);
   _Module.UpdateRegistryFromResource(IDR_PROXY_REGISTRY, FALSE, NULL);

	return S_OK;
}