sapling-util 0.1.0

Misc utilities used by Sapling.
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

/*
 * Copyright (c) Meta Platforms, Inc. and affiliates.
 *
 * This source code is licensed under the MIT license found in the
 * LICENSE file in the root directory of this source tree.
 */

use std::borrow::Cow;

use anyhow::Context;
use anyhow::Result;

pub fn hostname() -> String {
    if std::env::var_os("TESTTMP").is_some() || cfg!(test) {
        // Doesn't seem like we want to use the real hostname in tests.
        // Also, this may fix some debugruntest issues on mac due to whoami issues.
        "test-hostname".to_owned()
    } else {
        std::env::var_os(if cfg!(windows) {
            "COMPUTERNAME"
        } else if cfg!(target_os = "macos") {
            "HOST"
        } else {
            "HOSTNAME"
        })
        .and_then(|h| h.to_str().map(|s| s.to_string()))
        .unwrap_or("".to_owned())
    }
}

pub fn username() -> Result<String> {
    if std::env::var_os("TESTTMP").is_some() || cfg!(test) {
        Ok("test".to_owned())
    } else {
        std::env::var_os(if cfg!(windows) { "USERNAME" } else { "USER" })
            .context("to get username")
            .map(|k| k.to_string_lossy().to_string())
    }
}

pub fn shell_escape(args: &[String]) -> String {
    args.iter()
        .map(|a| shell_escape::escape(Cow::Borrowed(a.as_str())))
        .collect::<Vec<_>>()
        .join(" ")
}