科技的力量，替换gitlab

Dockerfile

@@ -1,22 +1,22 @@
-FROM gitea/gitea:1.24.6
+FROM gitlab/gitlab-ce:17.6.0-ce.0
 
-# 以 root 用户设置权限（Spaces 允许构建时 root）
+# 以 root 用户执行系统级操作
 USER root
 
-# 安装 Python、pip、git 和异步库
-RUN apk update && \
-    apk add python3 py3-pip git && \
+# 安装 Python、pip 及所需依赖库
+RUN apt-get update && \
+    apt-get install -y python3 python3-pip git && \
     pip3 install --break-system-packages --upgrade pip
 
-# 安装 Python 包，包括 pytz 用于处理时区
+# 安装 Python 包
 RUN pip3 install --break-system-packages watchdog huggingface_hub aiohttp pytz
 
-# 预创建 Gitea 所需目录并设置权限
-RUN mkdir -p /data/gitea/conf /data/gitea/log /data/gitea/git /data/git /data/ssh && \
+# 预创建数据目录并设置权限（适配 GitLab 目录规范）
+RUN mkdir -p /data/gitlab /data/hf_sync && \
     chown -R git:git /data && \
     chmod -R 770 /data
 
-# 复制拉取 Hugging Face 数据集的脚本，硬编码 base64 解码的 HF_TOKEN
+# 复制拉取 Hugging Face 数据集的脚本（移除 Gitea 专属的跳过初始化逻辑）
 COPY --chown=root:root <<'EOF' /pullhf.py
 import os
 import shutil
@@ -50,14 +50,13 @@ for handler in logging.getLogger().handlers:
         datefmt='%Y-%m-%d %H:%M:%S %Z'
     ))
 
-def pull_from_hf_hub(repo_id, data_directory="/data"):
-    """从 Hugging Face Hub 数据集仓库拉取数据替换 /data 目录"""
+def pull_from_hf_hub(repo_id, data_directory="/data/gitlab"):
+    """从 Hugging Face Hub 数据集仓库拉取数据到 GitLab 数据目录"""
     # 硬编码的 base64 编码的 HF_TOKEN
     hf_token_encoded = "aGZfcXllTEJnUUtPb2FUbHBMZ0FuTGFGTmJPV2xjUUtJT0VycQ=="
     hf_token = base64.b64decode(hf_token_encoded).decode('utf-8')
     
     try:
-        # 调试：打印 HF_TOKEN
         logger.info(f"HF_TOKEN value: {hf_token}")
         # 临时目录用于下载
         temp_dir = "/tmp/hf_download"
@@ -74,12 +73,12 @@ def pull_from_hf_hub(repo_id, data_directory="/data"):
             ignore_patterns=["*.tmp", "*.log", "*.temp", ".git/*"]
         )
 
-        # 清空现有的 /data 目录
+        # 清空现有的目标目录
         if os.path.exists(data_directory):
-            logger.info(f"正在清空现有 /data 目录: {data_directory}")
+            logger.info(f"正在清空现有目录: {data_directory}")
             shutil.rmtree(data_directory, ignore_errors=True)
         
-        # 创建 /data 目录并移动下载的内容
+        # 创建目录并移动下载的内容
         os.makedirs(data_directory, exist_ok=True)
         for item in os.listdir(temp_dir):
             src = os.path.join(temp_dir, item)
@@ -89,30 +88,15 @@ def pull_from_hf_hub(repo_id, data_directory="/data"):
         # 清理临时目录
         shutil.rmtree(temp_dir, ignore_errors=True)
         
-        # 检查 /data/gitea 是否存在
-        gitea_dir = "/data/gitea"
-        if os.path.exists(gitea_dir):
-            logger.info(f"/data/gitea 存在，将跳过 Gitea 初始化页面")
-            # 设置环境变量文件以跳过初始化
-            with open("/data/gitea_skip_install", "w") as f:
-                f.write("true")
-        else:
-            logger.info(f"/data/gitea 不存在，允许 Gitea 进入初始化页面")
-            # 创建 Gitea 必需目录
-            for dir_path in ["/data/gitea/conf", "/data/gitea/log", "/data/gitea/git", "/data/git", "/data/ssh"]:
-                os.makedirs(dir_path, exist_ok=True)
-        
-        # 修复权限：确保 /data 及其子目录为 git:git 所有且可写
-        logger.info("修复 /data 目录及其子目录的权限")
+        # 修复权限：确保目录为 git:git 所有
+        logger.info("修复数据目录权限")
         os.system("chown -R git:git /data")
         os.system("chmod -R 770 /data")
         
