Skeleton — surface count and recent items to the LLM

from __future__ import annotations

import logging

from imperal_sdk import Extension

log = logging.getLogger(__name__)

ext = Extension(
    "tasks",
    version="1.0.0",
    display_name="Tasks",
    description="Tasks extension — manage and track your tasks with AI assistance.",
    icon="icon.svg",
    actions_explicit=True,
)

# ── Skeleton: surface pending counts + recent items ───────────────────────────
#
# TTL=30: task counters must stay fresh after chat-initiated writes.
# alert=True: the web-kernel calls skeleton_alert_tasks when the snapshot changes.
# Production ref: tasks/skeleton.py

@ext.skeleton(
    "tasks",
    alert=True,
    ttl=30,
    description="Background: today/overdue/upcoming counts + recent tasks + active projects.",
)
async def skeleton_refresh_tasks(ctx) -> dict:
    """Refresh task counters and recent activity. Idempotent — safe per tick."""
    uid = ctx.user.imperal_id
    if not uid:
        return {"response": {"note": "no user on context"}}

    # Fetch counts and recent items via ctx.http to the extension backend.
    # Replace the URL and params with your own API contract.
    overdue_resp = await ctx.http.get(
        "http://task-service/v1/tasks",
        params={"user_id": uid, "filter": "overdue", "per_page": 200},
    )
    today_resp = await ctx.http.get(
        "http://task-service/v1/tasks",
        params={"user_id": uid, "filter": "due_today", "per_page": 200},
    )
    recent_resp = await ctx.http.get(
        "http://task-service/v1/tasks",
        params={"user_id": uid, "sort_by": "-updated", "per_page": 5},
    )

    def _count(resp) -> int:
        body = resp.json() if resp.ok else {}
        items = body if isinstance(body, list) else body.get("items", [])
        return len(items) if isinstance(items, list) else 0

    def _recent(resp) -> list[dict]:
        body = resp.json() if resp.ok else {}
        items = body if isinstance(body, list) else body.get("items", [])
        if not isinstance(items, list):
            return []
        return [
            {
                "task_id": t.get("id", ""),
                "title":   t.get("title", "")[:80],
                "done":    t.get("done", False),
                "due_date": (t.get("due_date") or "")[:10],
            }
            for t in items
        ]

    return {"response": {
        "overdue_count": _count(overdue_resp),
        "today_count":   _count(today_resp),
        "recent_tasks":  _recent(recent_resp),
    }}


# ── Alert tool: detect new overdue tasks ────────────────────────────────────
#
# Called by the web-kernel when the snapshot diff shows a change.
# Return "" for no alert; return a non-empty string to inject an ambient notice.

@ext.tool(
    "skeleton_alert_tasks",
    description="Alerts: overdue tasks, due-today, sudden spike in backlog.",
)
async def skeleton_alert_tasks(
    ctx,
    old: dict | None = None,
    new: dict | None = None,
) -> dict:
    if not new:
        return {"response": ""}
    overdue     = new.get("overdue_count", 0)
    old_overdue = (old or {}).get("overdue_count", 0)
    today       = new.get("today_count", 0)
    old_today   = (old or {}).get("today_count", 0)
    alerts: list[str] = []
    if overdue > 0 and overdue > old_overdue:
        delta = overdue - old_overdue
        alerts.append(
            f"{delta} new overdue task(s) (now {overdue} total)"
            if delta < overdue
            else f"{overdue} task(s) overdue"
        )
    if today > 0 and today != old_today:
        alerts.append(f"{today} task(s) due today")
    return {"response": " · ".join(alerts)}