irid-std 0.3.1

A replacement for std when running without a filesystem on the irid kernel
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68

use core::io::ErrorKind;

use alloc::{borrow::ToOwned, boxed::Box, collections::btree_map::BTreeMap, string::{String, ToString}, vec::Vec};

use crate::{ffi::{OsStr, OsString}, io::Error, path::{Path, PathBuf}, sync::{Mutex, MutexGuard}};

static ARGS: Mutex<Vec<Box<str>>> = Mutex::new(Vec::new());
static VARS: Mutex<BTreeMap<Box<str>, Box<str>>> = Mutex::new(BTreeMap::new());

pub fn args() -> impl Iterator<Item = String> {
    ARGS.lock().expect("args lock poisoned").iter().map(|x| x.to_string()).collect::<Vec<_>>().into_iter()
}

pub fn args_os() -> impl Iterator<Item = OsString> {
    args().map(OsString::from)
}

pub fn current_dir() -> Result<PathBuf, Error> {
    Ok(PathBuf::from(OsString::from(var("PWD").unwrap_or("/".to_owned()))))
}

pub fn set_current_dir(path: impl AsRef<Path>) -> Result<(), Error> {
    unsafe { set_var("PWD", path.as_ref()) };

    Ok(())
}

pub fn current_exe() -> Result<PathBuf, Error> {
    Ok(Path::new(&args().next().ok_or(Error::from(ErrorKind::NotFound))?).to_owned())
}

pub fn home_dir() -> Option<PathBuf> {
    Some(PathBuf::from(OsString::from(var("HOME")?)))
}

pub fn temp_dir() -> PathBuf {
    var("TMPDIR").map(|x| PathBuf::from(OsString::from(x))).unwrap_or_else(|| Path::new("/tmp").to_owned())
}

pub fn var(name: impl AsRef<OsStr>) -> Option<String> {
    VARS.lock().expect("vars lock poisoned").get(name.as_ref().as_str())
        .map(|x| x.to_string())
}

pub fn var_os(name: impl AsRef<OsStr>) -> Option<OsString> {
    var(name).map(OsString::from)
}

pub fn vars() -> impl Iterator<Item = (String, String)> {
    VARS.lock().expect("vars lock poisoned").iter().map(|(k, v)| (k.to_string(), v.to_string())).collect::<Vec<_>>().into_iter()
}

pub fn vars_os() -> impl Iterator<Item = (OsString, OsString)> {
    vars().map(|(a, b)| (OsString::from(a), OsString::from(b)))
}

pub unsafe fn set_var(key: impl AsRef<OsStr>, value: impl AsRef<OsStr>) {
    VARS.lock().expect("vars lock poisoned").insert(key.as_ref().as_str().to_owned().into_boxed_str(), value.as_ref().as_str().to_owned().into_boxed_str());
}

pub(crate) fn arg_lock() -> MutexGuard<'static, Vec<Box<str>>> {
    ARGS.lock().expect("failed to lock args")
}

pub(crate) fn var_lock() -> MutexGuard<'static, BTreeMap<Box<str>, Box<str>>> {
    VARS.lock().expect("failed to lock vars")
}