#!/usr/bin/env python3
"""
🌱 Canteen Seed Data Import Tool

CLI entry point for the seed data import pipeline.
Parses Cantaloupe Excel exports and writes to SQLite database.

Usage:
    python3 main.py seed-data/Machine_List.xlsx --output seed-data/assets.db
    python3 main.py seed-data/Machine_List.xlsx --output seed-data/assets.db --mode=update
    python3 main.py seed-data/Machine_List.xlsx -o seed-data/assets.db -m seed
"""

import argparse
import sys
import os

# Import pipeline modules
from parser import parse_excel
from db_writer import write_seed, write_update


def main():
    parser = argparse.ArgumentParser(
        description="🌱 Canteen Seed Data Import Tool",
        formatter_class=argparse.RawDescriptionHelpFormatter,
        epilog=(
            "Examples:\n"
            "  %(prog)s seed-data/Machine_List.xlsx -o seed-data/assets.db\n"
            "  %(prog)s seed-data/Machine_List.xlsx -o seed-data/assets.db --mode=update\n"
        ),
    )

    parser.add_argument(
        "input",
        type=str,
        help="Path to the input Excel file (Machine_List.xlsx)",
    )

    parser.add_argument(
        "-o", "--output",
        type=str,
        default="seed-data/assets.db",
        help="Output SQLite database path (default: seed-data/assets.db)",
    )

    parser.add_argument(
        "-m", "--mode",
        type=str,
        choices=["seed", "update"],
        default="seed",
        help="Import mode: 'seed' for first import (fresh insert), 'update' for incremental (default: seed)",
    )

    args = parser.parse_args()

    # Resolve input path
    input_path = os.path.abspath(args.input)
    if not os.path.exists(input_path):
        print(f"❌ Error: Input file not found: {input_path}")
        sys.exit(1)

    output_path = os.path.abspath(args.output)

    print("🌱 Canteen Seed Data Import Tool")
    print("═" * 50)
    print(f"  📄 Input:  {input_path}")
    print(f"  📁 Output: {output_path}")
    print(f"  🔧 Mode:   {args.mode}")
    print()

    # ── Step 1: Parse Excel ──
    print("📋 Parsing Excel file...")
    try:
        machines = parse_excel(input_path)
    except Exception as e:
        print(f"❌ Error parsing Excel file: {e}")
        sys.exit(1)

    print(f"  ✅ Parsed {len(machines)} machines from Excel")
    print()

    # ── Step 2: Write to database ──
    print("💾 Writing to database...")
    if args.mode == "seed":
        stats = write_seed(machines, output_path)
    else:
        stats = write_update(machines, output_path)

    print()
    print("═" * 50)
    print("✅ Import complete!")
    print(f"  Mode:      {args.mode}")
    print(f"  Records:   {stats.get('inserted', 0) + stats.get('updated', 0)} processed")
    if "inserted" in stats:
        print(f"  Inserted:  {stats['inserted']}")
    if "updated" in stats:
        print(f"  Updated:   {stats['updated']}")
    if "preserved" in stats:
        print(f"  Preserved: {stats['preserved']}")
    if "errors" in stats and stats["errors"]:
        print(f"  Errors:    {stats['errors']}")
    print("═" * 50)

    return 0


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