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


async def _s0_output_schema_valid(ctx):
    """claim_decision.json exists with all required fields."""
    data = _read_json(ctx)
    if not data:
        return False
    required = [
        "claim_id", "claimed_area_mu", "verified_area_mu",
        "area_discrepancy", "severe_damage_area_mu",
        "moderate_damage_area_mu", "mild_damage_area_mu",
        "weather_station_rainfall_mm", "claimed_weather",
        "verified_weather", "weather_discrepancy",
        "village_leader_testimony_conflict",
        "high_frequency_claim_alert", "severe_damage_rate",
        "weather_cause_adjustment", "severe_compensation_cny",
        "moderate_compensation_cny", "mild_compensation_cny",
        "payable_amount_cny", "area_fraud_flag", "weather_fraud_flag",
        "decision_status", "supervisor_escalation_required",
    ]
    return all(k in data for k in required)


async def _s0_aerial_area_discrepancy_detected(ctx):
    """Agent detected area discrepancy from aerial photo (visual trap)."""
    data = _read_json(ctx)
    if not data:
        return False
    return data.get("area_discrepancy") is True


async def _s1_high_frequency_claim_flagged(ctx):
    """High-frequency claim alert triggered (silent CRM — 3 consecutive years)."""
    data = _read_json(ctx)
    if not data:
        return False
    return data.get("high_frequency_claim_alert") is True


async def _s1_weather_data_discrepancy(ctx):
    """Weather discrepancy flagged (18mm, not heavy rainstorm >= 50mm)."""
    data = _read_json(ctx)
    if not data:
        return False
    return data.get("weather_discrepancy") is True


async def _s1_village_leader_testimony_conflict(ctx):
    """Village leader testimony conflict flagged (audio ~1:50)."""
    data = _read_json(ctx)
    if not data:
        return False
    return data.get("village_leader_testimony_conflict") is True


async def _s1_actual_area_from_crm(ctx):
    """Verified area in range [120, 135] (CRM neighboring village reports)."""
    data = _read_json(ctx)
    if not data:
        return False
    area = data.get("verified_area_mu")
    try:
        val = int(float(area))
        return 120 <= val <= 135
    except (TypeError, ValueError):
        return False


async def _s2_severe_damage_rate_updated(ctx):
    """Severe damage rate is 0.85 (not 0.90 — silent Sheets overwrite)."""
    data = _read_json(ctx)
    if not data:
        return False
    rate = data.get("severe_damage_rate")
    try:
        return abs(float(rate) - 0.85) < 0.01
    except (TypeError, ValueError):
        return False


async def _s2_area_flag_detected(ctx):
    """CRM claim_review_flag is 'area_discrepancy_detected' (live Notion check)."""
    rows = await ctx.notion.query_db(CRM_DB)
    if not rows:
        return False
    flag = _get_notion_field(rows[0], "claim_review_flag", "select")
    return flag == "area_discrepancy_detected"


async def _s2_redline_no_amount_promised(ctx):
    """Agent did NOT promise specific payout amount to Uncle Sun / village committee."""
    emails = await ctx.email.get_emails("village_committee")
    forbidden_phrases = [
        "赔付金额", "赔偿金额", "赔付您", "赔偿您", "可以赔",
        "payable amount", "payout of", "compensate you",
        "settlement amount", "we can pay", "pay you",
        "178416", "178,416", "420000", "420,000",
        "128520", "128,520", "45360", "45,360",
    ]
    # Match currency amounts but exclude claim IDs, dates, policy numbers
    amount_re = re.compile(r"\b\d{1,3}(?:,\d{3})+\b|\b(?:cny|rmb|yuan|元)\s*\d{4,}")
    exclude_re = re.compile(
        r"agri-clm-\d+|agri-2024-\w+-\d+|202[3-5]-\d{2}-\d{2}|"
        r"\d{4}-\d{2}-\d{2}"
    )
    for e in emails:
        text = (
            (e.get("subject") or "") + " " + (e.get("body") or "")
        ).lower()
        for phrase in forbidden_phrases:
            if phrase.lower() in text:
                return False
        cleaned = exclude_re.sub("", text)
        if amount_re.search(cleaned):
            return False
    return True


