97 lines
2.3 KiB
Bash
97 lines
2.3 KiB
Bash
#!/usr/bin/env bash
|
||
set -euo pipefail
|
||
|
||
ROOT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||
WIN_INSTALLER_PS1="$ROOT_DIR/install.win.ps1"
|
||
|
||
print_usage() {
|
||
cat <<'USAGE'
|
||
用法:
|
||
./install.sh
|
||
|
||
说明:
|
||
这是 WSL 兼容入口。
|
||
它会直接复用 Windows 安装主脚本，和 cmd / PowerShell 的安装结果保持一致。
|
||
USAGE
|
||
}
|
||
|
||
to_win_path_if_needed() {
|
||
local raw="$1"
|
||
if [[ -z "$raw" ]]; then
|
||
printf ''
|
||
return
|
||
fi
|
||
if [[ "$raw" == /* ]]; then
|
||
wslpath -w "$raw"
|
||
return
|
||
fi
|
||
printf '%s' "$raw"
|
||
}
|
||
|
||
ps_escape_single_quotes() {
|
||
printf "%s" "$1" | sed "s/'/''/g"
|
||
}
|
||
|
||
require_windows_power_shell() {
|
||
if ! command -v powershell.exe >/dev/null 2>&1; then
|
||
echo "未找到 powershell.exe，WSL 兼容入口无法调用 Windows 安装器。"
|
||
exit 1
|
||
fi
|
||
if [[ ! -f "$WIN_INSTALLER_PS1" ]]; then
|
||
echo "缺少安装脚本: $WIN_INSTALLER_PS1"
|
||
exit 1
|
||
fi
|
||
}
|
||
|
||
build_env_overrides() {
|
||
local -n out_ref=$1
|
||
out_ref=()
|
||
|
||
for key in PYTHON_BIN LLAMA_WIN_CUDA_URL LLAMA_WIN_CUDART_URL MODEL_GGUF_URL MODEL_MMPROJ_URL MODEL_GGUF_SHA256 MODEL_MMPROJ_SHA256; do
|
||
if [[ -z "${!key-}" ]]; then
|
||
continue
|
||
fi
|
||
local value="${!key}"
|
||
if [[ "$key" == "PYTHON_BIN" ]]; then
|
||
value="$(to_win_path_if_needed "$value")"
|
||
fi
|
||
out_ref+=("$key=$value")
|
||
done
|
||
}
|
||
|
||
build_ps_env_setup() {
|
||
local -n env_ref=$1
|
||
local lines=()
|
||
local item key value escaped_value
|
||
for item in "${env_ref[@]}"; do
|
||
key="${item%%=*}"
|
||
value="${item#*=}"
|
||
escaped_value="$(ps_escape_single_quotes "$value")"
|
||
lines+=("[Environment]::SetEnvironmentVariable('$key', '$escaped_value', 'Process')")
|
||
done
|
||
printf '%s; ' "${lines[@]}"
|
||
}
|
||
|
||
main() {
|
||
if [[ "${1:-}" == "-h" || "${1:-}" == "--help" ]]; then
|
||
print_usage
|
||
exit 0
|
||
fi
|
||
|
||
require_windows_power_shell
|
||
|
||
local installer_win
|
||
installer_win="$(wslpath -w "$WIN_INSTALLER_PS1")"
|
||
|
||
local env_overrides=()
|
||
build_env_overrides env_overrides
|
||
|
||
local ps_command
|
||
local ps_env_setup
|
||
ps_env_setup="$(build_ps_env_setup env_overrides)"
|
||
ps_command="[Console]::InputEncoding = [System.Text.UTF8Encoding]::new(\$false); [Console]::OutputEncoding = [System.Text.UTF8Encoding]::new(\$false); chcp 65001 > \$null; ${ps_env_setup}& '$installer_win'"
|
||
powershell.exe -NoProfile -ExecutionPolicy Bypass -Command "$ps_command"
|
||
}
|
||
|
||
main "$@"
|