src/VisVim/VisVim.cpp

   1 // VisVim.cpp : Defines the initialization routines for the DLL.
   2 //
   3
   4 #include "stdafx.h"
   5 #include <initguid.h>
   6 #include "VisVim.h"
   7 #include "DSAddIn.h"
   8 #include "Commands.h"
   9
  10 #ifdef _DEBUG
  11 #define new DEBUG_NEW
  12 #undef THIS_FILE
  13 static char THIS_FILE[] = __FILE__;
  14
  15 #endif
  16
  17 CComModule _Module;
  18
  19 BEGIN_OBJECT_MAP (ObjectMap)
  20 OBJECT_ENTRY (CLSID_DSAddIn, CDSAddIn)
  21 END_OBJECT_MAP ()
  22
  23 class CVisVimApp : public CWinApp
  24 {
  25     public:
  26         CVisVimApp ();
  27
  28         //{{AFX_VIRTUAL(CVisVimApp)
  29     public:
  30         virtual BOOL InitInstance ();
  31         virtual int ExitInstance ();
  32         //}}AFX_VIRTUAL
  33
  34         //{{AFX_MSG(CVisVimApp)
  35         //}}AFX_MSG
  36         DECLARE_MESSAGE_MAP ()
  37 };
  38
  39 BEGIN_MESSAGE_MAP (CVisVimApp, CWinApp)
  40 //{{AFX_MSG_MAP(CVisVimApp)
  41 //}}AFX_MSG_MAP
  42 END_MESSAGE_MAP ()
  43
  44 // The one and only CVisVimApp object
  45 CVisVimApp theApp;
  46
  47 CVisVimApp::CVisVimApp ()
  48 {
  49 }
  50
  51 BOOL CVisVimApp::InitInstance ()
  52 {
  53         _Module.Init (ObjectMap, m_hInstance);
  54         return CWinApp::InitInstance ();
  55 }
  56
  57 int CVisVimApp::ExitInstance ()
  58 {
  59         _Module.Term ();
  60         return CWinApp::ExitInstance ();
  61 }
  62
  63 // Special entry points required for inproc servers
  64 //
  65
  66 STDAPI DllGetClassObject (REFCLSID rclsid, REFIID riid, LPVOID * ppv)
  67 {
  68         AFX_MANAGE_STATE (AfxGetStaticModuleState ());
  69         return _Module.GetClassObject (rclsid, riid, ppv);
  70 }
  71
  72 STDAPI DllCanUnloadNow (void)
  73 {
  74         AFX_MANAGE_STATE (AfxGetStaticModuleState ());
  75         return (AfxDllCanUnloadNow () == S_OK && _Module.GetLockCount () == 0)
  76                 ? S_OK : S_FALSE;
  77 }
  78
  79 // By exporting DllRegisterServer, you can use regsvr32.exe
  80 //
  81 STDAPI DllRegisterServer (void)
  82 {
  83         AFX_MANAGE_STATE (AfxGetStaticModuleState ());
  84         HRESULT hRes;
  85
  86         // Registers object, typelib and all interfaces in typelib
  87         hRes = _Module.RegisterServer (TRUE);
  88         if (FAILED (hRes))
  89                 // Hack: When this fails we might be a normal user, while the
  90                 // admin already registered the module.  Returning S_OK then
  91                 // makes it work.  When the module was never registered it
  92                 // will soon fail in another way.
  93                 // old code: return hRes;
  94                 return S_OK;
  95
  96         _ATL_OBJMAP_ENTRY *pEntry = _Module.m_pObjMap;
  97         CRegKey key;
  98         LONG lRes = key.Open (HKEY_CLASSES_ROOT, _T ("CLSID"));
  99
 100         if (lRes == ERROR_SUCCESS)
 101         {
 102                 USES_CONVERSION;
 103                 LPOLESTR lpOleStr;
 104
 105                 StringFromCLSID (*pEntry->pclsid, &lpOleStr);
 106                 LPTSTR lpsz = OLE2T (lpOleStr);
 107
 108                 lRes = key.Open (key, lpsz);
 109                 if (lRes == ERROR_SUCCESS)
 110                 {
 111                         CString strDescription;
 112
 113                         strDescription.LoadString (IDS_VISVIM_DESCRIPTION);
 114                         key.SetKeyValue (_T ("Description"), strDescription);
 115                 }
 116                 CoTaskMemFree (lpOleStr);
 117         }
 118
 119         if (lRes != ERROR_SUCCESS)
 120                 hRes = HRESULT_FROM_WIN32 (lRes);
 121
 122         return hRes;
 123
 124 }
 125
 126 // DllUnregisterServer - Removes entries from the system registry
 127 //
 128 STDAPI DllUnregisterServer (void)
 129 {
 130         AFX_MANAGE_STATE (AfxGetStaticModuleState ());
 131
 132         HRESULT hRes = S_OK;
 133         _Module.UnregisterServer ();
 134         return hRes;
 135 }
 136
 137
 138 // Debugging support
 139
 140 // GetLastErrorDescription is used in the implementation of the VERIFY_OK
 141 //  macro, defined in stdafx.h.
 142
 143 #ifdef _DEBUG
 144
 145 void GetLastErrorDescription (CComBSTR & bstr)
 146 {
 147         CComPtr < IErrorInfo > pErrorInfo;
 148         if (GetErrorInfo (0, &pErrorInfo) == S_OK)
 149                 pErrorInfo->GetDescription (&bstr);
 150 }
 151
 152 #endif //_DEBUG