setup_native/source/win32/customactions/reg4msdoc/userregistrar.cxx

   1 /* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
   2 // UserRegistrar.cpp: Implementierung der Klasse UserRegistrar.
   3 //
   4 //////////////////////////////////////////////////////////////////////
   5
   6 #include "userregistrar.hxx"
   7 #include "registryvalueimpl.hxx"
   8 #include "windowsregistry.hxx"
   9 #include "registryexception.hxx"
  10
  11 #ifdef _MSC_VER
  12 #pragma warning(disable : 4350)
  13 #endif
  14
  15 //--------------------------------------
  16 /**
  17 */
  18 UserRegistrar::UserRegistrar(const RegistrationContextInformation& RegContext) :
  19     Registrar(RegContext)
  20 {
  21     RegistryKey RegKey = WindowsRegistry().GetCurrentUserKey();
  22     m_RootKey = RegKey->OpenSubKey(L"Software\\Classes");
  23 }
  24
  25 //###################################
  26 // Command
  27 //###################################
  28
  29 //--------------------------------------
  30 /**
  31 */
  32 void UserRegistrar::UnregisterAsHtmlEditorForInternetExplorer() const
  33 {
  34     Registrar::UnregisterAsHtmlEditorForInternetExplorer();
  35
  36     DeleteHtmFileAssociationKeys();
  37
  38     try
  39     {
  40         RegistryKey RegKey = m_RootKey->OpenSubKey(L"Applications");
  41         if ((0 == RegKey->GetSubValueCount()) && (0 == RegKey->GetSubKeyCount()))
  42         {
  43             RegKey->Close();
  44             m_RootKey->DeleteSubKey(L"Applications");
  45         }
  46     }
  47     catch(RegistryKeyNotFoundException&)
  48     {
  49     }
  50 }
  51
  52 //--------------------------------------
  53 /**
  54 */
  55 void UserRegistrar::RegisterAsDefaultShellHtmlEditor() const
  56 {
  57     RegistryKey LocalHtmKey = m_RootKey->CreateSubKey(L".htm");
  58
  59     if (!LocalHtmKey->HasValue(DEFAULT_VALUE_NAME))
  60     {
  61         RegistryKey HKCRKey = WindowsRegistry().GetClassesRootKey();
  62
  63         if (HKCRKey->HasSubKey(L".htm"))
  64         {
  65             RegistryKey RootHtmKey = HKCRKey->OpenSubKey(L".htm", false);
  66
  67             if (RootHtmKey->HasValue(DEFAULT_VALUE_NAME))
  68             {
  69                 RegistryValue RegVal = RootHtmKey->GetValue(DEFAULT_VALUE_NAME);
  70
  71                 std::wstring RootHtmFwdKey = RegVal->GetDataAsUniString();
  72
  73                 if (HKCRKey->HasSubKey(RootHtmFwdKey))
  74                 {
  75                     m_RootKey->CreateSubKey(RootHtmFwdKey);
  76                     LocalHtmKey->CopyValue(RootHtmKey, DEFAULT_VALUE_NAME);
  77                 }
  78             }
  79         }
  80     }
  81
  82     // calling base class method
  83     Registrar::RegisterAsDefaultShellHtmlEditor();
  84 }
  85
  86 //--------------------------------------
  87 /**
  88 */
  89 void UserRegistrar::UnregisterAsDefaultShellHtmlEditor() const
  90 {
  91     // calling base class method
  92     Registrar::UnregisterAsDefaultShellHtmlEditor();
  93     DeleteHtmFileAssociationKeys();
  94 }
  95
  96 //--------------------------------------
  97 /**
  98 */
  99 void UserRegistrar::UnregisterForMsOfficeApplication(
 100         const std::wstring& FileExtension) const
 101 {
 102     /// calling base class method
 103     Registrar::UnregisterForMsOfficeApplication(FileExtension);
 104
 105     if (m_RootKey->HasSubKey(FileExtension))
 106     {
 107         RegistryKey RegKey = m_RootKey->OpenSubKey(FileExtension);
 108
 109         if ((0 == RegKey->GetSubKeyCount()) && (0 == RegKey->GetSubValueCount()))
 110         {
 111             RegKey->Close();
 112             m_RootKey->DeleteSubKey(FileExtension);
 113         }
 114     }
 115 }
 116
 117 //--------------------------------------
 118 /**
 119 */
 120 RegistryKey UserRegistrar::GetRootKeyForDefHtmlEditorForIERegistration() const
 121 {
 122     return WindowsRegistry().GetCurrentUserKey();
 123 }
 124
 125 //--------------------------------------
 126 /**
 127 */
 128 void UserRegistrar::DeleteHtmFileAssociationKeys() const
 129 {
 130     // Later delete the created keys if they are empty and have not changed meanwhile.
 131     // Remeber: if we create a new registry key in the user part of the
 132     // registry, changes to that key via the merged key HKEY_CLASSES_ROOT
 133     // go into the user branch HKEY_CURRENT_USER and are not visible for other users.
 134     // so we must carefully detect if the keys have not changed in order to prevent accidentally
 135     // deleting a key and so destroying existing associations
 136     // See MSDN: "Merged View of HKEY_CLASSES_ROOT"
 137 }
 138
 139 /* vim:set shiftwidth=4 softtabstop=4 expandtab: */