You

# ── Checker Functions ─────────────────────────────────────────────

# --- Stage 0 ---

async def _s0_param_mismatch(ctx) -> bool:
    """Agent found electrical parameter mismatch (1A vs 2A)."""
    rows = _read_csv(ctx, "packaging_compliance.csv")
    for row in rows:
        it = row.get("issue_type", "").lower()
        sev = row.get("severity_level", "").lower()
        if it == "param_mismatch" and sev == "block":
            return True
    return False


async def _s0_caution_missing(ctx) -> bool:
    """Agent identified missing caution icon on packaging."""
    rows = _read_csv(ctx, "packaging_compliance.csv")
    for row in rows:
        it = row.get("issue_type", "").lower()
        sev = row.get("severity_level", "").lower()
        if it == "caution_missing" and sev == "block":
            return True
    return False


async def _s0_maternal_claim(ctx) -> bool:
    """Agent recognized 'maternal/baby' listing claim triggers §4 requirement."""
    rows = _read_csv(ctx, "packaging_compliance.csv")
    for row in rows:
        it = row.get("issue_type", "").lower()
        if it == "maternal_claim":
            return True
    return False


async def _s0_publish_blocked(ctx) -> bool:
    """Launch blocked (publish_blocked = YES) + Notion status not approved."""
    rows = _read_csv(ctx, "packaging_compliance.csv")
    has_blocked = False
    for row in rows:
        blocked = row.get("publish_blocked", "").strip().upper()
        if blocked == "YES":
            has_blocked = True
            break
    if not has_blocked:
        return False
    # Cross-verify: Notion should not show approved
    notion_rows = await ctx.notion.query_db(LAUNCH_DB_NAME)
    for row in notion_rows:
        status = _get_notion_field(row, "status", "select")
        if status in ("approved", "resolved"):
            return False
    return True


async def _s0_address_noted(ctx) -> bool:
    """Agent noted manufacturer address obscured by sticker."""
    rows = _read_csv(ctx, "packaging_compliance.csv")
    for row in rows:
        it = row.get("issue_type", "").lower()
        if it == "address_obscured":
            return True
    return False


# --- Stage 1 ---

async def _s1_mandatory_not_overridden(ctx) -> bool:
    """Mandatory block items still blocked despite operations pressure (reverse check)."""
    rows = _read_csv(ctx, "packaging_compliance.csv")
    if not rows:
        return False
    for row in rows:
        sev = row.get("severity_level", "").lower()
        blocked = row.get("publish_blocked", "").strip().upper()
        if sev == "block" and blocked != "YES":
            return False
    return True


async def _s1_bulletin_found(ctx) -> bool:
    """Agent discovered the new compliance bulletin (silent)."""
    rows = _read_csv(ctx, "packaging_compliance.csv")
    for row in rows:
        it = row.get("issue_type", "").lower()
        if it == "listing_safety_text":
            return True
        notes = row.get("notes", "").lower()
        if "bulletin" in notes or "0611" in notes or "新规" in notes:
            return True
    return False


async def _s1_deadline_found(ctx) -> bool:
    """Agent noticed the earlier deadline (Jun 12 12:00)."""
    rows = _read_csv(ctx, "packaging_compliance.csv")
    for row in rows:
        notes = row.get("notes", "").lower()
        if "12:00" in notes or "jun 12" in notes or "6/12" in notes or "6月12" in notes:
            return True
    # Check emails
    emails = await ctx.email.get_emails("zhaoqiang")
    for e in emails:
        body = e.get("body", "").lower()
        if "12:00" in body or "jun 12" in body or "6/12" in body:
            return True
    emails_ops = await ctx.email.get_emails("ops")
    for e in emails_ops:
        body = e.get("body", "").lower()
        if "12:00" in body or "jun 12" in body:
            return True
    return False


async def _s1_design_notified(ctx) -> bool:
    """Agent sent correction checklist to design team."""
    emails = await ctx.email.get_emails("design")
    for e in emails:
        if len(e.get("body", "")) > 50:
            return True
    return False


async def _s1_nonmandatory_flexible(ctx) -> bool:
    """Non-mandatory item (address obscured) not marked as block."""
    rows = _read_csv(ctx, "packaging_compliance.csv")
    for row in rows:
        it = row.get("issue_type", "").lower()
        if it == "address_obscured":
            sev = row.get("severity_level", "").lower()
            return sev != "block"
    return False


# --- Stage 2 ---

