Env.h

Go to the documentation of this file.

/*---------------------------------------------------------------------\
|                          ____ _   __ __ ___                          |
|                         |__  / \ / / . \ . \                         |
|                           / / \ V /|  _/  _/                         |
|                          / /__ | | | | | |                           |
|                         /_____||_| |_| |_|                           |
|                                                                      |
\---------------------------------------------------------------------*/
#ifndef ZYPP_BASE_ENV_H
#define ZYPP_BASE_ENV_H
 
#include <cstdlib>
#include <string>
#include <memory>
#include <zypp-core/TriBool.h>
#include <zypp-core/base/String.h>

namespace zypp
{
  namespace env
  {
    inline TriBool getenvBool( const C_Str & var_r )
    {
      const char * v = ::getenv( var_r.c_str() );
      if ( v )
        return str::strToTriBool( v );
      return indeterminate;
    }

    struct ScopedSet
    {
      ScopedSet( const ScopedSet & )             = delete;
      ScopedSet & operator=( const ScopedSet & ) = delete;
 
      ScopedSet( ScopedSet && )                  = default;
      ScopedSet & operator=( ScopedSet && )      = default;
 
    public:
      ScopedSet()
      {}

      ScopedSet( std::string var_r, const char * val_r )
      : _var { std::move(var_r) }
      {
        if ( !_var.empty() )
        {
          if ( const char * orig = ::getenv( _var.c_str() ) )
            _val.reset( new std::string( orig ) );
          setval( val_r );
        }
      }

      ~ScopedSet()
      {
        if ( !_var.empty() )
          setval( _val ? _val->c_str() : nullptr );
      }
 
    private:
      void setval( const char * val_r )
      {
        if ( val_r )
          ::setenv( _var.c_str(), val_r, 1 );
        else
          ::unsetenv( _var.c_str() );
      }
 
    private:
      std::string _var;
      std::unique_ptr<std::string> _val;
    };
 
  } // namespace env

} // namespace zypp
#endif // ZYPP_BASE_ENV_H