📦 da-x / vmess

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
73use std::collections::hash_map::DefaultHasher;
use std::convert::TryFrom;
use std::hash::{Hash, Hasher};
use std::path::PathBuf;

use super::Error;
use log::debug;

pub(crate) fn bash_stdout(cmd: String) -> Result<String, Error> {
    use std::process::Command;
    debug!("bash output: {:?}", cmd.trim());
    let out = Command::new("bash").arg("-c").arg(&cmd).output()?;
    if !out.status.success() {
        return Err(
            Error::CommandError(cmd, String::from_utf8_lossy(&out.stderr).into_owned()).into(),
        );
    }
    Ok(String::from_utf8(out.stdout)?)
}

pub(crate) fn adjust_path_by_env(path: PathBuf) -> PathBuf {
    if let Some(home) = dirs::home_dir() {
        let path = path
            .into_os_string()
            .into_string()
            .unwrap()
            .replace("$HOME", &home.into_os_string().into_string().unwrap());
        PathBuf::try_from(path).expect("cannot convert path")
    } else {
        path
    }
}

pub(crate) trait AddExtension {
    fn add_extension(&self, extension: impl AsRef<std::path::Path>) -> PathBuf;
}

impl AddExtension for PathBuf {
    fn add_extension(&self, extension: impl AsRef<std::path::Path>) -> PathBuf {
        let mut path = self.clone();
        match path.extension() {
            Some(ext) => {
                let mut ext = ext.to_os_string();
                ext.push(".");
                ext.push(extension.as_ref());
                path.set_extension(ext)
            }
            None => path.set_extension(extension.as_ref()),
        };

        path
    }
}

pub(crate) fn calculate_hash<T: Hash>(t: &T) -> u64 {
    let mut s = DefaultHasher::new();
    t.hash(&mut s);
    s.finish()
}

pub(crate) fn is_version_at_least(version_str: &str, min_version: &[u32]) -> bool {
    let version_numbers: Vec<u32> = version_str.trim().split('.')
        .map(|s| s.parse::<u32>().unwrap())
        .collect();

    version_numbers.as_slice() > min_version
}

#[macro_export]
macro_rules! ibash_stdout {
    ($($arg:tt)*) => { crate::utils::bash_stdout(f!($($arg)*)) }
}