#pragma once

#include "gp/exception.hpp"
#include "gp/functional/bind_front.hpp"
#include "gp/algorithms/move.hpp"
#include "gp/algorithms/tmp_manip.hpp"
#include "gp/utils/allocators/allocator.hpp"
#include "gp/functional/optional.hpp"

namespace gp{

	namespace details {
		template<typename pred, typename returner>
		auto ensure(pred a, returner b) {
			gp_config::assertion(a(), "could not ensure predicate");
			return b();
		}
	}

	template <typename fn>
	class function;

	template <typename ret, typename ...args>
	class function<ret(args...)>{
		using fn_ptr = char*;

		using invoke_fn_t = ret (*)(fn_ptr, args&&...);
		using condestruct_fn_t = void (*) (fn_ptr, fn_ptr);

		gp::optional<gp::reference_wrapper<allocator>> alloc;
		invoke_fn_t invokator;
		condestruct_fn_t condestructor;
		fn_ptr data_ptr;
		size_t data_size;


		template<typename func>
		static ret invoke(func* fn, args&&... fw_args) {
			return (*fn)(gp::forward<args>(fw_args)...);
		}

		template<typename func>
		static void condestruct(func* dest, func* src) {
			if(dest != nullptr) {
				new (dest) func(*src);
			} else {
				src->~func();
			}
		}

		static void nop_condestruct(char*, char*) {
			return;
		}

	public:
		function(allocator& alloc_v)
		: alloc(alloc_v)
		, invokator(nullptr)
		, condestructor(nullptr)
		, data_ptr(nullptr)
		, data_size(0)
		{}

		template<typename func>
		function(func f, allocator& alloc_v)
		: alloc(alloc_v)
		, invokator(reinterpret_cast<invoke_fn_t>(invoke<func>))
		, condestructor(reinterpret_cast<condestruct_fn_t>(condestruct<func>))
		, data_ptr(
			sizeof(func) <= sizeof(data_ptr) 
			? nullptr
			: details::ensure(
				[&](){return alloc.has_value();}, 
				[&](){return (char*)(alloc.value().get().allocate(sizeof(func)));}
			)
		)
		, data_size(
			sizeof(func)
		)
		{
			gp_config::assertion(data_size <= sizeof(func) || !(alloc.has_value() && data_ptr == nullptr), "allocator failed in function");
			this->condestructor(data_size <= sizeof(data_ptr) ? (char*)&data_ptr : data_ptr, reinterpret_cast<char*>(&f));
		}

		/*function(ret f_ptr(args...))
		: function(f_ptr, nullopt)
		{}*/

		template<typename func>
		function(func f, nullopt_t alloc_v)
		: alloc(alloc_v)
		, invokator(reinterpret_cast<invoke_fn_t>(invoke<func>))
		, condestructor(reinterpret_cast<condestruct_fn_t>(condestruct<func>))
		, data_ptr()
		, data_size(
			details::ensure(
				[&](){return sizeof(func) <= sizeof(data_ptr);}, 
				[&](){return sizeof(func);}
			)
		)
		{
			this->condestructor((char*)&data_ptr, reinterpret_cast<char*>(&f));
		}

		function(function const& rhs)
		: alloc(rhs.alloc)
		, invokator(rhs.invokator)
		, condestructor(rhs.condestructor)
		, data_ptr(
			rhs.data_size <= sizeof(data_ptr) 
			? 0 
			: details::ensure(
				[&](){return alloc.has_value();}, 
				[&](){return (char*)(alloc.value().get().allocate(rhs.data_size));}
			)
		)
		, data_size(rhs.data_size)
		{
			gp_config::assertion(data_size <= sizeof(data_ptr) || !(alloc.has_value() && data_ptr == nullptr), "allocator failed in function");
			if(
				data_size != 0
				and rhs.data_size != 0
			) this->condestructor(
				data_size <= sizeof(data_ptr) ? (char*)&data_ptr : data_ptr, 
				data_size <= sizeof(data_ptr) ? (char*)&(rhs.data_ptr) : rhs.data_ptr
			);
		}