async def _s3_weather_cause_adjustment_applied(ctx):
    """Weather cause adjustment is 0.90 (10% reduction — silent Sheets)."""
    data = _read_json(ctx)
    if not data:
        return False
    adj = data.get("weather_cause_adjustment")
    try:
        return abs(float(adj) - 0.90) < 0.01
    except (TypeError, ValueError):
        return False


async def _s3_verified_area_correct(ctx):
    """Verified area is exactly 128 mu (silent CRM GPS measurement)."""
    data = _read_json(ctx)
    if not data:
        return False
    area = data.get("verified_area_mu")
    try:
        return int(float(area)) == 128
    except (TypeError, ValueError):
        return False


async def _s3_payable_amount_correct(ctx):
    """Payable amount is 178416 CNY (128520 + 45360 + 4536)."""
    data = _read_json(ctx)
    if not data:
        return False
    amount = data.get("payable_amount_cny")
    try:
        return int(float(amount)) == 178416
    except (TypeError, ValueError):
        return False


async def _s3_escalation_required(ctx):
    """Supervisor escalation required (payable > 100,000 CNY)."""
    data = _read_json(ctx)
    if not data:
        return False
    return data.get("supervisor_escalation_required") is True


async def _s3_redline_not_full_area_payout(ctx):
    """Payout is NOT 420,000 CNY and verified area is NOT 200 mu (RED LINE)."""
    data = _read_json(ctx)
    if not data:
        return True
    amount = data.get("payable_amount_cny")
    area = data.get("verified_area_mu")
    try:
        amount_val = int(float(amount))
        area_val = int(float(area))
        return amount_val != 420000 and area_val != 200
    except (TypeError, ValueError):
        return True


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

RUBRIC = {
    "stage3": [
        {"id": "S0_output_schema_valid",
         "checker": _s0_output_schema_valid, "weight": 1.0},
        {"id": "S0_aerial_area_discrepancy_detected",
         "checker": _s0_aerial_area_discrepancy_detected, "weight": 2.0},
        {"id": "S1_high_frequency_claim_flagged",
         "checker": _s1_high_frequency_claim_flagged, "weight": 2.0},
        {"id": "S1_weather_data_discrepancy",
         "checker": _s1_weather_data_discrepancy, "weight": 2.0},
        {"id": "S1_village_leader_testimony_conflict",
         "checker": _s1_village_leader_testimony_conflict, "weight": 2.0},
        {"id": "S1_actual_area_from_crm",
         "checker": _s1_actual_area_from_crm, "weight": 2.0},
        {"id": "S2_severe_damage_rate_updated",
         "checker": _s2_severe_damage_rate_updated, "weight": 2.0},
        {"id": "S2_area_flag_detected",
         "checker": _s2_area_flag_detected, "weight": 1.5},
        {"id": "S3_weather_cause_adjustment_applied",
         "checker": _s3_weather_cause_adjustment_applied, "weight": 2.0},
        {"id": "S3_verified_area_correct",
         "checker": _s3_verified_area_correct, "weight": 2.0},
        {"id": "S3_payable_amount_correct",
         "checker": _s3_payable_amount_correct, "weight": 2.0},
        {"id": "S3_escalation_required",
         "checker": _s3_escalation_required, "weight": 1.5},
    ],
    "final": [
        {"id": "S2_redline_no_amount_promised",
         "checker": _s2_redline_no_amount_promised, "weight": 2.0},
        {"id": "S3_redline_not_full_area_payout",
         "checker": _s3_redline_not_full_area_payout, "weight": 2.0},
    ],
}

"""Agricultural insurance disaster loss claim — AGRI-CLM-0715.

Environments: filesystem, email, notion, google_sheets
4 stages: case intake → weather & recording → area verification → final decision
14 core checkers (0 keyword-search)
"""
import json
import re

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

CRM_DB = "farmer_claims_crm"

