iw5-mod/src/game/scripting/scheduler.cpp

#include "std_include.hpp"
#include "context.hpp"

namespace game
{
	namespace scripting
	{
		scheduler::scheduler(context* context) : context_(context)
		{
			const auto chai = this->context_->get_chai();

			chai->add(chaiscript::user_type<task_handle>(), "_task_handle");
			chai->add(chaiscript::constructor<task_handle()>(), "_task_handle");
			chai->add(chaiscript::constructor<task_handle(const task_handle&)>(), "_task_handle");

			chai->add(chaiscript::fun([](task_handle& lhs, const task_handle& rhs) -> task_handle&
			{
				return lhs = rhs;
			}), "=");

			chai->add(chaiscript::fun([this](const std::function<void()>& callback, const long long milliseconds) -> task_handle
			{
				return this->add(callback, milliseconds, true);
			}), "setTimeout");

			chai->add(chaiscript::fun([this](const std::function<void()>& callback, const long long milliseconds) -> task_handle
			{
				return this->add(callback, milliseconds, false);
			}), "setInterval");

			const auto clear = [this](const task_handle& handle)
			{
				this->remove(handle);
			};

			chai->add(chaiscript::fun(clear), "clear");
			chai->add(chaiscript::fun(clear), "clearTimeout");
			chai->add(chaiscript::fun(clear), "clearInterval");
		}

		void scheduler::run_frame()
		{
			for (const auto& task : this->tasks_)
			{
				const auto now = std::chrono::steady_clock::now();
				if ((now - task->last_execution) > task->delay)
				{
					task->last_execution = now;
					if (task->is_volatile)
					{
						this->tasks_.remove(task);
					}

					task->callback();
				}
			}
		}

		task_handle scheduler::add(const std::function<void()>& callback, const long long milliseconds,
		                           const bool is_volatile)
		{
			return this->add(callback, std::chrono::milliseconds(milliseconds), is_volatile);
		}

		task_handle scheduler::add(const std::function<void()>& callback, const std::chrono::milliseconds delay,
		                           const bool is_volatile)
		{
			task task;
			task.is_volatile = is_volatile;
			task.callback = callback;
			task.delay = delay;
			task.last_execution = std::chrono::steady_clock::now();
			task.id = ++this->current_task_id_;

			this->tasks_.add(task);

			return { task.id };
		}

		void scheduler::remove(const task_handle& handle)
		{
			for (const auto task : this->tasks_)
			{
				if(task->id == handle.id)
				{
					this->tasks_.remove(task);
					break;
				}
			}
		}
	}
}