deer-flow/backend/app/scheduler/notification_delivery.py

"""Outbox delivery worker for scheduled-task IM notifications (issue #4254).

The completion hook (``ScheduledTaskService._enqueue_run_notifications``)
only writes durable outbox rows; this worker owns the actual IM send.
Execution state and delivery state stay separated: the worker moves outbox
rows through ``pending -> sending -> sent|failed`` and never touches run or
task rows. Retries with backoff live in the repository (``mark_failed``);
the worker just claims what is due and reports outcomes.
"""

from __future__ import annotations

import asyncio
import inspect
import logging
import re
from collections.abc import Callable
from datetime import UTC, datetime, timedelta
from typing import Any

from app.channels.base import ChannelUnavailable

logger = logging.getLogger(__name__)

# Resolves a provider name (e.g. "wecom") to a running Channel instance, or
# None when the channel is not configured/running. Sync or async callables
# are both accepted (``ChannelService.get_channel`` is sync).
ChannelResolver = Callable[[str], Any]

# Resolves the run's final answer for a completed run, keyed
# ``(run_id, owner_user_id)``, or None when no summary is available. Sync or
# async callables are both accepted; failures fall back to the skeleton text.
SummaryResolver = Callable[[str, str | None], Any]

# Lists an owner's channel connections (``ChannelConnectionRepository
# .list_connections``), each a dict with ``provider``, ``external_account_id``
# and ``status``. Sync or async callables are both accepted. The worker uses
# it to confirm, at send time, that the delivery's target is still bound.
ConnectionResolver = Callable[[str], Any]

_ERROR_TEXT_LIMIT = 500
_SUMMARY_TEXT_LIMIT = 1000

# A claimed row stuck in "sending" longer than this is considered orphaned
# (process died between claim and the final status write) and flipped back
# to pending. Generous on purpose: a healthy send plus status write finishes
# in seconds, and resetting too eagerly would double-send live deliveries.
_STALE_SENDING_TIMEOUT_SECONDS = 600

# Matches Gateway ``_SHUTDOWN_HOOK_TIMEOUT_SECONDS``: this worker sits in
# front of external IM I/O, so an unbounded join would stall the whole
# lifespan shutdown path the channel-service bound was added to protect.
_STOP_TIMEOUT_SECONDS = 5.0


def _truncate(text: str, *, limit: int = _ERROR_TEXT_LIMIT) -> str:
    return text if len(text) <= limit else text[: limit - 1] + "\u2026"


_EGRESS_REDACTION = "[redacted]"
# Best-effort scrub before run content leaves for a third-party IM platform.
# Patterns mirror the high-confidence secret detectors in skillscan.
_EGRESS_SECRET_PATTERNS: tuple[re.Pattern[str], ...] = (
    re.compile(
        r"-----BEGIN [A-Z0-9 ]*PRIVATE KEY-----.*?-----END [A-Z0-9 ]*PRIVATE KEY-----",
        re.DOTALL,
    ),
    re.compile(r"\b(?:AKIA|ASIA)[A-Z0-9]{16}\b"),
    re.compile(r"\bgh[pousr]_[A-Za-z0-9_]{20,}\b"),
    re.compile(r"\bxox[baprs]-[A-Za-z0-9-]{20,}\b"),
    re.compile(r"\bsk-[A-Za-z0-9]{20,}\b"),
    re.compile(r"(?im)\b(token|password|passwd|api[_-]?key|secret|credential)s?\b\s*[:=]\s*[\"']?([^\"'\s#]+)"),
)


def redact_egress_text(text: str) -> str:
    """Scrub likely secrets from text before it is pushed to external IM."""
    if not text:
        return text
    redacted = text
    for pattern in _EGRESS_SECRET_PATTERNS:
        if pattern.groups:
            redacted = pattern.sub(lambda match: f"{match.group(1)}={_EGRESS_REDACTION}", redacted)
        else:
            redacted = pattern.sub(_EGRESS_REDACTION, redacted)
    return redacted