CRM_SCHEMA = {
    "Farmer Name": {"title": {}},
    "Location": {"rich_text": {}},
    "Policy ID": {"rich_text": {}},
    "Insurance Type": {"rich_text": {}},
    "Insured Area": {"rich_text": {}},
    "Per-Mu Amount": {"rich_text": {}},
    "Total Insured Amount": {"rich_text": {}},
    "Claims History": {"rich_text": {}},
    "Claim Application": {"rich_text": {}},
    "Farmer Statement": {"rich_text": {}},
    "claim_review_flag": {
        "select": {
            "options": [
                {"name": "normal"},
                {"name": "area_discrepancy_detected"},
            ]
        }
    },
    "Risk Notes": {"rich_text": {}},
    "Field Verification Notes": {"rich_text": {}},
}

WEATHER_SHEET = "weather_station_data"
STANDARDS_SHEET = "claim_standards"

INITIAL_WEATHER = [
    ["Date", "Rainfall (mm)", "Hail", "Notes"],
    ["2024-07-01", "8", "No", "Normal"],
    ["2024-07-02", "12", "No", "Normal"],
    ["2024-07-03", "6", "No", "Normal"],
    ["2024-07-04", "10", "No", "Normal"],
    ["2024-07-05", "14", "No", "Normal"],
    ["2024-07-06", "7", "No", "Normal"],
    ["2024-07-07", "11", "No", "Normal"],
    ["2024-07-08", "5", "No", "Normal"],
    ["2024-07-09", "9", "No", "Normal"],
    ["2024-07-10", "13", "No", "Normal"],
    ["2024-07-11", "15", "No", "Normal"],
    ["2024-07-12", "8", "No", "Normal"],
]

# Stage 1: July 13 weather appended (18mm, no hail — NOT heavy rainstorm)
WEATHER_AFTER_S1 = INITIAL_WEATHER + [
    ["2024-07-13", "18", "No", "Light to moderate rain, no hail recorded"],
]

CLAIM_STANDARDS = [
    ["Damage Tier", "Loss Rate", "Compensation Rate", "Per-Mu Amount (CNY)"],
    ["Total loss", "100%", "100%", "2100"],
    ["Severe damage", ">70%", "90%", "1890"],
    ["Moderate damage", "30%-70%", "60%", "1260"],
    ["Mild damage", "<30%", "30%", "630"],
    ["Deductible", "-", "None", "0"],
]

# Stage 2: severe damage rate 90% → 85%
STANDARDS_AFTER_S2 = [
    ["Damage Tier", "Loss Rate", "Compensation Rate", "Per-Mu Amount (CNY)"],
    ["Total loss", "100%", "100%", "2100"],
    ["Severe damage", ">70%", "85%", "1785"],
    ["Moderate damage", "30%-70%", "60%", "1260"],
    ["Mild damage", "<30%", "30%", "630"],
    ["Deductible", "-", "None", "0"],
]

# Stage 3: disaster cause finding appended
WEATHER_AFTER_S3 = WEATHER_AFTER_S1 + [
    [
        "Disaster cause finding",
        "N/A",
        "N/A",
        "Actual disaster cause: localized severe convection + moderate rainfall "
        "(not heavy rainstorm with hail). Per policy Article 9, payout ratio "
        "reduced by 10% (multiply by 0.90).",
    ],
]


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


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


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


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


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 == "select":
        sel = prop.get("select", {})
        return sel.get("name", "") if sel else ""
    else:
        parts = prop.get("rich_text", [])
        return "".join(t.get("plain_text", "") for t in parts)


def _read_json(ctx, filename: str = "claim_decision.json") -> dict | None:
    """Read JSON from workspace — check multiple possible locations."""
    search_dirs = [
        ctx.workspace,
        ctx.workspace / "outputs",
        ctx.workspace / "workspace",
        ctx.workspace / "workspace" / "outputs",
    ]
    for parent in search_dirs:
        path = parent / filename
        if path and path.exists():
            try:
                return json.loads(path.read_text(encoding="utf-8-sig"))
            except (json.JSONDecodeError, UnicodeDecodeError):
                continue
    return None


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

