121 lines
2.9 KiB
V
121 lines
2.9 KiB
V
// Copyright (c) 2019-2022 Alexander Medvednikov. All rights reserved.
|
|
// Use of this source code is governed by an MIT license
|
|
// that can be found in the LICENSE file.
|
|
module os
|
|
|
|
fn C.getenv(&char) &char
|
|
|
|
// C.GetEnvironmentStringsW & C.FreeEnvironmentStringsW are defined only on windows
|
|
fn C.GetEnvironmentStringsW() &u16
|
|
|
|
fn C.FreeEnvironmentStringsW(&u16) int
|
|
|
|
// `getenv` returns the value of the environment variable named by the key.
|
|
// If there is not one found, it returns an empty string ''.
|
|
pub fn getenv(key string) string {
|
|
return getenv_opt(key) or { '' }
|
|
}
|
|
|
|
// `getenv_opt` returns the value of the environment variable named by the key
|
|
// If there is not one found, it returns `none`.
|
|
[manualfree]
|
|
pub fn getenv_opt(key string) ?string {
|
|
unsafe {
|
|
$if windows {
|
|
kw := key.to_wide()
|
|
defer {
|
|
free(voidptr(kw))
|
|
}
|
|
s := C._wgetenv(kw)
|
|
if s == 0 {
|
|
return none
|
|
}
|
|
return string_from_wide(s)
|
|
} $else {
|
|
s := C.getenv(&char(key.str))
|
|
if s == voidptr(0) {
|
|
return none
|
|
}
|
|
// NB: C.getenv *requires* that the result be copied.
|
|
return cstring_to_vstring(s)
|
|
}
|
|
}
|
|
}
|
|
|
|
// os.setenv sets the value of an environment variable with `name` to `value`.
|
|
pub fn setenv(name string, value string, overwrite bool) int {
|
|
$if windows {
|
|
format := '$name=$value'
|
|
if overwrite {
|
|
unsafe {
|
|
return C._putenv(&char(format.str))
|
|
}
|
|
} else {
|
|
if getenv(name).len == 0 {
|
|
unsafe {
|
|
return C._putenv(&char(format.str))
|
|
}
|
|
}
|
|
}
|
|
return -1
|
|
} $else {
|
|
unsafe {
|
|
return C.setenv(&char(name.str), &char(value.str), overwrite)
|
|
}
|
|
}
|
|
}
|
|
|
|
// os.unsetenv clears an environment variable with `name`.
|
|
pub fn unsetenv(name string) int {
|
|
$if windows {
|
|
format := '$name='
|
|
return C._putenv(&char(format.str))
|
|
} $else {
|
|
return C.unsetenv(&char(name.str))
|
|
}
|
|
}
|
|
|
|
// See: https://linux.die.net/man/5/environ for unix platforms.
|
|
// See: https://docs.microsoft.com/bg-bg/windows/win32/api/processenv/nf-processenv-getenvironmentstrings
|
|
// os.environ returns a map of all the current environment variables
|
|
|
|
fn unix_environ() &&char {
|
|
// TODO: remove this helper function, when `&&char(C.environ)` works properly
|
|
return voidptr(C.environ)
|
|
}
|
|
|
|
pub fn environ() map[string]string {
|
|
mut res := map[string]string{}
|
|
$if windows {
|
|
mut estrings := C.GetEnvironmentStringsW()
|
|
mut eline := ''
|
|
for c := estrings; *c != 0; {
|
|
eline = unsafe { string_from_wide(c) }
|
|
eq_index := eline.index_byte(`=`)
|
|
if eq_index > 0 {
|
|
res[eline[0..eq_index]] = eline[eq_index + 1..]
|
|
}
|
|
unsafe {
|
|
c = c + eline.len + 1
|
|
}
|
|
}
|
|
C.FreeEnvironmentStringsW(estrings)
|
|
} $else {
|
|
start := unix_environ()
|
|
mut i := 0
|
|
for {
|
|
x := unsafe { start[i] }
|
|
if x == 0 {
|
|
break
|
|
}
|
|
eline := unsafe { cstring_to_vstring(x) }
|
|
eq_index := eline.index_byte(`=`)
|
|
if eq_index > 0 {
|
|
res[eline[0..eq_index]] = eline[eq_index + 1..]
|
|
}
|
|
i++
|
|
}
|
|
}
|
|
return res
|
|
}
|