-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup_backup.sh
189 lines (170 loc) · 7.82 KB
/
setup_backup.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
#!/bin/bash
echo -e "\033[1;94m远程备份脚本\033[0m"
echo -e "\033[1;94m项目地址：（https://github.com/small-haozi/backup-Remote）"
echo -e " "
echo "======================================================"
echo -e " "
# 创建存储配置和脚本的目录
BACKUP_DIR="/etc/back-Remote"
if [ ! -d "$BACKUP_DIR" ]; then
sudo mkdir -p $BACKUP_DIR
echo "已创建目录: $BACKUP_DIR"
fi
echo -e "\033[1;94m请输入服务器信息:\033[0m"
echo -e " "
read -p "远程服务器IP: " REMOTE_HOST
read -p "远程服务器用户名 [默认: root]: " REMOTE_USER
REMOTE_USER=${REMOTE_USER:-root}
read -p "远程服务器密码: " REMOTE_PASS
read -p "远程服务器端口 [默认: 22]: " REMOTE_PORT
REMOTE_PORT=${REMOTE_PORT:-22}
read -p "远程目录: " REMOTE_DIR
read -p "本地目录 [默认: /root/backup-$REMOTE_HOST]: " LOCAL_DIR
LOCAL_DIR=${LOCAL_DIR:-/root/backup-$REMOTE_HOST}
echo -e "\033[1;94m是否启用Telegram消息推送功能? (y/n):\033[0m"
read -p "选择: " ENABLE_TELEGRAM
if [ "$ENABLE_TELEGRAM" = "y" ]; then
echo -e "\033[1;94m请输入Telegram消息推送的配置信息:\033[0m"
read -p "Telegram bot token: " TELEGRAM_BOT_TOKEN
read -p "Telegram chat ID: " TELEGRAM_CHAT_ID
fi
echo -e " "
echo "======================================================"
echo -e "当前设置信息为："
echo -e "远程服务器IP: $REMOTE_HOST"
echo -e "远程服务器用户名: $REMOTE_USER"
echo -e "远程服务器密码: $REMOTE_PASS"
echo -e "远程服务器端口: $REMOTE_PORT"
echo -e "远程目录: $REMOTE_DIR"
echo -e "本地目录: $LOCAL_DIR"
if [ "$ENABLE_TELEGRAM" = "y" ]; then
echo -e "Telegram bot token: $TELEGRAM_BOT_TOKEN"
echo -e "Telegram chat ID: $TELEGRAM_CHAT_ID"
fi
echo -e "请仔细确认是否输入错误！！如需修改，请自行修改 $BACKUP_DIR/backup_config.conf 文件"
echo "======================================================"
echo -e " "
echo -e "\033[1;94m备份文件数量设置:\033[0m"
read -p "是否设置最大备份文件数量? (y/n): " SET_MAX_BACKUPS
if [ "$SET_MAX_BACKUPS" = "y" ]; then
read -p "最大备份文件数量: " MAX_BACKUPS
fi
# 删除旧的配置文件和脚本（如果存在）
sudo rm -f $BACKUP_DIR/backup_config.conf
sudo rm -f $BACKUP_DIR/back.sh
# 保存到配置文件
CONFIG_FILE="$BACKUP_DIR/backup_config.conf"
echo "# 配置文件" > $CONFIG_FILE
echo "REMOTE_HOST=$REMOTE_HOST # 远程服务器IP" >> $CONFIG_FILE
echo "REMOTE_USER=$REMOTE_USER # 远程服务器用户名，默认为root" >> $CONFIG_FILE
echo "REMOTE_PASS=$REMOTE_PASS # 远程服务器密码" >> $CONFIG_FILE
echo "REMOTE_PORT=$REMOTE_PORT # 远程服务器端口，默认为22" >> $CONFIG_FILE
echo "REMOTE_DIR=$REMOTE_DIR # 远程目录" >> $CONFIG_FILE
echo "LOCAL_DIR=$LOCAL_DIR # 本地目录，默认为/root/backup-\$REMOTE_HOST" >> $CONFIG_FILE
if [ "$SET_MAX_BACKUPS" = "y" ]; then
echo "MAX_BACKUPS=$MAX_BACKUPS # 最大备份文件数量" >> $CONFIG_FILE
fi
if [ "$ENABLE_TELEGRAM" = "y" ]; then
echo "TELEGRAM_BOT_TOKEN=$TELEGRAM_BOT_TOKEN # Telegram bot token" >> $CONFIG_FILE
echo "TELEGRAM_CHAT_ID=$TELEGRAM_CHAT_ID # Telegram chat ID" >> $CONFIG_FILE
fi
# 创建 back.sh 脚本
cat <<EOF > $BACKUP_DIR/back.sh
#!/bin/bash
# 加载配置文件
source $CONFIG_FILE
# 发送Telegram消息的函数
send_telegram_message() {
if [ "$ENABLE_TELEGRAM" = "y" ]; then
local message="$1"
local bot_token="$TELEGRAM_BOT_TOKEN" # 使用用户输入的Telegram bot token
local chat_id="$TELEGRAM_CHAT_ID" # 使用用户输入的chat ID
# 处理文件名，只显示开头三位和末尾三位
local filename="$2"
local filename_length=${#filename}
if [ $filename_length -gt 6 ]; then
local start=${filename:0:3}
local end=${filename: -3}
filename="$start...$end"
fi
message=$(echo "$message" | sed "s/{filename}/$filename/g")
curl -s -X POST "https://api.telegram.org/bot$bot_token/sendMessage" -d chat_id="$chat_id" -d text="$message"
fi
}
# 检查并安装 sshpass 和 rsync
if ! command -v sshpass &> /dev/null; then
echo -e "\033[0;33msshpass 未安装，正在安装...\033[0m"
sudo apt-get install -y sshpass
fi
if ! command -v rsync &> /dev/null; then
echo -e "\033[0;33mrsync 未安装，正在安装...\033[0m"
sudo apt-get install -y rsync
fi
# 创建本地备份目录（如果不存在）
sudo mkdir -p \$LOCAL_DIR
# 检查远程目录是否存在
if sshpass -p "\$REMOTE_PASS" ssh -o StrictHostKeyChecking=no -p \$REMOTE_PORT \$REMOTE_USER@\$REMOTE_HOST "[ -d \$REMOTE_DIR ]"; then
echo -e " "
echo "正在检查远程服务器目录: \$REMOTE_DIR"
echo -e " "
# 找到最新的文件
LATEST_FILE=\$(sshpass -p "\$REMOTE_PASS" ssh -o StrictHostKeyChecking=no -p \$REMOTE_PORT \$REMOTE_USER@\$REMOTE_HOST "find \$REMOTE_DIR -type f -printf '%T@ %p\n' | sort -nr | head -n 1 | cut -d' ' -f2")
LOCAL_LATEST_FILE=\$(find \$LOCAL_DIR -type f -printf '%T@ %p\n' | sort -nr | head -n 1 | cut -d' ' -f2)
REMOTE_LATEST_FILENAME=\$(basename "\$LATEST_FILE")
LOCAL_LATEST_FILENAME=\$(basename "\$LOCAL_LATEST_FILE")
echo "======================================================"
echo -e " "
echo "远程最新文件为: \$REMOTE_LATEST_FILENAME"
echo -e " "
echo "本地最新文件为: \$LOCAL_LATEST_FILENAME"
echo -e " "
echo "======================================================"
echo -e " "
# 比较文件名
if [ "\$REMOTE_LATEST_FILENAME" != "\$LOCAL_LATEST_FILENAME" ]; then
echo "发现新文件，正在同步..."
echo -e "======================================================"
# 使用 rsync 同步最新的文件到本地
sshpass -p "\$REMOTE_PASS" rsync -avz -e "ssh -o StrictHostKeyChecking=no -p \$REMOTE_PORT" --progress "\$REMOTE_USER@\$REMOTE_HOST:\$LATEST_FILE" "\$LOCAL_DIR"
[ "$ENABLE_TELEGRAM" = "y" ] && send_telegram_message "备份完成。最新文件：{filename}" "\$REMOTE_LATEST_FILENAME"
echo "备份完成。"
echo -e " "
echo "======================================================"
else
echo "最新文件已存在，无需备份。"
[ "$ENABLE_TELEGRAM" = "y" ] && send_telegram_message "最新文件已存在，无需备份。文件名：{filename}" "\$REMOTE_LATEST_FILENAME"
echo -e " "
echo "======================================================"
fi
else
echo "======================================================"
echo -e " "
echo "远程目录不存在。请检查 /etc/backup-Remote/backup_config.conf 文件中是否设置错误"
[ "$ENABLE_TELEGRAM" = "y" ] && send_telegram_message "远程目录不存在。请检查配置。"
echo -e " "
echo "======================================================"
fi
# 管理本地备份文件数量
if [ ! -z "\$MAX_BACKUPS" ]; then
cd \$LOCAL_DIR
if [ \$(ls -1 | wc -l) -gt \$MAX_BACKUPS ]; then
ls -t | tail -n +\$(expr \$MAX_BACKUPS + 1) | xargs rm -f
[ "$ENABLE_TELEGRAM" = "y" ] && send_telegram_message "超出最大备份文件数量，已删除最旧的备份文件。"
echo "超出最大备份文件数量，已删除最旧的备份文件。"
fi
fi
EOF
# 给 back.sh 脚本执行权限
sudo chmod +x $BACKUP_DIR/back.sh
echo "======================================================"
echo -e " "
echo -e "\033[1;94m配置文件和备份脚本已创建并保存在\033[0m \033[1;92m $BACKUP_DIR。\033[0m"
echo -e " "
# 创建符号链接
sudo ln -sf $BACKUP_DIR/back.sh /usr/local/bin/back.sh
echo -e "\033[1;94m已创建符号链接，可以从任何位置运行\033[0m \033[1;92m back.sh。\033[0m"
echo -e " "
echo "======================================================"
echo -e " "
# 安排在脚本执行完毕后删除自身
nohup bash -c "sleep 2; rm -- \"$0\"" &>/dev/null &