#!/usr/bin/env python3
"""
Adds the fields that carry real dates through receiving and ordering.

The reader has always read the slip's date, receiver, site name, customer
VAT and supplier registration, and the pipeline writes them to
dn_batch_notes, but the collection never had those fields, so PocketBase
dropped them on every save. Posting then fell back to "now" for
delivery_notes.date_received, which put every delivery in the month it was
posted instead of the month it arrived.

  dn_batch_notes:  date, received_by_name, site_name_raw, customer_vat_no,
                   supplier_registration_raw   (text, as the reader gives them)
  delivery_notes:  posted_at   (when the receipt was booked; date_received
                   becomes the slip's delivery date)
  purchase_orders: order_date  (the "PO Creation Date" printed on the PO;
                   the upload date says nothing about when it was ordered)

Idempotent. Run scripts/pb-backup-collection.py for each collection first.
"""
import sys

from pb_env import PB, http, superuser_token

TEXT = lambda name: {"name": name, "type": "text", "required": False}
WANTED = {
    "dn_batch_notes": [TEXT("date"), TEXT("received_by_name"), TEXT("site_name_raw"), TEXT("customer_vat_no"), TEXT("supplier_registration_raw")],
    "delivery_notes": [{"name": "posted_at", "type": "date", "required": False}],
    "purchase_orders": [{"name": "order_date", "type": "date", "required": False}],
}


def main():
    token = superuser_token()
    print(f"\n→ {PB}")
    for name, wanted in WANTED.items():
        code, coll = http("GET", f"/api/collections/{name}", token=token)
        if code != 200:
            sys.exit(f"GET {name} failed: {code} {coll}")
        fields = list(coll.get("fields", []))
        have = {f.get("name") for f in fields}
        add = [w for w in wanted if w["name"] not in have]
        if not add:
            print(f"  • {name}: nothing to add")
            continue
        code, resp = http("PATCH", f"/api/collections/{coll['id']}", {"fields": fields + add}, token)
        if code != 200:
            sys.exit(f"PATCH {name} failed: {code} {resp}")
        for a in add:
            print(f"  ✓ {name}: added {a['name']}")


if __name__ == "__main__":
    main()
    print("\nDone.")