apps/common.h

Go to the documentation of this file.

// =============================================================================
// =============================================================================

#ifndef HF_APP_COMMON_H
#define HF_APP_COMMON_H

#include <map>

#include "hanfun.h"

// =============================================================================
// Command API
// =============================================================================

struct ICommand
{
   protected:

   static std::map<std::string, ICommand *> registry;

   public:

   virtual const std::string &key() const = 0;

   virtual const std::string usage(bool format = false) const = 0;

   virtual void run(std::vector<std::string> &args) = 0;

   static void add(ICommand *command);

   static void remove(ICommand *command);

   static void run(std::string &cmd, std::vector<std::string> &args);

   static std::ostream &help(std::ostream &stream);
};

class Command: public ICommand
{
   protected:

   const std::string _key;    
   const std::string _usage;  

   public:

   Command(const char *__key, const char *__usage):
      _key(__key), _usage(__usage)
   {}

   const std::string &key() const
   {
      return _key;
   }

   const std::string usage(bool format = false) const
   {
      if (format)
      {
         std::string result = _usage;

         for (std::string::size_type pos = 0; (pos = result.find(":", pos)) != std::string::npos;
              pos += 3 - 1)
         {
            result.replace(pos, 1, " : ");
         }

         return result;
      }
      else
      {
         return _usage;
      }
   }
};

#define COMMAND(_name, _key, _help)              \
   struct Command_## _name: public Command       \
   {                                             \
      Command_##_name(): Command(_key, _help) {} \
                                                 \
      void run(std::vector<std::string>&args);   \
   };                                            \
   Command_##_name command##_name;               \
   void Command_##_name::run(std::vector<std::string>&args)

#define COMMAND_ADD(_name)            \
   {                                  \
      ICommand::add(&command##_name); \
   }

// =============================================================================
// Parser helpers
// =============================================================================

#define STRTOL(X)       strtol(X.c_str(), NULL, 10);

#define STRTOL_HEX(X)   strtol(X.c_str(), NULL, 16);

#endif /* HF_APP_COMMON_H */