source/configuration.cpp

   1 #include <ail/configuration.hpp>
   2
   3 namespace ail
   4 {
   5         configuration::configuration()
   6         {
   7         }
   8
   9         configuration::configuration(std::string const & file_name)
  10         {
  11                 load(file_name);
  12         }
  13
  14         bool configuration::load(std::string const & new_file_name)
  15         {
  16                 file_name = new_file_name;
  17
  18                 values.clear();
  19
  20                 std::string input;
  21                 bool success = read_file(file_name, input);
  22                 if(success == false)
  23                 {
  24                         return false;
  25                 }
  26
  27                 std::vector<std::string> lines = tokenise(input, "\n");
  28                 for(std::vector<std::string>::iterator i = lines.begin(), end = lines.end(); i != end; ++i)
  29                 {
  30                         std::string line = trim(*i);
  31
  32                         bool is_empty = line.empty();
  33
  34                         if(
  35                                 (is_empty == true)
  36                                         ||
  37                                 (
  38                                         (is_empty == false)
  39                                                 &&
  40                                         (line[0] == '#')
  41                                 )
  42                         )
  43                         {
  44                                 continue;
  45                         }
  46
  47                         std::size_t offset = line.find('=');
  48                         if(offset == std::string::npos)
  49                         {
  50                                 continue;
  51                         }
  52
  53                         std::string
  54                                 variable = right_trim(line.substr(0, offset)),
  55                                 value = left_trim(line.substr(offset + 1));
  56
  57                         if(
  58                                 (value.length() >= 2)
  59                                         &&
  60                                 (value[0] == '"')
  61                                         &&
  62                                 (*(value.end() - 1) == '"')
  63                         )
  64                         {
  65                                 value.erase(value.begin());
  66                                 value.erase(value.end() - 1);
  67                         }
  68
  69                         values[variable] = value;
  70                 }
  71
  72                 return true;
  73         }
  74
  75         bool configuration::read_string(std::string const & variable_name, std::string & output)
  76         {
  77                 std::map<std::string, std::string>::iterator search = values.find(variable_name);
  78                 if(search == values.end())
  79                         return false;
  80
  81                 output = search->second;
  82                 return true;
  83         }
  84
  85         std::string configuration::string(std::string const & variable_name)
  86         {
  87                 std::string output;
  88                 bool success = read_string(variable_name, output);
  89                 if(success == false)
  90                         throw exception("Unable to find string value \"" + variable_name + "\" in \"" + file_name + "\"");
  91                 return output;
  92         }
  93
  94         std::string configuration::string(std::string const & variable_name, std::string const & default_value)
  95         {
  96                 std::string output;
  97                 bool success = read_string(variable_name, output);
  98                 if(success == false)
  99                         return default_value;
 100
 101                 return output;
 102         }
 103 }