#ifndef __DEVICE_H__
#define __DEVICE_H__

#include <vector>
#include <memory>
#include <cstring>
#include "device/include/neodevice.h"
#include "communication/include/communication.h"
#include "third-party/concurrentqueue/concurrentqueue.h"

namespace icsneo {

class Device {
public:
	Device(neodevice_t neodevice = {}) {
		data = neodevice;
		data.device = this;
		setProductName("undefined");
	}
	virtual ~Device() {
		disableMessagePolling();
		close();
	}

	static std::string SerialNumToString(uint32_t serial);
	static uint32_t SerialStringToNum(const std::string& serial);
	static bool SerialStringIsNumeric(const std::string& serial);

	std::string getProductName() const { return data.type; }
	uint16_t getUSBProductId() const { return usbProductId; }
	std::string getSerial() const { return data.serial; }
	uint32_t getSerialNumber() const { return Device::SerialStringToNum(getSerial()); }
	const neodevice_t& getNeoDevice() const { return data; }

	virtual bool open();
	virtual bool close();
	virtual bool isOnline() const { return online; }
	virtual bool goOnline();
	virtual bool goOffline();

	// Message polling related functions
	void enableMessagePolling();
	bool disableMessagePolling();
	std::vector<std::shared_ptr<Message>> getMessages();
	bool getMessages(std::vector<std::shared_ptr<Message>>& container, size_t limit = 0);
	size_t getPollingMessageLimit() { return pollingMessageLimit; }
	void setPollingMessageLimit(size_t newSize) {
		pollingMessageLimit = newSize;
		enforcePollingMessageLimit();
	}

protected:
	uint16_t usbProductId = 0;
	bool online = false;
	int messagePollingCallbackID = 0;
	std::shared_ptr<Communication> com;

	neodevice_t& getWritableNeoDevice() { return data; }
	void setProductName(const std::string& newName) {
		#pragma warning( disable : 4996 )
		auto copied = newName.copy(data.type, sizeof(data.type) - 1);
		data.type[copied] = '\0';
	}

private:
	neodevice_t data;

	size_t pollingMessageLimit = 20000;
	moodycamel::ConcurrentQueue<std::shared_ptr<Message>> pollingContainer;
	void enforcePollingMessageLimit();
};

};

#endif