File size: 3,519 Bytes
632b0a7 | 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 | import os
import subprocess
from huggingface_hub import snapshot_download
# 绝对敏感：不同步
SENSITIVE_EXCLUDES = [
"auth.json",
".sandbox-secrets/**",
]
# 建议不同步的大缓存/易变目录（减少文件数 & vanished）
HEAVY_EXCLUDES = [
"plugins/cache/**",
"projects/**",
"shell-snapshots/**",
"statsig/**",
"todos/**",
"log/**",
"tmp/**",
"sessions/**",
]
def run_rsync(src: str, dst: str, excludes=None, delete=True):
excludes = excludes or []
os.makedirs(dst, exist_ok=True)
cmd = ["rsync", "-a"]
if delete:
cmd.append("--delete")
for ex in excludes:
cmd += ["--exclude", ex]
cmd += [src.rstrip("/") + "/", dst.rstrip("/") + "/"]
p = subprocess.run(cmd)
# rsync code 24: some files vanished (often due to changing caches). Treat as warning.
if p.returncode not in (0, 24):
raise RuntimeError(f"rsync failed with code {p.returncode}")
def main():
repo = os.getenv("DOTFILES_DATASET_REPO", "").strip()
if not repo:
print("DOTFILES_DATASET_REPO 未设置，跳过 dotfiles 同步")
return
token = os.getenv("HF_TOKEN") or os.getenv("HUGGINGFACE_HUB_TOKEN")
if not token:
print("HF_TOKEN 未设置（Private Dataset 必需），跳过 dotfiles 同步")
return
# 只下载你需要的目录，避免拉 1500+ 文件
allow = [
"ubuntu/home/.claude/**",
"ubuntu/home/.codex/**",
"dotfiles/home/.claude/**",
"dotfiles/home/.codex/**",
"ubuntu/project/**",
"dotfiles/project/**",
"project/**",
]
local_dir = snapshot_download(
repo_id=repo,
repo_type="dataset",
allow_patterns=allow,
token=token,
)
# 兼容你的结构：优先 ubuntu/home
cand_home = [
os.path.join(local_dir, "ubuntu", "home"),
os.path.join(local_dir, "dotfiles", "home"),
os.path.join(local_dir, "home"),
]
home_root = next((p for p in cand_home if os.path.isdir(p)), None)
if not home_root:
print("未找到 home 根目录（例如 ubuntu/home）")
return
claude_src = os.path.join(home_root, ".claude")
codex_src = os.path.join(home_root, ".codex")
# 只同步这两个目录，不碰 /home/user 其它文件
if os.path.isdir(claude_src):
print(f"Sync ~/.claude: {claude_src} -> /home/user/.claude")
run_rsync(claude_src, "/home/user/.claude", excludes=HEAVY_EXCLUDES, delete=True)
else:
print("未找到 .claude 目录，跳过")
if os.path.isdir(codex_src):
excludes = SENSITIVE_EXCLUDES + HEAVY_EXCLUDES
print(f"Sync ~/.codex: {codex_src} -> /home/user/.codex (exclude auth/secrets)")
run_rsync(codex_src, "/home/user/.codex", excludes=excludes, delete=True)
else:
print("未找到 .codex 目录，跳过")
# 可选：项目级配置（如果你未来加了）
cand_proj = [
os.path.join(local_dir, "ubuntu", "project"),
os.path.join(local_dir, "dotfiles", "project"),
os.path.join(local_dir, "project"),
]
proj_src = next((p for p in cand_proj if os.path.isdir(p)), None)
if proj_src:
print(f"Sync project: {proj_src} -> /home/user/work")
run_rsync(proj_src, "/home/user/work", excludes=[], delete=True)
else:
print("未找到 PROJECT dotfiles 源目录（可忽略）")
print("Dotfiles sync done.")
if __name__ == "__main__":
main() |