Function _verify_ui_state_settings

browser_utils/models/ui_state.py:16–94

Verify if the UI state settings are correct. Args: page: Playwright page object. req_id: Request ID for logging. Returns: dict: A dictionary containing the validation result.

(page: AsyncPage, req_id: str = "unknown")

async def _verify_ui_state_settings(page: AsyncPage, req_id: str = "unknown") -> dict:
    """
    Verify if the UI state settings are correct.

    Args:
        page: Playwright page object.
        req_id: Request ID for logging.

    Returns:
        dict: A dictionary containing the validation result.
    """
    # Don't set lifecycle phase names as request IDs - they appear as ghost prefixes
    # Only set actual request IDs (7-char alphanumerics)
    if req_id not in ("initial", "set_mod", "set_model", "reload", "unknown", ""):
        set_request_id(req_id)
    try:
        logger.debug("[State] Verifying UI state...")

        # Get current localStorage settings
        prefs_str = await page.evaluate(
            "() => localStorage.getItem('aiStudioUserPreference')"
        )

        if not prefs_str:
            logger.warning("localStorage.aiStudioUserPreference does not exist")
            return {
                "exists": False,
                "isAdvancedOpen": None,
                "areToolsOpen": None,
                "needsUpdate": True,
                "error": "localStorage not found",
            }

        try:
            prefs = json.loads(prefs_str)
            is_advanced_open = prefs.get("isAdvancedOpen")
            are_tools_open = prefs.get("areToolsOpen")

            # Check if update is needed
            needs_update = (is_advanced_open is not True) or (
                are_tools_open is not True
            )

            result = {
                "exists": True,
                "isAdvancedOpen": is_advanced_open,
                "areToolsOpen": are_tools_open,
                "needsUpdate": needs_update,
                "prefs": prefs,
            }

            if needs_update:
                logger.debug(
                    f"[State] State mismatch: adv={is_advanced_open}, tools={are_tools_open} (update needed)"
                )
            # No log needed when state is correct
            return result