#!/usr/bin/env python3
"""
将 ~/.local/share/opencode 上传到 Hugging Face Dataset 仓库。
需设置环境变量: HF_TOKEN, OPENCODE_DATASET_REPO。
"""
import os
import re
import sys

def _normalize_repo_id(value):
    """接受 repo_id 或完整 URL，返回 namespace/repo_name。"""
    if not value or not value.strip():
        return None
    value = value.strip()
    m = re.search(r"(?:huggingface\.co/datasets/|^)([\w.-]+/[\w.-]+)/?$", value)
    if m:
        return m.group(1)
    if "/" in value:
        return value
    return None

def main():
    token = os.environ.get("HF_TOKEN")
    raw = os.environ.get("OPENCODE_DATASET_REPO")
    repo_id = _normalize_repo_id(raw)
    data_dir = os.path.expanduser("~/.local/share/opencode")

    if not token or not repo_id:
        return 0

    if not os.path.isdir(data_dir):
        return 0

    try:
        from huggingface_hub import HfApi
    except ImportError:
        print("save: huggingface_hub not installed, skip save", file=sys.stderr)
        return 0

    try:
        api = HfApi(token=token)
        api.upload_folder(
            folder_path=data_dir,
            path_in_repo=".",
            repo_id=repo_id,
            repo_type="dataset",
            token=token,
        )
    except Exception as e:
        print(f"save: upload failed ({e})", file=sys.stderr)
        return 1
    return 0

if __name__ == "__main__":
    sys.exit(main())