/*
 * Copyright 2024 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
 *
 *     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 "comm/ble/gap_le_connection.h"
#include "comm/bt_lock.h"

#include "console/prompt.h"
#include "services/common/bluetooth/bluetooth_ctl.h"
#include "services/common/bluetooth/bluetooth_persistent_storage.h"
#include "services/common/bluetooth/bt_compliance_tests.h"
#include "services/common/bluetooth/local_id.h"
#include "services/common/bluetooth/pairability.h"
#include "services/common/shared_prf_storage/shared_prf_storage.h"
#include "util/string.h"

#include <bluetooth/bluetooth_types.h>
#include <bluetooth/bt_test.h>
#include <bluetooth/classic_connect.h>
#include <bluetooth/id.h>

#include <stdlib.h>

void command_bt_print_mac(void) {
  char addr_hex_str[BT_ADDR_FMT_BUFFER_SIZE_BYTES];
  bt_local_id_copy_address_hex_string(addr_hex_str);
  prompt_send_response(addr_hex_str);
}

//! Overrides the BD ADDR of the Bluetooth controller for test automation purposes
//! @param bd_addr String of 12 hex characters (6 bytes) of the Bluetooth device address
//! @note To undo the change, call this with all zeroes.
//! @note The change will take effect when the Bluetooth is (re)enabled.
void command_bt_set_addr(const char *bd_addr_str) {
  char buffer[32];
  BTDeviceAddress bd_addr;
  if (convert_bt_addr_hex_str_to_bd_addr(bd_addr_str, (uint8_t *) &bd_addr, sizeof(bd_addr))) {
    bt_driver_test_set_spoof_address(&bd_addr);
    prompt_send_response_fmt(buffer, 32, BT_DEVICE_ADDRESS_FMT, BT_DEVICE_ADDRESS_XPLODE(bd_addr));
  } else {
    prompt_send_response("?");
  }
}

//! @param bt_name A custom Bluetooth device name.
void command_bt_set_name(const char *bt_name) {
  bt_local_id_set_device_name(bt_name);
}

// BT FCC tests
void command_bt_test_start(void) {
  // take down the BT stack and put the OS in a mode where it will not
  // interfere with the BT testing.
  bt_test_start();
}

void command_bt_test_stop(void) {
  // restore the watch to normal operation
  bt_test_stop();
}

void command_bt_test_hci_passthrough(void) {
  bt_test_enter_hci_passthrough();
}

void command_bt_test_bt_sig_rf_mode(void) {
  if (bt_test_bt_sig_rf_test_mode()) {
    prompt_send_response("BT SIG RF Test Mode Enabled");
  } else {
    prompt_send_response("Failed to enter BT SIG RF Test Mode");
  }
}

void command_bt_prefs_wipe(void) {
  bt_driver_classic_disconnect(NULL);
  bt_persistent_storage_delete_all_pairings();
}

void command_bt_sprf_nuke(void) {
  shared_prf_storage_wipe_all();
#if RECOVERY_FW
  // Reset system to get caches (in s_intents, s_connections and controller-side caches) in sync.
  extern void factory_reset_set_reason_and_reset(void);
  factory_reset_set_reason_and_reset();
#endif
}

#ifdef RECOVERY_FW
void command_bt_status(void) {
  char buffer[64];

  prompt_send_response_fmt(buffer, sizeof(buffer), "Alive: %s",
                           bt_ctl_is_bluetooth_running() ? "yes" : "no");

  const char *prefix = "BT Chip Info: ";
  size_t prefix_length = strlen(prefix);
  strncpy(buffer, prefix, sizeof(buffer));
  bt_driver_id_copy_chip_info_string(buffer + prefix_length,
                                     sizeof(buffer) - prefix_length);
  prompt_send_response(buffer);

  char name[BT_DEVICE_NAME_BUFFER_SIZE];
  bt_lock();
  bool connected = bt_driver_classic_copy_connected_device_name(name);
  if (!connected) {
    // Try LE:
    GAPLEConnection *connection = gap_le_connection_any();
    if (connection) {
      const char *device_name = connection->device_name ?: "<Unknown>";
      strncpy(name, device_name, BT_DEVICE_NAME_BUFFER_SIZE);
      name[BT_DEVICE_NAME_BUFFER_SIZE - 1] = '\0';
      connected = true;
    }
  }
  bt_unlock();

  prompt_send_response_fmt(buffer, sizeof(buffer), "Connected: %s", connected ? "yes" : "no");
  if (connected) {
    prompt_send_response_fmt(buffer, sizeof(buffer), "Device: %s", name);
  }
}
#endif // RECOVERY_FW