def render_notification_text(delivery: dict[str, Any]) -> str:
    """Render a bounded markdown summary of the run outcome for IM push."""
    payload = delivery.get("payload") or {}
    event = delivery.get("event") or ""
    task_label = payload.get("task_title") or payload.get("task_id") or delivery.get("task_id")
    if event in {"run_unmet", "task_paused"}:
        label = "Scheduled task goal was not met" if event == "run_unmet" else "Scheduled task automatically paused"
        lines = [f"**{label}**", f"Task: `{task_label}`"]
        # These are host-defined result codes, never model/provider error text.
        reason = payload.get("reason_code")
        blockers = {"missing_evidence", "needs_user_input", "run_failed", "external_wait", "goal_not_met_yet"}
        allowed = (
            blockers
            | {f"blocked:{blocker}" for blocker in blockers}
            | {"no_verdict", "consecutive_unmet", "evaluator_failed", "max_continuations_reached", "no_progress_detected", "token_capped", "no_durable_end_of_turn", "thread_changed_after_evaluation", "thread_changed_before_continuation"}
        )
        lines.append(f"Reason: `{reason if isinstance(reason, str) and reason in allowed else 'unknown'}`")
        lines.append("See the DeerFlow workspace for the result and next steps.")
    elif event == "run_failed":
        lines = ["**Scheduled task failed**", f"Task: `{task_label}`"]
        # Do not forward raw error text to external IM: scheduled runs can
        # surface hostnames, paths, and token fragments in tracebacks. Users
        # can open the workspace for the full detail.
        lines.append("See the DeerFlow workspace for error details.")
    else:
        heading = "Scheduled task completed"
        if payload.get("relied_on_assumption") is True:
            heading += " — met, relying on stated assumptions"
        lines = [f"**{heading}**", f"Task: `{task_label}`"]
        # The result summary belongs to a successful outcome only: on failed
        # runs a partial answer would be misleading, so the error line above
        # stays the sole detail.
        summary = payload.get("result_summary")
        if isinstance(summary, str) and summary.strip():
            safe_summary = redact_egress_text(summary.strip())
            lines.append(f"Result: {_truncate(safe_summary, limit=_SUMMARY_TEXT_LIMIT)}")
    if delivery.get("run_id"):
        lines.append(f"Run: `{delivery['run_id']}`")
    return "\n".join(lines)


