mirror of
https://github.com/whyour/qinglong.git
synced 2026-06-28 02:45:08 +08:00
163 lines
4.4 KiB
Bash
163 lines
4.4 KiB
Bash
#!/usr/bin/env bash
|
||
|
||
# 前置依赖 nodejs、npm、python3
|
||
set -e
|
||
set -x
|
||
|
||
if [[ ! $QL_DIR ]]; then
|
||
npm_dir=$(npm root -g)
|
||
pnpm_dir=$(pnpm root -g)
|
||
if [[ -d "$npm_dir/@whyour/qinglong" ]]; then
|
||
QL_DIR="$npm_dir/@whyour/qinglong"
|
||
elif [[ -d "$pnpm_dir/@whyour/qinglong" ]]; then
|
||
QL_DIR="$pnpm_dir/@whyour/qinglong"
|
||
else
|
||
t '未找到 qinglong 模块，请先执行 npm i -g @whyour/qinglong 安装'
|
||
fi
|
||
|
||
if [[ $QL_DIR ]]; then
|
||
t '请先手动设置 export QL_DIR=%s，环境变量，并手动添加到系统环境变量，然后再次执行命令 qinglong 启动服务' "$QL_DIR"
|
||
fi
|
||
|
||
exit 1
|
||
fi
|
||
|
||
if [[ ! $QL_DATA_DIR ]]; then
|
||
t '请先手动设置数据存储目录 export QL_DATA_DIR 环境变量，目录必须以斜杠开头的绝对路径，并且以 /data 结尾，例如 /ql/data 并手动添加到系统环境变量'
|
||
exit 1
|
||
fi
|
||
|
||
if [[ $QL_DATA_DIR != */data ]]; then
|
||
t 'QL_DATA_DIR 必须以 /data 结尾，例如 /ql/data，如果有历史数据，请新建 data 目录，把历史数据放到 data 目录中'
|
||
exit 1
|
||
fi
|
||
|
||
command="$1"
|
||
|
||
if [[ $command != "reload" ]]; then
|
||
# 安装依赖
|
||
os_name="${QL_OS_TYPE:-}"
|
||
if [ -z "$os_name" ]; then
|
||
os_name=$(source /etc/os-release && echo "$ID")
|
||
fi
|
||
|
||
# 非 root 用户使用 sudo
|
||
SUDO=""
|
||
if [ "$(id -u)" -ne 0 ]; then
|
||
SUDO="sudo"
|
||
fi
|
||
|
||
case "$os_name" in
|
||
alpine)
|
||
$SUDO apk update
|
||
$SUDO apk add -f bash \
|
||
coreutils \
|
||
git \
|
||
curl \
|
||
wget \
|
||
tzdata \
|
||
perl \
|
||
openssl \
|
||
jq \
|
||
nginx \
|
||
openssh \
|
||
procps \
|
||
netcat-openbsd
|
||
;;
|
||
debian|ubuntu)
|
||
$SUDO apt-get update
|
||
$SUDO apt-get install -y git curl wget tzdata perl openssl jq nginx procps netcat-openbsd openssh-client
|
||
;;
|
||
*)
|
||
t '暂不支持此系统部署 %s' "$os_name"
|
||
exit 1
|
||
;;
|
||
esac
|
||
|
||
npm install -g pnpm@8.3.1 pm2 ts-node
|
||
fi
|
||
|
||
# 从 .env 文件读取 PYTHON_VENV_DIR（.env 是给 Node.js 用的，bash 需要手动读取）
|
||
if [[ -z "${PYTHON_VENV_DIR:-}" ]] && [[ -f "${QL_DIR}/.env" ]]; then
|
||
env_venv_dir=$(grep -E '^PYTHON_VENV_DIR=' "${QL_DIR}/.env" | tail -1 | cut -d'=' -f2 | xargs)
|
||
if [[ -n "$env_venv_dir" ]]; then
|
||
# 处理相对路径
|
||
if [[ "$env_venv_dir" != /* ]]; then
|
||
env_venv_dir="${QL_DIR}/${env_venv_dir}"
|
||
fi
|
||
export PYTHON_VENV_DIR="$env_venv_dir"
|
||
fi
|
||
fi
|
||
|
||
export PNPM_HOME=${QL_DIR}/data/dep_cache/node
|
||
|
||
# ===== Python venv =====
|
||
export PYTHON_VENV_DIR="${PYTHON_VENV_DIR:-${QL_DIR}/.venv}"
|
||
if [[ ! -f "${PYTHON_VENV_DIR}/bin/python3" ]]; then
|
||
echo "正在创建 Python venv: ${PYTHON_VENV_DIR}"
|
||
python3 -m venv "${PYTHON_VENV_DIR}"
|
||
fi
|
||
|
||
# 仅将 venv 的 bin 加入 PATH，Python 的 venv 机制自动处理包路径
|
||
export PATH=${PYTHON_VENV_DIR}/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:${PNPM_HOME}
|
||
export NODE_PATH=/usr/local/bin:/usr/local/lib/node_modules:${PNPM_HOME}/global/5/node_modules
|
||
export PIP_CACHE_DIR=${PYTHON_VENV_DIR}/pip
|
||
mkdir -p "${PIP_CACHE_DIR}"
|
||
|
||
if [[ $command != "reload" ]]; then
|
||
${PYTHON_VENV_DIR}/bin/pip3 install requests
|
||
fi
|
||
|
||
cd ${QL_DIR}
|
||
# 仅在 .env 不存在时从 .env.example 复制
|
||
if [[ ! -f .env ]] && [[ -f .env.example ]]; then
|
||
cp -f .env.example .env
|
||
fi
|
||
chmod 777 ${QL_DIR}/shell/*.sh
|
||
|
||
# 确保 task/ql 命令指向当前部署目录（覆盖 npm 全局安装可能指向旧路径的软链接）
|
||
ln -sf ${QL_DIR}/shell/task.sh /usr/local/bin/task 2>/dev/null || true
|
||
ln -sf ${QL_DIR}/shell/update.sh /usr/local/bin/ql 2>/dev/null || true
|
||
|
||
. ${QL_DIR}/shell/share.sh
|
||
. ${QL_DIR}/shell/env.sh
|
||
|
||
log_with_style() {
|
||
local level="$1"
|
||
local message="$2"
|
||
local timestamp=$(date '+%Y-%m-%d %H:%M:%S')
|
||
|
||
printf "\n[%s] [%7s] %s\n" "${timestamp}" "${level}" "${message}"
|
||
}
|
||
|
||
log_with_style "INFO" "🚀 1. 检测配置文件..."
|
||
import_config "$@"
|
||
make_dir /etc/nginx/conf.d
|
||
make_dir /run/nginx
|
||
fix_config
|
||
|
||
pm2 l &>/dev/null
|
||
|
||
log_with_style "INFO" "🔄 2. 启动 nginx..."
|
||
nginx -s reload 2>/dev/null || nginx -c /etc/nginx/nginx.conf
|
||
|
||
log_with_style "INFO" "⚙️ 3. 启动 pm2 服务..."
|
||
reload_pm2
|
||
|
||
if [[ $command != "reload" ]]; then
|
||
if [[ $AutoStartBot == true ]]; then
|
||
log_with_style "INFO" "🤖 4. 启动 bot..."
|
||
nohup ql bot >$dir_log/bot.log 2>&1 &
|
||
fi
|
||
|
||
if [[ $EnableExtraShell == true ]]; then
|
||
log_with_style "INFO" "🛠️ 5. 执行自定义脚本..."
|
||
nohup ql extra >$dir_log/extra.log 2>&1 &
|
||
fi
|
||
|
||
pm2 startup
|
||
pm2 save
|
||
fi
|
||
|
||
log_with_style "SUCCESS" "🎉 启动成功!"
|