407 lines
14 KiB
Bash
407 lines
14 KiB
Bash
#!/bin/sh
|
||
# ChuSQL 安装脚本（Linux / macOS / POSIX sh）。
|
||
#
|
||
# 三种来源，脚本自己判断走哪条：
|
||
#
|
||
# 1) 包内安装：解压一个发行包，脚本旁边就是 bin/，直接装本地文件
|
||
# ./install.sh --component web
|
||
#
|
||
# 2) 在线安装：脚本从 GitHub 取发行包（也可用 --url 指任意 URL 或本地归档）
|
||
# curl -fsSL https://raw.githubusercontent.com/TheBadRoger/ChuSQL/main/scripts/install.sh \
|
||
# | sh -s -- --component web
|
||
# sh install.sh --component web --url dist/chusql-web-linux-x86_64.tar.gz
|
||
#
|
||
# 3) 源码安装：拉仓库、本地编译（要有 git + cargo + stack + ghc）
|
||
# ./install.sh --component web --from-source
|
||
# ./install.sh --component web --from-source --source-dir ../ChuSQL
|
||
#
|
||
# 安装 = 释放到目标目录 + 配好环境变量 + 写全局配置，装完把包目录删掉。
|
||
# storage/engine 一定装；web 与 cli 二选一。
|
||
#
|
||
# 包目录结构（解压后就是这个样子）：
|
||
# bin/ chusql-storage（必装）、chusql-web 或 csql（二选一）
|
||
# static/ Web 前端静态资源（装 web 组件时带）
|
||
# scripts/ chusql.toml、init.sql
|
||
# csql-web.sh
|
||
# install.sh
|
||
|
||
set -eu
|
||
|
||
default_repo='TheBadRoger/ChuSQL'
|
||
|
||
usage() {
|
||
cat <<'EOF'
|
||
usage: ./install.sh --component web|cli [options]
|
||
|
||
--component web|cli which front end to install (required)
|
||
|
||
--install-dir DIR where to install (default: $HOME/.local/share/chusql)
|
||
--data-dir DIR where to keep data (default: <install-dir>/data)
|
||
--user NAME administrator name (default: root)
|
||
--password PW administrator password, empty means no password
|
||
(administrator-only sign-in)
|
||
--keep-package do not delete the downloaded/extracted package
|
||
|
||
where the files come from:
|
||
(default) a package next to this script, else the newest GitHub release
|
||
--url URL|PATH install from this archive (URL, file:// URL or local path)
|
||
--repo OWNER/NAME GitHub repository (default: TheBadRoger/ChuSQL)
|
||
--version TAG release tag to download / git ref to build (default: latest)
|
||
--from-source clone and build locally (needs git, cargo, stack, ghc)
|
||
--source-dir DIR build from this checkout instead of cloning
|
||
-h, --help this text
|
||
EOF
|
||
}
|
||
|
||
component=''
|
||
install_dir=''
|
||
data_dir=''
|
||
root_user='root'
|
||
root_password=''
|
||
keep_package='no'
|
||
repo="$default_repo"
|
||
version='latest'
|
||
url_arg=''
|
||
from_source='no'
|
||
source_dir=''
|
||
|
||
while [ $# -gt 0 ]; do
|
||
case "$1" in
|
||
--component) component="${2:-}"; shift 2 ;;
|
||
--install-dir) install_dir="${2:-}"; shift 2 ;;
|
||
--data-dir) data_dir="${2:-}"; shift 2 ;;
|
||
--user) root_user="${2:-}"; shift 2 ;;
|
||
--password) root_password="${2:-}"; shift 2 ;;
|
||
--keep-package) keep_package='yes'; shift ;;
|
||
--url) url_arg="${2:-}"; shift 2 ;;
|
||
--repo) repo="${2:-}"; shift 2 ;;
|
||
--version) version="${2:-}"; shift 2 ;;
|
||
--from-source) from_source='yes'; shift ;;
|
||
--source-dir) source_dir="${2:-}"; shift 2 ;;
|
||
-h|--help) usage; exit 0 ;;
|
||
*) echo "!! unknown option: $1" >&2; usage >&2; exit 1 ;;
|
||
esac
|
||
done
|
||
|
||
fail() { echo "!! $1" >&2; exit 1; }
|
||
step() { echo; echo "==> $1"; }
|
||
note() { echo " $1"; }
|
||
|
||
case "$component" in
|
||
web|cli) ;;
|
||
'') fail '--component is required (web or cli)' ;;
|
||
*) fail "--component must be web or cli, got: $component" ;;
|
||
esac
|
||
|
||
# ---- 本脚本在哪（管道进来时 $0 是 sh，这时没有本地包）----
|
||
self=$0
|
||
script_dir=''
|
||
case "$self" in
|
||
sh|bash|dash|ash|ksh|'-sh'|*/sh|*/bash|*/dash|*/ash|*/ksh) script_dir='' ;;
|
||
*) script_dir=$(CDPATH= cd -- "$(dirname -- "$self")" 2>/dev/null && pwd) || script_dir='' ;;
|
||
esac
|
||
|
||
[ -n "$install_dir" ] || install_dir="$HOME/.local/share/chusql"
|
||
[ -n "$data_dir" ] || data_dir="$install_dir/data"
|
||
|
||
tmp=''
|
||
cleanup() { if [ -n "$tmp" ]; then rm -rf "$tmp"; fi; }
|
||
new_tmp() {
|
||
tmp=$(mktemp -d "${TMPDIR:-/tmp}/chusql-install.XXXXXX") || fail 'cannot create a temporary directory'
|
||
# --keep-package 时不注册清理，临时目录就留着
|
||
if [ "$keep_package" = 'no' ]; then trap cleanup EXIT INT TERM HUP; fi
|
||
}
|
||
|
||
download() {
|
||
# download URL FILE
|
||
if command -v curl >/dev/null 2>&1; then
|
||
curl -fsSL -o "$2" "$1" || return 1
|
||
elif command -v wget >/dev/null 2>&1; then
|
||
wget -q -O "$2" "$1" || return 1
|
||
else
|
||
fail 'need curl or wget to download the package'
|
||
fi
|
||
}
|
||
|
||
verify_checksum() {
|
||
# verify_checksum ASSET_URL FILE —— 发布带 .sha256 就校验，没带就说明一下
|
||
sums="$tmp/asset.sha256"
|
||
if ! download "$1.sha256" "$sums" 2>/dev/null; then
|
||
note 'checksum no .sha256 published for this asset (skipped)'
|
||
return 0
|
||
fi
|
||
expected=$(awk 'NR==1 { print $1 }' "$sums")
|
||
if command -v sha256sum >/dev/null 2>&1; then
|
||
actual=$(sha256sum "$2" | awk '{ print $1 }')
|
||
elif command -v shasum >/dev/null 2>&1; then
|
||
actual=$(shasum -a 256 "$2" | awk '{ print $1 }')
|
||
else
|
||
fail 'no sha256sum/shasum available to verify the checksum'
|
||
fi
|
||
[ "$actual" = "$expected" ] || fail "checksum mismatch: expected $expected, got $actual"
|
||
note "checksum $expected (ok)"
|
||
}
|
||
|
||
unpack() {
|
||
# unpack ARCHIVE DEST
|
||
mkdir -p "$2"
|
||
case "$1" in
|
||
*.zip)
|
||
command -v unzip >/dev/null 2>&1 || fail 'unzip is required to open a .zip package'
|
||
unzip -q "$1" -d "$2"
|
||
;;
|
||
*)
|
||
tar -xzf "$1" -C "$2"
|
||
;;
|
||
esac
|
||
}
|
||
|
||
# 平台标签（全局：os、arch、ext、labels）
|
||
detect_platform() {
|
||
os=$(uname -s 2>/dev/null || echo unknown)
|
||
case "$os" in
|
||
Linux) os='linux' ;;
|
||
Darwin) os='macos' ;;
|
||
MINGW*|MSYS*|CYGWIN*) os='windows' ;;
|
||
esac
|
||
arch=$(uname -m 2>/dev/null || echo unknown)
|
||
case "$arch" in
|
||
x86_64|amd64) arch='x86_64' ;;
|
||
aarch64|arm64) arch='arm64' ;;
|
||
esac
|
||
if [ "$os" = 'windows' ]; then ext='zip'; else ext='tar.gz'; fi
|
||
labels="$os-$arch $os"
|
||
}
|
||
|
||
# 发行资源（全局：url、archive），成功返回 0
|
||
fetch_asset() {
|
||
asset="$1"
|
||
if [ "$version" = 'latest' ]; then
|
||
url="https://github.com/$repo/releases/latest/download/$asset"
|
||
else
|
||
url="https://github.com/$repo/releases/download/$version/$asset"
|
||
fi
|
||
archive="$tmp/$asset"
|
||
if download "$url" "$archive" 2>/dev/null; then
|
||
note "asset $asset"
|
||
return 0
|
||
fi
|
||
return 1
|
||
}
|
||
|
||
pack=''
|
||
url=''
|
||
archive=''
|
||
|
||
if [ "$from_source" = 'yes' ]; then
|
||
# ---------- 源码安装 ----------
|
||
if [ -z "$source_dir" ]; then
|
||
command -v git >/dev/null 2>&1 || fail 'git is required to clone the repository'
|
||
fi
|
||
command -v cargo >/dev/null 2>&1 || fail 'cargo (Rust) is required for --from-source'
|
||
command -v stack >/dev/null 2>&1 || fail 'stack (Haskell) is required for --from-source'
|
||
|
||
new_tmp
|
||
if [ -n "$source_dir" ]; then
|
||
src=$(CDPATH= cd -- "$source_dir" 2>/dev/null && pwd) || fail "no such directory: $source_dir"
|
||
note "source $src (using this checkout as is)"
|
||
else
|
||
src="$tmp/src"
|
||
if [ "$version" = 'latest' ]; then
|
||
step "Cloning https://github.com/$repo.git"
|
||
git clone --depth 1 "https://github.com/$repo.git" "$src" >/dev/null 2>&1 ||
|
||
fail "cannot clone https://github.com/$repo.git"
|
||
else
|
||
step "Cloning https://github.com/$repo.git at $version"
|
||
git clone --depth 1 --branch "$version" "https://github.com/$repo.git" "$src" >/dev/null 2>&1 ||
|
||
fail "cannot clone $repo at $version"
|
||
fi
|
||
note "source $src"
|
||
fi
|
||
[ -f "$src/chusql-storage/Cargo.toml" ] || fail "not a ChuSQL checkout (chusql-storage/Cargo.toml missing): $src"
|
||
[ -f "$src/scripts/chusql.toml" ] || fail "not a ChuSQL checkout (scripts/chusql.toml missing): $src"
|
||
|
||
if command -v pkg-config >/dev/null 2>&1 && ! pkg-config --exists zlib 2>/dev/null; then
|
||
note 'hint the Haskell zlib binding needs the C library: apt-get install zlib1g-dev (or dnf install zlib-devel)'
|
||
fi
|
||
|
||
step 'Building the Rust storage (cargo build --release)'
|
||
( cd "$src/chusql-storage" && cargo build --release ) || fail 'cargo build --release failed'
|
||
storage_bin="$src/chusql-storage/target/release/chusql-storage"
|
||
[ -f "$storage_bin" ] || fail "storage binary not found after the build: $storage_bin"
|
||
|
||
if [ "$component" = 'web' ]; then
|
||
comp_dir='chusql-web'; front_name='chusql-web'
|
||
else
|
||
comp_dir='chusql-cli'; front_name='csql'
|
||
fi
|
||
step "Building the Haskell binaries (stack build --fast in $comp_dir)"
|
||
note 'the first build downloads the Hackage index (about 139 MB)'
|
||
( cd "$src/$comp_dir" && stack build --fast ) || fail "stack build failed in $comp_dir"
|
||
stack_root=$( cd "$src/$comp_dir" && stack path --local-install-root 2>/dev/null | tr -d '\r' )
|
||
[ -n "$stack_root" ] || fail 'cannot ask stack for --local-install-root'
|
||
front_bin="$stack_root/bin/$front_name"
|
||
[ -f "$front_bin" ] || fail "front end binary not found after the build: $front_bin"
|
||
|
||
step 'Staging the package'
|
||
pack="$tmp/stage"
|
||
mkdir -p "$pack/bin" "$pack/scripts"
|
||
cp "$storage_bin" "$pack/bin/"
|
||
cp "$front_bin" "$pack/bin/"
|
||
for extra in "$stack_root/bin/"*.so "$stack_root/bin/"*.dll; do
|
||
if [ -f "$extra" ]; then cp "$extra" "$pack/bin/"; fi
|
||
done
|
||
cp "$src/scripts/chusql.toml" "$src/scripts/init.sql" "$pack/scripts/"
|
||
cp "$src/scripts/install.sh" "$pack/"
|
||
if [ "$component" = 'web' ]; then
|
||
[ -d "$src/chusql-web/static" ] || fail "static assets not found: $src/chusql-web/static"
|
||
mkdir -p "$pack/static"
|
||
cp -R "$src/chusql-web/static/." "$pack/static/"
|
||
cp "$src/scripts/csql-web.sh" "$pack/"
|
||
fi
|
||
elif [ -n "$url_arg" ]; then
|
||
# ---------- 指定 URL / 本地归档 ----------
|
||
new_tmp
|
||
archive="$tmp/package"
|
||
case "$url_arg" in
|
||
http://*|https://*|file://*) url="$url_arg" ;;
|
||
*)
|
||
[ -f "$url_arg" ] || fail "no such package: $url_arg"
|
||
url="file://$(CDPATH= cd -- "$(dirname -- "$url_arg")" && pwd)/$(basename -- "$url_arg")"
|
||
;;
|
||
esac
|
||
step "Downloading $url"
|
||
download "$url" "$archive" || fail "cannot download $url"
|
||
case "$url_arg" in
|
||
http://*|https://*|file://*) verify_checksum "$url" "$archive" ;;
|
||
esac
|
||
step 'Unpacking'
|
||
unpack "$archive" "$tmp/pkg"
|
||
pack="$tmp/pkg"
|
||
elif [ -n "$script_dir" ] && [ -f "$script_dir/bin/chusql-storage" ]; then
|
||
# ---------- 包内安装 ----------
|
||
pack="$script_dir"
|
||
else
|
||
# ---------- 从 GitHub 发行版下载 ----------
|
||
new_tmp
|
||
detect_platform
|
||
step "Looking for a ChuSQL package for $os-$arch (component $component) in $repo"
|
||
found=''
|
||
for label in $labels; do
|
||
if fetch_asset "chusql-$component-$label.$ext"; then
|
||
found="$asset"
|
||
break
|
||
fi
|
||
done
|
||
if [ -z "$found" ]; then
|
||
if fetch_asset "chusql-$component.$ext"; then found="$asset"; fi
|
||
fi
|
||
[ -n "$found" ] || fail "no package for $os-$arch in $repo ($version); try --url or --from-source"
|
||
verify_checksum "$url" "$archive"
|
||
step 'Unpacking'
|
||
unpack "$archive" "$tmp/pkg"
|
||
pack="$tmp/pkg"
|
||
fi
|
||
|
||
# ---- 校验包内容 ----
|
||
[ -f "$pack/bin/chusql-storage" ] || fail "package is incomplete: bin/chusql-storage not found in $pack"
|
||
if [ "$component" = 'web' ]; then
|
||
[ -f "$pack/bin/chusql-web" ] || fail 'package is incomplete: bin/chusql-web not found'
|
||
[ -d "$pack/static" ] || fail 'package is incomplete: static/ not found'
|
||
else
|
||
[ -f "$pack/bin/csql" ] || fail 'package is incomplete: bin/csql not found'
|
||
fi
|
||
[ -f "$pack/scripts/chusql.toml" ] || fail 'package is incomplete: scripts/chusql.toml not found'
|
||
|
||
echo 'ChuSQL installer'
|
||
echo " component $component"
|
||
echo " install to $install_dir"
|
||
echo " data dir $data_dir"
|
||
if [ -n "$root_password" ]; then
|
||
echo " root user $root_user"
|
||
else
|
||
echo " root user $root_user (no password: administrator-only sign-in)"
|
||
fi
|
||
|
||
# ---- 释放文件 ----
|
||
step 'Installing files'
|
||
mkdir -p "$install_dir/bin" "$install_dir/logs" "$data_dir"
|
||
cp -R "$pack/bin/." "$install_dir/bin/"
|
||
chmod +x "$install_dir/bin/"* 2>/dev/null || true
|
||
if [ "$component" = 'web' ]; then
|
||
mkdir -p "$install_dir/static"
|
||
cp -R "$pack/static/." "$install_dir/static/"
|
||
if [ -f "$pack/csql-web.sh" ]; then
|
||
cp "$pack/csql-web.sh" "$install_dir/"
|
||
chmod +x "$install_dir/csql-web.sh"
|
||
fi
|
||
fi
|
||
|
||
# ---- 写全局配置 ----
|
||
step 'Writing chusql.toml'
|
||
config_dir="${XDG_CONFIG_HOME:-$HOME/.config}/ChuSQL"
|
||
config_file="$config_dir/chusql.toml"
|
||
mkdir -p "$config_dir"
|
||
sed -e "s|^\(user[[:space:]]*=[[:space:]]*\).*|\1\"$root_user\"|" \
|
||
-e "s|^\(password[[:space:]]*=[[:space:]]*\).*|\1\"$root_password\"|" \
|
||
-e "s|^\(data_dir[[:space:]]*=[[:space:]]*\).*|\1\"$data_dir\"|" \
|
||
"$pack/scripts/chusql.toml" > "$config_file"
|
||
echo " config file $config_file"
|
||
if [ -f "$pack/scripts/init.sql" ]; then cp "$pack/scripts/init.sql" "$install_dir/"; fi
|
||
|
||
# ---- 命令入口 ----
|
||
step 'Creating commands'
|
||
if [ "$component" = 'web' ]; then
|
||
cat > "$install_dir/csql-web" <<'EOF'
|
||
#!/bin/sh
|
||
exec "$(CDPATH= cd -- "$(dirname -- "$0")" && pwd)/csql-web.sh" "$@"
|
||
EOF
|
||
chmod +x "$install_dir/csql-web"
|
||
else
|
||
cat > "$install_dir/csql" <<'EOF'
|
||
#!/bin/sh
|
||
exec "$(CDPATH= cd -- "$(dirname -- "$0")" && pwd)/bin/csql" "$@"
|
||
EOF
|
||
chmod +x "$install_dir/csql"
|
||
fi
|
||
|
||
# ---- 环境变量 ----
|
||
step 'Updating PATH'
|
||
profile="$HOME/.profile"
|
||
marker='# ChuSQL'
|
||
if [ -f "$profile" ] && grep -q "^$marker" "$profile"; then
|
||
echo "PATH entry already present in $profile"
|
||
else
|
||
{
|
||
echo ''
|
||
echo "$marker"
|
||
echo "PATH=\"\$PATH:$install_dir\""
|
||
echo "export PATH"
|
||
} >> "$profile"
|
||
echo "added $install_dir to PATH in $profile (login again or: . \"$profile\")"
|
||
fi
|
||
|
||
# ---- 删掉安装包 ----
|
||
if [ "$keep_package" = 'no' ]; then
|
||
step 'Removing the package'
|
||
case "$install_dir" in
|
||
"$pack"*|"$pack") echo "kept the package (install dir is inside it): $pack" ;;
|
||
*)
|
||
if [ -f "$pack/bin/chusql-storage" ]; then
|
||
( cd / && rm -rf "$pack" )
|
||
echo 'package removed'
|
||
else
|
||
echo "kept the package (not safe to delete automatically): $pack"
|
||
fi
|
||
;;
|
||
esac
|
||
else
|
||
step 'Keeping the package'
|
||
echo " package kept at $pack"
|
||
fi
|
||
|
||
step 'Done'
|
||
echo " chusql.toml $config_file"
|
||
echo " data $data_dir"
|
||
if [ "$component" = 'web' ]; then echo ' start with csql-web'; else echo ' start with csql'; fi
|