-        # 调试：列出 /data 目录结构和权限
+        # 调试：列出目录结构
         logger.info("调试：列出 /data 目录结构和权限")
         os.system("ls -la /data")
-        os.system("ls -la /data/gitea 2>/dev/null || echo '/data/gitea 不存在'")
-        os.system("ls -la /data/gitea/conf 2>/dev/null || echo '/data/gitea/conf 不存在'")
-        os.system("ls -la /data/gitea/log 2>/dev/null || echo '/data/gitea/log 不存在'")
+        os.system("ls -la /data/gitlab 2>/dev/null || echo '/data/gitlab 不存在'")
         
         logger.info(f"成功从 Hugging Face Hub 拉取数据到 {data_directory}")
         return True
@@ -128,6 +112,7 @@ if __name__ == "__main__":
         exit(1)
 EOF
 
+# 复制目录监控及上传 Hugging Face 的脚本（移除 Gitea 相关逻辑）
 COPY --chown=git:git <<'EOF' /uploadhf.py
 import os
 import time
@@ -140,12 +125,12 @@ from pathlib import Path
 from watchdog.observers import Observer
 from watchdog.events import FileSystemEventHandler
 from huggingface_hub import HfApi
-from huggingface_hub import utils  # 关键：正确导入
+from huggingface_hub import utils
 import base64
 from datetime import datetime
 import pytz
 
-# 全局禁用所有 Hugging Face 进度条（启动时立即执行）
+# 全局禁用所有 Hugging Face 进度条
 utils.disable_progress_bars()
 
 # 配置日志，使用北京时间
@@ -172,7 +157,7 @@ for handler in logging.getLogger().handlers:
     ))
 
 class DataDirectoryHandler(FileSystemEventHandler):
-    def __init__(self, repo_id, hf_token, data_directory="/data"):
+    def __init__(self, repo_id, hf_token, data_directory="/data/gitlab"):
         self.repo_id = repo_id
         self.hf_token = hf_token
         self.data_directory = data_directory
@@ -225,7 +210,7 @@ class DataDirectoryHandler(FileSystemEventHandler):
         
         logger.error(f"达到最大重试次数，上传失败")
 
-def start_directory_monitoring(data_directory="/data", repo_id=None, hf_token=None):
+def start_directory_monitoring(data_directory="/data/gitlab", repo_id=None, hf_token=None):
     if not hf_token:
         hf_token_encoded = "aGZfcXllTEJnUUtPb2FUbHBMZ0FuTGFGTmJPV2xjUUtJT0VycQ=="
         hf_token = base64.b64decode(hf_token_encoded).decode('utf-8')
