// uncomment the following line to use NimBLE library
// #define USE_NIMBLE

#ifndef ESP32_BLE_KEYBOARD_H
#define ESP32_BLE_KEYBOARD_H
#include "sdkconfig.h"
#if defined(CONFIG_BT_ENABLED)
#define USE_NIMBLE
#if __has_include(<NimBLEExtAdvertising.h>)
#define NIMBLE_V2_PLUS 1
#endif
#include "NimBLECharacteristic.h"
#include "NimBLEHIDDevice.h"
#ifdef NIMBLE_V2_PLUS
#include "NimBLEAdvertising.h"
#include "NimBLEServer.h"
#endif
#define BLEDevice NimBLEDevice
#define BLEServerCallbacks NimBLEServerCallbacks
#define BLECharacteristicCallbacks NimBLECharacteristicCallbacks
#define BLEHIDDevice NimBLEHIDDevice
#define BLECharacteristic NimBLECharacteristic
#define BLEAdvertising NimBLEAdvertising
#define BLEServer NimBLEServer

#include "Bad_Usb_Lib.h"
#include "Print.h"
#include "keys.h"

#define BLE_KEYBOARD_VERSION "0.0.4"
#define BLE_KEYBOARD_VERSION_MAJOR 0
#define BLE_KEYBOARD_VERSION_MINOR 0
#define BLE_KEYBOARD_VERSION_REVISION 4

class BleKeyboard : public BLEServerCallbacks, public BLECharacteristicCallbacks, public HIDInterface {
private:
    BLEHIDDevice *hid;
    BLECharacteristic *inputKeyboard;
    BLECharacteristic *outputKeyboard;
    BLECharacteristic *inputMediaKeys;
    BLEAdvertising *advertising;
    NimBLEConnInfo *activeConnection = nullptr;
    NimBLEServer *pServer;
    KeyReport _keyReport;
    MediaKeyReport _mediaKeyReport;
    String deviceName;
    String deviceManufacturer;
    uint8_t batteryLevel;
    bool connected = false;
    uint32_t _delay_ms = 7;
    void delay_ms(uint64_t ms);

    uint16_t vid = 0x05ac;
    uint16_t pid = 0x820a;
    uint16_t version = 0x0210;
    // Appearance list as shown in the 2.6.3 seccion of the list below
    // https://www.bluetooth.com/wp-content/uploads/Files/Specification/Assigned_Numbers.html
    uint16_t appearance = 0x03C1;

    const uint8_t *_asciimap;

public:
    BleKeyboard(
        String deviceName = "ESP32 Keyboard", String deviceManufacturer = "Espressif",
        uint8_t batteryLevel = 100
    );
    void begin(const uint8_t *layout = KeyboardLayout_en_US) override { begin(layout, HID_KEYBOARD); };
    void begin(const uint8_t *layout, uint16_t showAs);
    void setLayout(const uint8_t *layout = KeyboardLayout_en_US) { _asciimap = layout; }
    void end(void) override;
    void sendReport(KeyReport *keys);
    void sendReport(MediaKeyReport *keys);
    size_t press(uint8_t k) override;
    size_t press(const MediaKeyReport k);
    size_t release(uint8_t k) override;
    size_t release(const MediaKeyReport k);
    size_t write(uint8_t c) override;
    size_t write(const MediaKeyReport c);
    size_t write(const uint8_t *buffer, size_t size) override;
    void releaseAll(void) override;
    bool isConnected(void);
    void setBatteryLevel(uint8_t level);
    void setName(String deviceName);
    void setDelay(uint32_t ms);
    void setAppearence(uint16_t v) { appearance = v; }
    void setRandomUUID(void) { _randUUID = !_randUUID; };
    bool getRandomUUID() { return _randUUID; };
    uint16_t getAppearence() { return appearance; }

    void set_vendor_id(uint16_t vid);
    void set_product_id(uint16_t pid);
    void set_version(uint16_t version);

protected:
    bool _randUUID = false;
#ifndef NIMBLE_V2_PLUS
    virtual void onAuthenticationComplete(ble_gap_conn_desc *desc);
    virtual void onConnect(BLEServer *pServer) override;
    virtual void onDisconnect(BLEServer *pServer) override;
    virtual void onWrite(BLECharacteristic *me) override;
    virtual void
    onSubscribe(NimBLECharacteristic *pCharacteristic, ble_gap_conn_desc *desc, uint16_t subValue) override;
#else

    class ServerCallbacks : public NimBLEServerCallbacks {
    private:
        BleKeyboard *parent;

    public:
        ServerCallbacks(BleKeyboard *kb) : parent(kb) {}
        void onConnect(NimBLEServer *pServer, NimBLEConnInfo &connInfo) override;
        void onDisconnect(NimBLEServer *pServer, NimBLEConnInfo &connInfo, int reason) override;
        void onAuthenticationComplete(NimBLEConnInfo &connInfo) override;
    };
    class CharacteristicCallbacks : public NimBLECharacteristicCallbacks {
    private:
        BleKeyboard *parent;

    public:
        CharacteristicCallbacks(BleKeyboard *kb) : parent(kb) {}
        void onWrite(NimBLECharacteristic *pCharacteristic, NimBLEConnInfo &connInfo) override;
        void onSubscribe(
            NimBLECharacteristic *pCharacteristic, NimBLEConnInfo &connInfo, uint16_t subValue
        ) override;
    };
    uint8_t getSubscribedCount() { return m_subCount; }

private:
    uint8_t m_subCount{0};
#endif
};

#endif // CONFIG_BT_ENABLED
#endif // ESP32_BLE_KEYBOARD_H