/*
 * This file is part of the MicroPython project, http://micropython.org/
 *
 * The MIT License (MIT)
 *
 * Copyright (c) 2016 Glenn Ruben Bakke
 * Copyright (c) 2018 Artur Pacholec
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 * THE SOFTWARE.
 */

#include <stdio.h>
#include <string.h>

#include "ble.h"
#include "ble_drv.h"
#include "nrfx_power.h"
#include "nrf_nvic.h"
#include "nrf_sdm.h"
#include "py/runtime.h"
#include "shared-bindings/bleio/Adapter.h"

#include "supervisor/usb.h"

STATIC void softdevice_assert_handler(uint32_t id, uint32_t pc, uint32_t info) {
    mp_raise_msg_varg(&mp_type_AssertionError,
                      translate("Soft device assert, id: 0x%08lX, pc: 0x%08lX"), id, pc);
}

STATIC uint32_t ble_stack_enable(void) {
    nrf_clock_lf_cfg_t clock_config = {
        .source = NRF_CLOCK_LF_SRC_XTAL,
        .accuracy = NRF_CLOCK_LF_ACCURACY_20_PPM
    };

    uint32_t err_code = sd_softdevice_enable(&clock_config, softdevice_assert_handler);
    if (err_code != NRF_SUCCESS)
        return err_code;

    err_code = sd_nvic_EnableIRQ(SD_EVT_IRQn);
    if (err_code != NRF_SUCCESS)
        return err_code;

    // Start with no event handlers, etc.
    ble_drv_reset();

    uint32_t app_ram_start;
    app_ram_start = 0x20004000;

    ble_cfg_t ble_conf;
    ble_conf.conn_cfg.conn_cfg_tag = BLE_CONN_CFG_TAG_CUSTOM;
    ble_conf.conn_cfg.params.gap_conn_cfg.conn_count = BLE_GAP_CONN_COUNT_DEFAULT;
    ble_conf.conn_cfg.params.gap_conn_cfg.event_length = BLE_GAP_EVENT_LENGTH_DEFAULT;
    err_code = sd_ble_cfg_set(BLE_CONN_CFG_GAP, &ble_conf, app_ram_start);
    if (err_code != NRF_SUCCESS)
        return err_code;

    memset(&ble_conf, 0, sizeof(ble_conf));
    ble_conf.gap_cfg.role_count_cfg.periph_role_count = 1;
    ble_conf.gap_cfg.role_count_cfg.central_role_count = 1;
    err_code = sd_ble_cfg_set(BLE_GAP_CFG_ROLE_COUNT, &ble_conf, app_ram_start);
    if (err_code != NRF_SUCCESS)
        return err_code;

    memset(&ble_conf, 0, sizeof(ble_conf));
    ble_conf.conn_cfg.conn_cfg_tag = BLE_CONN_CFG_TAG_CUSTOM;
    ble_conf.conn_cfg.params.gatts_conn_cfg.hvn_tx_queue_size = MAX_TX_IN_PROGRESS;
    err_code = sd_ble_cfg_set(BLE_CONN_CFG_GATTS, &ble_conf, app_ram_start);
    if (err_code != NRF_SUCCESS)
        return err_code;

    err_code = sd_ble_enable(&app_ram_start);

    return err_code;
}

void common_hal_bleio_adapter_set_enabled(bool enabled) {
    const bool is_enabled = common_hal_bleio_adapter_get_enabled();

    // Don't enable or disable twice
    if ((is_enabled && enabled) || (!is_enabled && !enabled)) {
        return;
    }

    uint32_t err_code;
    if (enabled) {
        // The SD takes over the POWER module and will fail if the module is already in use.
        // Occurs when USB is initialized previously
        nrfx_power_uninit();

        err_code = ble_stack_enable();

        // Re-init USB hardware
        init_usb_hardware();
    } else {
        err_code = sd_softdevice_disable();

        // Re-init USB hardware
        init_usb_hardware();
    }

    if (err_code != NRF_SUCCESS) {
        mp_raise_OSError_msg(translate("Failed to change softdevice state"));
    }
}

bool common_hal_bleio_adapter_get_enabled(void) {
    uint8_t is_enabled;

    const uint32_t err_code = sd_softdevice_is_enabled(&is_enabled);
    if (err_code != NRF_SUCCESS) {
        mp_raise_OSError_msg(translate("Failed to get softdevice state"));
    }

    return is_enabled;
}

void common_hal_bleio_adapter_get_address(bleio_address_obj_t *address) {
    ble_gap_addr_t local_address;
    uint32_t err_code;

    common_hal_bleio_adapter_set_enabled(true);
    err_code = sd_ble_gap_addr_get(&local_address);

    if (err_code != NRF_SUCCESS) {
        mp_raise_OSError_msg(translate("Failed to get local address"));
    }

    address->type = local_address.addr_type;

    mp_buffer_info_t buf_info;
    mp_get_buffer_raise(address, &buf_info, MP_BUFFER_READ);
    memcpy(address->bytes, buf_info.buf, NUM_BLEIO_ADDRESS_BYTES);
}