mirror of
https://github.com/fish-shell/fish-shell.git
synced 2024-11-30 14:24:35 +08:00
219 lines
5.5 KiB
C++
219 lines
5.5 KiB
C++
/** \file env.h
|
|
Prototypes for functions for setting and getting environment variables.
|
|
*/
|
|
|
|
#ifndef FISH_ENV_H
|
|
#define FISH_ENV_H
|
|
|
|
#include <wchar.h>
|
|
#include <map>
|
|
|
|
#include "util.h"
|
|
#include "common.h"
|
|
|
|
/**
|
|
Flag for local (to the current block) variable
|
|
*/
|
|
#define ENV_LOCAL 1
|
|
|
|
/**
|
|
Flag for exported (to commands) variable
|
|
*/
|
|
#define ENV_EXPORT 2
|
|
|
|
/**
|
|
Flag for unexported variable
|
|
*/
|
|
#define ENV_UNEXPORT 16
|
|
|
|
/**
|
|
Flag for global variable
|
|
*/
|
|
#define ENV_GLOBAL 4
|
|
|
|
/**
|
|
Flag for variable update request from the user. All variable
|
|
changes that are made directly by the user, such as those from the
|
|
'set' builtin must have this flag set.
|
|
*/
|
|
#define ENV_USER 8
|
|
|
|
/**
|
|
Flag for universal variable
|
|
*/
|
|
#define ENV_UNIVERSAL 32
|
|
|
|
/**
|
|
Error code for trying to alter read-only variable
|
|
*/
|
|
enum
|
|
{
|
|
ENV_PERM = 1,
|
|
ENV_INVALID
|
|
}
|
|
;
|
|
|
|
/* A struct of configuration directories, determined in main() that fish will optionally pass to env_init.
|
|
*/
|
|
struct config_paths_t
|
|
{
|
|
wcstring data; // e.g. /usr/local/share
|
|
wcstring sysconf; // e.g. /usr/local/etc
|
|
wcstring doc; // e.g. /usr/local/share/doc/fish
|
|
wcstring bin; // e.g. /usr/local/bin
|
|
};
|
|
|
|
/**
|
|
Initialize environment variable data
|
|
*/
|
|
void env_init(const struct config_paths_t *paths = NULL);
|
|
|
|
/**
|
|
Destroy environment variable data
|
|
*/
|
|
void env_destroy();
|
|
|
|
|
|
/**
|
|
Set the value of the environment variable whose name matches key to val.
|
|
|
|
Memory policy: All keys and values are copied, the parameters can and should be freed by the caller afterwards
|
|
|
|
\param key The key
|
|
\param val The value
|
|
\param mode The type of the variable. Can be any combination of ENV_GLOBAL, ENV_LOCAL, ENV_EXPORT and ENV_USER. If mode is zero, the current variable space is searched and the current mode is used. If no current variable with the same name is found, ENV_LOCAL is assumed.
|
|
|
|
\returns 0 on suicess or an error code on failiure.
|
|
|
|
The current error codes are:
|
|
|
|
* ENV_PERM, can only be returned when setting as a user, e.g. ENV_USER is set. This means that the user tried to change a read-only variable.
|
|
* ENV_INVALID, the variable name or mode was invalid
|
|
*/
|
|
|
|
int env_set(const wcstring &key, const wchar_t *val, int mode);
|
|
|
|
|
|
/**
|
|
Return the value of the variable with the specified name. Returns 0
|
|
if the key does not exist. The returned string should not be
|
|
modified or freed. The returned string is only guaranteed to be
|
|
valid until the next call to env_get(), env_set(), env_push() or
|
|
env_pop() takes place.
|
|
*/
|
|
//const wchar_t *env_get( const wchar_t *key );
|
|
|
|
class env_var_t : public wcstring
|
|
{
|
|
private:
|
|
bool is_missing;
|
|
public:
|
|
static env_var_t missing_var(void);
|
|
env_var_t(const env_var_t &x) : wcstring(x), is_missing(x.is_missing) { }
|
|
env_var_t(const wcstring & x) : wcstring(x), is_missing(false) { }
|
|
env_var_t(const wchar_t *x) : wcstring(x), is_missing(false) { }
|
|
env_var_t() : wcstring(L""), is_missing(false) { }
|
|
bool missing(void) const
|
|
{
|
|
return is_missing;
|
|
}
|
|
bool missing_or_empty(void) const
|
|
{
|
|
return missing() || empty();
|
|
}
|
|
const wchar_t *c_str(void) const;
|
|
env_var_t &operator=(const env_var_t &s)
|
|
{
|
|
is_missing = s.is_missing;
|
|
wcstring::operator=(s);
|
|
return *this;
|
|
}
|
|
|
|
bool operator==(const env_var_t &s) const
|
|
{
|
|
if (is_missing && s.is_missing)
|
|
return true;
|
|
else if (s.is_missing || s.is_missing)
|
|
return false;
|
|
else
|
|
return *static_cast<const wcstring *>(this) == *static_cast<const wcstring *>(&s);
|
|
}
|
|
|
|
};
|
|
/**
|
|
Gets the variable with the specified name, or an empty string if it does not exist.
|
|
*/
|
|
env_var_t env_get_string(const wcstring &key);
|
|
|
|
/**
|
|
Returns 1 if the specified key exists. This can't be reliably done
|
|
using env_get, since env_get returns null for 0-element arrays
|
|
|
|
\param key The name of the variable to remove
|
|
\param mode the scope to search in. All scopes are searched if unset
|
|
*/
|
|
int env_exist(const wchar_t *key, int mode);
|
|
|
|
/**
|
|
Remove environemnt variable
|
|
|
|
\param key The name of the variable to remove
|
|
\param mode should be ENV_USER if this is a remove request from the user, 0 otherwise. If this is a user request, read-only variables can not be removed. The mode may also specify the scope of the variable that should be erased.
|
|
|
|
\return zero if the variable existed, and non-zero if the variable did not exist
|
|
*/
|
|
int env_remove(const wcstring &key, int mode);
|
|
|
|
/**
|
|
Push the variable stack. Used for implementing local variables for functions and for-loops.
|
|
*/
|
|
void env_push(int new_scope);
|
|
|
|
/**
|
|
Pop the variable stack. Used for implementing local variables for functions and for-loops.
|
|
*/
|
|
void env_pop();
|
|
|
|
/** Returns an array containing all exported variables in a format suitable for execv. */
|
|
char **env_export_arr(bool recalc);
|
|
void env_export_arr(bool recalc, null_terminated_array_t<char> &result);
|
|
|
|
/**
|
|
Returns all variable names.
|
|
*/
|
|
wcstring_list_t env_get_names(int flags);
|
|
|
|
|
|
|
|
/**
|
|
Update the PWD variable
|
|
directory
|
|
*/
|
|
int env_set_pwd();
|
|
|
|
class env_vars_snapshot_t
|
|
{
|
|
std::map<wcstring, wcstring> vars;
|
|
bool is_current() const;
|
|
|
|
public:
|
|
env_vars_snapshot_t(const wchar_t * const * keys);
|
|
env_vars_snapshot_t(void);
|
|
|
|
env_var_t get(const wcstring &key) const;
|
|
|
|
// Returns the fake snapshot representing the live variables array
|
|
static const env_vars_snapshot_t ¤t();
|
|
|
|
// vars necessary for highlighting
|
|
static const wchar_t * const highlighting_keys[];
|
|
};
|
|
|
|
extern bool g_log_forks;
|
|
extern int g_fork_count;
|
|
|
|
extern bool g_use_posix_spawn;
|
|
|
|
|
|
#endif
|