#!/usr/bin/env python3
"""Offline PostgreSQL backup/verify/recovery CLI for administrators."""
from __future__ import annotations
# Allow direct `python scripts/name.py` from a fresh source checkout.
import sys as _sys
from pathlib import Path as _Path
_sys.path.insert(0, str(_Path(__file__).resolve().parents[1]))
import argparse
import json
import os
from pathlib import Path
from app import pg_disaster_recovery as pg

def main(argv=None):
    p=argparse.ArgumentParser(description='SiteForge offline PostgreSQL disaster recovery')
    sub=p.add_subparsers(dest='operation',required=True)
    b=sub.add_parser('backup');b.add_argument('--data',required=True);b.add_argument('--output',required=True)
    b.add_argument('--confirmed-offline',action='store_true')
    v=sub.add_parser('verify');v.add_argument('archive')
    r=sub.add_parser('restore-new');r.add_argument('archive');r.add_argument('--data',required=True)
    r.add_argument('--confirmed-offline',action='store_true')
    args=p.parse_args(argv)
    try:
        if args.operation=='verify':value=pg.verify(args.archive)
        else:
            url=os.getenv('SITEFORGE_DATABASE_URL','')
            if args.operation=='backup':
                value=pg.backup(url,args.data,args.output,confirmed_offline=args.confirmed_offline)
            else:
                value=pg.restore_new(url,args.archive,args.data,confirmed_offline=args.confirmed_offline)
        print(json.dumps(value,ensure_ascii=False,indent=2))
        return 0
    except (pg.RecoveryError,OSError,ValueError) as exc:
        print('ERROR: '+str(exc),file=__import__('sys').stderr)
        return 1

if __name__=='__main__':raise SystemExit(main())