#  Copyright (C) 2023 The Android Open Source Project
#
#  Licensed under the Apache License, Version 2.0 (the "License");
#  you may not use this file except in compliance with the License.
#  You may obtain a copy of the License at
#
#       http://www.apache.org/licenses/LICENSE-2.0
#
#  Unless required by applicable law or agreed to in writing, software
#  distributed under the License is distributed on an "AS IS" BASIS,
#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
#  See the License for the specific language governing permissions and
#  limitations under the License.
#
#  Licensed under the Apache License, Version 2.0 (the "License");
#  you may not use this file except in compliance with the License.
#  You may obtain a copy of the License at
#
#       http://www.apache.org/licenses/LICENSE-2.0
#
#  Unless required by applicable law or agreed to in writing, software
#  distributed under the License is distributed on an "AS IS" BASIS,
#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
#  See the License for the specific language governing permissions and
#  limitations under the License.

# Lint as: python3

from __future__ import annotations

from collections.abc import Sequence
import dataclasses
import enum
import logging
from typing import Any, Optional, Union

WIFI_SNIPPET_PACKAGE_NAME = "com.google.snippet.wifi"
WIFI_DIRECT_SNIPPET_PACKAGE_NAME = 'com.google.snippet.wifi.direct'

ACTION_LISTENER_CALLBACK_EVENT = "WifiP2pManagerActionListenerCallback"
ACTION_LISTENER_ON_SUCCESS = 'onSuccess'
ACTION_LISTENER_ON_FAILURE = 'onFailure'
ACTION_LISTENER_FAILURE_REASON = 'reason'

EVENT_KEY_CALLBACK_NAME = 'callbackName'
EVENT_KEY_REASON = 'reason'
EVENT_KEY_P2P_DEVICE = 'p2pDevice'
EVENT_KEY_P2P_INFO = 'p2pInfo'
EVENT_KEY_P2P_GROUP = 'p2pGroup'
EVENT_KEY_PEER_LIST = 'peerList'

EXTRA_WIFI_P2P_GROUP = 'p2pGroupInfo'
EXTRA_WIFI_STATE = 'wifi_p2p_state'

ON_DEVICE_INFO_AVAILABLE = 'WifiP2pOnDeviceInfoAvailable'
ON_PERSISTENT_GROUP_INFO_AVAILABLE = 'onPersistentGroupInfoAvailable'
WIFI_P2P_CREATING_GROUP = 'CREATING_GROUP'
WIFI_P2P_CONNECTION_CHANGED_ACTION = (
    'android.net.wifi.p2p.CONNECTION_STATE_CHANGE'
)
WIFI_P2P_DISCOVERY_CHANGED_ACTION = (
    'android.net.wifi.p2p.DISCOVERY_STATE_CHANGE'
)
WIFI_P2P_PEERS_CHANGED_ACTION = 'android.net.wifi.p2p.PEERS_CHANGED'
WIFI_P2P_STATE_CHANGED_ACTION = 'android.net.wifi.p2p.STATE_CHANGED'
WIFI_P2P_THIS_DEVICE_CHANGED_ACTION = 'android.net.wifi.p2p.THIS_DEVICE_CHANGED'

ANONYMIZED_MAC_ADDRESS = '02:00:00:00:00:00'


@enum.unique
class ActionListenerOnFailure(enum.IntEnum):
  """Indicates the failure reason of the initiation of the action.

  https://developer.android.com/reference/android/net/wifi/p2p/WifiP2pManager.ActionListener#onFailure(int)
  """

  ERROR = 0
  P2P_UNSUPPORTED = 1
  BUSY = 2
  NO_SERVICE_REQUESTS = 3


@enum.unique
class Band(enum.IntEnum):
  """Indicates the band of the operating frequency.

  https://developer.android.com/reference/android/net/wifi/p2p/WifiP2pConfig#getGroupOwnerBand()
  """

  GROUP_OWNER_BAND_AUTO = 0
  GROUP_OWNER_BAND_2GHZ = 1
  GROUP_OWNER_BAND_5GHZ = 2


@enum.unique
class IpProvisioningMode(enum.IntEnum):
  """Indicates the IP provisioning mode.

  https://developer.android.com/reference/android/net/wifi/p2p/WifiP2pConfig#getGroupClientIpProvisioningMode()
  """

  GROUP_CLIENT_IP_PROVISIONING_MODE_IPV4_DHCP = 0
  GROUP_CLIENT_IP_PROVISIONING_MODE_IPV6_LINK_LOCAL = 1


@enum.unique
class ExtraWifiState(enum.IntEnum):
  """Indicates whether Wi-Fi p2p is enabled or disabled.

  https://developer.android.com/reference/android/net/wifi/p2p/WifiP2pManager#EXTRA_WIFI_STATE
  """

  WIFI_P2P_STATE_UNKNOWN = 0
  WIFI_P2P_STATE_DISABLED = 1
  WIFI_P2P_STATE_ENABLED = 2


