robotech 1.6.0

Backend service implementation for the RoboTech platform, providing RESTful APIs and business logic for web applications.
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

use crate::env::EnvError;
use std::env;
use std::path::PathBuf;
use std::sync::OnceLock;

/// 全局配置
pub static APP_ENV: OnceLock<AppEnv> = OnceLock::new();

#[derive(Debug)]
pub struct AppEnv {
    pub app_file_path: PathBuf,
    pub app_file_path_without_ext: PathBuf,
    pub app_dir: PathBuf,
    pub app_file_name: String,
    pub app_file_name_without_ext: String,
}

/// 初始化环境变量
pub fn init_env() -> Result<(), EnvError> {
    // 获取当前执行文件路径
    let app_file_path = env::current_exe().map_err(EnvError::GetAppPath)?;

    // 获取当前执行文件路径(不带后缀)
    let mut app_file_path_without_ext = app_file_path.clone();
    app_file_path_without_ext.pop();

    // 获取当前执行文件所在目录
    let mut app_dir = app_file_path.clone();
    app_dir.pop();

    // 获取当前执行文件名
    let app_file_name = app_file_path
        .file_name()
        .ok_or(EnvError::GetAppFileName())?
        .to_string_lossy()
        .to_string();

    // 获取当前执行文件名(不带后缀)
    let app_file_name_without_ext = app_file_path
        .file_stem()
        .ok_or(EnvError::GetAppFileName())?
        .to_string_lossy()
        .to_string();

    let env = AppEnv {
        app_file_path,
        app_file_path_without_ext,
        app_dir,
        app_file_name,
        app_file_name_without_ext,
    };

    APP_ENV.set(env).map_err(|_| EnvError::SetAppEnv())?;
    Ok(())
}