Update Dockerfile

Dockerfile

@@ -1,37 +1,61 @@
-# 使用原版镜像
 FROM ghcr.io/krau/saveany-bot:latest
 
-# 切换 root 用户进行安装和权限设置
+# 切换 root 用户安装依赖
 USER root
 
-# 安装 Python3 (用于运行保活的 Web 服务)
-# 尝试兼容 Alpine 和 Debian 两种系统的安装方式
+# 安装 Python3 (用于运行保活服务和启动脚本)
 RUN (apk add --no-cache python3 || (apt-get update && apt-get install -y python3))
 
-# 设置工作目录
 WORKDIR /app
 
-# 创建下载缓存目录并赋予 777 权限
-# 虽然你用了 WebDAV，但 bot 还是需要一个临时目录先下载再上传
+# 创建下载目录并赋予 777 权限
 RUN mkdir -p /app/downloads && chmod 777 /app/downloads
 
-# 创建启动脚本
-# 1. 把环境变量 CONFIG_CONTENT 写入 config.toml
-# 2. 创建假的 index.html
-# 3. 启动 Python Web Server (后台运行)
-# 4. 启动 saveany-bot (前台运行)
-RUN echo '#!/bin/sh' > /app/start.sh && \
-    echo 'echo "Generating config file..."' >> /app/start.sh && \
-    echo 'echo "$CONFIG_CONTENT" > /app/config.toml' >> /app/start.sh && \
-    echo 'echo "Saveany Bot is running..." > /app/index.html' >> /app/start.sh && \
-    echo 'echo "Starting Web Server for Keep-Alive..."' >> /app/start.sh && \
-    echo 'python3 -m http.server 7860 > /dev/null 2>&1 &' >> /app/start.sh && \
-    echo 'echo "Starting Bot..."' >> /app/start.sh && \
-    echo './saveany-bot' >> /app/start.sh && \
-    chmod +x /app/start.sh
+# 创建一个 Python 启动脚本 (main.py)
+# 这个脚本会做三件事：
+# 1. 从环境变量读取配置写入文件
+# 2. 启动 Web 服务器 (监听 7860)
+# 3. 启动 Bot (saveany-bot)
+RUN echo 'import os' > /app/main.py && \
+    echo 'import threading' >> /app/main.py && \
+    echo 'import http.server' >> /app/main.py && \
+    echo 'import socketserver' >> /app/main.py && \
+    echo 'import subprocess' >> /app/main.py && \
+    echo 'import time' >> /app/main.py && \
+    echo '' >> /app/main.py && \
+    echo '# 1. 生成配置文件' >> /app/main.py && \
+    echo 'config_content = os.environ.get("CONFIG_CONTENT", "")' >> /app/main.py && \
+    echo 'if config_content:' >> /app/main.py && \
+    echo '    print("Writing config.toml...")' >> /app/main.py && \
+    echo '    with open("/app/config.toml", "w") as f:' >> /app/main.py && \
+    echo '        f.write(config_content)' >> /app/main.py && \
+    echo 'else:' >> /app/main.py && \
+    echo '    print("WARNING: No CONFIG_CONTENT found!")' >> /app/main.py && \
+    echo '' >> /app/main.py && \
+    echo '# 2. 启动 Web Server (保活核心)' >> /app/main.py && \
+    echo 'def start_web():' >> /app/main.py && \
+    echo '    # 创建首页' >> /app/main.py && \
+    echo '    with open("index.html", "w") as f: f.write("Bot is running.")' >> /app/main.py && \
+    echo '    # 监听所有接口 0.0.0.0' >> /app/main.py && \
+    echo '    Handler = http.server.SimpleHTTPRequestHandler' >> /app/main.py && \
+    echo '    with socketserver.TCPServer(("0.0.0.0", 7860), Handler) as httpd:' >> /app/main.py && \
+    echo '        print("Web server started on port 7860")' >> /app/main.py && \
+    echo '        httpd.serve_forever()' >> /app/main.py && \
+    echo '' >> /app/main.py && \
+    echo '# 在后台线程启动 Web Server' >> /app/main.py && \
+    echo 't = threading.Thread(target=start_web)' >> /app/main.py && \
+    echo 't.daemon = True' >> /app/main.py && \
+    echo 't.start()' >> /app/main.py && \
+    echo '' >> /app/main.py && \
+    echo '# 3. 启动 Bot' >> /app/main.py && \
+    echo 'print("Starting Saveany Bot...")' >> /app/main.py && \
+    echo '# 给予 Web Server 一点启动时间' >> /app/main.py && \
+    echo 'time.sleep(2)' >> /app/main.py && \
+    echo '# 启动主程序 (假设二进制文件在当前目录)' >> /app/main.py && \
+    echo 'subprocess.run(["./saveany-bot"])' >> /app/main.py
 
 # 暴露端口
 EXPOSE 7860
 
-# 启动
-CMD ["/app/start.sh"]
+# 使用 Python 运行启动脚本
+CMD ["python3", "/app/main.py"]