/*!
*
* \author: einhornimmond
*
* \date: 04.04.19
*
* \brief: Config Basic Structure
*/

#ifndef DR_LUA_WEB_MODULE_STRUCTURES_MYSQL_QUERY_RESULT__H
#define DR_LUA_WEB_MODULE_STRUCTURES_MYSQL_QUERY_RESULT__H

#include <list>
#include <string.h>
#include "../lib/ErrorList.h"
#include "mysql.h"

#include "Poco/Data/Session.h"


enum MysqlRowType {
	MYSQL_ROW_STRING,
	MYSQL_ROW_INT, // 32 Bit
	MYSQL_ROW_LONG, // 64 Bit
	MYSQL_ROW_DECIMAL, // double
	MYSQL_ROW_TIMESTAMP,
	MYSQL_ROW_DATETIME,
	MYSQL_ROW_BINARY,
	MYSQL_ROW_NULL,
	MYSQL_ROW_TYPE_COUNT,
	MYSQL_ROW_TYPE_NONE
};

struct MysqlTableColumn
{
	MysqlTableColumn()
		: type(MYSQL_ROW_NULL), name("null") {}
	MysqlRowType type;
	std::string name;
};

class MysqlTableCell
{
public:
	MysqlTableCell(MysqlRowType type = MYSQL_ROW_NULL) : mType(type) {}

	inline bool isString() const { return mType == MYSQL_ROW_STRING; }
	inline bool isInt()  const { return mType == MYSQL_ROW_INT; }
	inline bool isLong() const { return mType == MYSQL_ROW_LONG;  }
	inline bool isDateTime() const { return mType == MYSQL_ROW_DATETIME; }
	inline bool isNull() const { return mType == MYSQL_ROW_NULL; }
	inline bool isDecimal() const { return mType == MYSQL_ROW_DECIMAL;  }
	inline bool isBinary() const { return mType == MYSQL_ROW_BINARY;  }
	inline MysqlRowType getType() const { return mType; }

	virtual operator const char*() const { return ""; }
	virtual operator Poco::Int32() const { return 0; }
	virtual operator Poco::Int64() const { return 0; }
	virtual operator unsigned char*() const { return nullptr; }
	virtual size_t size() { return 0; }
	
protected:
	MysqlRowType mType;
};

class MysqlTableCellString : public MysqlTableCell
{
public: 
	MysqlTableCellString(const char* content) : MysqlTableCell(MYSQL_ROW_STRING), mContent(content) {}
	MysqlTableCellString(const char* content, size_t count) : MysqlTableCell(MYSQL_ROW_STRING), mContent(content, count) {}


	virtual operator const char*() const { return mContent.data(); }
	virtual size_t size() { return mContent.size(); }
protected:
	std::string mContent;
};

class MysqlTableCellInt : public MysqlTableCell
{
public:
	MysqlTableCellInt(Poco::Int32 value) : MysqlTableCell(MYSQL_ROW_INT), mContent(value) {}

	virtual operator Poco::Int32() const { return mContent; }
protected:
	Poco::Int32 mContent;
};

class MysqlTableCellLong : public MysqlTableCell
{
public:
	MysqlTableCellLong(const Poco::Int64& value) : MysqlTableCell(MYSQL_ROW_LONG), mContent(value) {}

	virtual operator Poco::Int64() const { return mContent; }

protected: 
	Poco::Int64 mContent;
};

class MysqlTableCellTimestamp : public MysqlTableCell
{
public: 
	MysqlTableCellTimestamp(const time_t& timestamp) : MysqlTableCell(MYSQL_ROW_TIMESTAMP), mTimestamp(timestamp) {}

protected:
	time_t mTimestamp;
};

class MysqlTableCellDateTime : public MysqlTableCell
{
public:
	MysqlTableCellDateTime(const MYSQL_TIME& mysql_time) : MysqlTableCell(MYSQL_ROW_DATETIME), mMysqlTime(mysql_time) {}

protected:
	MYSQL_TIME mMysqlTime;
};

