]> git.saurik.com Git - apt.git/blobdiff - apt-pkg/contrib/cmndline.cc
switch from std::set to std::vector as it is way more simple, a bit
[apt.git] / apt-pkg / contrib / cmndline.cc
index 6b36325c7c71949e0905abedb45aa85db031feb4..5a994409663e299b6e4ca6991f1a13ba28950709 100644 (file)
@@ -1,20 +1,23 @@
 // -*- mode: cpp; mode: fold -*-
 // Description                                                         /*{{{*/
-// $Id: cmndline.cc,v 1.1 1998/09/22 05:30:26 jgg Exp $
+// $Id: cmndline.cc,v 1.15 2003/02/10 01:40:58 doogie Exp $
 /* ######################################################################
 
    Command Line Class - Sophisticated command line parser
    
+   This source is placed in the Public Domain, do with it what you will
+   It was originally written by Jason Gunthorpe <jgg@debian.org>.
+   
    ##################################################################### */
                                                                        /*}}}*/
 // Include files                                                       /*{{{*/
-#ifdef __GNUG__
-#pragma implementation "apt-pkg/cmndline.h"
-#endif
 #include <apt-pkg/cmndline.h>
 #include <apt-pkg/error.h>
-#include <strutl.h>
+#include <apt-pkg/strutl.h>
+
+#include <apti18n.h>    
                                                                        /*}}}*/
+using namespace std;
 
 // CommandLine::CommandLine - Constructor                              /*{{{*/
 // ---------------------------------------------------------------------
 CommandLine::CommandLine(Args *AList,Configuration *Conf) : ArgList(AList), 
                                  Conf(Conf), FileList(0)
 {
+}
+                                                                       /*}}}*/
+// CommandLine::~CommandLine - Destructor                              /*{{{*/
+// ---------------------------------------------------------------------
+/* */
+CommandLine::~CommandLine()
+{
+   delete [] FileList;
 }
                                                                        /*}}}*/
 // CommandLine::Parse - Main action member                             /*{{{*/
@@ -29,6 +40,7 @@ CommandLine::CommandLine(Args *AList,Configuration *Conf) : ArgList(AList),
 /* */
 bool CommandLine::Parse(int argc,const char **argv)
 {
+   delete [] FileList;
    FileList = new const char *[argc];
    const char **Files = FileList;
    int I;
@@ -47,7 +59,10 @@ bool CommandLine::Parse(int argc,const char **argv)
       
       // Double dash signifies the end of option processing
       if (*Opt == '-' && Opt[1] == 0)
+      {
+        I++;
         break;
+      }
       
       // Single dash is a short option
       if (*Opt != '-')
@@ -59,7 +74,7 @@ bool CommandLine::Parse(int argc,const char **argv)
            Args *A;
            for (A = ArgList; A->end() == false && A->ShortOpt != *Opt; A++);
            if (A->end() == true)
-              return _error->Error("Command line option '%c' [from %s] is not known.",*Opt,argv[I]);
+              return _error->Error(_("Command line option '%c' [from %s] is not known."),*Opt,argv[I]);
 
            if (HandleOpt(I,argc,argv,Opt,A) == false)
               return false;
@@ -79,12 +94,13 @@ bool CommandLine::Parse(int argc,const char **argv)
           stringcasecmp(Opt,OptEnd,A->LongOpt) != 0; A++);
       
       // Failed, look for a word after the first - (no-foo)
+      bool PreceedMatch = false;
       if (A->end() == true)
       {
         for (; Opt != OptEnd && *Opt != '-'; Opt++);
 
         if (Opt == OptEnd)
-           return _error->Error("Command line option %s is not understood",argv[I]);
+           return _error->Error(_("Command line option %s is not understood"),argv[I]);
         Opt++;
         
         for (A = ArgList; A->end() == false &&
@@ -92,18 +108,26 @@ bool CommandLine::Parse(int argc,const char **argv)
 
         // Failed again..
         if (A->end() == true && OptEnd - Opt != 1)
-           return _error->Error("Command line option %s is not understood",argv[I]);
-        
+           return _error->Error(_("Command line option %s is not understood"),argv[I]);
+
         // The option could be a single letter option prefixed by a no-..
-        for (A = ArgList; A->end() == false && A->ShortOpt != *Opt; A++);
-        
         if (A->end() == true)
-           return _error->Error("Command line option %s is not understood",argv[I]);
+        {
+           for (A = ArgList; A->end() == false && A->ShortOpt != *Opt; A++);
+           
+           if (A->end() == true)
+              return _error->Error(_("Command line option %s is not understood"),argv[I]);
+        }
+        
+        // The option is not boolean
+        if (A->IsBoolean() == false)
+           return _error->Error(_("Command line option %s is not boolean"),argv[I]);
+        PreceedMatch = true;
       }
       
       // Deal with it.
       OptEnd--;
-      if (HandleOpt(I,argc,argv,OptEnd,A) == false)
+      if (HandleOpt(I,argc,argv,OptEnd,A,PreceedMatch) == false)
         return false;
    }
    
