Spaces:
Paused
Paused
| """ | |
| Backup / Restore serwera Minecraft do Hugging Face Dataset. | |
| Usage: | |
| python3 sync_data.py restore | |
| python3 sync_data.py backup | |
| """ | |
| import os | |
| import sys | |
| import tarfile | |
| import shutil | |
| import tempfile | |
| from pathlib import Path | |
| from huggingface_hub import HfApi, hf_hub_download | |
| # ── configuration ──────────────────────────────────────────────────────────── | |
| DATASET_ID = os.environ.get("DATASET_ID", "natekkz/basketrush-data") | |
| HF_TOKEN = os.environ.get("HF_TOKEN", "") | |
| SERVER_DIR = Path("/home/user/minecraft/server") | |
| ARCHIVE_NAME = "server-backup.tar.gz" | |
| # Directories to back up (relative to SERVER_DIR) | |
| BACKUP_DIRS = [ | |
| "world", | |
| "plugins", | |
| "config", | |
| ] | |
| # Individual files to back up (relative to SERVER_DIR) | |
| BACKUP_FILES = [ | |
| "server.properties", | |
| "bukkit.yml", | |
| "spigot.yml", | |
| "commands.yml", | |
| "help.yml", | |
| "permissions.yml", | |
| "wepif.yml", | |
| "usercache.json", | |
| "whitelist.json", | |
| "banned-ips.json", | |
| "banned-players.json", | |
| "ops.json", | |
| "server-icon.png", | |
| "version_history.json", | |
| ] | |
| # playit.gg configuration (persisted across restarts) | |
| PLAYIT_TOML = Path.home() / ".config" / "playit" / "playit.toml" | |
| # ── restore ────────────────────────────────────────────────────────────────── | |
| def restore(): | |
| """Download and extract server backup from HF Dataset.""" | |
| if not HF_TOKEN: | |
| print("[sync] No HF_TOKEN set – skipping restore") | |
| return False | |
| print(f"[sync] Restoring from dataset {DATASET_ID}...") | |
| try: | |
| archive_path = hf_hub_download( | |
| repo_id=DATASET_ID, | |
| filename=ARCHIVE_NAME, | |
| repo_type="dataset", | |
| token=HF_TOKEN, | |
| ) | |
| except Exception as e: | |
| print(f"[sync] No backup found in dataset ({e}) – starting fresh") | |
| return False | |
| SERVER_DIR.mkdir(parents=True, exist_ok=True) | |
| # Safe extraction — validate paths | |
| with tarfile.open(archive_path, "r:gz") as tar: | |
| for member in tar.getmembers(): | |
| target = (SERVER_DIR / member.name).resolve() | |
| if not str(target).startswith(str(SERVER_DIR.resolve())): | |
| print(f"[sync] ⚠ Skipping unsafe path: {member.name}") | |
| continue | |
| tar.extract(member, path=SERVER_DIR) | |
| # Restore playit.gg config if present in backup | |
| playit_backup = SERVER_DIR / ".playit.toml" | |
| if playit_backup.exists(): | |
| PLAYIT_TOML.parent.mkdir(parents=True, exist_ok=True) | |
| shutil.copy2(playit_backup, PLAYIT_TOML) | |
| print("[sync] Restored playit.toml") | |
| print("[sync] ✅ Restore complete") | |
| return True | |
| # ── backup ─────────────────────────────────────────────────────────────────── | |
| def backup(): | |
| """Create a tarball of server data and upload to HF Dataset.""" | |
| if not HF_TOKEN: | |
| print("[sync] No HF_TOKEN set – skipping backup") | |
| return False | |
| print(f"[sync] Backing up to dataset {DATASET_ID}...") | |
| with tempfile.TemporaryDirectory() as tmp: | |
| archive_path = Path(tmp) / ARCHIVE_NAME | |
| with tarfile.open(archive_path, "w:gz") as tar: | |
| # Add directories | |
| for d in BACKUP_DIRS: | |
| src = SERVER_DIR / d | |
| if src.exists(): | |
| tar.add(str(src), arcname=d) | |
| print(f" + {d}/") | |
| # Add individual files | |
| for f in BACKUP_FILES: | |
| src = SERVER_DIR / f | |
| if src.exists(): | |
| tar.add(str(src), arcname=f) | |
| print(f" + {f}") | |
| # Include playit.gg config | |
| if PLAYIT_TOML.exists(): | |
| tar.add(str(PLAYIT_TOML), arcname=".playit.toml") | |
| print(" + .playit.toml") | |
| # Upload | |
| api = HfApi(token=HF_TOKEN) | |
| api.upload_file( | |
| path_or_fileobj=str(archive_path), | |
| path_in_repo=ARCHIVE_NAME, | |
| repo_id=DATASET_ID, | |
| repo_type="dataset", | |
| ) | |
| print("[sync] ✅ Backup uploaded") | |
| return True | |
| # ── CLI ────────────────────────────────────────────────────────────────────── | |
| if __name__ == "__main__": | |
| if len(sys.argv) < 2: | |
| print("Usage: python3 sync_data.py [restore|backup]") | |
| sys.exit(1) | |
| cmd = sys.argv[1].lower() | |
| if cmd == "restore": | |
| restore() | |
| elif cmd == "backup": | |
| backup() | |
| else: | |
| print(f"Unknown command: {cmd}") | |
| sys.exit(1) | |