/******************************************************************************
 *
 *  Copyright 2015 Google, Inc.
 *
 *  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.
 *
 ******************************************************************************/

#include "gatt/gatt_test.h"

#include "adapter/bluetooth_test.h"
#include "types/bluetooth/uuid.h"
#include "types/raw_address.h"

namespace bttest {

static GattTest* instance = nullptr;

void RegisterClientCallback(int status, int clientIf, const bluetooth::Uuid& /*app_uuid*/) {
  instance->status_ = status;
  instance->client_interface_id_ = clientIf;
  semaphore_post(instance->register_client_callback_sem_);
}

void ScanResultCallback(uint16_t /*ble_evt_type*/, uint8_t /*addr_type*/, RawAddress* /*bda*/,
                        uint8_t /*ble_primary_phy*/, uint8_t /*ble_secondary_phy*/,
                        uint8_t /*ble_advertising_sid*/, int8_t /*ble_tx_power*/, int8_t /*rssi*/,
                        uint16_t /*ble_periodic_adv_int*/, std::vector<uint8_t> /*adv_data*/,
                        RawAddress* /*original_bda*/) {
  semaphore_post(instance->scan_result_callback_sem_);
}

// GATT server callbacks
void RegisterServerCallback(int status, int server_if, const bluetooth::Uuid& /*uuid*/) {
  instance->status_ = status;
  instance->server_interface_id_ = server_if;
  semaphore_post(instance->register_server_callback_sem_);
}

void ServiceAddedCallback(int status, int server_if, const btgatt_db_element_t* service,
                          size_t /*service_count*/) {
  instance->status_ = status;
  instance->server_interface_id_ = server_if;
  instance->service_handle_ = service[0].attribute_handle;
  semaphore_post(instance->service_added_callback_sem_);
}

void ServiceStoppedCallback(int status, int server_if, int srvc_handle) {
  instance->status_ = status;
  instance->server_interface_id_ = server_if;
  instance->service_handle_ = srvc_handle;
  semaphore_post(instance->service_stopped_callback_sem_);
}

void ServiceDeletedCallback(int status, int server_if, int srvc_handle) {
  instance->status_ = status;
  instance->server_interface_id_ = server_if;
  instance->service_handle_ = srvc_handle;
  semaphore_post(instance->service_deleted_callback_sem_);
}

static const btgatt_scanner_callbacks_t scanner_callbacks = {
        .scan_result_cb = ScanResultCallback,
};

static const btgatt_client_callbacks_t client_callbacks = {
        .register_client_cb = RegisterClientCallback,
};

static const btgatt_server_callbacks_t server_callbacks = {
        .register_server_cb = RegisterServerCallback,
        .service_added_cb = ServiceAddedCallback,
        .service_stopped_cb = ServiceStoppedCallback,
        .service_deleted_cb = ServiceDeletedCallback,
};

static const btgatt_callbacks_t callbacks = {
        sizeof(btgatt_callbacks_t),
        &client_callbacks,
        &server_callbacks,
        &scanner_callbacks,
};

void GattTest::SetUp() {
  gatt_interface_ = nullptr;

  client_interface_id_ = 0;
  server_interface_id_ = 0;
  service_handle_ = 0;
  characteristic_handle_ = 0;
  descriptor_handle_ = 0;
  status_ = 0;

  BluetoothTest::SetUp();
  ASSERT_EQ(bt_interface()->enable(), BT_STATUS_SUCCESS);
  semaphore_wait(adapter_state_changed_callback_sem_);
  EXPECT_TRUE(GetState() == BT_STATE_ON);

  gatt_interface_ = reinterpret_cast<const btgatt_interface_t*>(
          bt_interface()->get_profile_interface(BT_PROFILE_GATT_ID));
  ASSERT_NE(nullptr, gatt_interface_);
  instance = this;
  auto status = gatt_interface_->init(&callbacks);
  ASSERT_EQ(status, BT_STATUS_SUCCESS);
}

void GattTest::TearDown() {
  instance = nullptr;
  gatt_interface_ = nullptr;

  ASSERT_EQ(bt_interface()->disable(), BT_STATUS_SUCCESS);
  semaphore_wait(adapter_state_changed_callback_sem_);
  BluetoothTest::TearDown();
}

const BleScannerInterface* GattTest::gatt_scanner_interface() { return gatt_interface_->scanner; }

const btgatt_client_interface_t* GattTest::gatt_client_interface() {
  return gatt_interface_->client;
}

const btgatt_server_interface_t* GattTest::gatt_server_interface() {
  return gatt_interface_->server;
}

}  // namespace bttest