useenv 1.4.0

Setup environment defined in a configuration file
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
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85

use std::path::PathBuf;
use std::{env, io};
use which::which;

use crate::colorize::Colorize;
use crate::context::Shell;

// The workflow and part of the code is copied from starship:
// https://starship.rs
// We use a two-phase init:
// - first phase is a simple command to the shell
// - second phase is a script that is executed by the shell

struct UsePath {
    native_path: PathBuf,
}
impl UsePath {
    fn init() -> io::Result<Self> {
        let exe_name = option_env!("CARGO_PKG_NAME").unwrap_or("use");
        let native_path = which(exe_name).or_else(|_| env::current_exe())?;
        Ok(Self { native_path })
    }
    fn str_path(&self) -> io::Result<&str> {
        let current_exe = self
            .native_path
            .to_str()
            .ok_or_else(|| io::Error::other("can't convert to str"))?;
        Ok(current_exe)
    }
    /// `PowerShell` specific path escaping
    fn sprint_pwsh(&self) -> io::Result<String> {
        self.str_path()
            .map(|s| s.replace('\'', "''"))
            .map(|s| format!("'{s}'"))
    }
    /// Command Shell specific path escaping
    fn sprint_cmdexe(&self) -> io::Result<String> {
        self.str_path().map(|s| format!("\"{s}\""))
    }
}

/// Print the init stub for the given shell
pub fn init_stub(shell: Shell) -> io::Result<()> {
    let use_path = UsePath::init()?;

    match shell {
        Shell::Powershell => print!(
            r#"Invoke-Expression (& {} init powershell --print-full-init | Out-String)"#,
            use_path.sprint_pwsh()?
        ),
        Shell::Cmd => print_script(CLINK_INIT, &use_path.sprint_cmdexe()?),
        _ => {
            eprintln!("{} Unsupported shell: {shell:?}", "error:".error());
            return Err(io::Error::other("Unsupported shell"));
        }
    }
    Ok(())
}

/// Print the main script for the given shell
/// This function is used when passing `--print-full-init` to the init command
pub fn init_main(shell: Shell) -> io::Result<()> {
    let use_path = UsePath::init()?;

    match shell {
        Shell::Powershell => print_script(POWERSHELL_INIT, &use_path.sprint_pwsh()?),
        Shell::Cmd => print_script(CLINK_INIT, &use_path.sprint_cmdexe()?),
        _ => {
            eprintln!("{} Unsupported shell: {shell:?}", "error:".error());
            return Err(io::Error::other("Unsupported shell"));
        }
    }
    Ok(())
}

/// Print the given script after replacing the placeholder with the given path
fn print_script(script: &str, path: &str) {
    // Escape path for Windows
    let path = path.replace("\\", "\\\\");
    let script = script.replace("::USE::", path.as_str());
    print!("{script}");
}

const POWERSHELL_INIT: &str = include_str!("use.ps1");
const CLINK_INIT: &str = include_str!("use.lua");