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 namespace std;
  27
  28 namespace lyx {
  29 namespace support {
  30
  31 string const getEnv(string const & envname)
  32 {
  33         // f.ex. what about error checking?
  34         char const * const ch = getenv(envname.c_str());
  35         return ch ? to_utf8(from_local8bit(ch)) : string();
  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         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         string const encoded = to_local8bit(from_utf8(value));
  64 #if defined (HAVE_SETENV)
  65         return ::setenv(name.c_str(), encoded.c_str(), 1) == 0;
  66 #elif defined (HAVE_PUTENV)
  67         static map<string, string> varmap;
  68         varmap[name] = name + '=' + encoded;
  69         return ::putenv(const_cast<char*>(varmap[name].c_str())) == 0;
  70 #else
  71 #error No environment-setting function has been defined.
  72 #endif
  73         return false;
  74 }
  75
  76
  77 void setEnvPath(string const & name, vector<string> const & env)
  78 {
  79         char const separator(os::path_separator());
  80         ostringstream ss;
  81         vector<string>::const_iterator const begin = env.begin();
  82         vector<string>::const_iterator const end = env.end();
  83         vector<string>::const_iterator it = begin;
  84         for (; it != end; ++it) {
  85                 if (it != begin)
  86                         ss << separator;
  87                 ss << os::external_path(*it);
  88         }
  89         setEnv(name, ss.str());
  90 }
  91
  92
  93 void prependEnvPath(string const & name, string const & prefix)
  94 {
  95         vector<string> env_var = getEnvPath(name);
  96
  97         typedef boost::char_separator<char> Separator;
  98         typedef boost::tokenizer<Separator> Tokenizer;
  99
 100         Separator const separator(string(1, os::path_separator()).c_str());
 101
 102         // Prepend each new element to the list, removing identical elements
 103         // that occur later in the list.
 104         Tokenizer const tokens(prefix, separator);
 105         vector<string> reversed_tokens(tokens.begin(), tokens.end());
 106
 107         typedef vector<string>::const_reverse_iterator token_iterator;
 108         token_iterator it = reversed_tokens.rbegin();
 109         token_iterator const end = reversed_tokens.rend();
 110         for (; it != end; ++it) {
 111                 vector<string>::iterator remove_it =
 112                         remove(env_var.begin(), env_var.end(), *it);
 113                 env_var.erase(remove_it, env_var.end());
 114                 env_var.insert(env_var.begin(), *it);
 115         }
 116
 117         setEnvPath(name, env_var);
 118 }
 119
 120 } // namespace support
 121 } // namespace lyx