class MysqlTableCellDecimal : public MysqlTableCell
{
public:
	MysqlTableCellDecimal(const double& value) : MysqlTableCell(MYSQL_ROW_DECIMAL), mDecimal(value) {}

protected:
	double mDecimal;
};

class MysqlTableCellBinary : public MysqlTableCell
{
public:
	MysqlTableCellBinary(const unsigned char* data, size_t _size) :MysqlTableCell(MYSQL_ROW_BINARY), m_data(nullptr), m_size(_size)  {
		m_data = (unsigned char*)malloc(m_size);
		memcpy(m_data, data, m_size);
	}
	~MysqlTableCellBinary() {
		if (m_data) free(m_data);
		m_data = nullptr;
		m_size = 0;
	}

	virtual operator unsigned char*() const { return m_data; }
	virtual size_t size() { return m_size; }
protected:
	unsigned char* m_data;
	size_t m_size;

};

class MysqlTable : public ErrorList
{
public: 
	MysqlTable(size_t fieldCount);
	~MysqlTable();

	bool setHeader(int index, const char* name, MysqlRowType type);
	inline MysqlRowType getHeaderType(int index) {
		if (index > 0 && index < mFieldCount) {
			return mHeader[index].type;
		}
		addError(new ParamError(__FUNCTION__, "invalid field index:", index));
		return MYSQL_ROW_TYPE_NONE;
	}
	inline const char* getHeaderName(int index) {
		if (index > 0 && index < mFieldCount) {
			return mHeader[index].name.data();
		}
		addError(new ParamError(__FUNCTION__, "invalid field index:", index));
		return nullptr;
	}
	inline bool addCellToCurrentRow(Poco::Int32 value) { return addCellToCurrentRow(new MysqlTableCellInt(value)); }
	inline bool addCellToCurrentRow(const Poco::Int64& value) { return addCellToCurrentRow(new MysqlTableCellLong(value)); }
	inline bool addCellToCurrentRow(const char* string) { return addCellToCurrentRow(new MysqlTableCellString(string)); }
	inline bool addCellToCurrentRow(const double& value) { return addCellToCurrentRow(new MysqlTableCellDecimal(value)); }
	inline bool addCellToCurrentRow(const unsigned char* bytes, size_t size) { return addCellToCurrentRow(new MysqlTableCellBinary(bytes, size)); }
	inline bool addCellToCurrentRowTime(const time_t& time) { return addCellToCurrentRow(new MysqlTableCellTimestamp(time)); }
	inline bool addCellToCurrentRow() { return addCellToCurrentRow(new MysqlTableCell); }

	//bool copyColumnValues()

	//! \brief free memory after not longer using it
	bool addCellToCurrentRow(MysqlTableCell* cell);
	inline void addRow() { mRows.push_back(new std::list<MysqlTableCell*>); }
	inline size_t getRowCount() const { return mRows.size(); }
	inline size_t getFieldCount() const { return mFieldCount; }
	inline MysqlRowType getRowType(int fieldIndex) const {
		if (fieldIndex >= mFieldCount || fieldIndex < 0) return MYSQL_ROW_TYPE_NONE;
		return mHeader[fieldIndex].type;
	}

	inline void setTableName(const char* tableName) { mTableName = tableName; }
	inline const char* getTableName() { return mTableName.data(); }

	int connectToStatement(Poco::Data::Statement* stmt, int rowIndex = 0);
	
	static size_t getFieldTypeSize(MysqlRowType type);

	static time_t parseFromMysqlDateTime(const char* mysql_date_time);

protected:
	size_t mFieldCount;
	MysqlTableColumn* mHeader;
	std::list<std::list<MysqlTableCell*>*> mRows;
	std::string mTableName;
};


#endif //DR_LUA_WEB_MODULE_STRUCTURES_MYSQL_QUERY_RESULT__H