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