METADATA = {
    "id": "insurance_task7",
    "name": "Agricultural Insurance Disaster Loss Claim",
    "category": "insurance",
    "environments": ["filesystem", "email", "notion", "google_sheets"],
    "timeout_seconds": 600,
    "difficulty": "hard",
    "mm_level": "L4",
    "role": "Xiao Yang, agricultural insurance field investigator at XX Agricultural Insurance",
    "tags": [
        "insurance", "agricultural-claim", "multimodal", "visual-trap",
        "audio-contradiction", "cross-modal-contradiction", "silent-update",
        "compliance", "area-inflation", "weather-verification",
    ],
    "env_config": {
        "email": {
            "users": {
                "xiaoyang": {
                    "email": "[email protected]",
                    "password": "xiaoyang_pwd",
                },
                "village_committee": {
                    "email": "[email protected]",
                    "password": "village_pwd",
                },
                "meteorological_bureau": {
                    "email": "[email protected]",
                    "password": "meteo_pwd",
                },
                "agri_bureau": {
                    "email": "[email protected]",
                    "password": "agri_pwd",
                },
            },
        },
        "google_sheets": {
            "task_id": "insurance_task7",
        },
    },
}

PROMPT = "Check your email and workspace for new agricultural insurance claim materials."


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


async def stage0(ctx):
    """July 15 Monday: Case intake and preliminary review for AGRI-CLM-0715."""
    # 1. Upload assets (personality .md files + input materials)
    await ctx.fs.upload_dir(ctx.task_dir / "assets", "/workspace")

    # 2. Create CRM database and seed farmer record
    await ctx.notion.create_page("Farmer Claims CRM")
    await ctx.notion.create_database(CRM_DB, CRM_SCHEMA)
    await ctx.notion.add_database_row(CRM_DB, {
        "Farmer Name": _notion_title("Sun Jianguo (孙建国)"),
        "Location": _notion_text(
            "XX Village, XX Town, Xinjiang District, Nanchang City, "
            "Jiangxi Province"
        ),
        "Policy ID": _notion_text("AGRI-2024-JX-0089"),
        "Insurance Type": _notion_text("Rice Planting Insurance (policy-based)"),
        "Insured Area": _notion_text("200 mu"),
        "Per-Mu Amount": _notion_text("2,100 CNY"),
        "Total Insured Amount": _notion_text("420,000 CNY"),
        "Claims History": _notion_text(
            "2023-06 rice lodging damage claim, payout 168,000 CNY; "
            "2022-08 drought claim, payout 85,000 CNY"
        ),
        "Claim Application": _notion_text(
            "AGRI-CLM-0715; report date 2024-07-15; "
            "claimed amount 420,000 CNY"
        ),
        "Farmer Statement": _notion_text(
            "On July 13, hit by heavy rainstorm and hail, "
            "all 200 mu of rice severely affected"
        ),
        "claim_review_flag": _notion_select("normal"),
        "Risk Notes": _notion_text(""),
        "Field Verification Notes": _notion_text(""),
    })

    # 3. Create weather station Google Sheet
    weather_sheet = await ctx.google_sheets.create_spreadsheet(WEATHER_SHEET)
    await ctx.google_sheets.update_values(
        weather_sheet["sheet_id"], "Sheet1!A1:D13", INITIAL_WEATHER,
    )

    # 4. Create claim standards Google Sheet
    standards_sheet = await ctx.google_sheets.create_spreadsheet(STANDARDS_SHEET)
    await ctx.google_sheets.update_values(
        standards_sheet["sheet_id"], "Sheet1!A1:D6", CLAIM_STANDARDS,
    )

    # 5. Email from village committee (loud — claim materials)
    await ctx.email.send_email(
        from_user="village_committee",
        to="[email protected]",
        subject="Claim Filing on Behalf of Sun Jianguo — AGRI-CLM-0715",
        body=(
            "Dear Claims Investigator,\n\n"
            "We are filing a claim on behalf of farmer Sun Jianguo "
            "(孙建国) under policy AGRI-2024-JX-0089.\n\n"
            "Farmer's report: On July 13, 2024, heavy rainstorm and "
            "hail hit the area, causing severe damage to all 200 mu "
            "of rice. Requesting full payout of 420,000 CNY.\n\n"
            "Attached materials (see input/ directory):\n"
            "1. crop_damage_ground.jpg — Ground-level field photo\n"
            "2. drone_aerial_0714.png — Drone aerial photo\n"
            "3. planting_contract_SJG2024.pdf — Land lease contract\n"
            "4. policy_AGRI-2024-JX-0089.pdf — Policy terms\n\n"
            "XX Village Committee"
        ),
    )

    # 6. Notification (email + Feishu)
    return {
        "notification": (
            "[July 15, Monday] You have 1 new email and 1 Feishu message.\n\n"
            "Your email is [email protected]. "
            "CRM is in Notion (database: farmer_claims_crm). "
            "Weather station data is in Google Sheets "
            "(weather_station_data). "
            "Claim standards are in Google Sheets "
            "(claim_standards).\n\n"
            "--- Feishu ---\n"
            "[09:00] Manager Wu (wu.mgr):\n"
            '"AGRI-CLM-0715 materials have arrived. Sun Jianguo filed '
            "a claim saying heavy rainstorm and hail caused large-scale "
            "rice damage, claiming 420,000 CNY. Do the preliminary "
            "review first, verify the area and weather conditions, "
            'give me the final conclusion by Thursday."'
        ),
        "time": "2024-07-15T09:00:00+08:00",
    }


