import json
from pathlib import Path

import click

import build_observed_products
import build_purchases
import review_products
from layer_helpers import read_csv_rows, write_csv_rows


SUMMARY_FIELDS = ["stage", "count"]


def read_rows_if_exists(path):
    path = Path(path)
    if not path.exists():
        return []
    return read_csv_rows(path)


def build_status_summary(
    giant_orders,
    giant_items,
    giant_enriched,
    costco_orders,
    costco_items,
    costco_enriched,
    purchases,
    resolutions,
):
    enriched_rows = giant_enriched + costco_enriched
    observed_rows = build_observed_products.build_observed_products(enriched_rows)
    queue_rows = review_products.build_review_queue(purchases, resolutions)

    unresolved_purchase_rows = [
        row
        for row in purchases
        if row.get("observed_product_id")
        and not row.get("canonical_product_id")
        and row.get("is_fee") != "true"
        and row.get("is_discount_line") != "true"
        and row.get("is_coupon_line") != "true"
    ]
    excluded_rows = [
        row
        for row in purchases
        if row.get("resolution_action") == "exclude"
    ]
    linked_purchase_rows = [row for row in purchases if row.get("canonical_product_id")]

    summary = [
        {"stage": "raw_orders", "count": len(giant_orders) + len(costco_orders)},
        {"stage": "raw_items", "count": len(giant_items) + len(costco_items)},
        {"stage": "enriched_items", "count": len(enriched_rows)},
        {"stage": "observed_products", "count": len(observed_rows)},
        {"stage": "review_queue_observed_products", "count": len(queue_rows)},
        {"stage": "canonical_linked_purchase_rows", "count": len(linked_purchase_rows)},
        {"stage": "final_purchase_rows", "count": len(purchases)},
        {"stage": "unresolved_purchase_rows", "count": len(unresolved_purchase_rows)},
        {"stage": "excluded_purchase_rows", "count": len(excluded_rows)},
        {
            "stage": "unresolved_not_in_review_rows",
            "count": len(
                [
                    row
                    for row in unresolved_purchase_rows
                    if row.get("observed_product_id")
                    not in {queue_row["observed_product_id"] for queue_row in queue_rows}
                ]
            ),
        },
    ]
    return summary


@click.command()
@click.option("--giant-orders-csv", default="giant_output/orders.csv", show_default=True)
@click.option("--giant-items-csv", default="giant_output/items.csv", show_default=True)
@click.option("--giant-enriched-csv", default="giant_output/items_enriched.csv", show_default=True)
@click.option("--costco-orders-csv", default="costco_output/orders.csv", show_default=True)
@click.option("--costco-items-csv", default="costco_output/items.csv", show_default=True)
@click.option("--costco-enriched-csv", default="costco_output/items_enriched.csv", show_default=True)
@click.option("--purchases-csv", default="combined_output/purchases.csv", show_default=True)
@click.option("--resolutions-csv", default="combined_output/review_resolutions.csv", show_default=True)
@click.option("--summary-csv", default="combined_output/pipeline_status.csv", show_default=True)
@click.option("--summary-json", default="combined_output/pipeline_status.json", show_default=True)
def main(
    giant_orders_csv,
    giant_items_csv,
    giant_enriched_csv,
    costco_orders_csv,
    costco_items_csv,
    costco_enriched_csv,
    purchases_csv,
    resolutions_csv,
    summary_csv,
    summary_json,
):
    summary_rows = build_status_summary(
        read_rows_if_exists(giant_orders_csv),
        read_rows_if_exists(giant_items_csv),
        read_rows_if_exists(giant_enriched_csv),
        read_rows_if_exists(costco_orders_csv),
        read_rows_if_exists(costco_items_csv),
        read_rows_if_exists(costco_enriched_csv),
        read_rows_if_exists(purchases_csv),
        read_rows_if_exists(resolutions_csv),
    )
    write_csv_rows(summary_csv, summary_rows, SUMMARY_FIELDS)
    summary_json_path = Path(summary_json)
    summary_json_path.parent.mkdir(parents=True, exist_ok=True)
    summary_json_path.write_text(json.dumps(summary_rows, indent=2), encoding="utf-8")
    for row in summary_rows:
        click.echo(f"{row['stage']}: {row['count']}")


if __name__ == "__main__":
    main()