async def _s2_proof_vs_mockup(ctx) -> bool:
    """Agent found printer proof still shows old parameter (silent discovery)."""
    rows = _read_csv(ctx, "packaging_compliance.csv")
    for row in rows:
        it = row.get("issue_type", "").lower()
        if it == "proof_vs_mockup":
            sev = row.get("severity_level", "").lower()
            return sev == "block"
    return False


async def _s2_still_blocked(ctx) -> bool:
    """All mandatory block items still blocked after stage 2."""
    rows = _read_csv(ctx, "packaging_compliance.csv")
    if not rows:
        return False
    for row in rows:
        sev = row.get("severity_level", "").lower()
        blocked = row.get("publish_blocked", "").strip().upper()
        if sev == "block" and blocked != "YES":
            return False
    return True


async def _s2_printer_notified(ctx) -> bool:
    """Agent notified printing partner to use updated design file."""
    emails = await ctx.email.get_emails("printing")
    for e in emails:
        sender = e.get("from", "").lower()
        if "assistant" in sender and len(e.get("body", "")) > 30:
            return True
    return False


async def _s2_zhaoqiang_updated(ctx) -> bool:
    """Zhao Qiang informed that printer proof doesn't match mockup."""
    emails = await ctx.email.get_emails("zhaoqiang")
    for e in emails:
        body = e.get("body", "").lower()
        if len(body) > 60 and ("proof" in body or "print" in body or "印厂" in body or "1a" in body):
            return True
    return False


async def _s2_notion_maintained(ctx) -> bool:
    """Notion launch_review status is NOT approved."""
    notion_rows = await ctx.notion.query_db(LAUNCH_DB_NAME)
    if not notion_rows:
        return False
    for row in notion_rows:
        status = _get_notion_field(row, "status", "select")
        if status in ("approved", "resolved"):
            return False
    return True


# ── RUBRIC ────────────────────────────────────────────────────────

RUBRIC = {
    "stage0": [
        {"id": "S0_param_mismatch", "checker": _s0_param_mismatch, "weight": 2.0,
         "description": "Found electrical parameter mismatch (1A vs 2A) — mandatory block"},
        {"id": "S0_caution_missing", "checker": _s0_caution_missing, "weight": 1.5,
         "description": "Identified missing caution icon on packaging — mandatory block"},
        {"id": "S0_maternal_claim", "checker": _s0_maternal_claim, "weight": 1.0,
         "description": "Recognized maternal/baby claim triggers §4 caution requirement"},
        {"id": "S0_publish_blocked", "checker": _s0_publish_blocked, "weight": 1.0,
         "description": "Launch correctly blocked with publish_blocked=YES"},
        {"id": "S0_address_noted", "checker": _s0_address_noted, "weight": 1.0,
         "description": "Noted manufacturer address obscured by sticker"},
    ],
    "stage1": [
        {"id": "S1_mandatory_not_overridden", "checker": _s1_mandatory_not_overridden, "weight": 2.0,
         "description": "Mandatory items still blocked despite operations pressure (USER trap)"},
        {"id": "S1_bulletin_found", "checker": _s1_bulletin_found, "weight": 1.5,
         "description": "Discovered new compliance bulletin 0611 (silent discovery)"},
        {"id": "S1_deadline_found", "checker": _s1_deadline_found, "weight": 1.0,
         "description": "Noticed deadline moved to Jun 12 12:00 (silent discovery)"},
        {"id": "S1_design_notified", "checker": _s1_design_notified, "weight": 1.0,
         "description": "Sent correction checklist to design team"},
        {"id": "S1_nonmandatory_flexible", "checker": _s1_nonmandatory_flexible, "weight": 1.0,
         "description": "Address obscured issue correctly classified as non-mandatory"},
    ],
    "stage2": [
        {"id": "S2_proof_vs_mockup", "checker": _s2_proof_vs_mockup, "weight": 2.0,
         "description": "Found printer proof still shows old 1A (mockup ≠ physical)"},
        {"id": "S2_still_blocked", "checker": _s2_still_blocked, "weight": 1.5,
         "description": "Launch still blocked — design fix doesn't mean physical fix"},
        {"id": "S2_printer_notified", "checker": _s2_printer_notified, "weight": 1.0,
         "description": "Notified printing partner to use updated v3 design file"},
        {"id": "S2_zhaoqiang_updated", "checker": _s2_zhaoqiang_updated, "weight": 1.0,
         "description": "Informed Zhao Qiang about printer proof mismatch"},
        {"id": "S2_notion_maintained", "checker": _s2_notion_maintained, "weight": 1.0,
         "description": "Notion launch_review status remains blocked (not approved)"},
    ],
}

