Update duokai.sh

duokai.sh

@@ -18,6 +18,9 @@ read -p "输入你的身份码: " id
 # 让用户输入想要创建的容器数量
 read -p "请输入你想要创建的节点数量，单IP限制最多5个节点: " container_count
 
+# 让用户输入想要分配的空间大小
+read -p "请输入你想要分配的存储空间大小（GB）: " storage_gb
+
 apt update
 
 # 检查 Docker 是否已安装
@@ -33,7 +36,7 @@ else
 fi
 
 # 拉取Docker镜像
-docker pull nezha123/titan-edge:1.1
+docker pull nezha123/titan-edge
 
 
 # 创建用户指定数量的容器
@@ -42,9 +45,13 @@ do
     # 为每个容器创建一个存储卷
     storage="titan_storage_$i"
     mkdir -p "$storage"
+    # 检查 ~/.titanedge 目录是否存在，如果不存在，则创建它
+    if [ ! -d "$HOME/.titanedge" ]; then
+        mkdir "$HOME/.titanedge"
+    fi
 
     # 运行容器，并设置重启策略为always
-    container_id=$(docker run -d --restart always -v "$PWD/$storage:/root/.titanedge/storage" --name "titan$i" nezha123/titan-edge:1.1)
+    container_id=$(docker run -d --restart always -v "$PWD/$storage:/root/.titanedge/storage" --name "titan$i" nezha123/titan-edge)
 
     echo "节点 titan$i 已经启动 容器ID $container_id"
 
@@ -55,4 +62,19 @@ do
         titan-edge bind --hash=$id https://api-test1.container1.titannet.io/api/v2/device/binding"
 done
 
+# 等待足够时间以确保所有容器都已启动并且config.toml文件已经生成
+echo "等待所有容器启动并生成配置文件..."
+sleep 60
+
+# 修改宿主机上的config.toml文件以设置StorageGB值
+config_path="$HOME/.titanedge/config.toml"
+if [ -f "$config_path" ]; then
+    # 修改文件，设置StorageGB
+    sed -i "s/#StorageGB = 64/StorageGB = $storage_gb/g" "$config_path"
+    echo "已将存储空间设置为 $storage_gb GB"
+else
+    echo "配置文件未找到，可能需要手动设置StorageGB。请检查容器是否正确启动并生成了配置文件。"
+fi
+
+
 echo "==============================所有节点均已设置并启动===================================."