"""
BigQuery Iceberg — Metadata Collect & Push (combined)
=====================================================
Convenience wrapper that runs collect_metadata.collect() followed by
push_metadata.push() in a single invocation. Supports
``--only-freshness-and-volume`` for fast periodic pushes.

Prerequisites:
  pip install google-cloud-bigquery pycarlo>=0.12.251
"""

from __future__ import annotations

import argparse
import os

from collect_metadata import _require_bq_identifier, collect
from push_metadata import push
from _safe_paths import safe_output_json_path


def main() -> None:
    parser = argparse.ArgumentParser(
        description="Collect BigQuery Iceberg metadata and push to Monte Carlo",
    )
    # Collection args
    parser.add_argument("--project-id", default=os.getenv("BIGQUERY_PROJECT_ID"))
    parser.add_argument("--datasets", nargs="+", default=None)
    parser.add_argument("--tables", nargs="+", default=None)
    parser.add_argument(
        "--only-freshness-and-volume",
        action="store_true",
        help="Skip field/schema collection — only collect freshness and volume.",
    )
    parser.add_argument("--manifest-file", default="metadata_output.json")

    # Push args
    parser.add_argument("--resource-uuid", default=os.getenv("MCD_RESOURCE_UUID"))
    parser.add_argument("--key-id", default=os.getenv("MCD_INGEST_ID"))
    parser.add_argument("--key-token", default=os.getenv("MCD_INGEST_TOKEN"))
    parser.add_argument("--batch-size", type=int, default=500)
    parser.add_argument("--push-result-file", default="metadata_push_result.json")

    args = parser.parse_args()

    if not args.project_id:
        parser.error("--project-id or BIGQUERY_PROJECT_ID env var is required")
    required_push = ["resource_uuid", "key_id", "key_token"]
    missing = [k for k in required_push if getattr(args, k) is None]
    if missing:
        parser.error(f"Missing required push arguments/env vars: {missing}")

    manifest_path = str(safe_output_json_path(args.manifest_file))
    push_result_path = str(safe_output_json_path(args.push_result_file))

    args.project_id = _require_bq_identifier(args.project_id, "project_id")
    args.datasets = [_require_bq_identifier(d, "dataset") for d in args.datasets or []] or None
    args.tables = [_require_bq_identifier(t, "table") for t in args.tables or []] or None

    collect(
        project_id=args.project_id,
        datasets=args.datasets,
        tables=args.tables,
        only_freshness_and_volume=args.only_freshness_and_volume,
        output_file=manifest_path,
    )

    push(
        input_file=manifest_path,
        resource_uuid=args.resource_uuid,
        key_id=args.key_id,
        key_token=args.key_token,
        batch_size=args.batch_size,
        output_file=push_result_path,
    )


if __name__ == "__main__":
    main()