#!/bin/bash
set -e

echo "=== Pic-Bed 一键安装 ==="
echo ""

# ========== 权限检测 ==========
if [ "$(id -u)" -eq 0 ]; then
  SUDO=""
  echo "👑 当前为 root 用户"
else
  if command -v sudo &> /dev/null; then
    SUDO="sudo"
    echo "🔐 非 root 用户，将使用 sudo"
  else
    echo "❌ 当前不是 root 用户，且系统未安装 sudo"
    echo "请切换到 root 用户后重新执行：su root"
    exit 1
  fi
fi

# ========== 架构检测 ==========
ARCH=$(uname -m)
case $ARCH in
  x86_64)   PLATFORM="linux-amd64" ;;
  aarch64)  PLATFORM="linux-arm64" ;;
  armv7l)   PLATFORM="linux-armv7" ;;
  *)
    echo ""
    echo "❌ 不支持的架构: $ARCH"
    echo "支持的架构: x86_64, aarch64, armv7l"
    exit 1
    ;;
esac
echo "📦 检测到架构: $ARCH ($PLATFORM)"

# ========== 获取最新版本 ==========
echo ""
echo "🔍 获取最新版本..."
VERSION=$(curl -s https://api.github.com/repos/renjiu13/Pic-bed/releases/latest | grep tag_name | cut -d '"' -f4)
if [ -z "$VERSION" ] || [ "$VERSION" = "null" ]; then
  echo "❌ 获取版本信息失败，请检查网络连接或 GitHub 是否可访问"
  exit 1
fi
echo "✨ 最新版本: $VERSION"

# ========== 安装目录 ==========
INSTALL_DIR="/opt/pic-bed"
echo ""
echo "📂 安装目录: $INSTALL_DIR"

# 检测是否已安装
SERVICE_RESTART_NEEDED=false
if [ -f "$INSTALL_DIR/pic-bed" ]; then
  OLD_VERSION=$($INSTALL_DIR/pic-bed -v 2>/dev/null | head -1 || echo "未知版本")
  echo "⚠️  检测到已安装版本: $OLD_VERSION"
  echo "将覆盖升级到 $VERSION"
  echo ""
  # 检测是否运行在 systemd 服务中
  if systemctl is-active --quiet pic-bed 2>/dev/null; then
    SERVICE_RESTART_NEEDED=true
    echo "🔄 检测到 systemd 服务正在运行，更新后将自动重启"
    echo ""
  fi
fi

# ========== 创建目录 ==========
$SUDO mkdir -p $INSTALL_DIR

# ========== 下载 ==========
echo "⬇️  下载中..."
$SUDO curl -L --progress-bar \
  "https://github.com/renjiu13/Pic-bed/releases/download/$VERSION/pic-bed-$PLATFORM" \
  -o $INSTALL_DIR/pic-bed
$SUDO chmod +x $INSTALL_DIR/pic-bed

# ========== 创建全局命令软链接（修复点） ==========
echo ""
echo "🔗 创建全局命令..."
$SUDO ln -sf $INSTALL_DIR/pic-bed /usr/local/bin/pic-bed
echo "✓ 已创建软链接: /usr/local/bin/pic-bed"

# ========== 配置 systemd 服务 ==========
SERVICE_CREATED=false
if command -v systemctl &> /dev/null; then
  SERVICE_FILE="/etc/systemd/system/pic-bed.service"
  
  # 如果服务文件不存在，自动创建
  if [ ! -f "$SERVICE_FILE" ]; then
    echo ""
    echo "🔧 检测到 systemd，自动配置开机自启服务..."
    
    $SUDO tee "$SERVICE_FILE" > /dev/null << 'EOF'
[Unit]
Description=Pic-bed - 极轻量图床
After=network.target

[Service]
Type=simple
User=root
WorkingDirectory=/opt/pic-bed
ExecStart=/opt/pic-bed/pic-bed
Restart=always
RestartSec=2
MemoryLimit=100M

[Install]
WantedBy=multi-user.target
EOF
    
    $SUDO systemctl daemon-reload
    $SUDO systemctl enable pic-bed
    SERVICE_CREATED=true
    echo "✓ systemd 服务已创建并设置开机自启"
  fi
fi

# ========== 重启服务（如果需要） ==========
if [ "$SERVICE_RESTART_NEEDED" = true ]; then
  echo ""
  echo "🔄 重启服务中..."
  $SUDO systemctl restart pic-bed
  echo "✓ 服务已重启"
fi

# ========== 完成 ==========
echo ""
echo "✅ 安装完成！"
echo ""
echo "📦 版本: $VERSION"
echo ""

if [ "$SERVICE_CREATED" = true ] || [ "$SERVICE_RESTART_NEEDED" = true ]; then
  echo "🚀 服务状态："
  echo "   systemctl status pic-bed   # 查看状态"
  echo "   systemctl start pic-bed    # 启动"
  echo "   systemctl stop pic-bed     # 停止"
  echo "   systemctl restart pic-bed  # 重启"
  echo "   journalctl -u pic-bed -f   # 查看日志"
else
  echo "🚀 启动服务："
  echo "   cd $INSTALL_DIR && ./pic-bed"
fi

echo ""
echo "⚙️  快捷命令："
echo "   pic-bed -v           # 查看版本"
echo "   pic-bed -h           # 查看帮助"
echo "   pic-bed check-update # 检查更新"
echo "   pic-bed update       # 在线更新"
echo "   pic-bed              # 启动服务"
echo ""
echo "⚙️  配置文件："
echo "   首次运行自动生成 $INSTALL_DIR/config.json"
echo ""
echo "🔄 在线更新："
echo "   pic-bed check-update  # 检查更新"
echo "   pic-bed update        # 一键更新"
echo ""