/*
 * Copyright 2017-2026 Espressif Systems (Shanghai) PTE LTD
 * Copyright 2020-2025 Ryan Powell <ryan@nable-embedded.io> and
 * esp-nimble-cpp, NimBLE-Arduino contributors.
 *
 * 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.
 */

#ifndef BLECONNINFO_H_
#define BLECONNINFO_H_

#if defined(CONFIG_NIMBLE_ENABLED)

#include <host/ble_gap.h>
#include "BLEAddress.h"

/**
 * @brief Connection information.
 */
class BLEConnInfo {
public:
  /** @brief Gets the over-the-air address of the connected peer */
  BLEAddress getAddress() const {
    return BLEAddress(m_desc.peer_ota_addr);
  }

  /** @brief Gets the ID address of the connected peer */
  BLEAddress getIdAddress() const {
    return BLEAddress(m_desc.peer_id_addr);
  }

  /** @brief Gets the connection handle (also known as the connection id) of the connected peer */
  uint16_t getConnHandle() const {
    return m_desc.conn_handle;
  }

  /** @brief Gets the connection interval for this connection (in 1.25ms units) */
  uint16_t getConnInterval() const {
    return m_desc.conn_itvl;
  }

  /** @brief Gets the supervision timeout for this connection (in 10ms units) */
  uint16_t getConnTimeout() const {
    return m_desc.supervision_timeout;
  }

  /** @brief Gets the allowable latency for this connection (unit = number of intervals) */
  uint16_t getConnLatency() const {
    return m_desc.conn_latency;
  }

  /** @brief Gets the maximum transmission unit size for this connection (in bytes) */
  uint16_t getMTU() const {
    return ble_att_mtu(m_desc.conn_handle);
  }

  /** @brief Check if we are in the master role in this connection */
  bool isMaster() const {
    return (m_desc.role == BLE_GAP_ROLE_MASTER);
  }

  /** @brief Check if we are in the slave role in this connection */
  bool isSlave() const {
    return (m_desc.role == BLE_GAP_ROLE_SLAVE);
  }

  /** @brief Check if we are connected to a bonded peer */
  bool isBonded() const {
    return (m_desc.sec_state.bonded == 1);
  }

  /** @brief Check if the connection in encrypted */
  bool isEncrypted() const {
    return (m_desc.sec_state.encrypted == 1);
  }

  /** @brief Check if the the connection has been authenticated */
  bool isAuthenticated() const {
    return (m_desc.sec_state.authenticated == 1);
  }

  /** @brief Gets the key size used to encrypt the connection */
  uint8_t getSecKeySize() const {
    return m_desc.sec_state.key_size;
  }

private:
  friend class BLEServer;
  friend class BLEClient;
  friend class BLECharacteristic;
  friend class BLEDescriptor;

  ble_gap_conn_desc m_desc{};
  BLEConnInfo(){};
  BLEConnInfo(ble_gap_conn_desc desc) {
    m_desc = desc;
  }
};
#endif

#endif  // BLECONNINFO_H_
