#pragma once

#include <string>

#include <sstream>

#include <vector>

#include <map>

#include <type_traits>

#include "ColorText.h"

#include "DataIdentity.h"

#include "LuaWrapper.h"

namespace df {

class DFHACK_EXPORT cur_lua_ostream_argument {

DFHack::color_ostream *out;

public:

cur_lua_ostream_argument(lua_State *state);

operator DFHack::color_ostream& () { return *out; }

};

// the std::is_enum_v alternative is because pushing is_primitive

// into all the enum identities would require changing codegen

template<typename RT>

concept isPrimitive = identity_traits<RT>::is_primitive || std::is_enum_v<RT> || std::is_void_v<RT>;

// handle call-by-reference function arguments by converting pointers to a reference

// will throw if lua code tries to pass a null pointer

template<typename T> requires std::is_reference_v<T>

T get_from_lua_state(lua_State* L, int idx)

}

// handle call-by-value function arguments. only semi-regular types are allowed

// (semi-regular covers copyable and default-constructible)

template<std::semiregular T>

T get_from_lua_state(lua_State* L, int idx)

}

template<isPrimitive RT, typename... AT, typename FT, typename... ET, std::size_t... I>

requires std::is_invocable_r_v<RT, FT, ET..., AT...>

void call_and_push_impl(lua_State* L, int base, std::index_sequence<I...>, FT fun, ET... extra)

}

template<isPrimitive RT, typename... AT, typename FT, typename... ET, typename indices = std::index_sequence_for<AT...> >

requires std::is_invocable_r_v<RT, FT, ET..., AT...>

void call_and_push(lua_State* L, int base, FT fun, ET... extra)

{

call_and_push_impl<RT, AT...>(L, base, indices{}, fun, extra...);

}

template<typename T> struct function_wrapper {};

template<isPrimitive RT, typename ...AT>

struct function_wrapper<RT(*)(DFHack::color_ostream&, AT...)> {

static const int num_args = sizeof...(AT);

static void execute(lua_State *L, int base, RT (fun)(DFHack::color_ostream& out, AT...)) {

cur_lua_ostream_argument out(L);

call_and_push<RT, AT...>(L, base, fun, out);

}

};

template<isPrimitive RT, typename ...AT>

struct function_wrapper<RT(*)(AT...)> {

static const int num_args = sizeof...(AT);

static void execute(lua_State *L, int base, RT (fun)(AT...)) {

call_and_push<RT, AT...>(L, base, fun);

}

};

template<isPrimitive RT, class CT, typename ...AT>

struct function_wrapper<RT(CT::*)(AT...)> {

static const int num_args = sizeof...(AT)+1;

static void execute(lua_State *L, int base, RT(CT::*mem_fun)(AT...)) {

};

};

template<isPrimitive RT, class CT, typename ...AT>

struct function_wrapper<RT(CT::*)(AT...) const> {

static const int num_args = sizeof...(AT)+1;

static void execute(lua_State *L, int base, RT(CT::*mem_fun)(AT...) const) {

};

};

template<isPrimitive RT, typename ...AT>

struct function_wrapper<RT(*)(DFHack::color_ostream&, AT...) noexcept> {

static const int num_args = sizeof...(AT);

static void execute(lua_State* L, int base, RT(fun)(DFHack::color_ostream& out, AT...)) {

cur_lua_ostream_argument out(L);

call_and_push<RT, AT...>(L, base, fun, out);

}

};

template<isPrimitive RT, typename ...AT>

struct function_wrapper<RT(*)(AT...) noexcept> {

static const int num_args = sizeof...(AT);

static void execute(lua_State* L, int base, RT(fun)(AT...)) {

call_and_push<RT, AT...>(L, base, fun);

}

};

template<isPrimitive RT, class CT, typename ...AT>

struct function_wrapper<RT(CT::*)(AT...) noexcept> {

static const int num_args = sizeof...(AT) + 1;

static void execute(lua_State* L, int base, RT(CT::* mem_fun)(AT...)) {

};

};

template<isPrimitive RT, class CT, typename ...AT>

struct function_wrapper<RT(CT::*)(AT...) const noexcept> {

static const int num_args = sizeof...(AT) + 1;

static void execute(lua_State* L, int base, RT(CT::* mem_fun)(AT...) const) {

};

};

template<typename T>

class function_identity : public function_identity_base {

};

template<typename T>

inline function_identity_base *wrap_function(T ptr, bool vararg = false) {

using RT = return_type<T>::type;

if constexpr (isPrimitive<RT>)

return new function_identity<T>(ptr, vararg);

else

return nullptr;

}}