		function(function&& rhs)
		: alloc(rhs.alloc)
		, invokator(rhs.invokator)
		, condestructor(rhs.condestructor)
		, data_ptr(rhs.data_size <= sizeof(data_ptr) ? nullptr : rhs.data_ptr)
		, data_size(rhs.data_size)
		{
			if(data_size != 0 && data_size <= sizeof(data_ptr)) {
				condestructor((char*)&data_ptr, (char*)&rhs.data_ptr);
				condestructor(nullptr, (char*)&rhs.data_ptr);
			}
			rhs.data_ptr = nullptr;
			rhs.data_size = 0;
		}

		~function(){
			if(data_size <= 0) {
				return;
			}
			if(data_ptr != nullptr && alloc.has_value()) {
				condestructor(nullptr, data_ptr);
				alloc.value().get().deallocate(data_ptr);
				data_ptr = nullptr;
			}
			if(data_size < sizeof(data_ptr))
			{
				condestructor(nullptr, (char*)&data_ptr);
			}
		}

		function& operator=(function&& rhs) {
			swap(alloc, rhs.alloc);
			swap(invokator, rhs.invokator);
			swap(condestructor, rhs.condestructor);
			swap(data_ptr,rhs.data_ptr);
			swap(data_size,rhs.data_size);
			return *this;
		}

		function& operator=(function& rhs) {
			alloc = rhs.alloc;
			invokator = rhs.invokator;
			condestructor = rhs.condestructor;

			/* Cleanup */

			if(data_size > sizeof(data_ptr)) {
				condestructor(nullptr, data_ptr);
				alloc.value().get().deallocate(data_ptr);
			} else if(data_size) {
				condestructor(nullptr, (char*)&data_ptr);
			}

			/* Reallocation */
			
			data_size = rhs.data_size;
			data_ptr = data_size <= sizeof(data_ptr) 
				? 0 
				: details::ensure(
					[&](){return alloc.has_value();}, 
					[&](){return (char*)(alloc.value().get().allocate(rhs.data_size));}
				);
			
			gp_config::assertion(!(alloc.has_value() && data_size > 8 && data_ptr != nullptr), "allocator failed in function");

			if(
				data_size > 0
			) this->condestructor(
				data_size <= sizeof(data_ptr) ? (char*)&data_ptr : data_ptr, 
				data_size <= sizeof(data_ptr) ? (char*)&rhs.data_ptr : rhs.data_ptr
			);
			return *this;
		}

		ret operator()(args&&... argv) {
			return invokator(data_size <= sizeof(data_ptr) ? (fn_ptr)&data_ptr : data_ptr, gp::forward<args>(argv)...);
		}

		bool ready() {
			return invokator != nullptr;
		}
	};

