// Copyright (C) 2017 Jérôme Leclercq
// This file is part of the "Nazara Engine - Lua scripting module"
// For conditions of distribution and use, see copyright notice in Config.hpp

#pragma once

#ifndef NAZARA_LUASTATE_HPP
#define NAZARA_LUASTATE_HPP

#include <Nazara/Prerequisites.hpp>
#include <Nazara/Core/MovablePtr.hpp>
#include <Nazara/Core/String.hpp>
#include <Nazara/Lua/Config.hpp>
#include <Nazara/Lua/Enums.hpp>
#include <cstddef>
#include <functional>
#include <type_traits>

struct lua_Debug;
struct lua_State;

namespace Nz
{
	class LuaCoroutine;
	class LuaInstance;
	class LuaState;
	class Stream;

	using LuaCFunction = int (*)(lua_State* internalState);
	using LuaFunction = std::function<int(LuaState& state)>;

	class NAZARA_LUA_API LuaState
	{
		public:
			LuaState(const LuaState&) = default;
			LuaState(LuaState&& instance) = default;
			~LuaState() = default;

			void ArgCheck(bool condition, unsigned int argNum, const char* error) const;
			void ArgCheck(bool condition, unsigned int argNum, const String& error) const;
			int ArgError(unsigned int argNum, const char* error) const;
			int ArgError(unsigned int argNum, const String& error) const;

			bool Call(unsigned int argCount);
			bool Call(unsigned int argCount, unsigned int resultCount);
			bool CallWithHandler(unsigned int argCount, int errorHandler);
			bool CallWithHandler(unsigned int argCount, unsigned int resultCount, int errorHandler);

			template<typename T> T Check(int* index) const;
			template<typename T> T Check(int* index, T defValue) const;
			void CheckAny(int index) const;
			bool CheckBoolean(int index) const;
			bool CheckBoolean(int index, bool defValue) const;
			template<typename T> T CheckBoundInteger(int index) const;
			template<typename T> T CheckBoundInteger(int index, T defValue) const;
			template<typename T> T CheckField(const char* fieldName, int tableIndex = -1) const;
			template<typename T> T CheckField(const String& fieldName, int tableIndex = -1) const;
			template<typename T> T CheckField(const char* fieldName, T defValue, int tableIndex = -1) const;
			template<typename T> T CheckField(const String& fieldName, T defValue, int tableIndex = -1) const;
			long long CheckInteger(int index) const;
			long long CheckInteger(int index, long long defValue) const;
			template<typename T> T CheckGlobal(const char* fieldName) const;
			template<typename T> T CheckGlobal(const String& fieldName) const;
			template<typename T> T CheckGlobal(const char* fieldName, T defValue) const;
			template<typename T> T CheckGlobal(const String& fieldName, T defValue) const;
			double CheckNumber(int index) const;
			double CheckNumber(int index, double defValue) const;
			void CheckStack(int space, const char* error = nullptr) const;
			void CheckStack(int space, const String& error) const;
			const char* CheckString(int index, std::size_t* length = nullptr) const;
			const char* CheckString(int index, const char* defValue, std::size_t* length = nullptr) const;
			void CheckType(int index, LuaType type) const;
			void* CheckUserdata(int index, const char* tname) const;
			void* CheckUserdata(int index, const String& tname) const;

			bool Compare(int index1, int index2, LuaComparison comparison) const;
			void Compute(LuaOperation operation) const;

			void Concatenate(int count) const;

			int CreateReference();
			void DestroyReference(int ref);

			String DumpStack() const;

			void Error(const char* message) const;
			void Error(const String& message) const;

			bool Execute(const String& code, int errorHandler = 0);
			bool ExecuteFromFile(const String& filePath, int errorHandler = 0);
			bool ExecuteFromMemory(const void* data, std::size_t size, int errorHandler = 0);
			bool ExecuteFromStream(Stream& stream, int errorHandler = 0);

			int GetAbsIndex(int index) const;
			LuaType GetField(const char* fieldName, int tableIndex = -1) const;
			LuaType GetField(const String& fieldName, int tableIndex = -1) const;
			LuaType GetGlobal(const char* name) const;
			LuaType GetGlobal(const String& name) const;
			inline lua_State* GetInternalState() const;
			inline String GetLastError() const;
			LuaType GetMetatable(const char* tname) const;
			LuaType GetMetatable(const String& tname) const;
			bool GetMetatable(int index) const;
			unsigned int GetStackTop() const;
			LuaType GetTable(int index = -2) const;
			LuaType GetTableRaw(int index = -2) const;
			LuaType GetType(int index) const;
			const char* GetTypeName(LuaType type) const;

