/*	EQEmu: EQEmulator

	Copyright (C) 2001-2026 EQEmu Development Team

	This program is free software; you can redistribute it and/or modify
	it under the terms of the GNU General Public License as published by
	the Free Software Foundation; either version 3 of the License, or
	(at your option) any later version.

	This program is distributed in the hope that it will be useful,
	but WITHOUT ANY WARRANTY; without even the implied warranty of
	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
	GNU General Public License for more details.

	You should have received a copy of the GNU General Public License
	along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
#pragma once

#ifdef LUA_EQEMU

#include "common/mysql_stmt.h"
#include "zone/quest_db.h"

#include "luabind/object.hpp"

namespace luabind { struct scope; }
luabind::scope lua_register_database();

class Lua_MySQLPreparedStmt;

class Lua_Database : public QuestDB
{
public:
	using QuestDB::QuestDB;

	void Close();
	Lua_MySQLPreparedStmt* Prepare(lua_State*, std::string query);
};

class Lua_MySQLPreparedStmt
{
public:
	Lua_MySQLPreparedStmt(lua_State* L, mysql::PreparedStmt&& stmt)
		: m_stmt(std::make_unique<mysql::PreparedStmt>(std::move(stmt)))
		, m_row_array(luabind::newtable(L))
		, m_row_hash(luabind::newtable(L)) {}

	void Close();
	void Execute(lua_State*);
	void Execute(lua_State*, luabind::object args);
	void SetOptions(luabind::object table_opts);
	luabind::object FetchArray(lua_State*);
	luabind::object FetchHash(lua_State*);

	// StmtResult functions accessible through this class to simplify api
	int ColumnCount();
	uint64_t LastInsertID();
	uint64_t RowCount();
	uint64_t RowsAffected();

private:
	std::unique_ptr<mysql::PreparedStmt> m_stmt;
	mysql::StmtResult m_res = {};
	luabind::object m_row_array; // perf: table cache for fetches
	luabind::object m_row_hash;
};

#endif // LUA_EQEMU