src/Utils/CmdLineParser.cpp

   1 // TortoiseGit - a Windows shell extension for easy version control
   2
   3 // Copyright (C) 2013 - TortoiseGit
   4 // Copyright (C) 2003-2006,2012 - Stefan Kueng
   5
   6 // This program is free software; you can redistribute it and/or
   7 // modify it under the terms of the GNU General Public License
   8 // as published by the Free Software Foundation; either version 2
   9 // of the License, or (at your option) any later version.
  10
  11 // This program is distributed in the hope that it will be useful,
  12 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14 // GNU General Public License for more details.
  15
  16 // You should have received a copy of the GNU General Public License
  17 // along with this program; if not, write to the Free Software Foundation,
  18 // 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  19 //
  20 #include "stdafx.h"
  21 #include "CmdLineParser.h"
  22 #include <locale>
  23 #include <algorithm>
  24
  25 const TCHAR CCmdLineParser::m_sDelims[] = _T("-/");
  26 const TCHAR CCmdLineParser::m_sQuotes[] = _T("\"");
  27 const TCHAR CCmdLineParser::m_sValueSep[] = _T(" :"); // don't forget space!!
  28
  29
  30 CCmdLineParser::CCmdLineParser(LPCTSTR sCmdLine)
  31 {
  32         if(sCmdLine)
  33         {
  34                 Parse(sCmdLine);
  35         }
  36 }
  37
  38 CCmdLineParser::~CCmdLineParser()
  39 {
  40         m_valueMap.clear();
  41 }
  42
  43 BOOL CCmdLineParser::Parse(LPCTSTR sCmdLine)
  44 {
  45         const stdstring sEmpty = _T("");                        //use this as a value if no actual value is given in commandline
  46         int nArgs = 0;
  47
  48         if(!sCmdLine)
  49                 return false;
  50
  51         m_valueMap.clear();
  52         m_sCmdLine = sCmdLine;
  53
  54         LPCTSTR sCurrent = sCmdLine;
  55
  56         for(;;)
  57         {
  58                 //format is  -Key:"arg"
  59
  60                 if (_tcslen(sCurrent) == 0)
  61                         break;          // no more data, leave loop
  62
  63                 LPCTSTR sArg = _tcspbrk(sCurrent, m_sDelims);
  64                 if(!sArg)
  65                         break; // no (more) delimiters found
  66                 sArg =  _tcsinc(sArg);
  67
  68                 if(_tcslen(sArg) == 0)
  69                         break; // ends with delim
  70
  71                 LPCTSTR sVal = _tcspbrk(sArg, m_sValueSep);
  72                 if(sVal == NULL)
  73                 {
  74                         stdstring Key(sArg);
  75                         std::transform(Key.begin(), Key.end(), Key.begin(), ::tolower);
  76                         m_valueMap.insert(CValsMap::value_type(Key, sEmpty));
  77                         break;
  78                 }
  79                 else
  80                 {
  81                         stdstring Key(sArg, (int)(sVal - sArg));
  82                         std::transform(Key.begin(), Key.end(), Key.begin(), ::tolower);
  83
  84                         LPCTSTR sQuote(NULL), sEndQuote(NULL);
  85                         if (_tcslen(sVal) > 0)
  86                         {
  87                                 if (sVal[0] != _T(' '))
  88                                         sVal = _tcsinc(sVal);
  89                                 else
  90                                 {
  91                                         while (_tcslen(sVal) > 0 && sVal[0] == _T(' '))
  92                                                 sVal = _tcsinc(sVal);
  93                                 }
  94
  95                                 LPCTSTR nextArg = _tcspbrk(sVal, m_sDelims);
  96
  97                                 sQuote = _tcspbrk(sVal, m_sQuotes);
  98
  99                                 if (nextArg == sVal)
 100                                 {
 101                                         // current key has no value, but a next key exist - so don't use next key as value of current one
 102                                         --sVal;
 103                                         sQuote = sVal;
 104                                         sEndQuote = sVal;
 105                                 }
 106                                 else if (nextArg != NULL && nextArg < sQuote)
 107                                 {
 108                                         // current key has a value w/o quotes, but next key one has value in quotes
 109                                         sQuote = sVal;
 110                                         sEndQuote = _tcschr(sQuote, _T(' '));
 111                                 }
 112                                 else
 113                                 {
 114                                         if(sQuote == sVal)
 115                                         {
 116                                                 // string with quotes (defined in m_sQuotes, e.g. '")
 117                                                 sQuote = _tcsinc(sVal);
 118                                                 sEndQuote = _tcspbrk(sQuote, m_sQuotes);
 119                                         }
 120                                         else
 121                                         {
 122                                                 sQuote = sVal;
 123                                                 sEndQuote = _tcschr(sQuote, _T(' '));
 124                                         }
 125                                 }
 126                         }
 127
 128                         if(sEndQuote == NULL)
 129                         {
 130                                 // no end quotes or terminating space, take the rest of the string to its end
 131                                 if (!Key.empty() && sQuote)
 132                                 {
 133                                         stdstring csVal(sQuote);
 134                                         m_valueMap.insert(CValsMap::value_type(Key, csVal));
 135                                 }
 136                                 break;
 137                         }
 138                         else
 139                         {
 140                                 // end quote
 141                                 if(!Key.empty())
 142                                 {
 143                                         stdstring csVal(sQuote, (int)(sEndQuote - sQuote));
 144                                         m_valueMap.insert(CValsMap::value_type(Key, csVal));
 145                                 }
 146                                 sCurrent = _tcsinc(sEndQuote);
 147                                 continue;
 148                         }
 149                 }
 150         }
 151
 152         return (nArgs > 0);             //TRUE if arguments were found
 153 }
 154
 155 CCmdLineParser::CValsMap::const_iterator CCmdLineParser::findKey(LPCTSTR sKey) const
 156 {
 157         stdstring s(sKey);
 158         std::transform(s.begin(), s.end(), s.begin(), ::tolower);
 159         return m_valueMap.find(s);
 160 }
 161
 162 BOOL CCmdLineParser::HasKey(LPCTSTR sKey) const
 163 {
 164         CValsMap::const_iterator it = findKey(sKey);
 165         if(it == m_valueMap.end())
 166                 return false;
 167         return true;
 168 }
 169
 170
 171 BOOL CCmdLineParser::HasVal(LPCTSTR sKey) const
 172 {
 173         CValsMap::const_iterator it = findKey(sKey);
 174         if(it == m_valueMap.end())
 175                 return false;
 176         if(it->second.empty())
 177                 return false;
 178         return true;
 179 }
 180
 181 LPCTSTR CCmdLineParser::GetVal(LPCTSTR sKey) const
 182 {
 183         CValsMap::const_iterator it = findKey(sKey);
 184         if (it == m_valueMap.end())
 185                 return 0;
 186         return it->second.c_str();
 187 }
 188
 189 LONG CCmdLineParser::GetLongVal(LPCTSTR sKey) const
 190 {
 191         CValsMap::const_iterator it = findKey(sKey);
 192         if (it == m_valueMap.end())
 193                 return 0;
 194         return _tstol(it->second.c_str());
 195 }
 196
 197 __int64 CCmdLineParser::GetLongLongVal(LPCTSTR sKey) const
 198 {
 199         CValsMap::const_iterator it = findKey(sKey);
 200         if (it == m_valueMap.end())
 201                 return 0;
 202         return _ttoi64(it->second.c_str());
 203 }
 204
 205 CCmdLineParser::ITERPOS CCmdLineParser::begin() const
 206 {
 207         return m_valueMap.begin();
 208 }
 209
 210 CCmdLineParser::ITERPOS CCmdLineParser::getNext(ITERPOS& pos, stdstring& sKey, stdstring& sValue) const
 211 {
 212         if (m_valueMap.end() == pos)
 213         {
 214                 sKey.clear();
 215                 return pos;
 216         }
 217         else
 218         {
 219                 sKey = pos->first;
 220                 sValue = pos->second;
 221                 return ++pos;
 222         }
 223 }
 224
 225 BOOL CCmdLineParser::isLast(const ITERPOS& pos) const
 226 {
 227         return (pos == m_valueMap.end());
 228 }