@@ -111,7 +135,9 @@ bool CommandLine::Parse(int argc,const char **argv)
    for (; I != argc; I++)
       *Files++ = argv[I];
    *Files = 0;
-   
+
+   SaveInConfig(argc, argv);
+
    return true;
 }
                                                                        /*}}}*/
@@ -121,7 +147,7 @@ bool CommandLine::Parse(int argc,const char **argv)
    and looks for specific patterns in the string, it gets tokanized
    -ruffly- like -*[yes|true|enable]-(o|longopt)[=][ ][argument] */
 bool CommandLine::HandleOpt(int &I,int argc,const char *argv[],
-                           const char *&Opt,Args *A)
+                           const char *&Opt,Args *A,bool PreceedMatch)
 {
    const char *Argument = 0;
    bool CertainArg = false;
@@ -136,7 +162,7 @@ bool CommandLine::HandleOpt(int &I,int argc,const char *argv[],
       
       // Equals was specified but we fell off the end!
       if (Opt[1] == '=' && Argument == 0)
-        return _error->Error("Option %s requires an argument.",argv[I]);
+        return _error->Error(_("Option %s requires an argument."),argv[I]);
       if (Opt[1] == '=')
         CertainArg = true;
         
@@ -152,20 +178,47 @@ bool CommandLine::HandleOpt(int &I,int argc,const char *argv[],
       else
         Argument = Opt + 1;
    }
-
+   
    // Option is an argument set
    if ((A->Flags & HasArg) == HasArg)
    {
       if (Argument == 0)
-        return _error->Error("Option %s requires an argument.",argv[I]);
+        return _error->Error(_("Option %s requires an argument."),argv[I]);
       Opt += strlen(Opt);
       I += IncI;
       
       // Parse a configuration file
       if ((A->Flags & ConfigFile) == ConfigFile)
         return ReadConfigFile(*Conf,Argument);
+
+      // Arbitrary item specification
+      if ((A->Flags & ArbItem) == ArbItem)
+      {
+        const char *J;
+        for (J = Argument; *J != 0 && *J != '='; J++);
+        if (*J == 0)
+           return _error->Error(_("Option %s: Configuration item specification must have an =<val>."),argv[I]);
+
+        // = is trailing
+        if (J[1] == 0)
+        {
+           if (I+1 >= argc)
+              return _error->Error(_("Option %s: Configuration item specification must have an =<val>."),argv[I]);
+           Conf->Set(string(Argument,J-Argument),string(argv[I++ +1]));
+        }
+        else
+           Conf->Set(string(Argument,J-Argument),string(J+1));
+        
+        return true;
+      }
       
-      Conf->Set(A->ConfName,Argument);
+      const char *I = A->ConfName;
+      for (; *I != 0 && *I != ' '; I++);
+      if (*I == ' ')
+        Conf->Set(string(A->ConfName,0,I-A->ConfName),string(I+1) + Argument);
+      else
+        Conf->Set(A->ConfName,string(I) + Argument);
+        
       return true;
    }
    
@@ -180,10 +233,10 @@ bool CommandLine::HandleOpt(int &I,int argc,const char *argv[],
         
         // Conversion failed and the argument was specified with an =s
         if (EndPtr == Argument && CertainArg == true)
-           return _error->Error("Option %s requires an integer argument, not '%s'",argv[I],Argument);
+           return _error->Error(_("Option %s requires an integer argument, not '%s'"),argv[I],Argument);
 
         // Conversion was ok, set the value and return
-        if (EndPtr != Argument)
+        if (EndPtr != 0 && EndPtr != Argument && *EndPtr == 0)
         {
            Conf->Set(A->ConfName,Value);
            Opt += strlen(Opt);
@@ -207,11 +260,16 @@ bool CommandLine::HandleOpt(int &I,int argc,const char *argv[],
       char Buffer[300];
       if (Argument == 0)
       {
-        if (strlen(argv[I]) >= sizeof(Buffer))
-           return _error->Error("Option '%s' is too long",argv[I]);
+        if (PreceedMatch == false)
+           break;
         
+        if (strlen(argv[I]) >= sizeof(Buffer))
+           return _error->Error(_("Option '%s' is too long"),argv[I]);
+
+        // Skip the leading dash
         const char *J = argv[I];
         for (; *J != 0 && *J == '-'; J++);
+        
         const char *JEnd = J;
         for (; *JEnd != 0 && *JEnd != '-'; JEnd++);
         if (*JEnd != 0)
@@ -225,14 +283,10 @@ bool CommandLine::HandleOpt(int &I,int argc,const char *argv[],
            break;
       }
 
-      // Check for positives
-      if (strcasecmp(Argument,"yes") == 0 ||
-         strcasecmp(Argument,"true") == 0 ||
-         strcasecmp(Argument,"with") == 0 ||
-         strcasecmp(Argument,"enable") == 0)
+      // Check for boolean
+      Sense = StringToBool(Argument);
+      if (Sense >= 0)
       {
-        Sense = 1;
-
         // Eat the argument     
         if (Argument != Buffer)
         {
@@ -242,25 +296,8 @@ bool CommandLine::HandleOpt(int &I,int argc,const char *argv[],
         break;
       }
 
-      // Check for negatives
-      if (strcasecmp(Argument,"no") == 0 ||
-         strcasecmp(Argument,"false") == 0 ||
-         strcasecmp(Argument,"without") == 0 ||
-         strcasecmp(Argument,"disable") == 0)
-      {
-        Sense = 0;
-        
-        // Eat the argument     
-        if (Argument != Buffer)
-        {
-           Opt += strlen(Opt);
-           I += IncI;
-        }       
-        break;
-      }
-      
       if (CertainArg == true)
-        return _error->Error("Sense %s is not understood, try true or false.",Argument);
+        return _error->Error(_("Sense %s is not understood, try true or false."),Argument);
       
       Argument = 0;
    }
@@ -278,3 +315,79 @@ bool CommandLine::HandleOpt(int &I,int argc,const char *argv[],
    return true;
 }
                                                                        /*}}}*/
+// CommandLine::FileSize - Count the number of filenames               /*{{{*/
+// ---------------------------------------------------------------------
+/* */
+unsigned int CommandLine::FileSize() const
+{
+   unsigned int Count = 0;
+   for (const char **I = FileList; I != 0 && *I != 0; I++)
+      Count++;
+   return Count;
+}
+                                                                       /*}}}*/
+// CommandLine::DispatchArg - Do something with the first arg          /*{{{*/
+// ---------------------------------------------------------------------
+/* */
+bool CommandLine::DispatchArg(Dispatch *Map,bool NoMatch)
+{
+   int I;
+   for (I = 0; Map[I].Match != 0; I++)
+   {
+      if (strcmp(FileList[0],Map[I].Match) == 0)
+      {
+        bool Res = Map[I].Handler(*this);
+        if (Res == false && _error->PendingError() == false)
+           _error->Error("Handler silently failed");
+        return Res;
+      }
+   }
+   
+   // No matching name
+   if (Map[I].Match == 0)
+   {
+      if (NoMatch == true)
+        _error->Error(_("Invalid operation %s"),FileList[0]);
+   }
+   
+   return false;
+}
+                                                                       /*}}}*/
+// CommandLine::SaveInConfig - for output later in a logfile or so     /*{{{*/
+// ---------------------------------------------------------------------
+/* We save the commandline here to have it around later for e.g. logging.
+   It feels a bit like a hack here and isn't bulletproof, but it is better
+   than nothing after all. */
+void CommandLine::SaveInConfig(unsigned int const &argc, char const * const * const argv)
+{
+   char cmdline[100 + argc * 50];
+   unsigned int length = 0;
+   bool lastWasOption = false;
+   bool closeQuote = false;
+   for (unsigned int i = 0; i < argc && length < sizeof(cmdline); ++i, ++length)
+   {
+      for (unsigned int j = 0; argv[i][j] != '\0' && length < sizeof(cmdline)-1; ++j, ++length)
+      {
+        cmdline[length] = argv[i][j];
+        if (lastWasOption == true && argv[i][j] == '=')
+        {
+           // That is possibly an option: Quote it if it includes spaces,
+           // the benefit is that this will eliminate also most false positives
+           const char* c = &argv[i][j+1];
+           for (; *c != '\0' && *c != ' '; ++c);
+           if (*c == '\0') continue;
+           cmdline[++length] = '"';
+           closeQuote = true;
+        }
+      }
+      if (closeQuote == true)
+        cmdline[length++] = '"';
+      // Problem: detects also --hello
+      if (cmdline[length-1] == 'o')
+        lastWasOption = true;
+      cmdline[length] = ' ';
+   }
+   cmdline[--length] = '\0';
+   _config->Set("CommandLine::AsString", cmdline);
+}
+                                                                       /*}}}*/