#!/bin/bash
# 浪浪AI VPS部署脚本
# 在Ubuntu 22.04 VPS上运行此脚本
#
# 前置条件:
# 1. 从 @BotFather 创建 Telegram Bot，获取 Token
# 2. 获取你的 Chat ID（给 @userinfobot 发消息）
#
# 用法:
#   chmod +x deploy.sh
#   ./deploy.sh

set -e

echo "=========================================="
echo "  浪浪AI 部署脚本"
echo "=========================================="

# 1. 安装依赖
echo "[1/6] 安装系统依赖..."
sudo apt-get update -qq
sudo apt-get install -y python3 python3-pip python3-venv

# 2. 创建目录
echo "[2/6] 创建项目目录..."
sudo mkdir -p /opt/langlang_ai
sudo cp -r . /opt/langlang_ai/
sudo chown -R $USER:$USER /opt/langlang_ai

# 3. 安装Python依赖
echo "[3/6] 安装Python依赖..."
cd /opt/langlang_ai
pip3 install --break-system-packages pandas numpy requests aiohttp

# 4. 配置环境变量
echo "[4/6] 配置Telegram..."
if [ -z "$TELEGRAM_BOT_TOKEN" ]; then
    read -p "Telegram Bot Token: " TELEGRAM_BOT_TOKEN
fi
if [ -z "$TELEGRAM_CHAT_ID" ]; then
    read -p "Telegram Chat ID: " TELEGRAM_CHAT_ID
fi

# 更新 systemd service
sudo sed -i "s/YOUR_BOT_TOKEN_HERE/$TELEGRAM_BOT_TOKEN/" deploy/langlang_ai.service
sudo sed -i "s/YOUR_CHAT_ID_HERE/$TELEGRAM_CHAT_ID/" deploy/langlang_ai.service

# 5. 安装 systemd 服务
echo "[5/6] 安装系统服务..."
sudo cp deploy/langlang_ai.service /etc/systemd/system/
sudo systemctl daemon-reload

# 6. 初始化历史数据并启动
echo "[6/6] 初始化数据并启动..."
cd /opt/langlang_ai
TELEGRAM_BOT_TOKEN=$TELEGRAM_BOT_TOKEN TELEGRAM_CHAT_ID=$TELEGRAM_CHAT_ID \
    python3 main.py --init --init-days 365 --once

sudo systemctl enable langlang_ai
sudo systemctl start langlang_ai

echo ""
echo "=========================================="
echo "  部署完成！"
echo "=========================================="
echo ""
echo "  管理命令:"
echo "    sudo systemctl status langlang_ai   # 查看状态"
echo "    sudo systemctl stop langlang_ai     # 停止"
echo "    sudo systemctl restart langlang_ai  # 重启"
echo "    sudo journalctl -u langlang_ai -f   # 查看日志"
echo ""
echo "  Telegram Bot 已启动，等待信号推送..."