ssphys/SSRep/Options.cpp

   1 // Options.cpp: implementation of the COptions class.
   2 //
   3 //////////////////////////////////////////////////////////////////////
   4
   5 #include "stdafx.h"
   6 #include "Options.h"
   7 #include "strstream"
   8
   9 //---------------------------------------------------------------------------
  10 #include "LeakWatcher.h"
  11
  12 #ifdef _DEBUG
  13 #define new DEBUG_NEW
  14 #undef THIS_FILE
  15 static char THIS_FILE[] = __FILE__;
  16 #endif
  17
  18 //////////////////////////////////////////////////////////////////////
  19 // Construction/Destruction
  20 //////////////////////////////////////////////////////////////////////
  21
  22 // TODO: replace the complete COptions class with a better design
  23
  24 COptions::COptions ()
  25   : m_bIntegrityCheck (false),
  26     m_bIncludeLabels (true),
  27     m_bOnlyLabels (false),
  28     m_bIncludeDeleted (false),
  29     m_bOnlyDeleted (false),
  30     m_bIncludeFiles (true),
  31     m_bOnlyFiles (false),
  32     m_bHelp (false),
  33     m_bAllRecords (false),
  34     m_pVersionFilter (NULL),
  35     m_Version (0),
  36     m_bForceOverwrite (false),
  37     m_bRecursive (false),
  38     m_bExtended (false),
  39     m_bPhys (false)
  40 {
  41
  42 }
  43
  44 COptions::~COptions ()
  45 {
  46   DELETE (m_pVersionFilter);
  47 }
  48
  49 void COptions::PrintUsage ()
  50 {
  51   std::cout << "ssrep v0.8:" << std::endl;
  52   std::cout << std::endl;
  53   std::cout << "usage:" << std::endl;
  54   std::cout << "general options: "  << std::endl;
  55   std::cout << " -Y           user,password"  << std::endl;
  56   std::cout << " -d           include deleted files" << std::endl;
  57   std::cout << " -d+            only deleted files" << std::endl;
  58   std::cout << " -f           include files (default)" << std::endl;
  59   std::cout << " -f+            only files" << std::endl;
  60   std::cout << " -f-            only projects" << std::endl;
  61   std::cout << " -v#          specific version number"  << std::endl;
  62   std::cout << std::endl;
  63   std::cout << std::endl;
  64   std::cout << "ssrep dir [file]      displays directory information about the project"  << std::endl;
  65 //  std::cout << "ssrep info [file]      displays basic information about the item"  << std::endl;
  66 //  std::cout << "options: "  << std::endl;
  67 //  std::cout << " -a           process all records within the file"  << std::endl;
  68 //  std::cout << "ssrep history [item]   displays the history of the specific item"  << std::endl;
  69 //  std::cout << "options: "  << std::endl;
  70 //  std::cout << " -L<-|+>      Filter versions by labels (-) all versions without labels"  << std::endl;
  71 //  std::cout << "                                        (+) only versions that have labels"  << std::endl;
  72 //  std::cout << " --del        include dead records"  << std::endl;
  73 //  std::cout << std::endl;
  74 //  std::cout << "ssrep get [item] [target]  retrieve a specific version of the item"  << std::endl;
  75 //  std::cout << "options: "  << std::endl;
  76 //  std::cout << " -f           force overwrite of target file" << std::endl;
  77 //  std::cout << " -v#          specifies the version to get"  << std::endl;
  78   std::cout << std::endl;
  79   std::cout << " -h --help    print help text" << std::endl;
  80 }
  81
  82 void COptions::ParseCommandLine (int argc, char* argv[])
  83 {
  84   for (int i = 1; i< argc; ++i)
  85   {
  86     if (argv[i][0] == '-')
  87     {
  88       if (IsArgChar (argv[i][1], '?'))
  89       {
  90         m_bHelp = true;
  91       }
  92       else if (IsArgChar (argv[i][1], 'a') || IsLongArgument (&argv[i][1], "all"))
  93       {
  94         m_bAllRecords = true;
  95       }
  96       else if (IsArgChar (argv[i][1], 'd'))
  97       {
  98         m_bIncludeDeleted = true;
  99         m_bOnlyDeleted = ToBool (argv[i][2], m_bOnlyDeleted);
 100       }
 101       else if (IsArgChar (argv[i][1], 'e'))
 102       {
 103         m_bExtended = true;
 104       }
 105       else if (IsArgChar (argv[i][1], 'f'))
 106       {
 107         m_bIncludeFiles = true;
 108         m_bOnlyFiles = ToBool (argv[i][2], m_bIncludeFiles );
 109       }
 110       else if (IsLongArgument (&argv[i][1], "force"))
 111       {
 112         m_bForceOverwrite = true;
 113       }
 114       else if (IsArgChar (argv[i][1], 'h') || IsLongArgument (&argv[i][1], "help"))
 115       {
 116         m_bHelp = true;
 117       }
 118       else if (IsArgChar (argv[i][1], 'i'))
 119       {
 120 // Ignore: Do not ask for input under any circumstances.
 121 //        m_bIgnore = true;
 122 //        m_bIgnore = ToBool (argv[i][2], m_bIgnore);
 123       }
 124       else if (IsArgChar (argv[i][1], 'l'))
 125       {
 126         m_bOnlyLabels = true;
 127         m_bIncludeLabels = ToBool (argv[i][2], m_bOnlyLabels);
 128       }
 129       else if (IsArgChar (argv[i][1], 'n') || IsLongArgument (&argv[i][1], "names"))
 130       {
 131         if (++i < argc)
 132           m_NamesDat = argv[i];
 133       }
 134       else if (IsArgChar (argv[i][1], 'o'))
 135       {
 136         m_LogFile = &argv[i][2];
 137       }
 138       else if (IsArgChar (argv[i][1], 'p'))
 139       {
 140         m_bPhys = true;
 141         m_bPhys = ToBool (argv[i][2], m_bPhys);
 142       }
 143
 144       else if (IsArgChar (argv[i][1], 'r') || IsLongArgument (&argv[i][1], "recursive"))
 145       {
 146         m_bRecursive = true;
 147         m_bRecursive = ToBool (argv[i][2], m_bRecursive);
 148       }
 149       else if (IsArgChar (argv[i][1], 's'))
 150       {
 151         m_SrcSafeIniPath = &argv[i][2];
 152       }
 153       else if (IsArgChar (argv[i][1], 'v') || IsLongArgument (&argv[i][1], "version"))
 154       {
 155         std::string filter;
 156         if (IsArgChar (argv[i][1], 'v'))
 157           filter = &argv[i][2];
 158         else if (++i < argc)
 159          filter = argv[i];
 160
 161         int start;
 162         char tilde;
 163         std::istrstream ist (filter.c_str (), filter.length());
 164         ist >> start;
 165         int end = start;
 166         ist >> tilde;
 167         ist >> end;
 168
 169         DELETE (m_pVersionFilter);
 170         m_pVersionFilter = new CVersionRangeFilter (end, start);
 171
 172         m_Version = start;
 173         m_VersionDate = filter;
 174       }
 175       else if (IsArgChar (argv[i][1], 'y'))
 176       {
 177         std::string userpw = &argv[i][2];
 178         m_User = userpw.substr(0, userpw.find (','));
 179         m_Password = userpw.substr(userpw.find (',')+1);
 180       }
 181     }
 182     else
 183     {
 184       if (m_Command.empty ())
 185       {
 186         m_Command = argv[i];
 187         // TODO: tolower bekommt ein (int) als parameter, std::string ist aber ein uchar --> Probleme mit negativen chars
 188         std::transform (m_Command.begin(), m_Command.end(), m_Command.begin(), tolower);
 189       }
 190       else
 191         m_Items.push_back (argv[i]);
 192     }
 193   }
 194
 195   if (m_Command == "get" && m_Items.size () > 1)
 196   {
 197     m_Dest = m_Items.back();
 198     m_Items.pop_back();
 199   }
 200 }
 201
 202
 203 bool COptions::IsLongArgument (const char* ch, const char* arg)
 204 {
 205   if (!ch)
 206     return false;
 207
 208   std::string str = (ch[0] == '-') ? &ch[1] : &ch[0];
 209   if (str == arg)
 210     return true;
 211
 212   return false;
 213 }
 214
 215 bool COptions::IsArgChar (const char ch, const char arg)
 216 {
 217   if (!ch)
 218     return false;
 219   else if (ch == arg)
 220     return true;
 221   else if (tolower(ch) == arg)
 222     return true;
 223   else if (toupper(ch) == arg)
 224     return true;
 225   return false;
 226 }
 227
 228 bool COptions::ToBool (const char ch, bool& val)
 229 {
 230   if (!ch)
 231     return val;
 232   if (ch == '-')
 233     val = false;
 234   else if (ch == '+')
 235     val = true;
 236
 237   return val;
 238 }