Function restoreFocusState

ui/src/composables/useStreamingState.ts:191–218

* Restore focus to previously focused element

()

   * Restore focus to previously focused element
   */
  function restoreFocusState(): void {
    if (!preserveFocus || !containerRef.value) return;

    const state = getPreservedState(surfaceId);

    if (state.focusedElementId) {
      const el = containerRef.value.querySelector(`#${state.focusedElementId}`)
        || containerRef.value.querySelector(`[data-component-id="${state.focusedElementId}"]`)
        || containerRef.value.querySelector(`[name="${state.focusedElementId}"]`);

      if (el && el instanceof HTMLElement) {
        // Use requestAnimationFrame to ensure DOM is ready
        requestAnimationFrame(() => {
          el.focus();

          // Restore cursor position for input elements
          if (el instanceof HTMLInputElement || el instanceof HTMLTextAreaElement) {
            const pendingValue = state.pendingInputValues.get(state.focusedElementId!);
            if (pendingValue !== undefined) {
              // Set selection to end of input
              const len = el.value.length;
              el.setSelectionRange(len, len);
            }
          }
        });
      }
    }
  }

  /**
   * Save all state before an update