Environment.cpp

Go to the documentation of this file.

#include "GaudiKernel/Environment.h"
#include "GaudiKernel/System.h"
#include <cstdlib>

namespace
{
  std::string i_resolve( std::string& source, int recursions )
  {
    if ( recursions > 0 ) {
      int lvl = 0, mx_lvl = 0;
      for ( const char *c = source.c_str(), *beg = nullptr; *c != 0; ++c ) {
        switch ( *c ) {
        case '$':
          if ( *( c + 1 ) == '{' ) {
            lvl++;
            if ( lvl > mx_lvl ) {
              mx_lvl = lvl;
              beg    = c;
            }
            c += 2;
          }
          break;
        case '}':
          if ( lvl == mx_lvl ) {
            std::string env( beg + 2, c - beg - 2 );
            std::string rep;
            if ( !System::getEnv( env.c_str(), rep ) ) rep = i_resolve( env, --recursions );
            if ( rep.length() ) {
              std::string e( beg, c - beg + 1 );
              size_t idx = std::string::npos;
              while ( ( idx = source.find( e ) ) != std::string::npos ) {
                source.replace( idx, e.length(), rep );
              }
              return i_resolve( source, --recursions );
            } else {
              // error: environment cannot be resolved....
              // Try to continue, but there is not too much hope.
            }
          }
          lvl--;
          break;
        default:
          break;
        }
      }
    }
    return source;
  }
}

StatusCode System::resolveEnv( const std::string& var, std::string& res, int recursions )
{
  std::string source = var;
  res                = i_resolve( source, recursions );
  if ( res.find( "${" ) == std::string::npos ) {
    return StatusCode::SUCCESS;
  }
  return StatusCode::FAILURE;
}

std::string System::homeDirectory()
{
  // Return the user's home directory.
  std::string home_dir = "./";
  // Try to replace the current value with the content of several
  // environment variables
  if ( !( System::getEnv( "home", home_dir ) || System::getEnv( "HOME", home_dir ) ) ) {
    // for Windows NT HOME might be defined as either $(HOMESHARE)/$(HOMEPATH)
    //                                         or     $(HOMEDRIVE)/$(HOMEPATH)
    if ( System::getEnv( "HOMESHARE", home_dir ) || System::getEnv( "HOMEDRIVE", home_dir ) ) {
      std::string path;
      if ( System::getEnv( "HOMEPATH", path ) ) home_dir += path;
    }
  }
  return home_dir;
}

std::string System::tempDirectory()
{
  // Return a user configured or systemwide directory to create
  // temporary files in.
  std::string dir;
  if ( System::getEnv( "TEMP", dir ) || System::getEnv( "TEMPDIR", dir ) || System::getEnv( "TEMP_DIR", dir ) ||
       System::getEnv( "TMP", dir ) || System::getEnv( "TMPDIR", dir ) || System::getEnv( "TMP_DIR", dir ) )
    return dir;
  else
    return homeDirectory();
}