import json
import frappe
from frappe.utils import today, getdate, add_days
from datetime import timedelta

@frappe.whitelist()
def get_pickups(date=None):
    """Fetch scheduled pickups with optional date filter.
    Returns pickups, calendar (next 30 days), and weekly chart data."""
    filters = []
    if date:
        filters.append(["Scheduled Pickup", "pickup_date", "=", date])

    fields = [
        "name", "pickup_date", "pickup_type", "status", "truck", "stop_order",
        "customer_number", "company_name",
        "contact_name", "contact_phone", "contact_email",
        "address_line", "city", "state", "zip_code",
        "latitude", "longitude",
        "estimated_items", "estimated_weight", "load_contents",
        "num_labels", "data_status", "red_r2",
        "notes", "legacy_notes", "needs_aor", "needs_cod",
    ]

    pickups = frappe.get_list("Scheduled Pickup",
        fields=fields,
        filters=filters if filters else None,
        order_by="pickup_date asc, stop_order asc",
        limit_page_length=500,
    )

    # Build calendar data (next 30 days)
    from_date = getdate(today())
    to_date = add_days(from_date, 30)

    all_pickups = frappe.get_list("Scheduled Pickup",
        fields=["pickup_date"],
        filters=[["Scheduled Pickup", "pickup_date", ">=", str(from_date)],
                 ["Scheduled Pickup", "pickup_date", "<=", str(to_date)]],
        limit_page_length=500,
    )

    pickup_counts = {}
    for p in all_pickups:
        d = p.get("pickup_date", "")
        if d:
            pickup_counts[d] = pickup_counts.get(d, 0) + 1

    calendar = []
    for i in range(30):
        d = add_days(from_date, i)
        ds = str(d)
        calendar.append({"date": ds, "count": pickup_counts.get(ds, 0)})

    # Build weekly chart data (last 12 weeks)
    weekly = []
    for i in range(11, -1, -1):
        week_start = add_days(from_date, -(from_date.weekday() + 7 * i))
        week_end = add_days(week_start, 6)
        count = 0
        for d_str, c in pickup_counts.items():
            try:
                d = getdate(d_str)
                if week_start <= d <= week_end:
                    count += c
            except (ValueError, TypeError):
                pass
        weekly.append({"label": week_start.strftime("%m/%d"), "count": count})

    return {
        "pickups": pickups,
        "calendar": calendar,
        "weekly": weekly,
    }


