#ifndef BLUETOOTH_H
#define BLUETOOTH_H

#include <boost/uuid/uuid.hpp>
#include <experimental/optional>
#include <iosfwd>
#include <stdexcept>
#include <vector>
#include <cstdint>
#include <map>

#include "ByteBuffer.h"

namespace trygvis {
namespace bluetooth {

using namespace std;
template<typename T>
using o = std::experimental::optional<T>;

struct SpecUuid {
public:
    SpecUuid(uint16_t value) : value(value) {
    }

    uint16_t value;
};

namespace uuids {
const SpecUuid PRIMARY_SERVICE = SpecUuid(0x2800);
const SpecUuid SECONDARY_SERVICE = SpecUuid(0x2801);
const SpecUuid CHARACTERISTIC = SpecUuid(0x2803);
}

class BluetoothAdapter;

class BluetoothDevice;

class BluetoothGattService;

class BluetoothGattCharacteristic;

typedef shared_ptr<BluetoothGattCharacteristic> BluetoothGattCharacteristicPtr;
typedef shared_ptr<BluetoothGattService> BluetoothGattServicePtr;

class BluetoothException : public runtime_error {
public:
    BluetoothException(const BluetoothAdapter *adapter, string const &what) :
        runtime_error(what), adapter(adapter), device(nullptr) {
    }

    BluetoothException(const BluetoothDevice *device, string const &what) :
        runtime_error(what), adapter(nullptr), device(device) {
    }

    BluetoothException(string const &what) :
        runtime_error(what), adapter(nullptr), device(nullptr) {
    }

    const BluetoothAdapter *adapter;
    const BluetoothDevice *device;
};

class Mac {
public:
    Mac(uint8_t _5, uint8_t _4, uint8_t _3, uint8_t _2, uint8_t _1, uint8_t _0) {
        bytes[5] = _5;
        bytes[4] = _4;
        bytes[3] = _3;
        bytes[2] = _2;
        bytes[1] = _1;
        bytes[0] = _0;
    };

    string str() const;

    bool operator==(Mac &other) const;

    bool operator!=(Mac &other) const;

    void copy(uint8_t &_5, uint8_t &_4, uint8_t &_3, uint8_t &_2, uint8_t &_1, uint8_t &_0) const;

    static Mac parseMac(string s);

    friend bool operator<(const Mac &a, const Mac &b);

private:
    uint8_t bytes[6];
};

template<typename T>
class Iterator {
public:
private:
};

template<typename T>
class Collection {
public:
    Iterator<T> begin();

    Iterator<T> end();
};

class BluetoothGattCharacteristic {
public:
    virtual ~BluetoothGattCharacteristic() {
    };

    virtual BluetoothGattServicePtr getService() const = 0;

    virtual uint16_t getHandle() const = 0;

    virtual const boost::uuids::uuid getUuid() const = 0;

    virtual uint8_t getProperties() const = 0;

    virtual uint16_t getValueHandle() const = 0;
};

class BluetoothGattService {
public:
    virtual ~BluetoothGattService() {
    };

    virtual BluetoothDevice &getDevice() const = 0;

    virtual boost::uuids::uuid getUuid() const = 0;

    virtual uint16_t getHandle() const = 0;

    virtual uint16_t getEndGroupHandle() const = 0;

    virtual vector<shared_ptr<BluetoothGattCharacteristic>> getCharacteristics() const = 0;

    virtual o<BluetoothGattCharacteristicPtr> findCharacteristic(boost::uuids::uuid uuid) = 0;
};

class BluetoothGatt {
public:
    BluetoothGatt();

    virtual ~BluetoothGatt();

    // No copying
    BluetoothGatt(const BluetoothGatt &) = delete;

    BluetoothGatt &operator=(const BluetoothGatt &) = delete;

    virtual BluetoothDevice &getDevice() const = 0;

    virtual bool isConnected() const = 0;

    virtual void writeValue(const BluetoothGattCharacteristicPtr &c, const ByteBuffer &bytes) = 0;

    virtual ByteBuffer readValue(const BluetoothGattCharacteristicPtr &c) = 0;

    virtual void discoverServices() = 0;

    virtual vector<shared_ptr<BluetoothGattService>> getServices() const = 0;

    virtual o<BluetoothGattServicePtr> findService(boost::uuids::uuid uuid) = 0;
};

class BluetoothDevice {
public:
    BluetoothDevice();

    virtual ~BluetoothDevice();

    virtual Mac const &getMac() = 0;

    virtual BluetoothAdapter &getAdapter() = 0;

    virtual shared_ptr<BluetoothGatt> connectGatt() = 0;
};

class BluetoothAdapter {
public:
    virtual Mac const &getMac() = 0;

    virtual void startScan() = 0;

    virtual void stopScan() = 0;

    virtual void runScan(std::function<void(const shared_ptr<BluetoothDevice> &device)>) = 0;

    virtual shared_ptr<BluetoothDevice> getDevice(Mac &mac) = 0;

protected:
    BluetoothAdapter();

    virtual ~BluetoothAdapter();
};

/**
* Right this is only RAII support to properly call shutdown().
*
* TODO: move getAdapter() here. Make this control all shutdowns.
*/
class BluetoothSystem {
public:
    BluetoothSystem();

    ~BluetoothSystem();

    shared_ptr<BluetoothAdapter> getAdapter(string name);

private:
    map<string, shared_ptr<BluetoothAdapter>> adapters;
};

enum AttPduType {
    ERROR = 0x00,
    INVALID_HANDLE = 0x01,
    READ_BY_TYPE_REQ = 0x08,
    READ_BY_TYPE_RES = 0x09,
    READ_REQ = 0x0a,
    READ_RES = 0x0b,
    READ_BY_GROUP_TYPE_REQ = 0x10,
    READ_BY_GROUP_TYPE_RES = 0x11,
    WRITE_REQ = 0x12,
    WRITE_RES = 0x13,
};

class AttributeData;

class AttPdu {
public:
    AttPdu(ByteBuffer &bytes);

    AttPdu(ByteBuffer &bytes, AttPduType type);

    AttPduType getType();

    static vector<AttributeData> parseReadByGroupType(ByteBuffer &bytes);

    static vector<AttributeData> parseReadByType(ByteBuffer &bytes);

    static void parseRead(ByteBuffer &bytes);

    static void parseWrite(ByteBuffer &bytes);

    static void makeReadByGroupType(ByteBuffer &bytes, uint16_t startHandle, uint16_t endHandle, SpecUuid uuid);

    static void makeReadByType(ByteBuffer &bytes, uint16_t startHandle, uint16_t endHandle, SpecUuid uuid);

    static void makeRead(ByteBuffer &bytes, uint16_t handle);

    static void makeWrite(ByteBuffer &req, uint16_t handle, const ByteBuffer &bytes);

private:
//    static void checkType(ByteBuffer &bytes, AttPduType type);

    static vector<AttributeData> parse(ByteBuffer &bytes, AttPduType type);

    ByteBuffer &bytes;
};

class AttributeData {
public:
    ~AttributeData();

    static AttributeData fromByteBuffer(ByteBuffer &value);

    const uint16_t handle;
    ByteBuffer value;

private:
    AttributeData(uint16_t handle, ByteBuffer value);
};

boost::uuids::uuid makeUuid(const boost::uuids::uuid base, uint8_t a, uint8_t b);

}
}

#endif