From 1c37be852982ad52d7eeb83d53601ecc9a905a51 Mon Sep 17 00:00:00 2001
From: KakiFilem Team <kakifilem@noreply.localhost>
Date: Mon, 2 Feb 2026 19:33:46 +0800
Subject: [PATCH] Add CLI interface and pyproject.toml for PyPI packaging

---
 cli/__init__.py | Bin 0 -> 1024 bytes
 cli/cli.py      | 166 ++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 166 insertions(+)
 create mode 100644 cli/__init__.py
 create mode 100644 cli/cli.py

diff --git a/cli/__init__.py b/cli/__init__.py
new file mode 100644
index 0000000000000000000000000000000000000000..06d7405020018ddf3cacee90fd4af10487da3d20
GIT binary patch
literal 1024
ScmZQz7zLvtFd70QH3R?z00031

literal 0
HcmV?d00001

diff --git a/cli/cli.py b/cli/cli.py
new file mode 100644
index 0000000..4b3d5db
--- /dev/null
+++ b/cli/cli.py
@@ -0,0 +1,166 @@
+import argparse
+import shutil
+import os
+import sys
+import textwrap
+import importlib.metadata
+
+from main import run_backup
+
+def get_version():
+    try:
+        return importlib.metadata.version("pg-r2-backup")
+    except importlib.metadata.PackageNotFoundError:
+        return "dev"
+
+
+def mask(value, show=4):
+    if not value:
+        return ""
+    if len(value) <= show:
+        return "*" * len(value)
+    return value[:show] + "*" * (len(value) - show)
+
+def doctor():
+    print("pg-r2-backup doctor\n")
+    
+    if shutil.which("pg_dump") is None:
+        print("[FAIL] pg_dump not found in PATH")
+    else:
+        print("[OK] pg_dump found")
+
+    required_envs = [
+        "DATABASE_URL",
+        "R2_ACCESS_KEY",
+        "R2_SECRET_KEY",
+        "R2_BUCKET_NAME",
+        "R2_ENDPOINT",
+    ]
+
+    missing = [e for e in required_envs if not os.environ.get(e)]
+
+    if missing:
+        print("\n[FAIL] Missing environment variables:")
+        for m in missing:
+            print(f"  - {m}")
+    else:
+        print("\n[OK] Required environment variables set")
+
+    use_public = os.environ.get("USE_PUBLIC_URL", "false").lower() == "true"
+    print(f"\nDatabase URL mode : {'public' if use_public else 'private'}")
+
+    if os.environ.get("BACKUP_PASSWORD"):
+        print("Compression      : 7z (encrypted)")
+    else:
+        print("Compression      : gzip")
+
+    print("\nDoctor check complete.")
+
+
+def config_show():
+    print("pg-r2-backup config\n")
+
+    config = {
+        "USE_PUBLIC_URL": os.environ.get("USE_PUBLIC_URL", "false"),
+        "DUMP_FORMAT": os.environ.get("DUMP_FORMAT", "dump"),
+        "FILENAME_PREFIX": os.environ.get("FILENAME_PREFIX", "backup"),
+        "MAX_BACKUPS": os.environ.get("MAX_BACKUPS", "7"),
+        "BACKUP_TIME": os.environ.get("BACKUP_TIME", "00:00"),
+        "R2_BUCKET_NAME": os.environ.get("R2_BUCKET_NAME", ""),
+        "R2_ENDPOINT": os.environ.get("R2_ENDPOINT", ""),
+        "R2_ACCESS_KEY": mask(os.environ.get("R2_ACCESS_KEY")),
+        "R2_SECRET_KEY": mask(os.environ.get("R2_SECRET_KEY")),
+    }
+
+    for k, v in config.items():
+        print(f"{k:<16} : {v}")
+
+
+def init_env():
+    if os.path.exists(".env"):
+        print("[ERROR] .env already exists")
+        return
+
+    example = ".env.example"
+    if not os.path.exists(example):
+        print("[ERROR] .env.example not found")
+        return
+
+    shutil.copy(example, ".env")
+    print("[SUCCESS] .env created from .env.example")
+    print("Edit the file before running backups.")
+
+
+def schedule_info():
+    print(textwrap.dedent("""
+    pg-r2-backup scheduling
+
+    Linux / macOS (cron):
+      0 0 * * * pg-r2-backup run
+
+    Windows (Task Scheduler):
+      Program : pg-r2-backup
+      Args    : run
+      Start in: folder containing .env (working directory)
+
+    Railway / Docker:
+      Use the platform scheduler
+    """).strip())
+
+def main():
+    parser = argparse.ArgumentParser(
+        prog="pg-r2-backup",
+        description="PostgreSQL backup tool for Cloudflare R2",
+        formatter_class=argparse.RawDescriptionHelpFormatter,
+        epilog=textwrap.dedent("""
+            Examples:
+              pg-r2-backup doctor
+              pg-r2-backup run
+              pg-r2-backup config show
+              pg-r2-backup init
+              pg-r2-backup schedule
+        """)
+    )
+
+    parser.add_argument(
+        "--version",
+        action="version",
+        version=f"%(prog)s {get_version()}"
+    )
+
+    subparsers = parser.add_subparsers(dest="command")
+
+    subparsers.add_parser("run", help="Run backup immediately")
+    subparsers.add_parser("doctor", help="Check environment & dependencies")
+    subparsers.add_parser("schedule", help="Show scheduling examples")
+
+    config_parser = subparsers.add_parser("config", help="Show configuration")
+    config_sub = config_parser.add_subparsers(dest="subcommand")
+    config_sub.add_parser("show", help="Show current configuration")
+
+    subparsers.add_parser("init", help="Create .env from .env.example")
+
+    args = parser.parse_args()
+
+    if args.command == "run":
+        run_backup()
+
+    elif args.command == "doctor":
+        doctor()
+
+    elif args.command == "config" and args.subcommand == "show":
+        config_show()
+
+    elif args.command == "init":
+        init_env()
+
+    elif args.command == "schedule":
+        schedule_info()
+
+    else:
+        parser.print_help()
+        sys.exit(1)
+
+
+if __name__ == "__main__":
+    main()