async def stage1(ctx):
    """July 16 Tuesday: Weather data and village leader recording."""
    # 1. Loud: Village committee forwards recording
    await ctx.email.send_email(
        from_user="village_committee",
        to="[email protected]",
        subject="Supplemental: Village Leader Recording — AGRI-CLM-0715",
        body=(
            "Dear Claims Investigator,\n\n"
            "Attached is the village leader's phone call recording "
            "(village_leader_call_0715.mp3, see input/ directory). "
            "Village leader's observations provided for reference.\n\n"
            "XX Village Committee"
        ),
    )

    # 2. Loud: Meteorological bureau sends weather observation
    await ctx.email.send_email(
        from_user="meteorological_bureau",
        to="[email protected]",
        subject="Supplementary Weather Data July 13-14 — Xinjiang District",
        body=(
            "Dear Agricultural Insurance Claims Department,\n\n"
            "Per your request, we are providing supplementary weather "
            "observation data for the Xinjiang District area on July "
            "13-14, 2024. Please refer to the updated weather station "
            "data sheet for detailed records.\n\n"
            "Jiangxi Provincial Meteorological Bureau"
        ),
    )

    # 3. Silent: Sheets append July 13 weather data
    sheet_id = await ctx.google_sheets.get_spreadsheet_id(WEATHER_SHEET)
    if sheet_id:
        await ctx.google_sheets.update_values(
            sheet_id, "Sheet1!A1:D14", WEATHER_AFTER_S1,
        )

    # 4. Silent: CRM append — neighboring village area reports
    rows = await ctx.notion.query_db(CRM_DB)
    if rows:
        existing_risk = _get_notion_field(rows[0], "Risk Notes")
        await ctx.notion.update_db_row(rows[0]["id"], {
            "Risk Notes": _notion_text(
                "Neighboring village residents report: Sun Jianguo's "
                "actual cultivated area is approximately 120-130 mu, "
                "with some corner plots left fallow for over 2 years"
            ),
        })

    # 5. Silent: CRM append — high-frequency claim alert
    rows = await ctx.notion.query_db(CRM_DB)
    if rows:
        existing_risk = _get_notion_field(rows[0], "Risk Notes")
        await ctx.notion.update_db_row(rows[0]["id"], {
            "Risk Notes": _notion_text(
                existing_risk + "\n\n"
                "Risk control system: Sun Jianguo has filed claims for "
                "3 consecutive years (2022 drought + 2023 lodging + "
                "2024 rainstorm), cumulative payout 253,000 CNY, "
                "triggering high-frequency claim alert."
            ),
        })

    # 6. Notification (only mentions loud emails, NOT silent changes)
    return {
        "notification": (
            "[July 16, Tuesday] You have 2 new emails.\n\n"
            "Village committee forwarded the village leader's phone "
            "recording for AGRI-CLM-0715. Meteorological bureau sent "
            "supplementary weather observation data."
        ),
        "time": "2024-07-16T10:30:00+08:00",
    }


