pairfile.cpp

/***********************************************************
 *   this code by Peter Semiletov is Public Domain         *
 **********************************************************/

#include <fstream>
#include <iostream>
#include <sstream>
#include <stdlib.h>
#include <string.h>
#include <cstring>
#include <algorithm>
#include <sys/stat.h>


#include "pairfile.h"
#include "libretta_utils.h"


using namespace std;

typedef std::map <string, string> TSPair;



bool file_exists (const string &name)
{
  if (name.empty())
     return false;

  struct stat buffer;
  return (stat (name.c_str(), &buffer) == 0);
}


void CPairFile::save()
{
  if (file_name.empty())
     return;

  ofstream myfile (file_name.c_str());
  if (! myfile.is_open())
      return; 
 
  TSPair::const_iterator end = values.end(); 

  for (TSPair::const_iterator it = values.begin(); it != end; it++)
      {
       myfile << it->first << "=" << it->second << '\n';
      }
  
  myfile.close();
}


void CPairFile::set_string (const string &key, const string &value)
{
  values[key] = value; 
}


void CPairFile::set_float (const string &key, float value)
{
  char buffer [64];
  //sprintf (buffer, "%f", value);
  snprintf (buffer, 64, "%f", value);
  string n (buffer);
  values[key] = n; 
}


float CPairFile::get_float (const string &key, float def_value)
{
  float result = def_value; 

  if (values[key].size() == 0)
     return result; 
  
  const char *s = values[key].c_str();
  if (s)
     result = atof (s);
   
  return result;
}


void CPairFile::set_int (const string &key, int value)
{
  char buffer [64];
  sprintf (buffer, "%d", value);
  string n (buffer);
  values[key] = n; 
}


bool CPairFile::get_bool (const string &key, bool def_value)
{
  bool result = def_value;

  if (values.find (key) == values.end())
     return result;

  if (values[key] == "true")
     return true;

  return result;
}



int CPairFile::get_int (const string &key, int def_value)
{

  int result = def_value;
  
  if (values.find (key) == values.end())
      return result;

  const char *s = values[key].c_str();
  if (s)
     result = atoi (s);
   
  return result;
}


size_t CPairFile::get_num (const string &key, unsigned long long def_value)
{
  size_t result = def_value;

  if (values.find (key) == values.end())
     return result;

  const char *s = values[key].c_str();
  if (! s)
     return result;

  char* end;
  result = strtoull (s, &end, 10);

  return result;
}


string CPairFile::get_string (const string &key, 
                              const string &def_value)
{
  string result = def_value;

  if (values.find (key) == values.end())
     return result;

  string s = values[key];
  if (! s.empty())
     result = s;
   
  return result;
}


CPairFile::CPairFile (const string &fname, bool from_data)
{
  if (fname.empty())
     return;

  if (from_data)
     {
      stringstream st (fname);
      string line;
     
      while (getline (st, line))
           {
            if (line.empty())
               continue;

            size_t pos = line.find ("=");
                 
            if (pos == string::npos)
                continue;
        
            if (pos > line.size())
                continue;
                    
            string a = line.substr (0, pos);
            string b = line.substr (pos + 1, line.size() - pos);
            
            values[a] = b;
           }
     
     return; 
    }
 

  file_name = fname;
  if (! file_exists(file_name))
     return;

  ifstream infile (file_name.c_str());

  if (infile.fail())
     {
    //  cout << "CPairFile::CPairFile - Could not open file: " << fname << endl;
      return;
     }

  string line;

  while (getline (infile, line))
        {
         size_t pos = line.find ("=");
         
         if (pos == string::npos)
            continue;
        
         //if (line[0] == '#')
           // continue;

         string a = line.substr (0, pos);
         string b = line.substr (pos + 1, line.size() - pos);

         values[a] = b;
        }

  infile.close();
};


CPairFile::CPairFile (int argc, char *argv[])
{
  stringstream st;

  for (int i = 0; i < argc; i++)
      {
       st << argv[i];
       st << endl;
      }

  string line;

  while (getline (st, line))
        {
         if (line.empty())
            continue;

         size_t pos = line.find ("=");

         if (pos == string::npos)
            {
             line.erase (0,2);
             values[line] = "true";
             continue;
            }

         if (pos >= line.size())
             continue;

         string a = line.substr (0, pos);
         a.erase (0,2);

         string b = line.substr (pos + 1, line.size() - pos);
         values[a] = b;
        }

}


CPairFile::CPairFile (vector <string> envars)
{
  for (size_t i = 0; i < envars.size(); i++)
      {
       const char* env_p = std::getenv (envars[i].c_str());
       if (env_p)
          {
           string param = envars[i];
           param.erase (0, 4);

           std::for_each (
                          param.begin(),
                          param.end(),
                          [](char & c) {
                                        c = ::tolower(c);
                                       });

           string value = env_p;
           values[param] = value;
         }
     }

}