#pragma once
#include "common.hpp"
#include "logger.hpp"
#include "script.hpp"

namespace big
{
	static void script_exception_handler(PEXCEPTION_POINTERS exp)
	{
		LOG_ERROR("Script threw an exception!");
		g_stackwalker.ShowCallstack(GetCurrentThread(), exp->ContextRecord);
	}

	script::script(func_t func, std::optional<std::size_t> stack_size) :
		m_func(func),
		m_script_fiber(nullptr),
		m_main_fiber(nullptr)
	{
		m_script_fiber = CreateFiber(stack_size.has_value() ? stack_size.value() : 0, [](void* param)
		{
			auto this_script = static_cast<script*>(param);
			this_script->fiber_func();
		}, this);
	}

	script::~script()
	{
		if (m_script_fiber)
			DeleteFiber(m_script_fiber);
	}

	void script::tick()
	{
		m_main_fiber = GetCurrentFiber();
		if (!m_wake_time.has_value() || m_wake_time.value() <= std::chrono::high_resolution_clock::now())
		{
			SwitchToFiber(m_script_fiber);
		}
	}

	void script::yield(std::optional<std::chrono::high_resolution_clock::duration> time)
	{
		if (time.has_value())
		{
			m_wake_time = std::chrono::high_resolution_clock::now() + time.value();
		}
		else
		{
			m_wake_time = std::nullopt;
		}

		SwitchToFiber(m_main_fiber);
	}

	script *script::get_current()
	{
		return static_cast<script*>(GetFiberData());
	}

	void script::fiber_func()
	{
		__try
		{
			[this]()
			{
				try
				{
					m_func();
				}
				catch (std::exception const &ex)
				{
					auto ex_class = typeid(ex).name() + 6;
					LOG_INFO("Script threw an C++ expection! {}: {}", ex_class, ex.what());
				}
				catch (...)
				{
					LOG_INFO("Script threw a C++ exception!");
				}
			}();
		}
		__except (script_exception_handler(GetExceptionInformation()), EXCEPTION_EXECUTE_HANDLER)
		{
			LOG_INFO("Script threw an exception!");
		}

		LOG_INFO("Script finished!");
		while (true)
		{
			yield();
		}
	}
}