(state as AppViewState & { requestUpdate?: () => void }).requestUpdate?.();

}

function focusChatSessionPickerSearch(state: AppViewState, surface: ChatSessionSelectSurface) {

const updateComplete = (state as AppViewState & { updateComplete?: Promise<unknown> })

.updateComplete;

const focus = () => {

document

.querySelector<HTMLInputElement>(

`[data-chat-session-picker-search="true"][data-chat-session-picker-search-surface="${surface}"]`,

)

?.focus();

};

if (updateComplete) {

void updateComplete.then(focus);

return;

}

setTimeout(focus, 0);

}

function openChatSessionPicker(state: AppViewState, surface: ChatSessionSelectSurface) {

state.chatSessionPickerOpen = true;

state.chatSessionPickerSurface = surface;

state.chatSessionPickerError = null;

requestHostUpdate(state);

focusChatSessionPickerSearch(state, surface);

}

function closeChatSessionPicker(state: AppViewState) {

}

});

it("focuses the inline search for the active picker surface", async () => {

it("keeps session trigger focus separate from inline search focus", async () => {

const { state } = createChatHeaderState();

const mobileContainer = document.createElement("div");

const desktopContainer = document.createElement("div");

expect(

mobileContainer.querySelector('input[data-chat-session-picker-search="true"]'),

).toBeTruthy();

expect(state.chatSessionPickerOpen).toBe(true);

expect(document.activeElement).not.toBe(desktopInput);

desktopInput!.focus();

await vi.waitFor(() => expect(document.activeElement).toBe(desktopInput));

expect(state.chatSessionPickerOpen).toBe(true);

expect(state.chatSessionPickerSurface).toBe("desktop");

} finally {

render(html``, mobileContainer);

render(html``, desktopContainer);