362 lines
8.7 KiB
Plaintext
362 lines
8.7 KiB
Plaintext
cat > r2-auto-setup.sh << 'EOF'
|
|
#!/bin/bash
|
|
|
|
# Cloudflare R2 一键配置脚本
|
|
R2_SETUP_VERSION="1.0"
|
|
|
|
# 颜色定义
|
|
RED='\033[0;31m'
|
|
GREEN='\033[0;32m'
|
|
YELLOW='\033[1;33m'
|
|
BLUE='\033[0;34m'
|
|
NC='\033[0m'
|
|
|
|
# 日志函数
|
|
log() {
|
|
echo -e "${GREEN}[$(date +'%Y-%m-%d %H:%M:%S')]${NC} $1"
|
|
}
|
|
|
|
warn() {
|
|
echo -e "${YELLOW}[警告]${NC} $1"
|
|
}
|
|
|
|
error() {
|
|
echo -e "${RED}[错误]${NC} $1"
|
|
}
|
|
|
|
info() {
|
|
echo -e "${BLUE}[信息]${NC} $1"
|
|
}
|
|
|
|
# 显示横幅
|
|
show_banner() {
|
|
clear
|
|
echo -e "${GREEN}"
|
|
echo "================================================"
|
|
echo " Cloudflare R2 一键配置脚本 v$R2_SETUP_VERSION"
|
|
echo "================================================"
|
|
echo -e "${NC}"
|
|
}
|
|
|
|
# 检查系统
|
|
check_system() {
|
|
log "检查系统环境..."
|
|
|
|
# 检查 curl
|
|
if ! command -v curl &> /dev/null; then
|
|
error "curl 未安装，正在安装..."
|
|
apt-get update && apt-get install -y curl || yum install -y curl
|
|
fi
|
|
|
|
# 检查 unzip
|
|
if ! command -v unzip &> /dev/null; then
|
|
warn "unzip 未安装，正在安装..."
|
|
apt-get install -y unzip || yum install -y unzip
|
|
fi
|
|
}
|
|
|
|
# 安装 rclone
|
|
install_rclone() {
|
|
if command -v rclone &> /dev/null; then
|
|
log "rclone 已安装，版本: $(rclone version | head -1)"
|
|
return 0
|
|
fi
|
|
|
|
log "安装 rclone..."
|
|
curl -s https://rclone.org/install.sh | sudo bash > /dev/null 2>&1
|
|
|
|
if command -v rclone &> /dev/null; then
|
|
log "✓ rclone 安装成功"
|
|
else
|
|
error "rclone 安装失败"
|
|
exit 1
|
|
fi
|
|
}
|
|
|
|
# 获取用户输入
|
|
get_user_input() {
|
|
echo
|
|
info "请提供 Cloudflare R2 配置信息："
|
|
echo
|
|
|
|
read -p "请输入配置名称 [默认: r2-storage]: " REMOTE_NAME
|
|
REMOTE_NAME=${REMOTE_NAME:-r2-storage}
|
|
|
|
read -p "请输入 Account ID: " ACCOUNT_ID
|
|
while [ -z "$ACCOUNT_ID" ]; do
|
|
error "Account ID 不能为空"
|
|
read -p "请输入 Account ID: " ACCOUNT_ID
|
|
done
|
|
|
|
read -p "请输入 Access Key ID: " ACCESS_KEY
|
|
while [ -z "$ACCESS_KEY" ]; do
|
|
error "Access Key ID 不能为空"
|
|
read -p "请输入 Access Key ID: " ACCESS_KEY
|
|
done
|
|
|
|
read -p "请输入 Secret Access Key: " SECRET_KEY
|
|
while [ -z "$SECRET_KEY" ]; do
|
|
error "Secret Access Key 不能为空"
|
|
read -p "请输入 Secret Access Key: " SECRET_KEY
|
|
done
|
|
|
|
read -p "请输入存储桶名称: " BUCKET_NAME
|
|
while [ -z "$BUCKET_NAME" ]; do
|
|
error "存储桶名称不能为空"
|
|
read -p "请输入存储桶名称: " BUCKET_NAME
|
|
done
|
|
|
|
ENDPOINT="https://${ACCOUNT_ID}.r2.cloudflarestorage.com"
|
|
}
|
|
|
|
# 配置 R2
|
|
configure_r2() {
|
|
log "配置 Cloudflare R2..."
|
|
|
|
# 删除已存在的配置
|
|
rclone config delete "$REMOTE_NAME" --quiet > /dev/null 2>&1
|
|
|
|
# 创建新配置
|
|
rclone config create "$REMOTE_NAME" s3 \
|
|
provider Cloudflare \
|
|
access_key_id "$ACCESS_KEY" \
|
|
secret_access_key "$SECRET_KEY" \
|
|
endpoint "$ENDPOINT" \
|
|
region auto \
|
|
acl private \
|
|
--non-interactive
|
|
|
|
if [ $? -eq 0 ]; then
|
|
log "✓ R2 配置成功"
|
|
else
|
|
error "R2 配置失败"
|
|
return 1
|
|
fi
|
|
}
|
|
|
|
# 测试配置
|
|
test_configuration() {
|
|
log "测试 R2 连接..."
|
|
|
|
# 测试存储桶访问
|
|
if rclone lsd "$REMOTE_NAME":"$BUCKET_NAME" > /dev/null 2>&1; then
|
|
log "✓ 存储桶访问测试成功"
|
|
return 0
|
|
fi
|
|
|
|
# 尝试创建测试目录
|
|
warn "存储桶访问失败，尝试创建测试目录..."
|
|
if rclone mkdir "$REMOTE_NAME":"$BUCKET_NAME"/r2-test > /dev/null 2>&1; then
|
|
log "✓ 目录创建测试成功"
|
|
rclone rmdir "$REMOTE_NAME":"$BUCKET_NAME"/r2-test > /dev/null 2>&1
|
|
return 0
|
|
else
|
|
error "配置测试失败"
|
|
warn "请检查："
|
|
warn "1. 存储桶名称是否正确"
|
|
warn "2. API 密钥权限是否足够"
|
|
warn "3. 网络连接是否正常"
|
|
return 1
|
|
fi
|
|
}
|
|
|
|
# 创建管理脚本
|
|
create_management_scripts() {
|
|
log "创建管理脚本..."
|
|
|
|
# 备份脚本
|
|
cat > r2-backup.sh << 'SCRIPTEOF'
|
|
#!/bin/bash
|
|
# Cloudflare R2 自动备份脚本
|
|
|
|
REMOTE="r2-storage"
|
|
BUCKET="${1:-default-bucket}"
|
|
BACKUP_SOURCE="${2:-/opt/1panel}"
|
|
BACKUP_PREFIX="backup"
|
|
DATE=$(date +%Y%m%d_%H%M%S)
|
|
BACKUP_FILE="/tmp/${BACKUP_PREFIX}_${DATE}.tar.gz"
|
|
LOG_FILE="/var/log/r2-backup.log"
|
|
|
|
log() {
|
|
echo "[$(date +'%Y-%m-%d %H:%M:%S')] $1" >> "$LOG_FILE"
|
|
echo "$1"
|
|
}
|
|
|
|
log "开始备份到 Cloudflare R2"
|
|
|
|
# 检查源目录
|
|
if [ ! -d "$BACKUP_SOURCE" ]; then
|
|
log "错误: 源目录不存在: $BACKUP_SOURCE"
|
|
exit 1
|
|
fi
|
|
|
|
# 创建备份
|
|
log "创建备份文件: $BACKUP_FILE"
|
|
tar -czf "$BACKUP_FILE" -C "$BACKUP_SOURCE" . 2>/dev/null
|
|
if [ $? -ne 0 ]; then
|
|
log "错误: 备份文件创建失败"
|
|
exit 1
|
|
fi
|
|
|
|
# 上传到 R2
|
|
log "上传到 Cloudflare R2..."
|
|
rclone copy "$BACKUP_FILE" "$REMOTE:$BUCKET/backups/"
|
|
if [ $? -eq 0 ]; then
|
|
log "✓ 备份成功: ${BACKUP_PREFIX}_${DATE}.tar.gz"
|
|
rm -f "$BACKUP_FILE"
|
|
else
|
|
log "错误: 上传到 R2 失败"
|
|
exit 1
|
|
fi
|
|
|
|
log "备份完成"
|
|
SCRIPTEOF
|
|
|
|
# 管理脚本
|
|
cat > r2-manage.sh << 'MANAGEEOF'
|
|
#!/bin/bash
|
|
|
|
REMOTE="r2-storage"
|
|
BUCKET="${1}"
|
|
COMMAND="${2}"
|
|
ARG1="${3}"
|
|
ARG2="${4}"
|
|
|
|
show_help() {
|
|
echo "Cloudflare R2 管理脚本"
|
|
echo "用法: $0 <存储桶> <命令> [参数]"
|
|
echo
|
|
echo "命令:"
|
|
echo " list 列出存储桶内容"
|
|
echo " ls 列出所有文件"
|
|
echo " size 显示存储桶大小"
|
|
echo " mkdir <目录名> 创建目录"
|
|
echo " upload <文件> [路径] 上传文件"
|
|
echo " download <文件> [路径] 下载文件"
|
|
echo " delete <文件> 删除文件"
|
|
echo " sync <本地路径> 同步目录"
|
|
echo
|
|
echo "示例:"
|
|
echo " $0 my-bucket list"
|
|
echo " $0 my-bucket upload file.txt"
|
|
echo " $0 my-bucket mkdir backups"
|
|
}
|
|
|
|
if [ -z "$BUCKET" ] || [ -z "$COMMAND" ]; then
|
|
show_help
|
|
exit 1
|
|
fi
|
|
|
|
case "$COMMAND" in
|
|
"list")
|
|
rclone lsd "$REMOTE:$BUCKET"
|
|
;;
|
|
"ls")
|
|
rclone ls "$REMOTE:$BUCKET"
|
|
;;
|
|
"size")
|
|
rclone size "$REMOTE:$BUCKET"
|
|
;;
|
|
"mkdir")
|
|
if [ -z "$ARG1" ]; then
|
|
echo "错误: 请指定目录名"
|
|
exit 1
|
|
fi
|
|
rclone mkdir "$REMOTE:$BUCKET/$ARG1"
|
|
;;
|
|
"upload")
|
|
if [ -z "$ARG1" ]; then
|
|
echo "错误: 请指定要上传的文件"
|
|
exit 1
|
|
fi
|
|
REMOTE_PATH="${ARG2:-/}"
|
|
rclone copy "$ARG1" "$REMOTE:$BUCKET$REMOTE_PATH"
|
|
;;
|
|
"download")
|
|
if [ -z "$ARG1" ]; then
|
|
echo "错误: 请指定要下载的文件"
|
|
exit 1
|
|
fi
|
|
LOCAL_PATH="${ARG2:-./}"
|
|
rclone copy "$REMOTE:$BUCKET/$ARG1" "$LOCAL_PATH"
|
|
;;
|
|
"delete")
|
|
if [ -z "$ARG1" ]; then
|
|
echo "错误: 请指定要删除的文件"
|
|
exit 1
|
|
fi
|
|
rclone delete "$REMOTE:$BUCKET/$ARG1"
|
|
;;
|
|
"sync")
|
|
if [ -z "$ARG1" ]; then
|
|
echo "错误: 请指定本地路径"
|
|
exit 1
|
|
fi
|
|
rclone sync "$ARG1" "$REMOTE:$BUCKET/"
|
|
;;
|
|
*)
|
|
show_help
|
|
;;
|
|
esac
|
|
MANAGEEOF
|
|
|
|
# 快速测试脚本
|
|
cat > r2-test.sh << 'TESTEOF'
|
|
#!/bin/bash
|
|
echo "测试 Cloudflare R2 连接..."
|
|
rclone lsd r2-storage:
|
|
echo
|
|
echo "测试完成"
|
|
TESTEOF
|
|
|
|
chmod +x r2-backup.sh r2-manage.sh r2-test.sh
|
|
log "✓ 管理脚本创建完成"
|
|
}
|
|
|
|
# 显示完成信息
|
|
show_success() {
|
|
echo
|
|
echo -e "${GREEN}"
|
|
echo "========================================"
|
|
echo " 配置完成！"
|
|
echo "========================================"
|
|
echo -e "${NC}"
|
|
echo
|
|
info "配置信息:"
|
|
echo " - 远程名称: $REMOTE_NAME"
|
|
echo " - 存储桶: $BUCKET_NAME"
|
|
echo " - Endpoint: $ENDPOINT"
|
|
echo
|
|
info "可用命令:"
|
|
echo " rclone lsd $REMOTE_NAME: 列出存储桶"
|
|
echo " rclone ls $REMOTE_NAME:$BUCKET_NAME 列出文件"
|
|
echo " ./r2-backup.sh 自动备份"
|
|
echo " ./r2-manage.sh $BUCKET_NAME list 管理存储桶"
|
|
echo " ./r2-test.sh 快速测试"
|
|
echo
|
|
warn "安全提醒: 请妥善保管 API 密钥！"
|
|
echo
|
|
}
|
|
|
|
# 主函数
|
|
main() {
|
|
show_banner
|
|
check_system
|
|
install_rclone
|
|
get_user_input
|
|
if configure_r2 && test_configuration; then
|
|
create_management_scripts
|
|
show_success
|
|
else
|
|
error "配置失败，请检查输入信息"
|
|
exit 1
|
|
fi
|
|
}
|
|
|
|
# 运行主函数
|
|
main "$@"
|
|
EOF
|
|
|
|
# 给脚本执行权限并运行
|
|
chmod +x r2-auto-setup.sh && ./r2-auto-setup.sh
|