#include "my-command.h"
#include "command.h"
#include "my-connection.h"
#include "my-error.h"
#include <boost/date_time/date.hpp>
#include <boost/date_time/gregorian_calendar.hpp>
#include <boost/date_time/time.hpp>
#include <boost/numeric/conversion/cast.hpp>
#include <cstdlib>
#include <cstring>
#include <mysql.h>
#include <type_traits>

namespace Glib {
	class ustring;
}

MySQL::Command::Command(const Connection * conn, const std::string & sql) :
	DB::Command(sql), c(conn), stmt(mysql_stmt_init(&conn->conn), &mysql_stmt_close), paramsNeedBinding(false)
{
	if (!stmt) {
		throw Error(&conn->conn);
	}
	if (mysql_stmt_prepare(stmt.get(), sql.c_str(), sql.length())) {
		throw Error(stmt.get());
	}
	binds.resize(mysql_stmt_param_count(stmt.get()));
	if (binds.size()) {
		paramsNeedBinding = true;
		for (auto & b : binds) {
			memset(&b, 0, sizeof(MYSQL_BIND));
			b.buffer_type = MYSQL_TYPE_NULL;
		}
	}
}

MySQL::Command::~Command()
{
	for (auto & b : binds) {
		// NOLINTNEXTLINE(hicpp-no-malloc)
		free(b.buffer);
	}
}

void *
MySQL::Command::realloc(void * buffer, size_t size)
{
	// NOLINTNEXTLINE(hicpp-no-malloc)
	void * newBuffer = ::realloc(buffer, size);
	if (buffer != newBuffer) {
		paramsNeedBinding = true;
	}
	return newBuffer;
}

template<typename B, enum_field_types b, typename P>
void
bindNumber(MYSQL_BIND & bind, const P & v)
{
	bind.buffer_type = b;
	// NOLINTNEXTLINE(hicpp-no-malloc)
	bind.buffer = realloc(bind.buffer, sizeof(B));
	*static_cast<B *>(bind.buffer) = boost::numeric_cast<B>(v);
	bind.is_unsigned = std::is_unsigned<P>::value;
}

void
MySQL::Command::bindParamI(unsigned int n, int v)
{
	bindNumber<int, MYSQL_TYPE_LONG>(binds[n], v);
}
void
MySQL::Command::bindParamI(unsigned int n, long int v)
{
	bindNumber<long long int, MYSQL_TYPE_LONGLONG>(binds[n], v);
}
void
MySQL::Command::bindParamI(unsigned int n, long long int v)
{
	bindNumber<long long int, MYSQL_TYPE_LONGLONG>(binds[n], v);
}
void
MySQL::Command::bindParamI(unsigned int n, unsigned int v)
{
	bindNumber<int, MYSQL_TYPE_LONG>(binds[n], v);
}
void
MySQL::Command::bindParamI(unsigned int n, long unsigned int v)
{
	bindNumber<long long int, MYSQL_TYPE_LONGLONG>(binds[n], v);
}
void
MySQL::Command::bindParamI(unsigned int n, long long unsigned int v)
{
	bindNumber<long long int, MYSQL_TYPE_LONGLONG>(binds[n], v);
}
void
MySQL::Command::bindParamB(unsigned int n, bool v)
{
	bindNumber<char, MYSQL_TYPE_TINY>(binds[n], (v ? 1 : 0));
}
void
MySQL::Command::bindParamF(unsigned int n, double v)
{
	bindNumber<double, MYSQL_TYPE_DOUBLE>(binds[n], v);
}
void
MySQL::Command::bindParamF(unsigned int n, float v)
{
	bindNumber<float, MYSQL_TYPE_FLOAT>(binds[n], v);
}
void
MySQL::Command::bindParamS(unsigned int n, const Glib::ustring & s)
{
	bindParamS(n, std::string_view {s.data(), s.bytes()});
}

void
MySQL::Command::bindParamS(unsigned int n, const std::string_view s)
{
	binds[n].buffer_type = MYSQL_TYPE_STRING;
	// NOLINTNEXTLINE(hicpp-no-malloc)
	binds[n].buffer = realloc(binds[n].buffer, s.length());
	s.copy(static_cast<char *>(binds[n].buffer), s.length(), 0);
	binds[n].buffer_length = s.length();
	binds[n].is_unsigned = false;
}
void
MySQL::Command::bindParamT(unsigned int n, const boost::posix_time::ptime v)
{
	binds[n].buffer_type = MYSQL_TYPE_DATETIME;
	// NOLINTNEXTLINE(hicpp-no-malloc)
	binds[n].buffer = realloc(binds[n].buffer, sizeof(MYSQL_TIME));
	MYSQL_TIME & ts = *static_cast<MYSQL_TIME *>(binds[n].buffer);
	ts.year = v.date().year();
	ts.month = v.date().month();
	ts.day = v.date().day();
	ts.hour = static_cast<unsigned int>(v.time_of_day().hours());
	ts.minute = static_cast<unsigned int>(v.time_of_day().minutes());
	ts.second = static_cast<unsigned int>(v.time_of_day().seconds());
	ts.second_part = static_cast<long unsigned int>(v.time_of_day().total_microseconds() % 1000000U);
	ts.neg = false;
}
void
MySQL::Command::bindParamT(unsigned int n, const boost::posix_time::time_duration v)
{
	binds[n].buffer_type = MYSQL_TYPE_TIME;
	// NOLINTNEXTLINE(hicpp-no-malloc)
	binds[n].buffer = realloc(binds[n].buffer, sizeof(MYSQL_TIME));
	MYSQL_TIME & ts = *static_cast<MYSQL_TIME *>(binds[n].buffer);
	ts.year = 0;
	ts.month = 0;
	ts.day = 0;
	ts.hour = static_cast<unsigned int>(v.hours());
	ts.minute = static_cast<unsigned int>(v.minutes());
	ts.second = static_cast<unsigned int>(v.seconds());
	ts.second_part = static_cast<long unsigned int>(v.total_microseconds() % 1000000U);
	ts.neg = (v.is_negative() ? 1 : 0);
}
void
MySQL::Command::bindNull(unsigned int n)
{
	binds[n].buffer_type = MYSQL_TYPE_NULL;
	binds[n].buffer = nullptr;
	// NOLINTNEXTLINE(hicpp-no-malloc)
	free(binds[n].buffer);
}

void
MySQL::Command::bindParams()
{
	if (paramsNeedBinding) {
		if (mysql_stmt_bind_param(stmt.get(), &binds.front())) {
			throw Error(stmt.get());
		}
	}
}