			void Insert(int index) const;

			bool IsOfType(int index, LuaType type) const;
			bool IsOfType(int index, const char* tname) const;
			bool IsOfType(int index, const String& tname) const;
			bool IsValid(int index) const;

			bool Load(const String& code);
			bool LoadFromFile(const String& filePath);
			bool LoadFromMemory(const void* data, std::size_t size);
			bool LoadFromStream(Stream& stream);

			long long Length(int index) const;
			std::size_t LengthRaw(int index) const;

			void MoveTo(LuaState* instance, int n) const;

			LuaCoroutine NewCoroutine();
			bool NewMetatable(const char* str);
			bool NewMetatable(const String& str);
			bool Next(int index = -2) const;

			void Pop(unsigned int n = 1U) const;

			template<typename T> int Push(T arg) const;
			template<typename T, typename T2, typename... Args> int Push(T firstArg, T2 secondArg, Args... args) const;
			void PushBoolean(bool value) const;
			void PushCFunction(LuaCFunction func, unsigned int upvalueCount = 0) const;
			template<typename T> void PushField(const char* name, T&& arg, int tableIndex = -2) const;
			template<typename T> void PushField(const String& name, T&& arg, int tableIndex = -2) const;
			void PushFunction(LuaFunction func) const;
			template<typename R, typename... Args, typename... DefArgs> void PushFunction(R(*func)(Args...), DefArgs&&... defArgs) const;
			template<typename T> void PushGlobal(const char* name, T&& arg);
			template<typename T> void PushGlobal(const String& name, T&& arg);
			template<typename T> void PushInstance(const char* tname, T&& instance) const;
			template<typename T, typename... Args> void PushInstance(const char* tname, Args&&... args) const;
			void PushInteger(long long value) const;
			void PushLightUserdata(void* value) const;
			void PushMetatable(const char* str) const;
			void PushMetatable(const String& str) const;
			void PushNil() const;
			void PushNumber(double value) const;
			void PushReference(int ref) const;
			void PushString(const char* str) const;
			void PushString(const char* str, std::size_t size) const;
			void PushString(const String& str) const;
			void PushTable(std::size_t sequenceElementCount = 0, std::size_t arrayElementCount = 0) const;
			void* PushUserdata(std::size_t size) const;
			void PushValue(int index) const;

			bool RawEqual(int index1, int index2) const;

			void Remove(int index) const;
			void Replace(int index) const;

			void SetField(const char* name, int tableIndex = -2) const;
			void SetField(const String& name, int tableIndex = -2) const;
			void SetGlobal(const char* name);
			void SetGlobal(const String& name);
			void SetMetatable(const char* tname) const;
			void SetMetatable(const String& tname) const;
			void SetMetatable(int index) const;
			void SetTable(int index = -3) const;
			void SetTableRaw(int index = -3) const;

			bool ToBoolean(int index) const;
			long long ToInteger(int index, bool* succeeded = nullptr) const;
			double ToNumber(int index, bool* succeeded = nullptr) const;
			const void* ToPointer(int index) const;
			const char* ToString(int index, std::size_t* length = nullptr) const;
			void* ToUserdata(int index) const;
			void* ToUserdata(int index, const char* tname) const;
			void* ToUserdata(int index, const String& tname) const;

			void Traceback(const char* message = nullptr, int level = 0);

			LuaState& operator=(const LuaState&) = default;
			LuaState& operator=(LuaState&& instance) = default;

			static int GetIndexOfUpValue(int upValue);
			static LuaInstance& GetInstance(lua_State* internalState);
			static inline LuaState GetState(lua_State* internalState);

		protected:
			LuaState(lua_State* internalState);

			template<typename T> std::enable_if_t<std::is_signed<T>::value, T> CheckBounds(int index, long long value) const;
			template<typename T> std::enable_if_t<std::is_unsigned<T>::value, T> CheckBounds(int index, long long value) const;
			virtual bool Run(int argCount, int resultCount, int errHandler);

			static int ProxyFunc(lua_State* internalState);

			MovablePtr<lua_State> m_state;
			String m_lastError;
	};
}

#include <Nazara/Lua/LuaState.inl>

#endif // NAZARA_LUASTATE_HPP