rosapps/sysutils/regexpl/ShellCommandDeleteValue.cpp

   1 /* $Id$
   2  *
   3  * regexpl - Console Registry Explorer
   4  *
   5  * Copyright (C) 2000-2005 Nedko Arnaudov <nedko@users.sourceforge.net>
   6  *
   7  * This program is free software; you can redistribute it and/or modify
   8  * it under the terms of the GNU General Public License as published by
   9  * the Free Software Foundation; either version 2 of the License, or
  10  * (at your option) any later version.
  11  *
  12  * This program is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15  * GNU General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU General Public License
  18  * along with this program; see the file COPYING.  If not, write to
  19  * the Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  20  * Boston, MA 02111-1307, USA.
  21  */
  22
  23 // ShellCommandDeleteValue.cpp: implementation of the CShellCommandDeleteValue class.
  24 //
  25 //////////////////////////////////////////////////////////////////////
  26
  27 #include "ph.h"
  28 #include "ShellCommandDeleteValue.h"
  29 #include "RegistryExplorer.h"
  30 #include "Pattern.h"
  31
  32 #define DV_CMD                  _T("DV")
  33 #define DV_CMD_LENGTH           COMMAND_LENGTH(DV_CMD)
  34 #define DV_CMD_SHORT_DESC       DV_CMD _T(" command is used to delete value.\n")
  35
  36 //////////////////////////////////////////////////////////////////////
  37 // Construction/Destruction
  38 //////////////////////////////////////////////////////////////////////
  39
  40 CShellCommandDeleteValue::CShellCommandDeleteValue(CRegistryTree& rTree):m_rTree(rTree)
  41 {
  42 }
  43
  44 CShellCommandDeleteValue::~CShellCommandDeleteValue()
  45 {
  46 }
  47
  48 BOOL CShellCommandDeleteValue::Match(const TCHAR *pszCommand)
  49 {
  50         return _tcsicmp(pszCommand,DV_CMD) == 0;
  51 }
  52
  53 int CShellCommandDeleteValue::Execute(CConsole &rConsole, CArgumentParser& rArguments)
  54 {
  55   rArguments.ResetArgumentIteration();
  56   TCHAR *pszCommandItself = rArguments.GetNextArgument();
  57
  58   TCHAR *pszParameter;
  59   TCHAR *pszValueFull = NULL;
  60   BOOL blnHelp = FALSE;
  61
  62   if ((_tcsnicmp(pszCommandItself,DV_CMD _T(".."),DV_CMD_LENGTH+2*sizeof(TCHAR)) == 0)||
  63       (_tcsnicmp(pszCommandItself,DV_CMD _T("\\"),DV_CMD_LENGTH+1*sizeof(TCHAR)) == 0))
  64   {
  65     pszValueFull = pszCommandItself + DV_CMD_LENGTH;
  66   }
  67   else if (_tcsnicmp(pszCommandItself,DV_CMD _T("/"),DV_CMD_LENGTH+1*sizeof(TCHAR)) == 0)
  68   {
  69     pszParameter = pszCommandItself + DV_CMD_LENGTH;
  70     goto CheckValueArgument;
  71   }
  72
  73   while((pszParameter = rArguments.GetNextArgument()) != NULL)
  74   {
  75 CheckValueArgument:
  76     if ((_tcsicmp(pszParameter,_T("/?")) == 0)
  77         ||(_tcsicmp(pszParameter,_T("-?")) == 0))
  78     {
  79       blnHelp = TRUE;
  80       break;
  81     }
  82     else if (!pszValueFull)
  83     {
  84       pszValueFull = pszParameter;
  85     }
  86     else
  87     {
  88       rConsole.Write(_T("Bad parameter: "));
  89       rConsole.Write(pszParameter);
  90       rConsole.Write(_T("\n"));
  91     }
  92   }
  93
  94   CRegistryKey Key;
  95   TCHAR *pszValueNamePattern;
  96   const TCHAR *pszPath;
  97
  98   if (blnHelp)
  99   {
 100     rConsole.Write(GetHelpString());
 101     return 0;
 102   }
 103
 104   if (pszValueFull)
 105   {
 106     if (_tcscmp(pszValueFull,_T("\\")) == 0)
 107       goto CommandNAonRoot;
 108
 109     TCHAR *pchSep = _tcsrchr(pszValueFull,_T('\\'));
 110     pszValueNamePattern = pchSep?(pchSep+1):(pszValueFull);
 111     pszPath = pchSep?pszValueFull:_T(".");
 112
 113     if (pchSep)
 114       *pchSep = 0;
 115   }
 116   else
 117   {
 118     pszValueNamePattern = _T("");
 119     pszPath = _T(".");
 120   }
 121
 122   {
 123     size_t s = _tcslen(pszValueNamePattern);
 124     if (s && (pszValueNamePattern[0] == _T('\"'))&&(pszValueNamePattern[s-1] == _T('\"')))
 125     {
 126       pszValueNamePattern[s-1] = 0;
 127       pszValueNamePattern++;
 128     }
 129   }
 130
 131   if (!m_rTree.GetKey(pszPath,KEY_QUERY_VALUE|KEY_SET_VALUE,Key))
 132   {
 133     rConsole.Write(m_rTree.GetLastErrorDescription());
 134     return 0;
 135   }
 136
 137   if (!Key.IsRoot())
 138   {     // not root key ???
 139     TCHAR Buffer[254];
 140     DWORD dwMaxValueNameLength;
 141     LONG nError = Key.GetMaxValueNameLength(dwMaxValueNameLength);
 142     if (nError != ERROR_SUCCESS)
 143     {
 144       _stprintf(Buffer,_T("Cannot query info about %s key. Error is %u\n"),Key.GetKeyName(),(unsigned int)nError);
 145       rConsole.Write(Buffer);
 146       return 0;
 147     }
 148
 149     TCHAR *pszValueName = new TCHAR[dwMaxValueNameLength];
 150     if (!pszValueName)
 151     {
 152       rConsole.Write("Out of memory.");
 153       return 0;
 154     }
 155
 156     Key.InitValueEnumeration(pszValueName,dwMaxValueNameLength,NULL,0,NULL);
 157
 158     while ((nError = Key.GetNextValue()) == ERROR_SUCCESS)
 159     {
 160       if (PatternMatch(pszValueNamePattern,pszValueName))
 161       {
 162         nError = Key.DeleteValue(pszValueName);
 163         if (nError != ERROR_SUCCESS)
 164         {
 165           _stprintf(Buffer,_T("Cannot delete value. Error is %u\n"),(unsigned int)nError);
 166           rConsole.Write(Buffer);
 167         }
 168         else
 169         {
 170           InvalidateCompletion();
 171         }
 172         Key.InitValueEnumeration(pszValueName,dwMaxValueNameLength,NULL,0,NULL); // reset iteration
 173       }
 174     }
 175   } // if (pKey)
 176   else
 177   {
 178 CommandNAonRoot:
 179     rConsole.Write(DV_CMD COMMAND_NA_ON_ROOT);
 180   }
 181
 182   return 0;
 183 }
 184
 185 const TCHAR * CShellCommandDeleteValue::GetHelpString()
 186 {
 187         return DV_CMD_SHORT_DESC
 188                         _T("Syntax: ") DV_CMD _T(" [<PATH>][<VALUE_NAME>] [/?]\n\n")
 189                         _T("    <PATH>       - Optional relative path of key which value will be delete.\n")
 190                         _T("    <VALUE_NAME> - Name pattern of key's value. Default is key's default value.\n")
 191                         _T("    /? - This help.\n\n");
 192 }
 193
 194 const TCHAR * CShellCommandDeleteValue::GetHelpShortDescriptionString()
 195 {
 196         return DV_CMD_SHORT_DESC;
 197 }