class NotificationDeliveryWorker:
    """Polls the notification outbox and pushes due deliveries over IM."""

    def __init__(
        self,
        *,
        delivery_repo,
        resolve_channel: ChannelResolver,
        poll_interval_seconds: int = 5,
        batch_size: int = 10,
        resolve_run_summary: SummaryResolver | None = None,
        resolve_connections: ConnectionResolver | None = None,
        stale_sending_timeout_seconds: int = _STALE_SENDING_TIMEOUT_SECONDS,
        stop_timeout_seconds: float = _STOP_TIMEOUT_SECONDS,
    ) -> None:
        self._delivery_repo = delivery_repo
        self._resolve_channel = resolve_channel
        self._resolve_run_summary = resolve_run_summary
        self._resolve_connections = resolve_connections
        self._poll_interval_seconds = poll_interval_seconds
        self._batch_size = batch_size
        self._stale_sending_timeout_seconds = stale_sending_timeout_seconds
        self._stop_timeout_seconds = stop_timeout_seconds
        self._task: asyncio.Task | None = None
        self._stop = asyncio.Event()

    async def start(self) -> None:
        if self._task is not None:
            return
        self._stop.clear()
        self._task = asyncio.create_task(self._run_loop())

    async def stop(self) -> None:
        if self._task is None:
            return
        self._stop.set()
        task = self._task
        self._task = None
        try:
            await asyncio.wait_for(task, timeout=self._stop_timeout_seconds)
        except TimeoutError:
            logger.warning(
                "Notification delivery worker stop exceeded %.1fs; cancelling in-flight poll",
                self._stop_timeout_seconds,
            )
            task.cancel()
            try:
                await task
            except asyncio.CancelledError:
                pass

    async def run_once(self, *, now: datetime) -> None:
        await self._recover_stale_sending(now)
        rows = await self._delivery_repo.claim_due_deliveries(now=now, limit=self._batch_size)
        for row in rows:
            try:
                await self._deliver(row)
            except Exception:
                # One poisoned row must not abort the loop and strand the
                # rest of the claimed batch in "sending". Best-effort
                # mark_failed; if even that raises, the stale reset above
                # reclaims the row on a later poll.
                logger.exception("Notification delivery %s crashed; isolating from the rest of the batch", row.get("id"))
                try:
                    await self._delivery_repo.mark_failed(row["id"], claim_token=row.get("claim_token"), error="delivery crashed before completion")
                except Exception:
                    logger.warning("Could not mark crashed delivery %s as failed; stale reset will recover it", row.get("id"), exc_info=True)

    async def _recover_stale_sending(self, now: datetime) -> None:
        """Lease-style reconciliation, run before every claim (the first poll
        after startup is covered too). Mirrors how the scheduler recovers
        stale active runs."""
        try:
            reset = await self._delivery_repo.reset_stale_sending_rows(
                now=now,
                timeout=timedelta(seconds=self._stale_sending_timeout_seconds),
            )
            if reset:
                logger.info("Recovered %s notification deliveries stuck in 'sending'", reset)
        except Exception:
            logger.warning("Failed to reset stale sending rows; retrying next poll", exc_info=True)

    async def _resolve_summary(self, delivery: dict[str, Any]) -> str | None:
        """Best-effort lookup of the run's final answer at delivery time.

        Read at delivery time, not enqueue time, so retried deliveries see
        the freshest value and the outbox payload stays skeleton-only. Any
        failure degrades to the skeleton notification instead of blocking it.
        """
        if self._resolve_run_summary is None:
            return None
        run_id = delivery.get("run_id")
        if not run_id:
            return None
        try:
            summary = self._resolve_run_summary(run_id, delivery.get("owner_user_id"))
            if inspect.isawaitable(summary):
                summary = await summary
        except Exception:
            logger.warning("Failed to resolve run summary for run %s; sending skeleton notification", run_id, exc_info=True)
            return None
        if isinstance(summary, str) and summary.strip():
            return summary
        return None

    async def _target_still_connected(self, delivery: dict[str, Any]) -> bool | None:
        """Return whether the delivery's ``(provider, target)`` is still a
        ``connected`` identity of its owner, or None when that could not be
        determined (no resolver wired, or the lookup failed).

        The outbox row was written while the identity was connected, but
        delivery can run up to a day later (counted retries, then parking),
        and ``disconnect_connection`` only flips the connection row. The
        lookup is scoped to the row's own ``owner_user_id`` so it cannot
        match another owner's binding of the same external account.
        """
        if self._resolve_connections is None:
            return None
        owner_user_id = delivery.get("owner_user_id")
        provider = delivery.get("provider")
        target = delivery.get("target")
        if not owner_user_id or not provider or not target:
            return None
        try:
            connections = self._resolve_connections(owner_user_id)
            if inspect.isawaitable(connections):
                connections = await connections
        except Exception:
            logger.warning("Failed to check channel connections for delivery %s; parking it", delivery.get("id"), exc_info=True)
            return None
        if connections is None:
            connections = []
        if not isinstance(connections, (list, tuple)):
            # An unexpected shape is a lookup problem, not evidence that the
            # binding is gone: iterating a dict or a str would "find" nothing
            # and drop the row for good.
            logger.warning("Channel connection lookup for delivery %s returned %s, not a list; parking it", delivery.get("id"), type(connections).__name__)
            return None
        for connection in connections:
            if not isinstance(connection, dict):
                continue
            # The resolver is called with the row's owner; a row that names a
            # different owner is skipped as well, so a resolver that is not
            # owner-scoped still cannot make another user's binding count.
            row_owner = connection.get("owner_user_id")
            if row_owner is not None and row_owner != owner_user_id:
                continue
            if connection.get("provider") == provider and connection.get("external_account_id") == target and connection.get("status") == "connected":
                return True
        return False

    async def _deliver(self, delivery: dict[str, Any]) -> None:
        delivery_id = delivery["id"]
        provider = delivery.get("provider") or ""
        if self._resolve_connections is not None:
            # Re-check the binding at delivery time, before anything else: a
            # target the owner has disconnected since enqueue must never
            # receive the push, however long the row waited, and it must not
            # sit in parking for a day either when the channel is also down
            # (disconnecting a provider revokes its connections and stops its
            # channel together). A lookup failure parks instead, so a
            # transient store error neither drops nor sends the row.
            still_connected = await self._target_still_connected(delivery)
            if still_connected is False:
                await self._delivery_repo.mark_failed(
                    delivery_id,
                    claim_token=delivery.get("claim_token"),
                    error=f"target is no longer a connected {provider} identity of its owner",
                    terminal=True,
                )
                return
            if still_connected is None:
                await self._delivery_repo.mark_failed(delivery_id, claim_token=delivery.get("claim_token"), error="could not verify the target's channel connection", count_attempt=False)
                return
        # Re-check channel liveness at delivery time, not enqueue time: the
        # channel may have been disabled or disconnected after the outbox row
        # was written, and the row stays retryable for when it comes back.
        channel = self._resolve_channel(provider)
        if inspect.isawaitable(channel):
            channel = await channel
        if channel is None or not getattr(channel, "is_running", True):
            # Channel outage is not the delivery's fault: park the row
            # without consuming its retry budget so it survives an
            # hours-long outage and delivers once the channel returns.
            await self._delivery_repo.mark_failed(delivery_id, claim_token=delivery.get("claim_token"), error=f"channel '{provider}' is not running", count_attempt=False)
            return
        enriched = delivery
        if delivery.get("event") == "run_completed":
            summary = await self._resolve_summary(delivery)
            if summary is not None:
                # Copy before enriching: the claimed row must not be mutated
                # in place (its payload is the durable outbox snapshot).
                enriched = dict(delivery)
                enriched["payload"] = {**(delivery.get("payload") or {}), "result_summary": summary}
        try:
            await channel.send_notification(
                target=delivery.get("target") or "",
                text_markdown=render_notification_text(enriched),
            )
        except ChannelUnavailable as exc:
            await self._delivery_repo.mark_failed(delivery_id, claim_token=delivery.get("claim_token"), error=str(exc), count_attempt=False)
            return
        except Exception as exc:
            await self._delivery_repo.mark_failed(delivery_id, claim_token=delivery.get("claim_token"), error=str(exc))
            return
        await self._delivery_repo.mark_sent(delivery_id, claim_token=delivery.get("claim_token"))

    async def _run_loop(self) -> None:
        while not self._stop.is_set():
            try:
                await self.run_once(now=datetime.now(UTC))
            except Exception:
                # A transient DB error must not kill the poller task for the
                # rest of the process life (same policy as the scheduler).
                logger.exception("Notification delivery poll failed; retrying next interval")
            try:
                await asyncio.wait_for(
                    self._stop.wait(),
                    timeout=self._poll_interval_seconds,
                )
            except TimeoutError:
                continue