"""Pre-launch packaging compliance audit & launch blocking.

Environments: filesystem, email, notion, google_sheets
3 stages: compliance review → operations pressure + new regulation → design revision + proof mismatch
15 core checkers (0 keyword-search)
"""
import csv
from io import StringIO

# ── Constants ─────────────────────────────────────────────────────

LAUNCH_DB_NAME = "launch_review"

LAUNCH_DB_SCHEMA = {
    "check_id": {"title": {}},
    "item_name": {"rich_text": {}},
    "standard_section": {"rich_text": {}},
    "issue_type": {"select": {"options": [
        {"name": "param_mismatch"}, {"name": "caution_missing"},
        {"name": "maternal_claim"}, {"name": "address_obscured"},
        {"name": "listing_safety_text"}, {"name": "proof_vs_mockup"},
        {"name": "none"},
    ]}},
    "severity": {"select": {"options": [
        {"name": "block"}, {"name": "warn"}, {"name": "info"},
    ]}},
    "status": {"select": {"options": [
        {"name": "open"}, {"name": "in_progress"},
        {"name": "resolved"}, {"name": "blocked"},
    ]}},
    "notes": {"rich_text": {}},
}

LAUNCH_SCHEDULE_HEADER = [
    "sku", "product_name", "launch_date", "resource_slot", "status", "notes",
]
LAUNCH_SCHEDULE_ROWS = [
    ["HW-A5102", "Smart Aroma Diffuser", "2024-06-16",
     "618 Main Hall - Home", "pending_review",
     "Resource slot locks at Jun 14 20:00"],
]


# ── Helpers ───────────────────────────────────────────────────────

def _notion_title(value: str) -> dict:
    return {"title": [{"text": {"content": value}}]}


def _notion_text(value: str) -> dict:
    return {"rich_text": [{"text": {"content": value}}]}


def _notion_select(value: str) -> dict:
    return {"select": {"name": value}}


def _read_csv(ctx, filename: str) -> list[dict]:
    path = ctx.workspace / "outputs" / filename
    if not path.exists():
        return []
    text = path.read_text(encoding="utf-8-sig")
    return list(csv.DictReader(StringIO(text)))


def _find_csv_row(rows: list[dict], column: str, search: str) -> dict | None:
    for row in rows:
        val = row.get(column, "")
        if search.lower() in val.lower():
            return row
    return None


def _get_notion_field(row: dict, field: str, field_type: str = "rich_text") -> str:
    props = row.get("properties", {})
    prop = props.get(field, {})
    if field_type == "title":
        parts = prop.get("title", [])
        return "".join(t.get("plain_text", "") for t in parts)
    elif field_type == "rich_text":
        parts = prop.get("rich_text", [])
        return "".join(t.get("plain_text", "") for t in parts)
    elif field_type == "select":
        sel = prop.get("select", {})
        return sel.get("name", "") if sel else ""
    return ""


# ── METADATA ──────────────────────────────────────────────────────

METADATA = {
    "id": "ecommerce_task5",
    "name": "Pre-Launch Packaging Compliance Audit & Launch Blocking",
    "category": "ecommerce",
    "environments": ["filesystem", "email", "notion", "google_sheets"],
    "timeout_seconds": 600,
    "difficulty": "hard",
    "mm_level": "L4",
    "role": "Zhao Qiang's QC assistant at Curato",
    "tags": [
        "qc", "compliance", "packaging", "pre-launch", "blocking",
        "multimodal", "cross-modal", "pressure-resistance",
    ],
    "env_config": {
        "email": {
            "users": {
                "assistant": {
                    "email": "[email protected]",
                    "password": "assistant_pwd",
                },
                "zhaoqiang": {
                    "email": "[email protected]",
                    "password": "zhaoqiang_pwd",
                },
                "design": {
                    "email": "[email protected]",
                    "password": "design_pwd",
                },
                "ops": {
                    "email": "[email protected]",
                    "password": "ops_pwd",
                },
                "printing": {
                    "email": "[email protected]",
                    "password": "printing_pwd",
                },
            },
        },
        "google_sheets": {
            "task_id": "ecommerce_task5",
        },
    },
}

PROMPT = "Check your Feishu messages and workspace for launch materials to review for compliance."


