#!/usr/bin/env python3
"""
Snapshot the currently configured TiDB connection into /tmp/tidb_instance.json.

This script is kept for backward compatibility with older local workflows that
expected a file-based DB handoff. It no longer provisions TiDB Cloud Zero.
"""

import json
import os
import sys


def getenv(name: str, fallback: str = "") -> str:
    return os.environ.get(name, fallback)


def main():
    host = getenv("TIDB_HOST", getenv("DB_HOST"))
    port = int(getenv("TIDB_PORT", getenv("DB_PORT", "4000")))
    user = getenv("TIDB_USER", getenv("DB_USERNAME"))
    password = getenv("TIDB_PASSWORD", getenv("DB_PASSWORD"))
    database = getenv("TIDB_DATABASE", getenv("DB_DATABASE", "nexuscrm"))

    if not host or not user:
        print(
            "❌ Missing DB connection. Set TIDB_* or DB_* environment variables before running setup_tidb.py.",
            file=sys.stderr,
        )
        sys.exit(1)

    result_data = {
        "host": host,
        "port": port,
        "user": user,
        "password": password,
        "database": database,
    }

    with open("/tmp/tidb_instance.json", "w") as f:
        json.dump(result_data, f, indent=2)

    print("✅ Wrote configured TiDB connection to /tmp/tidb_instance.json")
    print(f"   host={host}:{port}")
    print(f"   user={user}")
    print(f"   database={database}")


if __name__ == "__main__":
    main()