]> Dogcows Code - chaz/yoink/blobdiff - src/Moof/Settings.cc
extreme refactoring
[chaz/yoink] / src / Moof / Settings.cc
diff --git a/src/Moof/Settings.cc b/src/Moof/Settings.cc
new file mode 100644 (file)
index 0000000..d13302d
--- /dev/null
@@ -0,0 +1,135 @@
+
+/*******************************************************************************
+
+ Copyright (c) 2009, Charles McGarvey
+ All rights reserved.
+ Redistribution   and   use  in  source  and  binary  forms,  with  or  without
+ modification, are permitted provided that the following conditions are met:
+   * Redistributions  of  source  code  must retain the above copyright notice,
+     this list of conditions and the following disclaimer.
+   * Redistributions  in binary form must reproduce the above copyright notice,
+     this  list of conditions and the following disclaimer in the documentation
+     and/or other materials provided with the distribution.
+ THIS  SOFTWARE  IS  PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+ AND  ANY  EXPRESS  OR  IMPLIED  WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+ IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+ DISCLAIMED.  IN  NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
+ FOR  ANY  DIRECT,  INDIRECT,  INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+ DAMAGES  (INCLUDING,  BUT  NOT  LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
+ SERVICES;  LOSS  OF  USE,  DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
+ CAUSED  AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
+ OR  TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+ OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+
+*******************************************************************************/
+
+#include <sstream>
+#include <cstdlib>             // getenv
+#include <cstring>             // strchr
+
+#include <boost/algorithm/string.hpp>
+
+#include "Settings.hh"
+
+
+namespace Mf {
+
+
+Settings::Settings(int argc, char* argv[])
+{
+       parseArgs(argc, argv);
+}
+
+
+void Settings::parseArgs(int argc, char* argv[])
+{
+       for (int i = 1; i < argc; i++)
+       {
+               char* where = strchr(argv[i], (int)'=');
+
+               if (where)
+               {
+                       std::string key(argv[i], (size_t)(where - argv[i]));
+                       std::string stringValue(where + 1);
+
+                       std::stringstream stream;
+                       stream << stringValue << std::endl;
+
+                       Deserializer deserializer(stream);
+
+                       try
+                       {
+                               SerializablePtr value(deserializer.deserialize());
+                               map_[key] = value;
+                       }
+                       catch (std::exception e)
+                       {
+                               // it doesn't deserialize to anything we know, so just store it
+                               // as a string
+                               map_[key] = SerializablePtr(new SerializableString(stringValue));
+                       }
+               }
+       }
+}
+
+
+void Settings::loadFromFile(const std::string& filePath, bool precedence)
+{
+       std::vector<std::string> paths;
+       boost::split(paths, filePath, boost::is_any_of(":"));
+
+       loadFromFiles(paths, precedence);
+}
+
+void Settings::loadFromFiles(const std::vector<std::string>& filePaths,
+               bool precedence)
+{
+       std::vector<std::string>::const_iterator it;
+
+       char* home = getenv("HOME");
+
+       for (it = filePaths.begin(); it != filePaths.end(); it++)
+       {
+               std::string path = *it;
+
+               if (home)
+               {
+                       boost::replace_first(path, "$HOME", home);
+               }
+
+               Deserializer deserializer(*it, true);
+
+               std::cout << "Looking for a config file at " << path << std::endl;
+               try
+               {
+                       SerializablePtr obj = deserializer.deserialize();
+                       std::map<std::string,SerializablePtr> map;
+                       if (obj && obj->get(map))
+                       {
+                               if (!precedence)
+                               {
+                                       map_.insert(map.begin(), map.end());
+                               }
+                               else
+                               {
+                                       map.insert(map_.begin(), map_.end());
+                                       map_ = map;
+                               }
+                       }
+               }
+               catch (Deserializer::Exception e)
+               {
+                       std::cerr << "Cannot load settings from " << *it <<
+                               " because an exception was thrown: " << e.what() << std::endl;
+               }
+       }
+}
+
+
+} // namepsace Mf
+
+/** vim: set ts=4 sw=4 tw=80: *************************************************/
+
This page took 0.024973 seconds and 4 git commands to generate.