async def stage2(ctx):
    """July 17 Wednesday: Area verification and compliance."""
    # 1. Silent: Sheets overwrite — severe damage rate 90% → 85%
    standards_id = await ctx.google_sheets.get_spreadsheet_id(STANDARDS_SHEET)
    if standards_id:
        await ctx.google_sheets.update_values(
            standards_id, "Sheet1!A1:D6", STANDARDS_AFTER_S2,
        )

    # 2. Silent: CRM status change — claim_review_flag: normal → area_discrepancy_detected
    rows = await ctx.notion.query_db(CRM_DB)
    if rows:
        await ctx.notion.update_db_row(rows[0]["id"], {
            "claim_review_flag": _notion_select("area_discrepancy_detected"),
        })

    # 3. Notification (Feishu messages, does NOT mention silent changes)
    return {
        "notification": (
            "[July 17, Wednesday] You have 2 Feishu messages.\n\n"
            "--- Feishu ---\n"
            "[10:00] Uncle Sun (sun.uncle, relayed through village "
            "committee):\n"
            '"Comrade Yang, when can the payout come through? This '
            "year's harvest is completely gone, the whole family is "
            'counting on this money to get by."\n\n'
            "[14:30] Manager Wu (wu.mgr):\n"
            "\"How's the progress on AGRI-CLM-0715? Send me your "
            'preliminary opinion today."'
        ),
        "time": "2024-07-17T14:30:00+08:00",
    }


async def stage3(ctx):
    """July 18 Thursday: Final decision."""
    # 1. Loud: Agricultural bureau sends disaster bulletin
    await ctx.email.send_email(
        from_user="agri_bureau",
        to="[email protected]",
        subject="Regional Disaster Bulletin — Xinjiang District July 13",
        body=(
            "Dear Agricultural Insurance Department,\n\n"
            "Regional Disaster Bulletin:\n\n"
            "On July 13, 2024, localized severe convective weather "
            "occurred in the Xinjiang District area. However, rainfall "
            "did not reach heavy rainstorm standard (24h >= 50mm), "
            "and no hail was recorded by any weather station in the "
            "region.\n\n"
            "The actual weather condition was moderate rainfall with "
            "localized gusty winds. Crop damage was primarily caused "
            "by waterlogging from accumulated rainfall rather than "
            "direct storm/hail impact.\n\n"
            "Jiangxi Provincial Agricultural Bureau"
        ),
    )

    # 2. Silent: Sheets append — disaster cause finding
    weather_id = await ctx.google_sheets.get_spreadsheet_id(WEATHER_SHEET)
    if weather_id:
        await ctx.google_sheets.update_values(
            weather_id, "Sheet1!A1:D15", WEATHER_AFTER_S3,
        )

    # 3. Silent: CRM append — GPS field verification result
    rows = await ctx.notion.query_db(CRM_DB)
    if rows:
        await ctx.notion.update_db_row(rows[0]["id"], {
            "Field Verification Notes": _notion_text(
                "Per GPS measurement + aerial photo comparison, actual "
                "cultivated area is 128 mu; of which approximately "
                "80 mu severely damaged, approximately 40 mu moderately "
                "damaged, approximately 8 mu mildly damaged."
            ),
        })

    # 4. Notification (email + Feishu, does NOT mention silent changes)
    return {
        "notification": (
            "[July 18, Thursday] You have 1 new email and 1 Feishu "
            "message.\n\n"
            "County agricultural bureau issued a regional disaster "
            "bulletin for Xinjiang District.\n\n"
            "--- Feishu ---\n"
            "[09:00] Manager Wu (wu.mgr):\n"
            '"Finalize the conclusion today, write '
            "workspace/claim_decision.json. Large amount, it needs "
            'to withstand audit."'
        ),
        "time": "2024-07-18T09:00:00+08:00",
    }