Refresh a panel after a write

from __future__ import annotations

from pydantic import BaseModel, Field

from imperal_sdk import Extension, ActionResult, ui

ext = Extension(
    "my-ext",
    display_name="My Extension",
    description="An extension demonstrating panel refresh after a write action.",
    actions_explicit=True,
)

# ── Shared stub store — replace with your real persistence layer ───────────
_STORE: dict[str, dict] = {
    "note-1": {"id": "note-1", "title": "First note",  "body": "Original body."},
    "note-2": {"id": "note-2", "title": "Second note", "body": "Original body."},
}


# ── Pydantic params model (must be module-scope — federal invariant) ───────

class SaveNoteParams(BaseModel):
    note_id: str = Field(
        description="UUID of the note to save. Obtain from list_notes — never invent.",
    )
    body: str = Field(
        description="The full updated body of the note.",
    )


# ── chat.function — save and refresh ──────────────────────────────────────

@ext.chat.function(  # type: ignore[attr-defined]
    "save_note",
    description="Save the full body of an existing note.",
    action_type="write",
    chain_callable=True,
    effects=["note.update"],
)
async def save_note(ctx: object, params: SaveNoteParams) -> ActionResult:
    # Replace with your real persistence call:
    # await ctx.http.patch(f"/notes/{params.note_id}", json={"body": params.body})
    if params.note_id not in _STORE:
        return ActionResult(
            status="error",
            summary=f"Note {params.note_id!r} not found.",
        )

    _STORE[params.note_id]["body"] = params.body

    # refresh_panels lists bare panel_ids (no "__panel__" prefix).
    # "sidebar" → the host calls __panel__sidebar on Path A (direct button call).
    # On Path B (chat-initiated), the host fires refreshAll() regardless of this
    # list — see the callout above.
    return ActionResult(
        status="ok",
        summary="Note saved.",
        refresh_panels=["sidebar"],
    )


# ── Left sidebar — shown here for a complete runnable example ─────────────

@ext.panel(
    "sidebar",
    slot="left",
    title="Notes",
    icon="🗒️",
    default_width=280,
    min_width=200,
    max_width=500,
)
async def sidebar(ctx: object, **kwargs: object) -> object:
    notes = list(_STORE.values())
    return ui.Stack(
        children=[
            ui.List(
                items=[
                    ui.ListItem(
                        id=note["id"],
                        title=note["title"],
                        on_click=ui.Call("__panel__editor", note_id=note["id"]),
                    )
                    for note in notes
                ]
            )
        ],
        gap=2,
    )


# ── Center editor — panel that triggers save_note ─────────────────────────

@ext.panel(
    "editor",
    slot="center",
    center_overlay=True,
    title="Editor",
    icon="📝",
)
async def editor(ctx: object, note_id: str = "", **kwargs: object) -> object:
    if not note_id:
        return ui.Empty(message="Select a note to edit", icon="📄")

    note = _STORE.get(note_id)
    if note is None:
        return ui.Error(message="Note not found")

    return ui.Stack(
        children=[
            ui.Text(note["title"], variant="h2"),
            ui.RichEditor(
                content=note["body"],
                # The on_change calls save_note directly. After save_note returns
                # ActionResult(refresh_panels=["sidebar"]), the host re-fetches
                # the sidebar to reflect any list-level changes (e.g., updated title).
                on_change=ui.Call("save_note", note_id=note_id),
            ),
        ],
        gap=4,
    )