#!/usr/bin/env python3
"""arq-sendgrid — SendGrid API bridge.
v0: arq-sendgrid stats get [--days N]
"""
import argparse, json, os, sys, urllib.request, urllib.error
from datetime import UTC, datetime, timedelta
from pathlib import Path
sys.path.insert(0, str(Path(__file__).parent))
from _arq_provider_base import sops_extract, call_with_audit, print_json, handle_meta_flags

PROVIDER = "sendgrid"
REQUIRED_SCOPES: dict[str, list[str]] = {
    # SendGrid API Key permission for /v3/stats.
    "stats get": ["stats.read"],
}
def _key(): return os.environ.get("SENDGRID_API_KEY") or sops_extract('["arqera_twin_admin"]["sendgrid"]["value"]') or sops_extract('["sendgrid"]["sendgrid_api_key"]')
def _get(path):
    k = _key()
    if not k: return 401, "no sendgrid key"
    req = urllib.request.Request(f"https://api.sendgrid.com{path}", headers={"Authorization": f"Bearer {k}"})
    try:
        with urllib.request.urlopen(req, timeout=30) as r: return r.status, json.loads(r.read())
    except urllib.error.HTTPError as e: return e.code, e.read().decode("utf-8","ignore")
    except Exception as e: return 500, str(e)

def _stats(a):
    end = datetime.now(UTC).date()
    start = end - timedelta(days=a.days)
    c, d = _get(f"/v3/stats?start_date={start}&end_date={end}&aggregated_by=day")
    if c != 200: sys.stderr.write(f"HTTP {c}: {d}\n"); return 1
    return print_json(d)

def main():
    handle_meta_flags(PROVIDER, REQUIRED_SCOPES)
    p = argparse.ArgumentParser(prog="arq-sendgrid"); s = p.add_subparsers(dest="cmd", required=True)
    ss = s.add_parser("stats"); ss2 = ss.add_subparsers(dest="action", required=True)
    ssg = ss2.add_parser("get"); ssg.add_argument("--days", type=int, default=30); ssg.set_defaults(func=_stats, verb="stats get")
    args = p.parse_args()
    return call_with_audit(PROVIDER, args.verb, args.func, args)

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