aBER0724 commited on
Commit
10a7680
·
verified ·
1 Parent(s): 436a37b

Update sync_data.sh

Browse files
Files changed (1) hide show
  1. sync_data.sh +61 -35
sync_data.sh CHANGED
@@ -6,6 +6,10 @@ if [ -z "$G_NAME" ] || [ -z "$G_TOKEN" ]; then
6
  exit 1
7
  fi
8
 
 
 
 
 
9
  # 从 WebDAV 恢复配置的函数
10
  restore_from_webdav() {
11
  if [ -z "$WEBDAV_URL" ] || [ -z "$WEBDAV_USERNAME" ] || [ -z "$WEBDAV_PASSWORD" ]; then
@@ -20,7 +24,7 @@ restore_from_webdav() {
20
  YESTERDAY_FILENAME="mcp_backup_$(date -d '1 day ago' +'%m_%d').tar.gz"
21
 
22
  # 创建临时恢复目录
23
- RESTORE_DIR="/mcp-proxy-server/config/webdav_restore"
24
  mkdir -p "$RESTORE_DIR"
25
 
26
  # 尝试下载今天的备份,如果失败则尝试昨天的
@@ -49,25 +53,29 @@ restore_from_webdav() {
49
  echo "解压缩备份文件成功"
50
 
51
  # 恢复配置文件
52
- if [ -d "config" ]; then
53
  echo "恢复 config 目录..."
54
  mkdir -p /mcp-proxy-server/config
55
  cp -r config/* /mcp-proxy-server/config/
56
  echo "config 目录恢复成功"
 
 
57
  fi
58
 
59
- if [ -d "tools" ]; then
60
  echo "恢复 tools 目录..."
61
  mkdir -p /tools
62
  cp -r tools/* /tools/
63
  echo "tools 目录恢复成功"
 
 
64
  fi
65
 
66
  echo "从 WebDAV 恢复配置文件完成"
67
  } || {
68
  echo "解压缩备份文件失败"
69
  }
70
- cd ../..
71
  fi
72
 
73
  # 清理临时目录
@@ -80,11 +88,11 @@ IFS='/' read -r GITHUB_USER GITHUB_REPO <<< "$G_NAME"
80
 
81
  # 构建 GitHub 仓库的克隆 URL,包含令牌
82
  REPO_URL="https://${G_TOKEN}@github.com/${G_NAME}.git"
83
- mkdir -p /mcp-proxy-server/config/github_data
84
 
85
  # 克隆仓库
86
- echo "正在克隆仓库……"
87
- git clone "$REPO_URL" /mcp-proxy-server/config/github_data || {
88
  echo "克隆失败,请检查 G_NAME 和 G_TOKEN 是否正确。"
89
  exit 1
90
  }
@@ -114,17 +122,28 @@ if [ "$RESTORE_SOURCE" = "webdav" ]; then
114
  restore_from_webdav
115
  elif [ "$RESTORE_SOURCE" = "github" ] || [ "$RESTORE_SOURCE" = "auto" ]; then
116
  # 检查并恢复配置文件(从 GitHub)
117
- if [ -d /mcp-proxy-server/config/github_data/config ] && [ -d /mcp-proxy-server/config/github_data/tools ]; then
118
  echo "从 GitHub 仓库中恢复配置文件..."
119
  # 创建目标目录
120
  mkdir -p /mcp-proxy-server/config
121
  mkdir -p /tools
122
 
123
- # 复制配置文件
124
- cp -r /mcp-proxy-server/config/github_data/config/* /mcp-proxy-server/config/ 2>/dev/null || echo "config 目录为空或复制失败"
125
- cp -r /mcp-proxy-server/config/github_data/tools/* /tools/ 2>/dev/null || echo "tools 目录为空或复制失败"
 
 
 
 
 
 
 
 
 
 
 
126
 
127
- echo "从 GitHub 仓库中拉取配置文件成功"
128
  else
129
  echo "GitHub 仓库中未找到 config 或 tools 目录"
130
  if [ "$RESTORE_SOURCE" = "auto" ]; then
@@ -145,7 +164,7 @@ sync_data() {
145
  # 1. 同步到 GitHub
146
  echo "正在开始同步"
147
  # 进入仓库目录
148
- cd /mcp-proxy-server/config/github_data
149
  # 配置 Git 用户信息
150
  git config user.name "AutoSync Bot"
151
  git config user.email "[email protected]"
@@ -160,17 +179,18 @@ sync_data() {
160
  mkdir -p ./config
161
  mkdir -p ./tools
162
 
163
- # 复制配置文件(如果存在)
164
- if [ -d "/mcp-proxy-server/config" ]; then
165
- cp -r /mcp-proxy-server/config/* ./config/ 2>/dev/null || echo "config 目录为空"
 
166
  else
167
- echo "/mcp-proxy-server/config 目录不存在"
168
  fi
169
 
170
- if [ -d "/tools" ]; then
171
- cp -r /tools/* ./tools/ 2>/dev/null || echo "tools 目录为空"
172
  else
173
- echo "/tools 目录不存在"
174
  fi
175
 
176
  # 检查是否有变化
@@ -192,8 +212,7 @@ sync_data() {
192
  }
193
  }
194
  # 返回上级目录
195
- cd ..
196
- cd ..
197
 
198
  # 2. 同步到 WebDAV
199
  if [ -z "$WEBDAV_URL" ] || [ -z "$WEBDAV_USERNAME" ] || [ -z "$WEBDAV_PASSWORD" ]; then
@@ -202,37 +221,45 @@ sync_data() {
202
  echo "同步到 WebDAV..."
203
  FILENAME="mcp_backup_$(date +'%m_%d').tar.gz"
204
 
205
- # 创建临时备份压缩包
206
- TEMP_DIR="/mcp-proxy-server/config/temp_backup"
 
207
  mkdir -p "$TEMP_DIR"
208
 
209
  # 复制要备份的目录到临时目录
210
- if [ -d "/mcp-proxy-server/config" ]; then
211
- cp -r /mcp-proxy-server/config "$TEMP_DIR/"
 
 
 
 
212
  fi
213
- if [ -d "/tools" ]; then
214
- cp -r /tools "$TEMP_DIR/"
 
 
 
215
  fi
216
 
217
  # 创建压缩包
218
- if [ "$(ls -A $TEMP_DIR)" ]; then
219
  cd "$TEMP_DIR"
220
  tar -czf "../$FILENAME" .
221
- cd ../../..
222
 
223
  # 使用 curl 进行文件上传
224
- curl -T "/mcp-proxy-server/config/$FILENAME" --user "$WEBDAV_USERNAME:$WEBDAV_PASSWORD" "$WEBDAV_URL/$FILENAME" && {
225
  echo "WebDAV 上传成功"
226
  } || {
227
  echo "WebDAV 上传失败,等待重试..."
228
  sleep 10
229
- curl -T "/mcp-proxy-server/config/$FILENAME" --user "$WEBDAV_USERNAME:$WEBDAV_PASSWORD" "$WEBDAV_URL/$FILENAME" || {
230
  echo "重试失败,放弃webdav上传。"
231
  }
232
  }
233
 
234
  # 清理临时文件
235
- rm -f "/mcp-proxy-server/config/$FILENAME"
236
  else
237
  echo "没有找到要备份的文件,跳过 WebDAV 同步"
238
  fi
@@ -243,8 +270,7 @@ sync_data() {
243
 
244
  else
245
  # 返回上级目录
246
- cd ..
247
- cd ..
248
  echo "GitHub: 没有检测到配置文件变化"
249
  fi
250
  # 3. 等待统一的时间间隔
 
6
  exit 1
7
  fi
8
 
9
+ # 使用独立的工作目录,避免与配置目录冲突
10
+ WORK_DIR="/tmp/mcp_sync_work"
11
+ GITHUB_DATA_DIR="$WORK_DIR/github_data"
12
+
13
  # 从 WebDAV 恢复配置的函数
14
  restore_from_webdav() {
15
  if [ -z "$WEBDAV_URL" ] || [ -z "$WEBDAV_USERNAME" ] || [ -z "$WEBDAV_PASSWORD" ]; then
 
24
  YESTERDAY_FILENAME="mcp_backup_$(date -d '1 day ago' +'%m_%d').tar.gz"
25
 
26
  # 创建临时恢复目录
27
+ RESTORE_DIR="$WORK_DIR/webdav_restore"
28
  mkdir -p "$RESTORE_DIR"
29
 
30
  # 尝试下载今天的备份,如果失败则尝试昨天的
 
53
  echo "解压缩备份文件成功"
54
 
55
  # 恢复配置文件
56
+ if [ -d "config" ] && [ "$(ls -A config 2>/dev/null)" ]; then
57
  echo "恢复 config 目录..."
58
  mkdir -p /mcp-proxy-server/config
59
  cp -r config/* /mcp-proxy-server/config/
60
  echo "config 目录恢复成功"
61
+ else
62
+ echo "备份中没有 config 目录或目录为空"
63
  fi
64
 
65
+ if [ -d "tools" ] && [ "$(ls -A tools 2>/dev/null)" ]; then
66
  echo "恢复 tools 目录..."
67
  mkdir -p /tools
68
  cp -r tools/* /tools/
69
  echo "tools 目录恢复成功"
70
+ else
71
+ echo "备份中没有 tools 目录或目录为空"
72
  fi
73
 
74
  echo "从 WebDAV 恢复配置文件完成"
75
  } || {
76
  echo "解压缩备份文件失败"
77
  }
78
+ cd - > /dev/null
79
  fi
80
 
81
  # 清理临时目录
 
88
 
89
  # 构建 GitHub 仓库的克隆 URL,包含令牌
90
  REPO_URL="https://${G_TOKEN}@github.com/${G_NAME}.git"
91
+ mkdir -p "$GITHUB_DATA_DIR"
92
 
93
  # 克隆仓库
94
+ echo "正在克隆仓库到 $GITHUB_DATA_DIR ……"
95
+ git clone "$REPO_URL" "$GITHUB_DATA_DIR" || {
96
  echo "克隆失败,请检查 G_NAME 和 G_TOKEN 是否正确。"
97
  exit 1
98
  }
 
122
  restore_from_webdav
123
  elif [ "$RESTORE_SOURCE" = "github" ] || [ "$RESTORE_SOURCE" = "auto" ]; then
124
  # 检查并恢复配置文件(从 GitHub)
125
+ if [ -d "$GITHUB_DATA_DIR/config" ] && [ -d "$GITHUB_DATA_DIR/tools" ]; then
126
  echo "从 GitHub 仓库中恢复配置文件..."
127
  # 创建目标目录
128
  mkdir -p /mcp-proxy-server/config
129
  mkdir -p /tools
130
 
131
+ # 复制配置文件(检查是否为空)
132
+ if [ "$(ls -A $GITHUB_DATA_DIR/config 2>/dev/null)" ]; then
133
+ cp -r "$GITHUB_DATA_DIR/config"/* /mcp-proxy-server/config/
134
+ echo "config 目录恢复成功"
135
+ else
136
+ echo "GitHub 仓库中的 config 目录为空"
137
+ fi
138
+
139
+ if [ "$(ls -A $GITHUB_DATA_DIR/tools 2>/dev/null)" ]; then
140
+ cp -r "$GITHUB_DATA_DIR/tools"/* /tools/
141
+ echo "tools 目录恢复成功"
142
+ else
143
+ echo "GitHub 仓库中的 tools 目录为空"
144
+ fi
145
 
146
+ echo "从 GitHub 仓库中拉取配置文件完成"
147
  else
148
  echo "GitHub 仓库中未找到 config 或 tools 目录"
149
  if [ "$RESTORE_SOURCE" = "auto" ]; then
 
164
  # 1. 同步到 GitHub
165
  echo "正在开始同步"
166
  # 进入仓库目录
167
+ cd "$GITHUB_DATA_DIR"
168
  # 配置 Git 用户信息
169
  git config user.name "AutoSync Bot"
170
  git config user.email "[email protected]"
 
179
  mkdir -p ./config
180
  mkdir -p ./tools
181
 
182
+ # 复制配置文件(如果存在且非空)
183
+ if [ -d "/mcp-proxy-server/config" ] && [ "$(ls -A /mcp-proxy-server/config 2>/dev/null | grep -v github_data)" ]; then
184
+ # 排除 github_data 目录,避免嵌套
185
+ rsync -av --exclude='github_data' --exclude='temp_backup' --exclude='webdav_restore' /mcp-proxy-server/config/ ./config/ 2>/dev/null && echo "config 目录同步成功" || echo "config 目录同步失败"
186
  else
187
+ echo "/mcp-proxy-server/config 目录不存在或为空"
188
  fi
189
 
190
+ if [ -d "/tools" ] && [ "$(ls -A /tools 2>/dev/null)" ]; then
191
+ cp -r /tools/* ./tools/ 2>/dev/null && echo "tools 目录同步成功" || echo "tools 目录同步失败"
192
  else
193
+ echo "/tools 目录不存在或为空"
194
  fi
195
 
196
  # 检查是否有变化
 
212
  }
213
  }
214
  # 返回上级目录
215
+ cd - > /dev/null
 
216
 
217
  # 2. 同步到 WebDAV
218
  if [ -z "$WEBDAV_URL" ] || [ -z "$WEBDAV_USERNAME" ] || [ -z "$WEBDAV_PASSWORD" ]; then
 
221
  echo "同步到 WebDAV..."
222
  FILENAME="mcp_backup_$(date +'%m_%d').tar.gz"
223
 
224
+ # 创建临时备份压缩包(使用独立目录)
225
+ TEMP_DIR="$WORK_DIR/temp_backup"
226
+ rm -rf "$TEMP_DIR" # 清理之前的临时目录
227
  mkdir -p "$TEMP_DIR"
228
 
229
  # 复制要备份的目录到临时目录
230
+ BACKUP_CREATED=0
231
+
232
+ if [ -d "/mcp-proxy-server/config" ] && [ "$(ls -A /mcp-proxy-server/config 2>/dev/null | grep -v github_data | grep -v temp_backup | grep -v webdav_restore)" ]; then
233
+ echo "备份 config 目录..."
234
+ mkdir -p "$TEMP_DIR/config"
235
+ rsync -av --exclude='github_data' --exclude='temp_backup' --exclude='webdav_restore' /mcp-proxy-server/config/ "$TEMP_DIR/config/" 2>/dev/null && BACKUP_CREATED=1
236
  fi
237
+
238
+ if [ -d "/tools" ] && [ "$(ls -A /tools 2>/dev/null)" ]; then
239
+ echo "备份 tools 目录..."
240
+ mkdir -p "$TEMP_DIR/tools"
241
+ cp -r /tools/* "$TEMP_DIR/tools/" 2>/dev/null && BACKUP_CREATED=1
242
  fi
243
 
244
  # 创建压缩包
245
+ if [ "$BACKUP_CREATED" = "1" ] && [ "$(ls -A $TEMP_DIR)" ]; then
246
  cd "$TEMP_DIR"
247
  tar -czf "../$FILENAME" .
248
+ cd - > /dev/null
249
 
250
  # 使用 curl 进行文件上传
251
+ curl -T "$WORK_DIR/$FILENAME" --user "$WEBDAV_USERNAME:$WEBDAV_PASSWORD" "$WEBDAV_URL/$FILENAME" && {
252
  echo "WebDAV 上传成功"
253
  } || {
254
  echo "WebDAV 上传失败,等待重试..."
255
  sleep 10
256
+ curl -T "$WORK_DIR/$FILENAME" --user "$WEBDAV_USERNAME:$WEBDAV_PASSWORD" "$WEBDAV_URL/$FILENAME" || {
257
  echo "重试失败,放弃webdav上传。"
258
  }
259
  }
260
 
261
  # 清理临时文件
262
+ rm -f "$WORK_DIR/$FILENAME"
263
  else
264
  echo "没有找到要备份的文件,跳过 WebDAV 同步"
265
  fi
 
270
 
271
  else
272
  # 返回上级目录
273
+ cd - > /dev/null
 
274
  echo "GitHub: 没有检测到配置文件变化"
275
  fi
276
  # 3. 等待统一的时间间隔