scripts/debug/zxdb_cli/daemon/handlers/attach.py

# Copyright 2026 The Fuchsia Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

from __future__ import annotations

from typing import TYPE_CHECKING, Any

from pydap.models import AttachRequestArguments
from shared.protocol.attach import (
    COMMAND_NAME,
    AttachRequest,
)
from shared.protocol.base import Response

__all__ = ["COMMAND_NAME", "handle"]

if TYPE_CHECKING:
    from daemon.daemon import Daemon


async def handle(daemon: Daemon, req: AttachRequest) -> Response[Any]:
    if not daemon.zxdb_writer:
        return Response(
            success=False, message="Not connected to zxdb DAP server"
        )

    # AttachRequestArguments is a generic DAP model. Zxdb-specific arguments
    # (like "process") must be passed via extra_fields, which are flattened
    # during serialization.
    attach_args = AttachRequestArguments(
        restart=None, extra_fields={"process": req.filter}
    )

    try:
        resp = await daemon.dap_client.attach(attach_args)
        return Response(success=True, body=resp.dump_dap())
    except Exception as e:
        return Response(success=False, message=f"Failed to attach: {e}")