68 lines
2.2 KiB
Rust
68 lines
2.2 KiB
Rust
use anyhow::{Context, Result, bail};
|
||
use std::env;
|
||
use std::path::{Path, PathBuf};
|
||
use tracing::debug;
|
||
pub struct Layout {
|
||
pub base_dir: PathBuf,
|
||
pub bin_dir: PathBuf,
|
||
pub tools_dir: PathBuf,
|
||
pub lua_file: PathBuf,
|
||
}
|
||
|
||
impl Layout {
|
||
/// 自动解析目录布局：
|
||
/// 1. 优先使用环境变量 RSHIM_HOME
|
||
/// 2. 兜底回退到当前垫片可执行文件所在目录推断 (exe -> bin -> root)
|
||
pub fn discover(current_exe: &Path) -> Result<Self> {
|
||
// 策略 1: 环境变量优先
|
||
if let Ok(home_val) = env::var("MIMIC_HOME") {
|
||
let trimmed = home_val.trim();
|
||
if !trimmed.is_empty() {
|
||
debug!(home = %trimmed, "检测到 MIMIC_HOME，采用环境变量配置");
|
||
return Self::from_base_dir(PathBuf::from(trimmed));
|
||
}
|
||
}
|
||
|
||
// 策略 2: 相对路径自动推断兜底
|
||
debug!("未配置 MIMIC_HOME，尝试从当前可执行文件路径推断根目录");
|
||
Self::from_executable(current_exe)
|
||
}
|
||
|
||
/// 基于确定的根目录构建完整布局
|
||
fn from_base_dir(base_dir: PathBuf) -> Result<Self> {
|
||
if !base_dir.is_dir() {
|
||
bail!("指定的根目录不存在或不是有效目录: [{}]", base_dir.display());
|
||
}
|
||
|
||
let bin_dir = base_dir.join("bin");
|
||
let tools_dir = base_dir.join("tools");
|
||
let lua_file = base_dir.join("mimic.lua");
|
||
|
||
Ok(Self {
|
||
base_dir,
|
||
bin_dir,
|
||
tools_dir,
|
||
lua_file,
|
||
})
|
||
}
|
||
|
||
/// 从当前可执行文件解析 shim 安装目录布局
|
||
fn from_executable(exe_path: &Path) -> Result<Self> {
|
||
// let exe_path = exe_path.as_ref();
|
||
let bin_dir = exe_path
|
||
.parent()
|
||
.with_context(|| format!("无法获取程序 [{}] 的父级 bin 目录", exe_path.display()))?
|
||
.to_path_buf();
|
||
|
||
debug!("bin_dir目录 {}", bin_dir.display());
|
||
|
||
let base_dir = bin_dir
|
||
.parent()
|
||
.with_context(|| format!("无法获取 bin 目录 [{}] 的父级 root 目录", bin_dir.display()))?
|
||
.to_path_buf();
|
||
debug!("base_dir 目录 {}", base_dir.display());
|
||
|
||
Self::from_base_dir(base_dir)
|
||
}
|
||
}
|