src/support/environment.cpp

   1 /**
   2  * \file environment.cpp
   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
  17 #include "support/docstring.h"
  18 #include "support/os.h"
  19
  20 #include <boost/tokenizer.hpp>
  21
  22 #include <cstdlib>
  23 #include <map>
  24 #include <sstream>
  25
  26 using std::string;
  27 using std::vector;
  28
  29
  30 namespace lyx {
  31 namespace support {
  32
  33 string const getEnv(string const & envname)
  34 {
  35         // f.ex. what about error checking?
  36         char const * const ch = getenv(envname.c_str());
  37         return ch ? to_utf8(from_local8bit(ch)) : string();
  38 }
  39
  40
  41 vector<string> const getEnvPath(string const & name)
  42 {
  43         typedef boost::char_separator<char> Separator;
  44         typedef boost::tokenizer<Separator> Tokenizer;
  45
  46         string const env_var = getEnv(name);
  47         Separator const separator(string(1, os::path_separator()).c_str());
  48         Tokenizer const tokens(env_var, separator);
  49         Tokenizer::const_iterator it = tokens.begin();
  50         Tokenizer::const_iterator const end = tokens.end();
  51
  52         std::vector<string> vars;
  53         for (; it != end; ++it)
  54                 vars.push_back(os::internal_path(*it));
  55
  56         return vars;
  57 }
  58
  59
  60 bool setEnv(string const & name, string const & value)
  61 {
  62         // CHECK Look at and fix this.
  63         // f.ex. what about error checking?
  64
  65         string const encoded = to_local8bit(from_utf8(value));
  66 #if defined (HAVE_SETENV)
  67         return ::setenv(name.c_str(), encoded.c_str(), true);
  68
  69 #elif defined (HAVE_PUTENV)
  70         static std::map<string, char *> varmap;
  71
  72         string envstr = name + '=' + encoded;
  73         char * newptr = new char[envstr.size() + 1];
  74         envstr.copy(newptr, envstr.length());
  75         newptr[envstr.length()] = '\0';
  76         int const retval = ::putenv(newptr);
  77
  78         char * oldptr = varmap[name];
  79         if (oldptr)
  80                 delete oldptr;
  81         varmap[name] = newptr;
  82         return retval == 0;
  83
  84 #else
  85 #error No environment-setting function has been defined.
  86 #endif
  87         return false;
  88 }
  89
  90
  91 void setEnvPath(string const & name, vector<string> const & env)
  92 {
  93         char const separator(os::path_separator());
  94         std::ostringstream ss;
  95         vector<string>::const_iterator const begin = env.begin();
  96         vector<string>::const_iterator const end = env.end();
  97         vector<string>::const_iterator it = begin;
  98         for (; it != end; ++it) {
  99                 if (it != begin)
 100                         ss << separator;
 101                 ss << os::external_path(*it);
 102         }
 103         setEnv(name, ss.str());
 104 }
 105
 106
 107 void prependEnvPath(string const & name, string const & prefix)
 108 {
 109         vector<string> env_var = getEnvPath(name);
 110
 111         typedef boost::char_separator<char> Separator;
 112         typedef boost::tokenizer<Separator> Tokenizer;
 113
 114         Separator const separator(string(1, os::path_separator()).c_str());
 115
 116         // Prepend each new element to the list, removing identical elements
 117         // that occur later in the list.
 118         Tokenizer const tokens(prefix, separator);
 119         vector<string> reversed_tokens(tokens.begin(), tokens.end());
 120
 121         typedef vector<string>::const_reverse_iterator token_iterator;
 122         token_iterator it = reversed_tokens.rbegin();
 123         token_iterator const end = reversed_tokens.rend();
 124         for (; it != end; ++it) {
 125                 vector<string>::iterator remove_it =
 126                         std::remove(env_var.begin(), env_var.end(), *it);
 127                 env_var.erase(remove_it, env_var.end());
 128                 env_var.insert(env_var.begin(), *it);
 129         }
 130
 131         setEnvPath(name, env_var);
 132 }
 133
 134 } // namespace support
 135 } // namespace lyx