#include <BLEDevice.h>
#include <BLEUtils.h>
#include <BLEScan.h>
#include <BLESecurity.h>

const char* LOG_TAG = "ESP32OmBle";

const BLEUUID serviceUUID("ecbe3980-c9a2-11e1-b1bd-0002a5d5c51b");
const BLEUUID CommandChannelUUID("b305b680-aee7-11e1-a730-0002a5d5c51b");

void notifyCallback(
  BLERemoteCharacteristic* pBLERemoteCharacteristic,
  uint8_t* pData,
  size_t length,
  bool isNotify) {
    Serial.print("Notify callback for characteristic ");
    Serial.print(pBLERemoteCharacteristic->getUUID().toString().c_str());
    Serial.print(" of data length ");
    Serial.println(length);
    Serial.print("data: ");
    for(int i = 0; i<length; i++){
      Serial.print(pData[i], HEX);
    }
    
}

class OmronBleSecCallbacks : public BLESecurityCallbacks {
  uint32_t onPassKeyRequest(){
    ESP_LOGE(LOG_TAG, "onPassKeyRequest");
    return 0000;
  }
  void onPassKeyNotify(uint32_t pass_key){
    ESP_LOGE(LOG_TAG, "The passkey Notify number:%d", pass_key);
  }
  bool onConfirmPIN(uint32_t pass_key){
    ESP_LOGI(LOG_TAG, "The passkey YES/NO number:%d", pass_key);
    vTaskDelay(5000);
    return true;
  }
  bool onSecurityRequest(){
    ESP_LOGI(LOG_TAG, "Security Request");
    return true;
  }
  void onAuthenticationComplete(esp_ble_auth_cmpl_t auth_cmpl){
    if(auth_cmpl.success){
      ESP_LOGI(LOG_TAG, "Auth success, remote BD_ADDR:");
      esp_log_buffer_hex(LOG_TAG, auth_cmpl.bd_addr, sizeof(auth_cmpl.bd_addr));
      ESP_LOGI(LOG_TAG, "address type = %d", auth_cmpl.addr_type);
      ESP_LOGI(LOG_TAG, "auth mode = %d", auth_cmpl.auth_mode);
    }else{
      ESP_LOGI(LOG_TAG, "auth fail reason: %d", auth_cmpl.fail_reason);
    }
    
  }
};

void setup() {
}

void loop() {

  Serial.begin(115200);
  ESP_LOGI(LOG_TAG, "ESP32 bridge online");
  BLEDevice::init("ESP32");
  BLEDevice::setEncryptionLevel(ESP_BLE_SEC_ENCRYPT);
  BLEDevice::setSecurityCallbacks(new OmronBleSecCallbacks());

  //setup encryption for successful pairing
  BLESecurity *securityP = new BLESecurity();
  securityP->setAuthenticationMode(ESP_LE_AUTH_REQ_SC_BOND); //also available ESP_LE_AUTH_NO_BOND, ESP_LE_AUTH_REQ_SC_MITM, ESP_LE_AUTH_BOND, ESP_LE_AUTH_NO_BOND
  ESP_LOGI(LOG_TAG, "try bonding");
  securityP->setCapability(ESP_IO_CAP_NONE);
  securityP->setInitEncryptionKey(ESP_BLE_ENC_KEY_MASK | ESP_BLE_ID_KEY_MASK);
  securityP->setRespEncryptionKey(ESP_BLE_ENC_KEY_MASK | ESP_BLE_ID_KEY_MASK);
  securityP->setKeySize(16);
  securityP->setRespEncryptionKey(ESP_BLE_ENC_KEY_MASK | ESP_BLE_ID_KEY_MASK);

  BLEClient* clientP = BLEDevice::createClient();
  BLEAddress omronMacAddress = BLEAddress("11:22:33:44:55:66");
  clientP->connect(omronMacAddress);
  BLERemoteService* RemoteServiceP = clientP->getService(serviceUUID);
  BLERemoteCharacteristic* unlockChannelCharacteristicP = RemoteServiceP->getCharacteristic(CommandChannelUUID);
  
  if(unlockChannelCharacteristicP->canNotify()){
    unlockChannelCharacteristicP->registerForNotify(notifyCallback);
  }else{
      ESP_LOGI(LOG_TAG, "noNotify possible");
  }
  uint8_t bytes[] = {0x02, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00};
  unlockChannelCharacteristicP->writeValue(bytes, sizeof(bytes), true);
  
  delay(200000);
}