// Copyright 2022 Google LLC
//
// 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
//
//     https://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.

#include "internal/platform/implementation/g3/ble.h"

#include <algorithm>
#include <memory>
#include <optional>
#include <string>
#include <utility>
#include <vector>

#include "absl/functional/any_invocable.h"
#include "absl/log/check.h"
#include "absl/status/status.h"
#include "absl/status/statusor.h"
#include "absl/strings/escaping.h"
#include "absl/strings/str_cat.h"
#include "absl/strings/string_view.h"
#include "absl/synchronization/mutex.h"
#include "internal/platform/borrowable.h"
#include "internal/platform/byte_array.h"
#include "internal/platform/cancellation_flag_listener.h"
#include "internal/platform/count_down_latch.h"
#include "internal/platform/exception.h"
#include "internal/platform/implementation/ble.h"
#include "internal/platform/implementation/bluetooth_adapter.h"
#include "internal/platform/implementation/g3/bluetooth_adapter.h"
#include "internal/platform/logging.h"
#include "internal/platform/mac_address.h"
#include "internal/platform/medium_environment.h"
#include "internal/platform/prng.h"
#include "internal/platform/uuid.h"

namespace nearby {
namespace g3 {

namespace {

using ::nearby::api::ble::BleAdvertisementData;
using ::nearby::api::ble::TxPowerLevel;

std::string TxPowerLevelToName(TxPowerLevel power_mode) {
  switch (power_mode) {
    case TxPowerLevel::kUltraLow:
      return "UltraLow";
    case TxPowerLevel::kLow:
      return "Low";
    case TxPowerLevel::kMedium:
      return "Medium";
    case TxPowerLevel::kHigh:
      return "High";
    case TxPowerLevel::kUnknown:
      return "Unknown";
  }
}

}  // namespace

api::ble::BlePeripheral::UniqueId BleSocket::GetRemotePeripheralId() {
  BleSocket* remote_socket = GetRemoteSocket();
  if (remote_socket == nullptr || remote_socket->adapter_ == nullptr ||
      remote_socket->adapter_->GetBleMedium() == nullptr) {
    return 0LL;
  }
  BleMedium* medium =
      dynamic_cast<BleMedium*>(remote_socket->adapter_->GetBleMedium());
  if (medium == nullptr) {
    return 0LL;
  }
  return medium->GetAdapter().GetUniqueId();
}

std::unique_ptr<api::ble::BleSocket> BleServerSocket::Accept() {
  absl::MutexLock lock(mutex_);
  while (!closed_ && pending_sockets_.empty()) {
    cond_.Wait(&mutex_);
  }
  // whether or not we were running in the wait loop, return early if closed.
  if (closed_) return {};
  auto* remote_socket =
      pending_sockets_.extract(pending_sockets_.begin()).value();
  CHECK(remote_socket);

  auto local_socket = std::make_unique<BleSocket>(adapter_);
  local_socket->Connect(*remote_socket);
  remote_socket->Connect(*local_socket);
  cond_.SignalAll();
  return local_socket;
}

bool BleServerSocket::Connect(BleSocket& socket) {
  absl::MutexLock lock(mutex_);
  if (closed_) return false;
  if (socket.IsConnected()) {
    LOG(WARNING) << "Failed to connect to Ble server socket: already connected";
    return true;  // already connected.
  }
  // add client socket to the pending list
  pending_sockets_.insert(&socket);
  cond_.SignalAll();
  while (!socket.IsConnected()) {
    cond_.Wait(&mutex_);
    if (closed_) return false;
  }
  return true;
}

void BleServerSocket::SetCloseNotifier(absl::AnyInvocable<void()> notifier) {
  absl::MutexLock lock(mutex_);
  close_notifier_ = std::move(notifier);
}

BleServerSocket::~BleServerSocket() {
  absl::MutexLock lock(mutex_);
  DoClose();
}

Exception BleServerSocket::Close() {
  absl::MutexLock lock(mutex_);
  return DoClose();
}

Exception BleServerSocket::DoClose() {
  bool should_notify = !closed_;
  closed_ = true;
  if (should_notify) {
    cond_.SignalAll();
    if (close_notifier_) {
      auto notifier = std::move(close_notifier_);
      mutex_.unlock();
      // Notifier may contain calls to public API, and may cause deadlock, if
      // mutex_ is held during the call.
      notifier();
      mutex_.lock();
    }
  }
  return {Exception::kSuccess};
}

BleMedium::BleMedium(BluetoothAdapter& adapter) : adapter_(adapter) {
  CHECK(&adapter_);
  adapter_.SetBleMedium(this);
  is_extended_advertisements_available_ =
      MediumEnvironment::Instance().IsBleExtendedAdvertisementsAvailable();

  MediumEnvironment::Instance().RegisterBleMedium(*this,
                                                  adapter_.GetUniqueId());
}

BleMedium::~BleMedium() {
  adapter_.SetBleMedium(nullptr);
  MediumEnvironment::Instance().UnregisterBleMedium(*this);
}

bool BleMedium::StartAdvertising(
    const BleAdvertisementData& advertising_data,
    api::ble::AdvertiseParameters advertise_parameters) {
  LOG(INFO)
      << "G3 Ble StartAdvertising: advertising_data.is_extended_advertisement="
      << advertising_data.is_extended_advertisement
      << ", advertising_data.service_data size="
      << advertising_data.service_data.size() << ", tx_power_level="
      << TxPowerLevelToName(advertise_parameters.tx_power_level)
      << ", is_connectable=" << advertise_parameters.is_connectable;
  if (advertising_data.is_extended_advertisement &&
      !IsExtendedAdvertisementsAvailable()) {
    LOG(INFO)
        << "G3 Ble StartAdvertising does not support extended advertisement";
    return false;
  }

  absl::MutexLock lock(mutex_);
  MediumEnvironment::Instance().UpdateBleMediumForAdvertising(
      /*enabled=*/true, *this, adapter_.GetUniqueId(), advertising_data);
  return true;
}

bool BleMedium::StopAdvertising() {
  LOG(INFO) << "G3 Ble StopAdvertising";
  absl::MutexLock lock(mutex_);

  BleAdvertisementData empty_advertisement_data = {};
  MediumEnvironment::Instance().UpdateBleMediumForAdvertising(
      /*enabled=*/false, *this, adapter_.GetUniqueId(),
      empty_advertisement_data);
  return true;
}

std::unique_ptr<BleMedium::AdvertisingSession> BleMedium::StartAdvertising(
    const api::ble::BleAdvertisementData& advertising_data,
    api::ble::AdvertiseParameters advertise_parameters,
    BleMedium::AdvertisingCallback callback) {
  LOG(INFO)
      << "G3 Ble StartAdvertising: advertising_data.is_extended_advertisement="
      << advertising_data.is_extended_advertisement
      << ", advertising_data.service_data size="
      << advertising_data.service_data.size() << ", tx_power_level="
      << TxPowerLevelToName(advertise_parameters.tx_power_level)
      << ", is_connectable=" << advertise_parameters.is_connectable;
  if (advertising_data.is_extended_advertisement &&
      !IsExtendedAdvertisementsAvailable()) {
    LOG(INFO)
        << "G3 Ble StartAdvertising does not support extended advertisement";
    return nullptr;
  }

  if (callback.start_advertising_result) {
    callback.start_advertising_result(absl::OkStatus());
  }
  absl::MutexLock lock(mutex_);
  MediumEnvironment::Instance().UpdateBleMediumForAdvertising(
      /*enabled=*/true, *this, adapter_.GetUniqueId(), advertising_data);
  return std::make_unique<AdvertisingSession>(
      AdvertisingSession{.stop_advertising = [this] {
        return StopAdvertising()
                   ? absl::OkStatus()
                   : absl::InternalError("Failed to stop advertising");
      }});
}

bool BleMedium::StartScanning(const Uuid& service_uuid,
                              TxPowerLevel tx_power_level,
                              ScanCallback callback) {
  LOG(INFO) << "G3 Ble StartScanning";
  auto internal_session_id = Prng().NextUint32();
  absl::MutexLock lock(mutex_);
  MediumEnvironment::Instance().UpdateBleMediumForScanning(
      /*enabled=*/true, service_uuid, internal_session_id,
      {.advertisement_found_cb = std::move(callback.advertisement_found_cb)},
      *this);
  scanning_internal_session_ids_.insert({service_uuid, internal_session_id});
  return true;
}

bool BleMedium::StartMultipleServicesScanning(
    const std::vector<Uuid>& service_uuids,
    api::ble::TxPowerLevel tx_power_level, ScanCallback callback) {
  LOG(INFO) << "G3 Ble StartMultipleServicesScanning";

  absl::MutexLock lock(mutex_);
  scan_callback_ = std::move(callback);
  for (const auto& service_uuid : service_uuids) {
    auto internal_session_id = Prng().NextUint32();
    ScanCallback multiple_scan_callback = {
        .advertisement_found_cb =
            [this](api::ble::BlePeripheral::UniqueId peripheral_id,
                   BleAdvertisementData advertisement_data) {
              scan_callback_.advertisement_found_cb(peripheral_id,
                                                    advertisement_data);
            }};

    MediumEnvironment::Instance().UpdateBleMediumForScanning(
        /*enabled=*/true, service_uuid, internal_session_id,
        {.advertisement_found_cb =
             std::move(multiple_scan_callback.advertisement_found_cb)},
        *this);
    scanning_internal_session_ids_.insert({service_uuid, internal_session_id});
  }

  return true;
}

bool BleMedium::StopScanning() {
  LOG(INFO) << "G3 Ble StopScanning";
  absl::MutexLock lock(mutex_);
  for (auto element : scanning_internal_session_ids_) {
    MediumEnvironment::Instance().UpdateBleMediumForScanning(
        /*enabled=*/false,
        /*service_uuid=*/element.first,
        /*internal_session_id*/ element.second,
        /*callback=*/{}, *this);
  }
  return true;
}

std::unique_ptr<BleMedium::ScanningSession> BleMedium::StartScanning(
    const Uuid& service_uuid, TxPowerLevel tx_power_level,
    BleMedium::ScanningCallback callback) {
  LOG(INFO) << "G3 Ble StartScanning";
  auto internal_session_id = Prng().NextUint32();

  {
    absl::MutexLock lock(mutex_);

    MediumEnvironment::Instance().UpdateBleMediumForScanning(
        /*enabled=*/true, service_uuid, internal_session_id,
        std::move(callback), *this);
    scanning_internal_session_ids_.insert({service_uuid, internal_session_id});
  }
  return std::make_unique<ScanningSession>(ScanningSession{
      .stop_scanning =
          [this, service_uuid = service_uuid,
           internal_session_id = internal_session_id]() {
            absl::MutexLock lock(mutex_);
            if (scanning_internal_session_ids_.find(
                    {service_uuid, internal_session_id}) ==
                scanning_internal_session_ids_.end()) {
              // can't find the provided internal session.
              return absl::NotFoundError(
                  "can't find the provided internal session");
            }
            MediumEnvironment::Instance().UpdateBleMediumForScanning(
                /*enabled=*/false, service_uuid, internal_session_id,
                /*callback=*/{}, *this);
            scanning_internal_session_ids_.erase(
                {service_uuid, internal_session_id});
            return absl::OkStatus();
          },
  });
}

std::unique_ptr<api::ble::GattServer> BleMedium::StartGattServer(
    api::ble::ServerGattConnectionCallback callback) {
  return std::make_unique<GattServer>(*this, std::move(callback));
}

bool BleMedium::IsStopped(Borrowable<api::ble::GattServer*> server) {
  auto borrowed = server.Borrow();
  if (!borrowed) {
    return true;
  }
  BleMedium::GattServer* gatt_server = static_cast<GattServer*>(*borrowed);
  return gatt_server->IsStopped();
}

std::unique_ptr<api::ble::GattClient> BleMedium::ConnectToGattServer(
    api::ble::BlePeripheral::UniqueId peripheral_id,
    TxPowerLevel tx_power_level,
    api::ble::ClientGattConnectionCallback callback) {
  Borrowable<api::ble::GattServer*> server =
      MediumEnvironment::Instance().GetGattServer(peripheral_id);
  if (IsStopped(server)) {
    LOG(WARNING) << "No GATT server found for " << peripheral_id;
    return nullptr;
  }
  return std::make_unique<GattClient>(peripheral_id, server,
                                      std::move(callback));
}

bool BleMedium::IsExtendedAdvertisementsAvailable() {
  return is_extended_advertisements_available_;
}

std::optional<api::ble::BlePeripheral::UniqueId>
BleMedium::RetrieveBlePeripheralIdFromNativeId(
    const std::string& ble_peripheral_native_id) {
  MacAddress mac_address;
  if (!MacAddress::FromString(ble_peripheral_native_id, mac_address)) {
    return std::nullopt;
  }

  if (MediumEnvironment::Instance()
          .FindBlePeripheral(mac_address.address())
          .IsSet()) {
    return mac_address.address();
  }

  return std::nullopt;
}

BleMedium::GattServer::GattServer(
    BleMedium& medium, api::ble::ServerGattConnectionCallback callback)
    : medium_(medium), callback_(std::move(callback)) {
  BluetoothAdapter& adapter = medium.GetAdapter();
  MediumEnvironment::Instance().RegisterGattServer(
      medium_, adapter.GetUniqueId(), lender_.GetBorrowable());
}

BleMedium::GattServer::~GattServer() {
  Stop();
  lender_.Release();
  MediumEnvironment::Instance().UnregisterGattServer(medium_);
}
std::optional<api::ble::GattCharacteristic>
BleMedium::GattServer::CreateCharacteristic(
    const Uuid& service_uuid, const Uuid& characteristic_uuid,
    api::ble::GattCharacteristic::Permission permission,
    api::ble::GattCharacteristic::Property property) {
  absl::MutexLock lock(mutex_);
  api::ble::GattCharacteristic characteristic = {.uuid = characteristic_uuid,
                                                 .service_uuid = service_uuid};
  characteristics_[characteristic] = absl::NotFoundError("value not set");
  return characteristic;
}

bool BleMedium::GattServer::DiscoverBleMediumGattCharacteristics(
    const Uuid& service_uuid, const std::vector<Uuid>& characteristic_uuids) {
  absl::MutexLock lock(mutex_);
  auto contains = [&](const Uuid& characteristic_uuid) {
    return std::find(characteristic_uuids.begin(), characteristic_uuids.end(),
                     characteristic_uuid) != characteristic_uuids.end();
  };
  for (const auto& it : characteristics_) {
    const api::ble::GattCharacteristic& characteristic = it.first;
    if (characteristic.service_uuid == service_uuid &&
        contains(characteristic.uuid)) {
      return true;
    }
  }
  return false;
}

bool BleMedium::GattServer::UpdateCharacteristic(
    const api::ble::GattCharacteristic& characteristic,
    const nearby::ByteArray& value) {
  absl::MutexLock lock(mutex_);
  LOG(INFO) << "G3 Ble GattServer UpdateCharacteristic, characteristic=("
            << characteristic.service_uuid.Get16BitAsString() << ","
            << std::string(characteristic.uuid)
            << "), value = " << absl::BytesToHexString(value.data());
  characteristics_[characteristic] = value;
  return true;
}

absl::Status BleMedium::GattServer::NotifyCharacteristicChanged(
    const api::ble::GattCharacteristic& characteristic, bool confirm,
    const ByteArray& new_value) {
  absl::MutexLock lock(mutex_);
  LOG(INFO) << "G3 Ble GattServer NotifyCharacteristicChanged, characteristic=("
            << characteristic.service_uuid.Get16BitAsString() << ","
            << std::string(characteristic.uuid)
            << "), new_value = " << absl::BytesToHexString(new_value.data());
  absl::Status status = absl::NotFoundError(
      "Characteristic not subscribed to receive notification.");
  for (auto& it : subscribers_) {
    const SubscriberKey& key = it.first;
    if (key.second == characteristic) {
      it.second(new_value.AsStringView());
      status = absl::OkStatus();
    }
  }
  return status;
}

absl::StatusOr<ByteArray> BleMedium::GattServer::ReadCharacteristic(
    const api::ble::BlePeripheral::UniqueId remote_device_id,
    const api::ble::GattCharacteristic& characteristic, int offset) {
  {
    absl::MutexLock lock(mutex_);
    const auto it = characteristics_.find(characteristic);
    if (it == characteristics_.end()) {
      return absl::FailedPreconditionError(
          absl::StrCat(characteristic, " not found"));
    } else if (it->second.ok()) {
      return it->second;
    }
  }
  absl::StatusOr<ByteArray> result;
  CountDownLatch latch(1);
  callback_.on_characteristic_read_cb(
      remote_device_id, characteristic, offset,
      [&](absl::StatusOr<absl::string_view> data) {
        if (data.ok()) {
          result = ByteArray(std::string(*data));
        } else {
          result = data.status();
        }
        latch.CountDown();
      });
  latch.Await();
  return result;
}

absl::Status BleMedium::GattServer::WriteCharacteristic(
    const api::ble::BlePeripheral::UniqueId remote_device_id,
    const api::ble::GattCharacteristic& characteristic, int offset,
    absl::string_view data) {
  if (HasCharacteristic(characteristic)) {
    absl::Status result;
    CountDownLatch latch(1);
    callback_.on_characteristic_write_cb(remote_device_id, characteristic,
                                         offset, data,
                                         [&](absl::Status status) {
                                           result = status;
                                           latch.CountDown();
                                         });
    latch.Await();
    return result;
  }
  return absl::FailedPreconditionError(
      absl::StrCat(characteristic, " not found"));
}

bool BleMedium::GattServer::AddCharacteristicSubscription(
    const api::ble::BlePeripheral::UniqueId remote_device_id,
    const api::ble::GattCharacteristic& characteristic,
    absl::AnyInvocable<void(absl::string_view value)> callback) {
  absl::MutexLock lock(mutex_);
  const auto it = characteristics_.find(characteristic);
  if (it != characteristics_.end()) {
    subscribers_[SubscriberKey(remote_device_id, characteristic)] =
        std::move(callback);
    return true;
  }
  return false;
}

bool BleMedium::GattServer::RemoveCharacteristicSubscription(
    const api::ble::BlePeripheral::UniqueId remote_device_id,
    const api::ble::GattCharacteristic& characteristic) {
  absl::MutexLock lock(mutex_);
  const auto it = characteristics_.find(characteristic);
  if (it != characteristics_.end()) {
    subscribers_.erase(SubscriberKey(remote_device_id, characteristic));
    return true;
  }
  return false;
}

bool BleMedium::GattServer::HasCharacteristic(
    const api::ble::GattCharacteristic& characteristic) {
  absl::MutexLock lock(mutex_);
  return characteristics_.find(characteristic) != characteristics_.end();
}

void BleMedium::GattServer::Connect(GattClient* client) {
  absl::MutexLock lock(mutex_);
  connected_clients_.push_back(client);
}

void BleMedium::GattServer::Disconnect(GattClient* client) {
  absl::MutexLock lock(mutex_);
  connected_clients_.erase(
      std::remove(connected_clients_.begin(), connected_clients_.end(), client),
      connected_clients_.end());
}

void BleMedium::GattServer::Stop() {
  if (stopped_) return;
  absl::MutexLock lock(mutex_);
  stopped_ = true;
  for (auto& client : connected_clients_) {
    client->OnServerDisconnected();
  }
  characteristics_.clear();
}

BleMedium::GattClient::GattClient(
    api::ble::BlePeripheral::UniqueId peripheral_id,
    Borrowable<api::ble::GattServer*> gatt_server,
    api::ble::ClientGattConnectionCallback callback)
    : peripheral_id_(peripheral_id),
      gatt_server_(gatt_server),
      callback_(std::move(callback)) {
  Borrowed<api::ble::GattServer*> borrowed = gatt_server_.Borrow();
  if (borrowed) {
    BleMedium::GattServer* gatt_server =
        static_cast<BleMedium::GattServer*>(*borrowed);
    gatt_server->Connect(this);
  }
}

BleMedium::GattClient::~GattClient() {
  Borrowed<api::ble::GattServer*> borrowed = gatt_server_.Borrow();
  if (borrowed) {
    BleMedium::GattServer* gatt_server =
        static_cast<BleMedium::GattServer*>(*borrowed);
    gatt_server->Disconnect(this);
  }
}

bool BleMedium::GattClient::DiscoverServiceAndCharacteristics(
    const Uuid& service_uuid, const std::vector<Uuid>& characteristic_uuids) {
  LOG(INFO)
      << "G3 Ble GattClient DiscoverServiceAndCharacteristics, service_uuid="
      << service_uuid.Get16BitAsString();
  absl::MutexLock lock(mutex_);
  if (!is_connection_alive_) {
    return false;
  }
  Borrowed<api::ble::GattServer*> borrowed = gatt_server_.Borrow();
  if (!borrowed) {
    return false;
  }
  BleMedium::GattServer* gatt_server =
      static_cast<BleMedium::GattServer*>(*borrowed);
  return gatt_server->DiscoverBleMediumGattCharacteristics(
      service_uuid, characteristic_uuids);
}

std::optional<api::ble::GattCharacteristic>
BleMedium::GattClient::GetCharacteristic(const Uuid& service_uuid,
                                         const Uuid& characteristic_uuid) {
  absl::MutexLock lock(mutex_);
  LOG(INFO) << "G3 Ble GattClient GetCharacteristic, service_uuid="
            << service_uuid.Get16BitAsString()
            << ", characteristic_uuid=" << std::string(characteristic_uuid);
  if (!is_connection_alive_) {
    return std::nullopt;
  }
  // clang-format off
  api::ble::GattCharacteristic characteristic = {
      .uuid = characteristic_uuid,
      .service_uuid = service_uuid};
  // clang-format on
  Borrowed<api::ble::GattServer*> borrowed = gatt_server_.Borrow();
  if (!borrowed) {
    return std::nullopt;
  }
  BleMedium::GattServer* gatt_server =
      static_cast<BleMedium::GattServer*>(*borrowed);
  if (!gatt_server->HasCharacteristic(characteristic)) {
    LOG(WARNING) << "G3 Ble GattClient GetCharacteristic, characteristic=("
                 << characteristic.service_uuid.Get16BitAsString() << ","
                 << std::string(characteristic.uuid)
                 << ") not registered on GATT server";
    return std::nullopt;
  }
  LOG(INFO) << "G3 Ble GattClient GetCharacteristic, found characteristic=("
            << characteristic.service_uuid.Get16BitAsString() << ","
            << std::string(characteristic.uuid) << ")";

  return characteristic;
}

std::optional<std::string> BleMedium::GattClient::ReadCharacteristic(
    const api::ble::GattCharacteristic& characteristic) {
  absl::MutexLock lock(mutex_);
  if (!is_connection_alive_) {
    return std::nullopt;
  }
  Borrowed<api::ble::GattServer*> borrowed = gatt_server_.Borrow();
  if (!borrowed) {
    return std::nullopt;
  }
  BleMedium::GattServer* gatt_server =
      static_cast<BleMedium::GattServer*>(*borrowed);
  absl::StatusOr<ByteArray> value =
      gatt_server->ReadCharacteristic(peripheral_id_, characteristic,
                                      /*offset=*/0);
  if (!value.ok()) {
    LOG(INFO) << "G3 Ble ReadCharacteristic failed, characteristic=("
              << characteristic.service_uuid.Get16BitAsString() << ","
              << std::string(characteristic.uuid) << "), " << value.status();
    return std::nullopt;
  }
  LOG(INFO) << "G3 Ble ReadCharacteristic, characteristic=("
            << characteristic.service_uuid.Get16BitAsString() << ","
            << std::string(characteristic.uuid)
            << "), value = " << absl::BytesToHexString(value->data());
  return value->string_data();
}

bool BleMedium::GattClient::WriteCharacteristic(
    const api::ble::GattCharacteristic& characteristic, absl::string_view value,
    api::ble::GattClient::WriteType write_type) {
  absl::MutexLock lock(mutex_);
  if (!is_connection_alive_) {
    return false;
  }
  Borrowed<api::ble::GattServer*> borrowed = gatt_server_.Borrow();
  if (!borrowed) {
    return false;
  }
  BleMedium::GattServer* gatt_server =
      static_cast<BleMedium::GattServer*>(*borrowed);
  LOG(INFO) << "G3 Ble WriteCharacteristic, characteristic=("
            << characteristic.service_uuid.Get16BitAsString() << ","
            << std::string(characteristic.uuid)
            << "), value = " << absl::BytesToHexString(value);
  absl::Status status =
      gatt_server->WriteCharacteristic(peripheral_id_, characteristic,
                                       /*offset=*/0, value);
  if (!status.ok()) {
    LOG(WARNING) << "WriteCharacteristic failed with " << status;
  }
  return status.ok();
}

bool BleMedium::GattClient::SetCharacteristicSubscription(
    const api::ble::GattCharacteristic& characteristic, bool enable,
    absl::AnyInvocable<void(absl::string_view value)>
        on_characteristic_changed_cb) {
  absl::MutexLock lock(mutex_);
  if (!is_connection_alive_) {
    return false;
  }
  Borrowed<api::ble::GattServer*> borrowed = gatt_server_.Borrow();
  if (!borrowed) {
    return false;
  }
  BleMedium::GattServer* gatt_server =
      static_cast<BleMedium::GattServer*>(*borrowed);
  LOG(INFO) << "G3 Ble SetCharacteristicSubscription, characteristic=("
            << characteristic.service_uuid.Get16BitAsString() << ","
            << std::string(characteristic.uuid) << "), enable = " << enable;
  if (enable) {
    return gatt_server->AddCharacteristicSubscription(
        peripheral_id_, characteristic,
        std::move(on_characteristic_changed_cb));
  } else {
    return gatt_server->RemoveCharacteristicSubscription(peripheral_id_,
                                                         characteristic);
  }
}

void BleMedium::GattClient::Disconnect() {
  bool was_alive = is_connection_alive_.exchange(false);
  if (!was_alive) return;
  LOG(INFO) << "G3 Ble GattClient Disconnect";
  Borrowed<api::ble::GattServer*> borrowed = gatt_server_.Borrow();
  if (borrowed) {
    BleMedium::GattServer* gatt_server =
        static_cast<BleMedium::GattServer*>(*borrowed);
    gatt_server->Disconnect(this);
  }
}

void BleMedium::GattClient::OnServerDisconnected() {
  bool was_alive = is_connection_alive_.exchange(false);
  if (!was_alive) return;
  LOG(INFO) << "G3 Ble GattServer disconnected";
  if (callback_.disconnected_cb != nullptr) {
    callback_.disconnected_cb();
  }
}

std::unique_ptr<api::ble::BleServerSocket> BleMedium::OpenServerSocket(
    const std::string& service_id) {
  auto server_socket = std::make_unique<BleServerSocket>(&GetAdapter());
  server_socket->SetCloseNotifier([this, service_id]() {
    absl::MutexLock lock(mutex_);
    server_sockets_.erase(service_id);
  });
  LOG(INFO) << "G3 Ble Adding server socket: medium=" << this
            << ", service_id=" << service_id;
  absl::MutexLock lock(mutex_);
  server_sockets_.insert({service_id, server_socket.get()});
  return server_socket;
}

std::unique_ptr<api::ble::BleL2capServerSocket>
BleMedium::OpenL2capServerSocket(const std::string& service_id) {
  // TODO(mingshiouwu): add more codes for g3 testing.
  return nullptr;
}

std::unique_ptr<api::ble::BleSocket> BleMedium::Connect(
    const std::string& service_id, TxPowerLevel tx_power_level,
    api::ble::BlePeripheral::UniqueId remote_peripheral_id,
    CancellationFlag* cancellation_flag) {
  LOG(INFO) << "G3 Ble Connect [self]: medium=" << this
            << ", adapter=" << &GetAdapter()
            << ", peripheral id=" << adapter_.GetUniqueId()
            << ", service_id=" << service_id;
  // First, find an instance of remote medium, that exposed this peripheral.
  BleMedium* remote_medium = dynamic_cast<BleMedium*>(
      MediumEnvironment::Instance().FindBleMedium(remote_peripheral_id));
  if (remote_medium == nullptr) {
    LOG(INFO) << "Peripheral not found, id= " << remote_peripheral_id;
    return nullptr;
  }

  BleServerSocket* remote_server_socket = nullptr;
  LOG(INFO) << "G3 Ble Connect [peer]: medium=" << remote_medium
            << ", peripheral=" << &remote_peripheral_id
            << ", service_id=" << service_id;
  // Then, find our server socket context in this medium.
  {
    absl::MutexLock medium_lock(remote_medium->mutex_);
    auto item = remote_medium->server_sockets_.find(service_id);
    remote_server_socket =
        item != remote_medium->server_sockets_.end() ? item->second : nullptr;
    if (remote_server_socket == nullptr) {
      LOG(ERROR) << "G3 Ble Failed to find Ble Server socket: service_id="
                 << service_id;
      return nullptr;
    }
  }

  if (cancellation_flag->Cancelled()) {
    LOG(ERROR) << "G3 BLE Connect: Has been cancelled: "
                  "service_id="
               << service_id;
    return nullptr;
  }

  CancellationFlagListener listener(cancellation_flag,
                                    [&remote_server_socket]() {
                                      LOG(INFO) << "G3 Ble Cancel Connect.";
                                      if (remote_server_socket != nullptr) {
                                        remote_server_socket->Close();
                                      }
                                    });

  auto socket = std::make_unique<BleSocket>(&GetAdapter());
  // Finally, Request to connect to this socket.
  if (!remote_server_socket->Connect(*socket)) {
    LOG(ERROR) << "G3 Ble Failed to connect to existing Ble "
                  "Server socket: service_id="
               << service_id;
    return nullptr;
  }
  LOG(INFO) << "G3 Ble Connect to socket=" << socket.get();
  return socket;
}

}  // namespace g3
}  // namespace nearby