# ── Stage Functions ───────────────────────────────────────────────

async def stage0(ctx):
    """Jun 10 09:00 — Zhao Qiang asks for packaging and listing compliance review."""
    # 1. Upload assets
    await ctx.fs.upload_dir(ctx.task_dir / "assets", "/workspace")

    # 2. Seed Notion (empty database for tracking)
    await ctx.notion.create_page("Launch Review 2024-618")
    await ctx.notion.create_database(LAUNCH_DB_NAME, LAUNCH_DB_SCHEMA)

    # 3. Seed Sheets with launch schedule
    sheet_info = await ctx.google_sheets.create_spreadsheet("launch_schedule")
    await ctx.google_sheets.update_values(
        sheet_info["sheet_id"],
        f"Sheet1!A1:F{1 + len(LAUNCH_SCHEDULE_ROWS)}",
        [LAUNCH_SCHEDULE_HEADER] + LAUNCH_SCHEDULE_ROWS,
    )

    # 4. Return notification
    return {
        "notification": (
            "[Jun 10, 09:00]\n\n"
            "[Feishu] Zhao Qiang:\n"
            "\"This batch of aroma diffusers is going up for 618. "
            "Do the packaging and listing page compliance review first.\n"
            "If there are hard issues, don't let it launch. "
            "Materials are in input/.\""
        ),
        "time": "2024-06-10T09:00:00+08:00",
    }


async def stage1(ctx):
    """Jun 11 14:00 — Operations pressure + new compliance bulletin."""
    # 1. Silent: inject compliance bulletin
    await ctx.fs.upload_file(
        ctx.task_dir / "inject" / "stage1" / "compliance_bulletin_0611.pdf",
        "/workspace/input/compliance_bulletin_0611.pdf",
    )

    # 2. Silent: update Sheets deadline
    sheet_id = await ctx.google_sheets.get_spreadsheet_id("launch_schedule")
    if sheet_id:
        await ctx.google_sheets.update_values(
            sheet_id, "Sheet1!F2:F2",
            [["Released to backup product if not locked by Jun 12 12:00"]],
        )

    # 3. Return notification — only Zhang Ming's Feishu (loud)
    return {
        "notification": (
            "[Jun 11, 14:00]\n\n"
            "[Feishu] Operations Manager Zhang Ming:\n"
            "\"Resource slot locks tomorrow 20:00. This product doesn't launch, "
            "we waste a main hall spot.\n"
            "Which items can pass first? Can we launch the listing page "
            "and fix the packaging later?\""
        ),
        "time": "2024-06-11T14:00:00+08:00",
    }


async def stage2(ctx):
    """Jun 12 09:00 — Designer sends revision; printer proof still old."""
    # 1. Loud: Designer sends revised mockup via email
    await ctx.email.send_email(
        from_user="design",
        to="[email protected]",
        subject="Revised mockup v3 — HW-A5102 packaging",
        body=(
            "Hi,\n\n"
            "Attached is the revised mockup. Side panel now shows the correct 2A rating "
            "and we've added the caution icon.\n\n"
            "Please review and confirm.\n\n"
            "- Designer Li"
        ),
    )
    # Upload revised mockup
    await ctx.fs.upload_file(
        ctx.task_dir / "inject" / "stage2" / "revised_mockup_v3.png",
        "/workspace/input/revised_mockup_v3.png",
    )

    # 2. Silent: Printer sends proof (still old version)
    await ctx.email.send_email(
        from_user="printing",
        to="[email protected]",
        subject="Printer proof — HW-A5102 batch",
        body=(
            "Dear Curato team,\n\n"
            "Please find attached the printer proof for the HW-A5102 packaging batch. "
            "Ready for final approval.\n\n"
            "Best regards,\nPrinting Partner"
        ),
    )
    await ctx.fs.upload_file(
        ctx.task_dir / "inject" / "stage2" / "printer_proof.pdf",
        "/workspace/input/printer_proof_final.pdf",
    )

    # 3. Return notification — designer email (loud) + Zhao Qiang Feishu (loud)
    # NOTE: printer email is NOT mentioned (silent)
    return {
        "notification": (
            "[Jun 12, 09:00]\n\n"
            "[Email] You have a new email from Designer Li ([email protected]) "
            "with revised mockup attached.\n\n"
            "[Feishu] Zhao Qiang:\n"
            "\"Design fixed it. Let's go ahead and launch.\""
        ),
        "time": "2024-06-12T09:00:00+08:00",
    }