ServiceMainOptions.h

Go to the documentation of this file.

  /* Copyright 2006-2013 Savarese Software Research Corporation.
   *
   * Licensed under the Apache License, Version 2.0 (the "License");
   * you may not use this file except in compliance with the License.
   * You may obtain a copy of the License at
   *
   * http://www.savarese.com/software/ApacheLicense-2.0
   *
   * Unless required by applicable law or agreed to in writing, software
   * distributed under the License is distributed on an "AS IS" BASIS,
   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
   * See the License for the specific language governing permissions and
   * limitations under the License.
   */
  
  #ifndef __SSRC_WISP_UTILITY_SERVICE_MAIN_OPTIONS_H
  #define __SSRC_WISP_UTILITY_SERVICE_MAIN_OPTIONS_H
  
  #include <ssrc/wisp/utility/BasicServiceOptions.h>
  
  #include <fstream>
  
  __BEGIN_NS_SSRC_WISP_UTILITY
  
  class ServiceMainOptions : public BasicServiceOptions {
   boost::program_options::options_description _custom_description;
  
  protected:
  
   ServiceMainOptions() :
   BasicServiceOptions(), _custom_description("Wisp Service"),
   wisp_version(false)
   {
   using namespace boost::program_options;
  
   _custom_description.add_options()
   ("config,f", value<std::vector<std::string> >()->composing(),
   "Specify a configuration file from which to read additional options.")
   ("wisp-version",
   "Print Wisp version with which the service was compiled.");
  
   description.add(_custom_description);
   }
  
   virtual void notify(boost::program_options::variables_map & vm) {
   namespace po = boost::program_options;
  
   if(vm.count("config") > 0) {
   std::vector<std::string> config =
   vm.find("config")->second.as<std::vector<std::string> >();
  
   for(std::vector<std::string>::iterator file = config.begin(); file != config.end(); ++file) {
   std::ifstream ifs(file->c_str());
  
   if(ifs.is_open())
   po::store(po::parse_config_file(ifs, description), vm);
   else
   throw std::invalid_argument(*file);
   }
   }
  
   BasicServiceOptions::notify(vm);
   wisp_version = (vm.count("wisp-version") > 0);
   }
  
  public:
   bool wisp_version;
  
   virtual ~ServiceMainOptions() = default;
  };
  
  __END_NS_SSRC_WISP_UTILITY
  
  #endif