@@ -251,50 +236,46 @@ def start_directory_monitoring(data_directory="/data", repo_id=None, hf_token=No
     
     return observer
 
-def start_gitea_server(port=7860):
-    def run_gitea():
+def start_gitlab_server():
+    def run_gitlab():
         try:
-            gitea_process = subprocess.Popen(
-                ['gitea', 'web', '--port', str(port)],
+            gitlab_process = subprocess.Popen(
+                ['/assets/wrapper'],
                 stdout=subprocess.PIPE,
                 stderr=subprocess.PIPE,
                 universal_newlines=True
             )
-            logger.info(f"Gitea 服务器已启动，端口: {port}")
-            logger.info(f"访问地址: http://localhost:{port}")
+            logger.info("GitLab 服务已启动")
             
             while True:
-                output = gitea_process.stdout.readline()
-                if output == '' and gitea_process.poll() is not None:
+                output = gitlab_process.stdout.readline()
+                if output == '' and gitlab_process.poll() is not None:
                     break
                 if output:
-                    logger.info(f"Gitea: {output.strip()}")
+                    logger.info(f"GitLab: {output.strip()}")
             
-            return_code = gitea_process.poll()
+            return_code = gitlab_process.poll()
             if return_code != 0:
-                error_output = gitea_process.stderr.read()
-                logger.error(f"Gitea 服务器异常退出，返回码: {return_code}")
+                error_output = gitlab_process.stderr.read()
+                logger.error(f"GitLab 服务器异常退出，返回码: {return_code}")
                 logger.error(f"错误信息: {error_output}")
             else:
-                logger.info("Gitea 服务器正常退出")
+                logger.info("GitLab 服务器正常退出")
                 
-        except FileNotFoundError:
-            logger.error("未找到 gitea 命令，请确保 Gitea 已正确安装")
         except Exception as e:
-            logger.error(f"启动 Gitea 服务器时发生错误: {e}")
+            logger.error(f"启动 GitLab 服务器时发生错误: {e}")
     
-    gitea_thread = threading.Thread(target=run_gitea)
-    gitea_thread.daemon = True
-    gitea_thread.start()
+    gitlab_thread = threading.Thread(target=run_gitlab)
+    gitlab_thread.daemon = True
+    gitlab_thread.start()
     
-    return gitea_thread
+    return gitlab_thread
 
 async def main():
     CONFIG = {
-        "data_directory": "/data",
+        "data_directory": "/data/gitlab",
         "repo_id": os.getenv("REPO_ID", "02engine/02gitea"),
         "hf_token": os.getenv('HF_TOKEN'),
-        "gitea_port": 7860
     }
     
     max_retries = 5
@@ -310,7 +291,7 @@ async def main():
                 logger.error("pullhf.py 执行失败，退出")
                 exit(1)
             
-            gitea_thread = start_gitea_server(CONFIG["gitea_port"])
+            gitlab_thread = start_gitlab_server()
             observer = start_directory_monitoring(
                 data_directory=CONFIG["data_directory"],
                 repo_id=CONFIG["repo_id"],
@@ -318,8 +299,8 @@ async def main():
             )
             
             logger.info("所有服务已启动完成！")
-            logger.info("目录监控: /data to Hugging Face Hub")
-            logger.info("Gitea 服务: http://localhost:7860")
+            logger.info("目录监控: /data/gitlab to Hugging Face Hub")
+            logger.info("GitLab 服务已启动，可通过对应端口访问")
             logger.info("按 Ctrl+C 停止所有服务")
             
             try:
@@ -348,32 +329,23 @@ if __name__ == "__main__":
     asyncio.run(main())
 EOF
 
-# 设置工作目录
-WORKDIR /data
-
-# 暴露 Spaces 默认端口（文档用途，实际由 $PORT 控制）
-EXPOSE 7860
-
-# Gitea 环境变量：适配 Spaces 端口，禁用 SSH，动态决定是否跳过安装页面
-RUN echo '#!/bin/sh' > /set_gitea_env.sh && \
-    echo 'if [ -f /data/gitea_skip_install ]; then' >> /set_gitea_env.sh && \
-    echo '  export GITEA__server__STARTUP_DISABLE_INSTALL_FORM=true' >> /set_gitea_env.sh && \
-    echo 'else' >> /set_gitea_env.sh && \
-    echo '  export GITEA__server__STARTUP_DISABLE_INSTALL_FORM=false' >> /set_gitea_env.sh && \
-    echo 'fi' >> /set_gitea_env.sh && \
-    echo 'exec "$@"' >> /set_gitea_env.sh && \
-    chmod +x /set_gitea_env.sh
-
-ENV GITEA__database__DB_TYPE=sqlite3 \
-    GITEA__database__PATH=/data/gitea/gitea.db \
-    GITEA__server__ROOT_URL=https://deep-sea-02gitea.hf.space/ \
-    GITEA__server__HTTP_ADDR=0.0.0.0 \
-    GITEA__server__HTTP_PORT=7860 \
-    GITEA__server__DISABLE_SSH=true \
-    USER=git
-
-# 切换到 git 用户（非 root）
+# 配置 GitLab 环境变量（基础必要配置）
+ENV GITLAB_OMNIBUS_CONFIG="external_url 'https://deep-sea-02gitlab.hf.space'; \
+    gitlab_rails['time_zone'] = 'Asia/Shanghai'; \
+    gitlab_rails['db_adapter'] = 'postgresql'; \
+    gitlab_rails['db_database'] = 'gitlabhq_production'; \
+    gitlab_rails['db_username'] = 'gitlab'; \
+    gitlab_rails['db_password'] = 'gitlabdbpass'; \
+    gitlab_rails['db_host'] = 'localhost'; \
+    gitlab_rails['db_port'] = 5432; \
+    unicorn['port'] = 8080; \
+    nginx['listen_port'] = 7860;"
+
+# 暴露 GitLab 服务端口（适配 Spaces 默认端口）
+EXPOSE 7860 22 8080
+
+# 切换到 git 用户
 USER git
 
-# 运行 Gitea，显式指定 Spaces 默认端口
-CMD ["/set_gitea_env.sh", "python3", "/uploadhf.py"]
+# 启动同步脚本并托管 GitLab 服务
+CMD ["python3", "/uploadhf.py"]