@enum.unique
class WifiP2pDeviceStatus(enum.IntEnum):
  """Represents status code for WifiP2pDevice.and

  https://developer.android.com/reference/android/net/wifi/p2p/WifiP2pDevice#constants_1
  """

  CONNECTED = 0
  INVITED = 1
  FAILED = 2
  AVAILABLE = 3
  UNAVAILABLE = 4


@enum.unique
class WpsInfo(enum.IntEnum):
  """Represents Wi-Fi Protected Setup.

  https://developer.android.com/reference/android/net/wifi/WpsInfo
  """

  PBC = 0
  DISPLAY = 1
  KEYPAD = 2
  LABEL = 3
  INVALID = 4


@dataclasses.dataclass(frozen=True)
class WifiP2pConfig:
  """Represents a Wi-Fi P2p configuration for setting up a connection.

  https://developer.android.com/reference/android/net/wifi/p2p/WifiP2pConfig
  """

  persistent_mode: Optional[bool] = None
  device_address: Optional[str] = None
  group_client_ip_provisioning_mode: Optional[IpProvisioningMode] = None
  group_operating_band: Optional[Band] = None
  group_operating_frequency: Optional[int] = None
  network_name: Optional[str] = None
  passphrase: Optional[str] = None
  wps_setup: WpsInfo | None = None

  def to_dict(self) -> dict[str, Union[bool, int, str]]:
    """Converts this WifiP2pConfig to a dictionary."""
    return {
        k: v.value if isinstance(v, enum.Enum) else v
        for k, v in self.__dict__.items()
        if v is not None
    }


@dataclasses.dataclass
class WifiP2pDevice:
  """Represents a Wi-Fi p2p device.

  https://developer.android.com/reference/android/net/wifi/p2p/WifiP2pDevice
  """

  device_name: str
  device_address: str
  is_group_owner: bool
  status: int
  primary_device_type: str
  secondary_device_type: str

  @classmethod
  def from_dict(cls, device: dict[str, Any]) -> WifiP2pDevice:
    """Generates a WifiP2pDevice object from a dictionary."""
    logging.debug(
        "Converting following snippet event data to WifiP2pDevice: %s",
        device,
    )
    return WifiP2pDevice(
        device_name=device["deviceName"],
        device_address=device["deviceAddress"],
        is_group_owner=device["isGroupOwner"],
        status=device["status"],
        primary_device_type=device["primaryDeviceType"],
        secondary_device_type=device["secondaryDeviceType"],
    )

  @classmethod
  def from_dict_list(
      cls, devices: list[dict[str, Any]]
  ) -> Sequence[WifiP2pDevice]:
    """Generates WifiP2pDevice objects from a list of dictionary."""
    return (cls.from_dict(device) for device in devices)


@dataclasses.dataclass(frozen=True)
class WifiP2pInfo:
  """Represents a connection information about a Wi-Fi p2p group.

  https://developer.android.com/reference/android/net/wifi/p2p/WifiP2pInfo
  """

  group_formed: bool
  group_owner_address: str
  is_group_owner: bool

  @classmethod
  def from_dict(cls, info: dict[str, Any]) -> WifiP2pInfo:
    """Generates a WifiP2pInfo object from a dictionary."""
    return WifiP2pInfo(
        group_formed=info["groupFormed"],
        group_owner_address=info["groupOwnerAddress"],
        is_group_owner=info["isGroupOwner"],
    )


@dataclasses.dataclass(frozen=True)
class WifiP2pGroup:
  """Represents a Wi-Fi p2p group.

  https://developer.android.com/reference/android/net/wifi/p2p/WifiP2pGroup
  """

  frequency: int
  interface: str
  network_id: int
  network_name: str
  owner: WifiP2pDevice
  passphrase: str
  is_group_owner: bool

  @classmethod
  def from_dict(cls, group: dict[str, Any]) -> WifiP2pGroup:
    """Generates a WifiP2pGroup object from a dictionary."""
    return WifiP2pGroup(
        frequency=group['frequency'],
        interface=group['interface'],
        network_id=group['networkId'],
        network_name=group['networkName'],
        owner=WifiP2pDevice.from_dict(group['owner']),
        passphrase=group['passphrase'],
        is_group_owner=group['isGroupOwner'],
    )

  @classmethod
  def from_dict_list(
      cls, groups: list[dict[str, Any]]
  ) -> Sequence[WifiP2pGroup]:
    """Generates WifiP2pGroup objects from a list of dictionary."""
    return (cls.from_dict(group) for group in groups)