"""Account pages associated with Fitbit Companion App testing."""
from typing import List

from blueberry.utils.ui_pages import ui_core
from blueberry.utils.ui_pages import ui_node
from blueberry.utils.ui_pages.fitbit_companion import constants


class AccountPage(ui_core.UIPage):
  """Fitbit Companion App's account page."""

  ACTIVITY = f'{constants.PKG_NAME}/com.fitbit.settings.ui.AccountActivity'
  PAGE_RID = f'{constants.PKG_NAME_ID}/account_recycler'
  NODE_UER_AVATOR_RID = f'{constants.PKG_NAME_ID}/userAvatar'
  NODE_BACK_CONTENT_DESC = 'Navigate up'
  NODE_ADD_DEV_RID = f'{constants.PKG_NAME_ID}/add_device_img'
  NODE_DEVICE_RID = f'{constants.PKG_NAME_ID}/device_name'

  def add_device(self) -> ui_core.UIPage:
    """Goes to page to add new device.

    Returns:
      The transformed page.

    Raises:
      errors.UIError: Fail to get the target node.
    """
    return self.click_node_by_rid(self.NODE_ADD_DEV_RID)

  def back(self) -> ui_core.UIPage:
    """Goes back to Fitbit Companion App's home page.

    Returns:
      The transformed page.

    Raises:
      errors.UIError: Fail to get target node.
    """
    return self.click_node_by_content_desc(self.NODE_BACK_CONTENT_DESC)

  def get_paired_devices(self) -> List[ui_node.UINode]:
    """Gets all paired devices.

    Returns:
      The list of node representing the paired device.
    """
    return self.get_all_nodes_by_rid(self.NODE_DEVICE_RID)


class PairedDeviceDetailPage(ui_core.UIPage):
  """Fitbit Companion App's page of paired device."""

  PAGE_RID = f'{constants.PKG_NAME_ID}/unpair'

  def unpair(self) -> ui_core.UIPage:
    """Unpairs device.

    Returns:
      The transformed page.

    Raises:
      errors.UIError: Fail to find the target node.
    """
    # TODO(user): We need to consider the situation while device
    #   sync now which cause unpair to fail.
    return self.click_node_by_rid(self.PAGE_RID)


class UnpairConfirmPage(ui_core.UIPage):
  """Fitbit Companion App's page to confirm the action of unpairing."""

  PAGE_TEXT = 'UNPAIR'

  def confirm(self) -> ui_core.UIPage:
    """Confirms the action of unpairing.

    Returns:
      The transformed page.
    """
    self.click_node_by_text(self.PAGE_TEXT)
    self.ctx.expect_page(AccountPage)
    return self.ctx.page