@frappe.whitelist()
def auto_route(date=None):
    """Auto-assign pickups to trucks based on capacity and proximity."""
    if not date:
        date = today()

    pickups = frappe.get_list("Scheduled Pickup",
        filters={"pickup_date": date},
        fields=["name", "company_name", "estimated_items", "estimated_weight",
                "latitude", "longitude", "pickup_type"],
        limit_page_length=200,
    )

    if not pickups:
        return {"success": True, "assigned": 0}

    trucks = ["Truck 1", "Truck 2", "Truck 3"]
    sorted_p = sorted(pickups, key=lambda p: (float(p.get("latitude") or 0), float(p.get("longitude") or 0)))
    n = len(sorted_p)
    assigned = 0

    for i, p in enumerate(sorted_p):
        if p.get("pickup_type") == "Drop-off":
            truck = ""
        else:
            truck = trucks[i % 3] if n <= 3 else trucks[min(i * 3 // n, 2)]

        doc = frappe.get_doc("Scheduled Pickup", p["name"])
        doc.truck = truck
        doc.status = "Routed" if truck else "Scheduled"
        doc.stop_order = i + 1
        doc.save()
        assigned += 1

    frappe.db.commit()
    return {"success": True, "assigned": assigned}


@frappe.whitelist()
def get_checkins():
    """Fetch completed check-ins."""
    checkins = frappe.get_list("Scheduled Pickup",
        filters={"status": ["in", ["Complete", "In Progress"]]},
        fields=["name", "pickup_date", "pickup_type", "status",
                "company_name", "customer_number",
                "estimated_items", "estimated_weight", "load_contents",
                "data_status", "red_r2", "notes"],
        order_by="pickup_date desc",
        limit_page_length=100,
    )
    return {"checkins": checkins}


@frappe.whitelist()
def cor_report():
    """Generate Certificate of Recycling report."""
    pickups = frappe.get_list("Scheduled Pickup",
        filters={"status": "Complete"},
        fields=["name", "pickup_date", "company_name", "customer_number",
                "estimated_items", "estimated_weight", "load_contents",
                "data_status", "red_r2", "needs_aor", "needs_cod"],
        order_by="pickup_date desc",
        limit_page_length=200,
    )

    html = "<!DOCTYPE html><html><head><title>CoR Report</title>"
    html += "<style>body{font-family:Arial,sans-serif;margin:40px;}"
    html += "table{border-collapse:collapse;width:100%;}"
    html += "th,td{border:1px solid #ddd;padding:8px;text-align:left;font-size:13px;}"
    html += "th{background:#2F5496;color:#fff;}h1{color:#2F5496;}</style></head><body>"
    html += "<h1>Certificate of Recycling (CoR) Report</h1>"
    html += "<p>Generated: " + frappe.utils.now() + "</p>"
    html += "<p>Total completed loads: " + str(len(pickups)) + "</p>"

    if pickups:
        html += "<table><tr><th>Date</th><th>Customer</th><th>Items</th><th>Weight</th>"
        html += "<th>Contents</th><th>Data Status</th><th>RED/R2</th><th>AoR</th><th>CoD</th></tr>"
        for p in pickups:
            html += "<tr><td>" + str(p.get("pickup_date", "")) + "</td>"
            html += "<td>" + str(p.get("company_name", "")) + "</td>"
            html += "<td>" + str(p.get("estimated_items", "")) + "</td>"
            html += "<td>" + str(p.get("estimated_weight", "")) + "</td>"
            html += "<td>" + str(p.get("load_contents", "")) + "</td>"
            html += "<td>" + str(p.get("data_status", "")) + "</td>"
            html += "<td>" + str(p.get("red_r2", "")) + "</td>"
            html += "<td>" + ("✓" if p.get("needs_aor") else "") + "</td>"
            html += "<td>" + ("✓" if p.get("needs_cod") else "") + "</td></tr>"
        html += "</table>"
    else:
        html += "<p>No completed loads found.</p>"

    html += "</body></html>"
    frappe.local.response["type"] = "html"
    frappe.local.response["page_content"] = html


@frappe.whitelist()
def print_route_sheet(date=None):
    """Generate printable route sheet."""
    if not date:
        date = today()

    filters = {"pickup_date": date} if date else {}
    pickups = frappe.get_list("Scheduled Pickup",
        filters=filters,
        fields=["name", "pickup_date", "pickup_type", "status", "truck", "stop_order",
                "company_name", "contact_name", "contact_phone", "contact_email",
                "address_line", "city", "state", "zip_code",
                "estimated_items", "estimated_weight", "load_contents",
                "data_status", "red_r2", "needs_aor", "needs_cod", "notes"],
        order_by="truck asc, stop_order asc",
        limit_page_length=200,
    )

    trucks = {}
    unassigned = []
    for p in pickups:
        t = p.get("truck", "")
        if t and t != "Unassigned":
            trucks.setdefault(t, []).append(p)
        else:
            unassigned.append(p)

    html = "<!DOCTYPE html><html><head><title>Route Sheet</title>"
    html += "<style>body{font-family:Arial,sans-serif;margin:30px;}"
    html += "table{border-collapse:collapse;width:100%;margin-bottom:12px;}"
    html += "th,td{border:1px solid #999;padding:6px 10px;text-align:left;font-size:12px;}"
    html += "th{background:#2F5496;color:#fff;}h1{color:#2F5496;font-size:20px;}"
    html += ".truck-header{background:#f0f0f0;padding:8px;font-weight:700;font-size:14px;border:1px solid #ccc;}"
    html += "@media print{body{margin:10px;}}</style></head><body>"
    html += "<h1>Route Sheet — " + str(date or "Today") + "</h1>"

    for truck_name, stops in sorted(trucks.items()):
        html += '<div class="truck-header">🚛 ' + truck_name + " — " + str(len(stops)) + " stops</div>"
        html += "<table><tr><th>#</th><th>Customer</th><th>Address</th><th>Contact</th>"
        html += "<th>Items</th><th>Weight</th><th>Data</th><th>RED/R2</th><th>AoR</th><th>CoD</th><th>Notes</th></tr>"
        for i, s in enumerate(stops, 1):
            addr = str(s.get("address_line", "")) + ", " + str(s.get("city", "")) + ", " + str(s.get("state", "")) + " " + str(s.get("zip_code", ""))
            html += "<tr><td>" + str(i) + "</td><td>" + str(s.get("company_name", "")) + "</td>"
            html += "<td>" + addr + "</td><td>" + str(s.get("contact_name", "")) + "<br>" + str(s.get("contact_phone", "")) + "</td>"
            html += "<td>" + str(s.get("estimated_items", "")) + "</td><td>" + str(s.get("estimated_weight", "")) + "</td>"
            html += "<td>" + str(s.get("data_status", "")) + "</td><td>" + str(s.get("red_r2", "")) + "</td>"
            html += "<td>" + ("✓" if s.get("needs_aor") else "") + "</td>"
            html += "<td>" + ("✓" if s.get("needs_cod") else "") + "</td>"
            html += "<td>" + str(s.get("notes", "")) + "</td></tr>"
        html += "</table>"

    if unassigned:
        html += "<h2>Unassigned</h2><table><tr><th>Customer</th><th>Address</th><th>Notes</th></tr>"
        for s in unassigned:
            html += "<tr><td>" + str(s.get("company_name", "")) + "</td>"
            html += "<td>" + str(s.get("address_line", "")) + "</td>"
            html += "<td>" + str(s.get("notes", "")) + "</td></tr>"
        html += "</table>"

    html += "</body></html>"
    frappe.local.response["type"] = "html"
    frappe.local.response["page_content"] = html


@frappe.whitelist()
def print_green_sheet(date=None):
    """Generate Green Sheet printout.
    ⚠️ TODO: Green Sheet template needs to be filled in with the actual form layout."""
    if not date:
        date = today()

    filters = {"pickup_date": date} if date else {}
    pickups = frappe.get_list("Scheduled Pickup",
        filters=filters,
        fields=["name", "pickup_date", "pickup_type", "company_name",
                "contact_name", "contact_phone", "address_line", "city", "state", "zip_code",
                "estimated_items", "estimated_weight", "load_contents",
                "data_status", "red_r2", "needs_aor", "needs_cod", "notes"],
        order_by="truck asc, stop_order asc",
        limit_page_length=200,
    )

    html = "<!DOCTYPE html><html><head><title>Green Sheet</title>"
    html += "<style>body{font-family:Arial,sans-serif;margin:30px;}"
    html += "h1{color:#2E7D32;font-size:20px;border-bottom:2px solid #2E7D32;padding-bottom:4px;}"
    html += ".field-row{display:flex;margin:4px 0;}.field-label{font-weight:700;width:180px;}"
    html += ".field-value{flex:1;border-bottom:1px dotted #ccc;min-height:20px;}"
    html += ".pickup-block{border:1px solid #2E7D32;border-radius:6px;padding:12px;margin:12px 0;}"
    html += ".placeholder{color:#999;font-style:italic;font-size:11px;}"
    html += "@media print{body{margin:10px;}}</style></head><body>"
    html += "<h1>🟢 Green Sheet — " + str(date or "Today") + "</h1>"
    html += '<p class="placeholder">⚠️ This is a PLACEHOLDER template. Replace with actual Green Sheet layout once the form spec is provided.</p>'

    for p in pickups:
        html += '<div class="pickup-block">'
        fields = [
            ("Customer", p.get("company_name", "")),
            ("Contact", p.get("contact_name", "")),
            ("Phone", p.get("contact_phone", "")),
            ("Address", str(p.get("address_line", "")) + ", " + str(p.get("city", "")) + ", " + str(p.get("state", "")) + " " + str(p.get("zip_code", ""))),
            ("Est. Items", p.get("estimated_items", "")),
            ("Est. Weight", p.get("estimated_weight", "")),
            ("Load Contents", p.get("load_contents", "")),
            ("Data Status", p.get("data_status", "")),
            ("RED/R2", p.get("red_r2", "")),
            ("Needs AoR", "✓" if p.get("needs_aor") else ""),
            ("Needs CoD", "✓" if p.get("needs_cod") else ""),
            ("Notes", p.get("notes", "")),
        ]
        for label, val in fields:
            html += '<div class="field-row"><div class="field-label">' + label + ':</div><div class="field-value">' + str(val) + '</div></div>'
        html += '</div>'

    html += "</body></html>"
    frappe.local.response["type"] = "html"
    frappe.local.response["page_content"] = html


@frappe.whitelist()
def print_labels(date=None):
    """Generate printable labels."""
    if not date:
        date = today()

    filters = {"pickup_date": date} if date else {}
    pickups = frappe.get_list("Scheduled Pickup",
        filters=filters,
        fields=["name", "company_name", "pickup_date", "num_labels", "data_status", "red_r2"],
        limit_page_length=200,
    )

    html = "<!DOCTYPE html><html><head><title>Labels</title>"
    html += "<style>body{font-family:Arial,sans-serif;margin:10px;}"
    html += ".label{border:2px solid #000;width:3in;height:1.5in;display:inline-block;margin:4px;padding:6px;font-size:11px;page-break-inside:avoid;}"
    html += ".label-co{font-weight:700;font-size:14px;}.label-date{font-size:10px;color:#666;}"
    html += ".label-status{font-size:10px;margin-top:2px;}"
    html += "@media print{body{margin:0;}}</style></head><body>"

    for p in pickups:
        n = p.get("num_labels") or 1
        for _ in range(n):
            html += '<div class="label">'
            html += '<div class="label-co">' + str(p.get("company_name", "")) + '</div>'
            html += '<div class="label-date">' + str(p.get("pickup_date", "")) + '</div>'
            html += '<div class="label-status">' + str(p.get("data_status", "")) + " | " + str(p.get("red_r2", "")) + '</div>'
            html += '</div>'

    html += "</body></html>"
    frappe.local.response["type"] = "html"
    frappe.local.response["page_content"] = html