#!/bin/bash

# 定义变量
APT_SOURCE="https://apt.feinian.net"
DISTRO="stable"  # 假设你的发行版是 stable，可根据实际情况修改
COMPONENT="main" # 假设组件是 main，可根据实际情况修改
SOURCE_LIST="/etc/apt/sources.list.d/feinian-apt.list"

# 检查是否以 root 权限运行
if [ "$EUID" -ne 0 ]; then
    echo "请以 root 权限运行此脚本（使用 sudo）"
    exit 1
fi

# 添加 APT 源
echo "deb $APT_SOURCE $DISTRO $COMPONENT" > "$SOURCE_LIST"
echo "已添加 APT 源到 $SOURCE_LIST"

# 如果你的源提供了 GPG 公钥（可选）
# 假设公钥地址是 https://apt.feinian.net/key.asc（请确认实际地址）
GPG_KEY_URL="$APT_SOURCE/key.asc"
if curl --output /dev/null --silent --head --fail "$GPG_KEY_URL"; then
    echo "正在下载并添加 GPG 公钥..."
    curl -fsSL "$GPG_KEY_URL" | apt-key add -
    if [ $? -eq 0 ]; then
        echo "GPG 公钥添加成功"
    else
        echo "警告：GPG 公钥添加失败，请手动检查 $GPG_KEY_URL"
    fi
else
    echo "未找到 GPG 公钥，跳过签名验证（不推荐）"
fi

# 更新 APT 索引
echo "正在更新 APT 索引..."
apt update

if [ $? -eq 0 ]; then
    echo "APT 源添加并更新成功！"
else
    echo "APT 更新失败，请检查网络或源配置"
    exit 1
fi

exit 0
