src/support/environment.C

   1 /**
   2  * \file environment.C
   3  * This file is part of LyX, the document processor.
   4  * Licence details can be found in the file COPYING.
   5  *
   6  * \author Angus Leeming
   7  * \author João Luis M. Assirati
   8  * \author Lars Gullik Bjønnes
   9  *
  10  * Full author contact details are available in file CREDITS.
  11  */
  12
  13 #include <config.h>
  14
  15 #include "support/environment.h"
  16 #include "support/os.h"
  17
  18 #include <boost/tokenizer.hpp>
  19
  20 #include <cstdlib>
  21 #include <map>
  22
  23
  24 using std::string;
  25 using std::vector;
  26
  27 namespace lyx {
  28 namespace support {
  29
  30 string const getEnv(string const & envname)
  31 {
  32         // f.ex. what about error checking?
  33         char const * const ch = getenv(envname.c_str());
  34         string const envstr = !ch ? "" : ch;
  35         return envstr;
  36 }
  37
  38
  39 vector<string> const getEnvPath(string const & name)
  40 {
  41         typedef boost::char_separator<char> Separator;
  42         typedef boost::tokenizer<Separator> Tokenizer;
  43
  44         string const env_var = getEnv(name);
  45         Separator const separator(string(1, os::path_separator()).c_str());
  46         Tokenizer const tokens(env_var, separator);
  47         Tokenizer::const_iterator it = tokens.begin();
  48         Tokenizer::const_iterator const end = tokens.end();
  49
  50         std::vector<string> vars;
  51         for (; it != end; ++it)
  52                 vars.push_back(os::internal_path(*it));
  53
  54         return vars;
  55 }
  56
  57
  58 bool setEnv(string const & name, string const & value)
  59 {
  60         // CHECK Look at and fix this.
  61         // f.ex. what about error checking?
  62
  63 #if defined (HAVE_SETENV)
  64         int const retval = ::setenv(name.c_str(), value.c_str(), true);
  65
  66 #elif defined (HAVE_PUTENV)
  67         static std::map<string, char *> varmap;
  68
  69         string envstr = name + '=' + value;
  70         char * newptr = new char[envstr.size() + 1];
  71         envstr.copy(newptr, envstr.length());
  72         newptr[envstr.length()] = '\0';
  73         int const retval = ::putenv(newptr);
  74
  75         char * oldptr = varmap[name];
  76         if (oldptr)
  77                 delete oldptr;
  78         varmap[name] = newptr;
  79
  80 #else
  81 #error No environment-setting function has been defined.
  82 #endif
  83         return retval == 0;
  84 }
  85
  86
  87 void setEnvPath(string const & name, vector<string> const & env)
  88 {
  89         char const separator(os::path_separator());
  90         std::ostringstream ss;
  91         vector<string>::const_iterator it = env.begin();
  92         vector<string>::const_iterator const end = env.end();
  93         for (; it != end; ++it) {
  94                 if (ss.tellp() > 0)
  95                         ss << separator;
  96                 ss << os::external_path(*it);
  97         }
  98         setEnv(name, ss.str());
  99 }
 100
 101
 102 void prependEnvPath(string const & name, string const & prefix)
 103 {
 104         vector<string> env_var = getEnvPath(name);
 105
 106         typedef boost::char_separator<char> Separator;
 107         typedef boost::tokenizer<Separator> Tokenizer;
 108
 109         Separator const separator(string(1, os::path_separator()).c_str());
 110
 111         // Prepend each new element to the list, removing identical elements
 112         // that occur later in the list.
 113         Tokenizer const tokens(prefix, separator);
 114         vector<string> reversed_tokens(tokens.begin(), tokens.end());
 115
 116         typedef vector<string>::const_reverse_iterator token_iterator;
 117         token_iterator it = reversed_tokens.rbegin();
 118         token_iterator const end = reversed_tokens.rend();
 119         for (; it != end; ++it) {
 120                 vector<string>::iterator remove_it =
 121                         std::remove(env_var.begin(), env_var.end(), *it);
 122                 env_var.erase(remove_it, env_var.end());
 123                 env_var.insert(env_var.begin(), *it);
 124         }
 125
 126         setEnvPath(name, env_var);
 127 }
 128
 129 } // namespace support
 130 } // namespace lyx