Added mechanism to understand passed Option value.

[command.git] / include / option.h

diff --git a/include/option.h b/include/option.h
index 241754d973b274286360e1772b9d038c2ccfecb3..dba1c16818c4aa65ce05fecd9b79b048e68cac4c 100644 (file)
--- a/include/option.h
+++ b/include/option.h
@@ -1,22 +1,38 @@
 #ifndef __COMMAND_OPTION_H
 #define __COMMAND_OPTION_H
 
+#include <string>
+#include <sstream>
+#include <stdexcept>
+
+#include "parameter.h"
+
 namespace command {
     /**
      * Class responsible for handling commandline options.
      * Options are non-required, named parameters of program.
      *
      * Example:
-     *  ./myprog OptionName OptionValue
+     *  ./myprog OptionName=OptionValue
      */
     template<typename OptionType>
     class Option
-        : Argument<OptionType> {
+        : public Parameter, public Callable<OptionType>  {
+    public:
+        typedef std::string OptionName;
     protected:
         /**
-         * Option name
+         * Current Option name
+         */
+        OptionName name;
+
+        /**
+         * Current Option value
          */
-        std::string name;
+        OptionType value;
+
+        /** Variable indicating if current Option was already used or not */
+        bool used = false;
 
     public:
         /**
@@ -26,8 +42,70 @@ namespace command {
          * @param description Description of current Option
          * @param function Function used to handle current Option.
          */
-        Argument(std::string name, std::string description, FunctionType function)
-            : name(name), Argument<OptionType>(description, function) {
+        Option(std::string name, const std::string & description, void (*function)(OptionType))
+            : Parameter(description), Callable<OptionType>(function), name(name) {
+        }
+
+        /**
+         *
+         */
+        virtual ~Option() { }
+
+        /**
+         *
+         */
+        virtual void handle() {
+            this->call(value);
+        }
+
+        /**
+         * Method used for checking if Option understands given user value.
+         * If so current Option is flagged as used and no more checks against
+         * it will be done in future.
+         *
+         * Passed value should be in form of:
+         *      OptionName=OptionValue
+         *
+         * If no equal sign is after OptionName part,
+         * std::invalid_argument exception with appropriate message is thrown
+         *
+         * If conversion of OptionValue part to OptionType failed,
+         * std::invalid_argument exception with appropriate message is thrown
+         *
+         * @param argv command line value against which test will be made.
+         *  User value should be in format: OptionName=OptionValue.
+         *
+         * @return If passed argv succesfully detected OptionName part as a
+         *  current option and its OptionValue part has been succesfully
+         *  converted to OptionType, returns true and Option is set as used one.
+         *  Otherwise returns false and can be used to check against next value.
+         *
+         * @throw std::invalid_argument when OptionName part has no equal sign
+         *  after itself
+         * @throw std::invalid_argument when OptionValue part failed conversion
+         *  to OptionType
+         */
+        virtual bool understand(const std::string & argv) {
+            if ((!used) &&
+                (argv.find(name) == 0)) {
+                std::size_t pos = argv.find("=");
+                if (pos != name.size()) {
+                    throw std::invalid_argument("Option: " + name + " requires value but no one has been provided");
+                }
+
+                std::stringstream ss;
+
+                ss << argv.substr(pos + 1);
+                ss >> value;
+
+                if (ss.fail()) {
+                    throw std::invalid_argument("Value for option: " + name + " failed conversion to the required type");
+                }
+
+                used = true;
+                return true;
+            }
+            return false;
         }
     };
 }