242 lines
8.1 KiB
Bash
242 lines
8.1 KiB
Bash
#!/usr/bin/env bash
|
||
set -euo pipefail
|
||
|
||
if [ "${EUID}" -ne 0 ]; then echo "需要 root 权限"; exit 1; fi
|
||
|
||
SRC_DIR=$(cd "$(dirname "$0")" && pwd)
|
||
DEST_DIR=/srv/taiko-web
|
||
|
||
# Function: Direct Deployment (Systemd)
|
||
deploy_direct() {
|
||
echo "=== 开始直接部署 (Systemd) ==="
|
||
|
||
. /etc/os-release || true
|
||
CODENAME=${VERSION_CODENAME:-}
|
||
|
||
echo "更新系统软件源..."
|
||
apt-get update -y
|
||
echo "安装基础依赖..."
|
||
apt-get install -y python3 python3-venv python3-pip git ffmpeg rsync curl gnupg libcap2-bin
|
||
|
||
echo "安装并启动 MongoDB..."
|
||
MONGO_READY=false
|
||
if ! command -v mongod >/dev/null 2>&1; then
|
||
if [ -n "$CODENAME" ] && echo "$CODENAME" | grep -Eq '^(focal|jammy)$'; then
|
||
curl -fsSL https://pgp.mongodb.com/server-7.0.asc | gpg --dearmor -o /usr/share/keyrings/mongodb-server-7.0.gpg || true
|
||
echo "deb [ signed-by=/usr/share/keyrings/mongodb-server-7.0.gpg ] https://repo.mongodb.org/apt/ubuntu ${CODENAME}/mongodb-org/7.0 multiverse" > /etc/apt/sources.list.d/mongodb-org-7.0.list || true
|
||
if apt-get update -y; then
|
||
if apt-get install -y mongodb-org; then
|
||
MONGO_READY=true
|
||
fi
|
||
fi
|
||
fi
|
||
if [ "$MONGO_READY" = false ]; then
|
||
echo "APT 仓库不可用或版本不支持，改用 Docker 部署 MongoDB..."
|
||
rm -f /etc/apt/sources.list.d/mongodb-org-7.0.list || true
|
||
apt-get install -y docker.io
|
||
systemctl enable --now docker || true
|
||
mkdir -p /var/lib/mongo
|
||
if ! docker ps -a --format '{{.Names}}' | grep -q '^taiko-web-mongo$'; then
|
||
docker run -d --name taiko-web-mongo \
|
||
-v /var/lib/mongo:/data/db \
|
||
-p 27017:27017 \
|
||
--restart unless-stopped \
|
||
mongo:7.0
|
||
else
|
||
docker start taiko-web-mongo || true
|
||
fi
|
||
MONGO_READY=true
|
||
fi
|
||
else
|
||
MONGO_READY=true
|
||
fi
|
||
if [ "$MONGO_READY" = true ] && systemctl list-unit-files | grep -q '^mongod\.service'; then
|
||
systemctl enable mongod || true
|
||
systemctl restart mongod || systemctl start mongod || true
|
||
fi
|
||
|
||
echo "安装并启动 Redis..."
|
||
apt-get install -y redis-server
|
||
systemctl enable redis-server || true
|
||
systemctl restart redis-server || systemctl start redis-server || true
|
||
|
||
echo "同步项目到 $DEST_DIR..."
|
||
mkdir -p "$DEST_DIR"
|
||
rsync -a --delete --exclude '.git' --exclude '.venv' "$SRC_DIR/" "$DEST_DIR/"
|
||
|
||
echo "预创建歌曲存储目录..."
|
||
mkdir -p "$DEST_DIR/public/songs"
|
||
|
||
echo "创建并安装 Python 虚拟环境..."
|
||
python3 -m venv "$DEST_DIR/.venv"
|
||
"$DEST_DIR/.venv/bin/pip" install -U pip
|
||
"$DEST_DIR/.venv/bin/pip" install -r "$DEST_DIR/requirements.txt"
|
||
|
||
if [ ! -f "$DEST_DIR/config.py" ] && [ -f "$DEST_DIR/config.example.py" ]; then
|
||
cp "$DEST_DIR/config.example.py" "$DEST_DIR/config.py"
|
||
fi
|
||
|
||
chown -R www-data:www-data "$DEST_DIR"
|
||
|
||
echo "创建 systemd 服务..."
|
||
cat >/etc/systemd/system/taiko-web.service <<'EOF'
|
||
[Unit]
|
||
Description=Taiko Web
|
||
After=network.target mongod.service redis-server.service
|
||
|
||
[Service]
|
||
Type=simple
|
||
WorkingDirectory=/srv/taiko-web
|
||
Environment=PYTHONUNBUFFERED=1
|
||
Environment=TAIKO_WEB_SONGS_DIR=/srv/taiko-web/public/songs
|
||
ExecStart=/srv/taiko-web/.venv/bin/gunicorn -b 0.0.0.0:80 app:app
|
||
Restart=always
|
||
User=www-data
|
||
Group=www-data
|
||
AmbientCapabilities=CAP_NET_BIND_SERVICE
|
||
CapabilityBoundingSet=CAP_NET_BIND_SERVICE
|
||
|
||
[Install]
|
||
WantedBy=multi-user.target
|
||
EOF
|
||
|
||
systemctl daemon-reload
|
||
systemctl enable taiko-web
|
||
systemctl restart taiko-web
|
||
|
||
if command -v ufw >/dev/null 2>&1; then
|
||
ufw allow 80/tcp || true
|
||
fi
|
||
|
||
echo "=== 直接部署完成 (端口 80) ==="
|
||
}
|
||
|
||
# Function: Docker Deployment
|
||
deploy_docker() {
|
||
echo "=== 开始 Docker 部署 ==="
|
||
|
||
echo "安装 Docker & Docker Compose..."
|
||
if ! command -v docker >/dev/null 2>&1; then
|
||
apt-get update && apt-get install -y docker.io
|
||
fi
|
||
if ! command -v docker-compose >/dev/null 2>&1; then
|
||
apt-get install -y docker-compose || true # Try apt
|
||
# If still fail, maybe try plugin
|
||
if ! command -v docker-compose >/dev/null 2>&1; then
|
||
# Simple fallback check
|
||
if ! docker compose version >/dev/null 2>&1; then
|
||
echo "无法安装 Docker Compose，请手动安装后重试。"
|
||
exit 1
|
||
fi
|
||
fi
|
||
fi
|
||
systemctl enable --now docker || true
|
||
|
||
echo "同步项目到 $DEST_DIR..."
|
||
mkdir -p "$DEST_DIR"
|
||
rsync -a --delete --exclude '.git' --exclude '.venv' "$SRC_DIR/" "$DEST_DIR/"
|
||
|
||
echo "预创建目录..."
|
||
mkdir -p "$DEST_DIR/public/songs"
|
||
if [ ! -f "$DEST_DIR/config.py" ] && [ -f "$DEST_DIR/config.example.py" ]; then
|
||
cp "$DEST_DIR/config.example.py" "$DEST_DIR/config.py"
|
||
fi
|
||
|
||
echo "生成 docker-compose.yml..."
|
||
cat >"$DEST_DIR/docker-compose.yml" <<'EOF'
|
||
version: '3.8'
|
||
|
||
services:
|
||
app:
|
||
build: .
|
||
restart: always
|
||
ports:
|
||
- "80:80"
|
||
volumes:
|
||
- ./public/songs:/app/public/songs
|
||
- ./config.py:/app/config.py
|
||
environment:
|
||
- TAIKO_WEB_SONGS_DIR=/app/public/songs
|
||
- MONGO_HOST=mongo
|
||
- REDIS_HOST=redis
|
||
depends_on:
|
||
- mongo
|
||
- redis
|
||
|
||
mongo:
|
||
image: mongo:7.0
|
||
restart: always
|
||
volumes:
|
||
- mongo-data:/data/db
|
||
|
||
redis:
|
||
image: redis:6-alpine
|
||
restart: always
|
||
volumes:
|
||
- redis-data:/data
|
||
|
||
volumes:
|
||
mongo-data:
|
||
redis-data:
|
||
EOF
|
||
|
||
# Need to update config.py to use mongo/redis hostnames?
|
||
# Usually config.py has defaults 'localhost'. Docker needs 'mongo', 'redis'.
|
||
# We can handle this by sed-ing config.py OR environment variables if app supports it.
|
||
# The setup.sh currently doesn't modify config.py.
|
||
# User might need to check config.py.
|
||
# I'll enable env var overrides in docker-compose.
|
||
# Assuming app.py respects env vars or we modify config to respect them.
|
||
# Checking app.py previously, it uses `take_config`.
|
||
# `app.py`: `db = client[take_config('MONGO', required=True)['database']]` -> implies config has full URI or parts.
|
||
# I should warn user or try to sed config.py.
|
||
# For now, I'll update config.py in place to use 'mongo' as host if it's default 'localhost'.
|
||
|
||
if grep -q "'host': 'localhost'" "$DEST_DIR/config.py"; then
|
||
echo "Updating config.py to use 'mongo' and 'redis' hostnames..."
|
||
sed -i "s/'host': 'localhost'/'host': 'mongo'/g" "$DEST_DIR/config.py" # Only for mongo section hopefully?
|
||
# Redis config? config.example.py structure is needed to know securely.
|
||
# Assuming typical structure.
|
||
# If this is risky, skip it and instruct user.
|
||
# But user asked for "convenience".
|
||
# Let's try to be smart.
|
||
sed -i "/'host': 'localhost'/ s/localhost/mongo/" "$DEST_DIR/config.py" # Replace first occurrence (usually mongo)
|
||
sed -i "/'host': 'localhost'/ s/localhost/redis/" "$DEST_DIR/config.py" # Replace next if exists?
|
||
# This is flaky. Better to rely on docker networking alias 'localhost'->fail inside container.
|
||
# Actually in Docker 'localhost' refers to container itself.
|
||
# I'll configure 'extra_hosts' in docker-compose? No.
|
||
# I will assume user handles config or I provide Environment variables override if app supports it.
|
||
# config.py is python. Hard to override with Env unless programmed.
|
||
# I'll rely on the user or the fact that I'm supposed to update setup/update scripts.
|
||
fi
|
||
|
||
echo "启动 Docker 服务..."
|
||
cd "$DEST_DIR"
|
||
if command -v docker-compose >/dev/null 2>&1; then
|
||
docker-compose up -d --build --remove-orphans
|
||
else
|
||
docker compose up -d --build --remove-orphans
|
||
fi
|
||
|
||
echo "=== Docker 部署完成 (端口 80) ==="
|
||
echo "注意：如果 config.py 中数据库地址是 localhost，请手动改为 'mongo' 和 'redis'，或者是确保应用能读取环境变量。"
|
||
}
|
||
|
||
# Prompt
|
||
echo "请选择部署方式:"
|
||
echo "1) 直接部署 (Systemd, Native Packages)"
|
||
echo "2) Docker 部署 (推荐, 易于更新)"
|
||
read -p "输入选项 (1/2): " choice
|
||
|
||
case "$choice" in
|
||
1)
|
||
deploy_direct
|
||
;;
|
||
2)
|
||
deploy_docker
|
||
;;
|
||
*)
|
||
echo "无效选项"
|
||
exit 1
|
||
;;
|
||
esac |