my_file_parser.cpp

Go to the documentation of this file.

#include <file_parser.hh>
#include <string.h>
#include <stdio.h>

#if 0
// Need an ostream inserter for the special icstring type.
inline std::ostream& operator<<( std::ostream& strm, const FileParser::icstring& s )
{
  return strm << std::string(s.data(),s.length() );
}
#endif

/****************************************************************/
/**************************** File Parser ***********************/

static char error[256];

/******************/
/*** constructor ***/

FileParser::FileParser( std::string const file )
{
  if ( ! load( file ) )
    {
      sprintf( error, "File '%s' was not found.\n", file.c_str() );
      fprintf( stderr, "%s", error );
      throw error;
    }
}


/******************/
/*** destructor ***/

FileParser::~FileParser()
{
}

/************/
/*** load ***/
int
FileParser::load( std::string const file )
{
  FILE *fd = fopen( file.c_str(), "r" );
  if ( ! fd )
    return 0;

  f_lst.clear();
  f_seclst.clear();

  char *sec = NULL;
  SectionList::iterator secit;

  const int size = 1024;
  char line[size];

  while ( fgets( line, size, fd ) ) {
    // Find the first # which indicates the start of comment.
    char *sep = strchr( line,'#' );
    // If it's at the beginning of the line
    // loop.
    if ( sep == line )
      continue;

    // If it's not at the beginning of the line,
    // assign it to "null", terminating the string,
    // and effectively commenting to end of line.
    if ( sep )
      *sep = '\0';

    // Find the start of the key.
    // Trim the whitespace from the front of the line.
    char *start_key = line;
    while ( isspace( *start_key ) && (*start_key != '\0') ) {
      start_key++;
    }

    // Nothing but whitespace.  Loop.
    if ( *start_key == '\0' )
      continue;

    // Are we inside a section [section_name]?
    char *sec_ini = strchr( line, '[' );
    char *sec_end = strchr( line, ']' );
    if ( sec_ini && sec_end ) {

      sec = sec_ini+1;
      // Left trim the whitespace.
      while ( isspace( *sec ) && ( *sec != '\0' ) ) {
  sec++;
      }
      // Right trim the value.
      char *tmp = sec + strlen( sec ) - 1;
      while ( isspace( *tmp ) || ( *tmp != ']' ) )
  *tmp-- = '\0';
      ItemList lst;
      f_seclst.push_back( std::make_pair( sec, lst ) );
      secit = f_seclst.end() -1;
      continue;
    }

    // Separate key and value
    sep = strchr( line, '=' );
    if ( sep ) {
      // Find the start of the value.
      char *start_val = sep+1;
      // Left trim the whitespace.
      while ( isspace( *start_val ) && ( *start_val != '\0' ) ) {
  start_val++;
      }
      // No value?  loop.
      if ( *start_val == '\0' )
  continue;

      // Null terminate the key.
      *sep-- = '\0';
      // and right trim it.
      while( isspace( *sep ) )
  *sep-- = '\0';

      // Right trim the value.
      sep = start_val + strlen( start_val ) - 1;
      // Remember the \0 we substitued for the '='?
      // use that to terminate the loop.
      while ( isspace( *sep ) && ( *sep != '\0' ) )
  *sep-- = '\0';
      
      if ( *sep == '\0' )
  continue;
      
      if ( sec != NULL ) {

        (*secit).second.push_back( std::make_pair( start_key, start_val ) );
      }
      else {

        f_lst.push_back( std::make_pair( start_key, start_val ) );
      }
    }
  }
  
  fclose( fd );
  return 1;
}


/************/
/*** _get ***/
std::string
FileParser::_get( std::string const key, ItemList lst ) const
{
  ItemList::const_iterator it = lst.begin();

  while ( it != lst.end() ) {

    if ( it->first == key.c_str() ) {

      return it->second;
    }

    ++it;
  }

  return "";
}


/***********/
/*** get ***/
std::string
FileParser::get( std::string const key ) const
{
  return _get( key, f_lst );
}


/***********/
/*** get ***/
std::string
FileParser::get( std::string const section, std::string const key ) const
{
  SectionList::const_iterator it = f_seclst.begin();

  while ( it != f_seclst.end() ) {

    if ( it->first == section.c_str() ) {

      return _get( key, it->second );
    }

    ++it;
  }

  return "";
}


/**************/
/*** _count ***/
int 
FileParser::_count( std::string const key, ItemList lst ) const
{
  int n = 0;

  ItemList::const_iterator it = lst.begin();

  while ( it != lst.end() ) {

    if ( it->first == key.c_str() ) {

      ++n;
    }

    ++it;
  }

  return n;
}


/*************/
/*** count ***/
int
FileParser::count( std::string const key ) const
{
  return _count( key, f_lst );
}


/*************/
/*** count ***/
int
FileParser::count( std::string const section, std::string const key ) const
{
  SectionList::const_iterator it = f_seclst.begin();

  while ( it != f_seclst.end() ) {

    if (  it->first == section.c_str() ) {

      return _count( key, it->second );
    }

    ++it;
  }

  return 0;
}


/****************/
/*** _get All ***/
std::vector<std::string>
FileParser::_getAll( std::string const key, ItemList lst ) const
{
  std::vector<std::string> values;
  ItemList::const_iterator it = lst.begin();

  while ( it != lst.end() ) {

    if ( it->first == key.c_str() ) {

      values.push_back( it->second );
    }

    ++it;
  }
  return values;
}


/***************/
/*** get All ***/
std::vector<std::string>
FileParser::getAll( std::string const key ) const
{
  return _getAll( key, f_lst );
}


/***************/
/*** get All ***/
std::vector<std::string>
FileParser::getAll( std::string const section, std::string const key ) const
{
  SectionList::const_iterator it = f_seclst.begin();

  while ( it != f_seclst.end() ) {

    if (  it->first == section.c_str() ) {

      return _getAll( key, it->second );
    }

    ++it;
  }

  std::vector<std::string> values;
  return values;
}