#!/bin/bash

###############################################################################
# systemd 服务安装脚本
###############################################################################

# 颜色定义
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
RED='\033[0;31m'
NC='\033[0m'

echo -e "${YELLOW}=========================================${NC}"
echo -e "${YELLOW}  安装 systemd 服务${NC}"
echo -e "${YELLOW}=========================================${NC}"
echo ""

# 检查是否为root
if [ "$EUID" -ne 0 ]; then 
    echo -e "${RED}请使用 sudo 运行此脚本${NC}"
    exit 1
fi

PROJECT_DIR="/var/www/ai_baijiahao"
SERVICE_FILE="baijiahao.service"

# 复制服务文件
echo "复制服务文件到 /etc/systemd/system/..."
cp ${PROJECT_DIR}/${SERVICE_FILE} /etc/systemd/system/

# 重载 systemd
echo "重载 systemd 配置..."
systemctl daemon-reload

# 启用服务（开机自启）
echo "启用服务开机自启..."
systemctl enable baijiahao

echo ""
echo -e "${GREEN}=========================================${NC}"
echo -e "${GREEN}  服务安装完成！${NC}"
echo -e "${GREEN}=========================================${NC}"
echo ""
echo "常用命令："
echo -e "  启动服务: ${YELLOW}sudo systemctl start baijiahao${NC}"
echo -e "  停止服务: ${YELLOW}sudo systemctl stop baijiahao${NC}"
echo -e "  重启服务: ${YELLOW}sudo systemctl restart baijiahao${NC}"
echo -e "  查看状态: ${YELLOW}sudo systemctl status baijiahao${NC}"
echo -e "  查看日志: ${YELLOW}sudo journalctl -u baijiahao -f${NC}"
echo -e "  禁用自启: ${YELLOW}sudo systemctl disable baijiahao${NC}"
echo ""