	/*template <typename ret, typename ...args>
	class function<ret(args...)>{
		struct virtual_callable
		{
			virtual void inplace_move(char*) = 0;
			virtual virtual_callable* all_copy() = 0;
			virtual void inplace_copy(char*) = 0;
			virtual virtual_callable* all_move() = 0;
			virtual ~virtual_callable() = default;
			virtual ret operator() (args...) = 0;
		};

		template<typename fn>
		class callable final : public virtual_callable{
			typename gp::remove_reference<fn>::type internal_representation;
		public:
			callable(const fn func)
			: internal_representation{gp::move(func)}
			{}

			callable(callable&) = default;
			callable(callable&&) = default;

			virtual ~callable() override = default;

			virtual void inplace_copy(char* ptr) override {
				new(ptr) callable(*this);
			}

			virtual virtual_callable* all_copy() override {
				return new callable(*this);
			}	

			virtual void inplace_move(char* ptr) override {
				new(ptr) callable(gp::move(*this));
			}

			virtual virtual_callable* all_move() override {
				return new callable(gp::move(*this));
			}	

			ret operator() (args... arg_list) override
			{
				return internal_representation(arg_list...);
			}
		};

		// tweak a way to store a size in there for trivial copy
		enum state_t : uint8_t{
			INACTIVE = 0,
			ACTIVE = 1,
			NO_SOO = 0,
			SOO = 2
		};

		state_t state{};
		union{
			virtual_callable* functor = nullptr;
			char inplace[16];
		} self;
		
	public:
		template <typename T>
		function& operator=(T& t) 
		{
			if(state & ACTIVE)
			{
				if(state & SOO)
				{
					((virtual_callable*)self.inplace)->~virtual_callable();
				}
				else
				{
					delete self.functor;
				}
			}
			if(!(t.state & ACTIVE))
			{
				state = INACTIVE;
				return;
			}
			if constexpr (!std::is_same_v<T, function<ret(args...)>>) {
				if constexpr (sizeof(callable<T>) <= sizeof(self))
				{
					new((void*)self.inplace) callable<T>(t);
					state = (state_t)(ACTIVE | SOO);
				}
				else
				{
					self.functor = new callable<T>(t);
					state = (state_t)(ACTIVE | NO_SOO);
				}
			} else {
				if(t.state & SOO)
				{
					auto& ref = t.self.functor;
					ref->inplace_copy((char*)&self);
					state = (state_t)(ACTIVE | SOO);
				}
				else
				{
					self.functor = t.self.functor->all_copy();
					state = (state_t)(ACTIVE | NO_SOO);
				}
			}
		}

		function() 
		{
			state = INACTIVE;
		}

		template<typename T>
		function<>(T& t) 
		{
			if constexpr (!std::is_same_v<T, function<ret(args...)>>) {
				if constexpr (sizeof(callable<T>) <= sizeof(self))
				{
					new((void*)self.inplace) callable<T>(t);
					state = (state_t)(ACTIVE | SOO);
				}
				else
				{
					self.functor = new callable<T>(t);
					state = (state_t)(ACTIVE | NO_SOO);
				}
			} else {
				if(t.state & SOO)
				{
					t.self.functor->inplace_copy(self.inplace);
					state = (state_t)(ACTIVE | SOO);
				}
				else
				{
					self.functor = t.self.functor->all_copy();
					state = (state_t)(ACTIVE | NO_SOO);
				}
			}
		}

		template <typename T>
		function(T&& t) 
		{
			if constexpr (!std::is_same_v<T, function<ret(args...)>>) {
				if constexpr (sizeof(callable<T>) <= sizeof(self))
				{
					new((void*)self.inplace) callable<T>(gp::move(t));
					state = (state_t)(ACTIVE | SOO);
				}
				else
				{
					self.functor = new callable<T>(gp::move(t));
					state = (state_t)(ACTIVE | NO_SOO);
				}
			} else {
				if(t.state & SOO)
				{
					auto& ref = t.self.functor;
					ref->inplace_move((char*)&self);
					state = (state_t)(ACTIVE | SOO);
				}
				else
				{
					self.functor = t.self.functor->all_move();
					state = (state_t)(ACTIVE | NO_SOO);
				}
			}
		}

		ret operator()(args... arg_list) const {
			if constexpr (gp_config::has_exceptions)
			{
				if(!(state & ACTIVE))
				{
					throw bad_functor{};
				}
			}
			if(state & SOO)
			{
				return (*(virtual_callable*)&self)(arg_list...);
			}
			else
			{
				return (*self.functor)(arg_list...);
			}
			
		}

		~function()
		{
			if(state & ACTIVE)
			{
				if(state & SOO)
				{
					((virtual_callable*)&self)->~virtual_callable();
				}
				else
				{
					delete self.functor;
				}
			}
			
		}
	};*/

}