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

use std::env;
use std::os::unix::net::UnixStream;

use anyhow::Result;
use lazy_static::lazy_static;
use regex::Regex;

use crate::{conn::Conn, fid, fsys};

pub fn dial(addr: &str) -> Result<Conn> {
    let stream = UnixStream::connect(addr)?;
    Conn::new(stream)
}

pub fn dial_service(service: &str) -> Result<Conn> {
    let ns = namespace();
    dial((ns + "/" + service).as_str())
}

pub fn mount_service(service: &str) -> Result<fsys::Fsys> {
    let mut conn = dial_service(service)?;
    let fsys = conn.attach(fid::get_user(), "".to_string())?;
    Ok(fsys)
}

// namespace returns the path to the name space directory.
pub fn namespace() -> String {
    if let Ok(val) = env::var("NAMESPACE") {
        return val;
    }
    let mut disp = if let Ok(val) = env::var("DISPLAY") {
        val
    } else {
        // No $DISPLAY? Use :0.0 for non-X11 GUI (OS X).
        String::from(":0.0")
    };

    lazy_static! {
        static ref DOT_ZERO: Regex = Regex::new(r"\A(.*:\d+)\.0\z").unwrap();
    }
    // Canonicalize: xxx:0.0 => xxx:0.
    if let Some(m) = DOT_ZERO.captures(disp.as_str()) {
        disp = m.get(1).unwrap().as_str().to_string();
    }
    format!("/tmp/ns.{}.{}", env::var("USER").unwrap(), disp)
}