#include <chrono>
#include <iostream>
#include <string>
#include <thread>

#include "librf/librf.h"

using namespace librf;

template<class _Ctype>
static void callback_get_long(int64_t val, _Ctype&& cb)
{
	using namespace std::chrono;
	std::thread([val, cb = std::forward<_Ctype>(cb)]
		{
			std::this_thread::sleep_for(500ms);
			cb(val * val);
		}).detach();
}

//这种情况下，没有生成 frame-context，因此，并没有promise_type被内嵌在frame-context里
static future_t<int64_t> async_get_long(int64_t val)
{
	awaitable_t<int64_t> awaitable;
	callback_get_long(val, [awaitable](int64_t val)
	{
		awaitable.set_value(val);
	});
	return awaitable.get_future();
}

static future_t<int64_t> wait_get_long(int64_t val)
{
	val = co_await async_get_long(val);
	co_return val;
}

//这种情况下，会生成对应的 frame-context，一个promise_type被内嵌在frame-context里
static future_t<int64_t> resumable_get_long(int64_t val)
{
	std::cout << val << std::endl;
	val = co_await wait_get_long(val);
	std::cout << val << std::endl;
	val = co_await wait_get_long(val);
	std::cout << val << std::endl;
	val = co_await wait_get_long(val);
	std::cout << val << std::endl;
	co_return val;
}

static future_t<int64_t> loop_get_long(int64_t val)
{
	std::cout << val << std::endl;
	for (int i = 0; i < 5; ++i)
	{
		val = co_await async_get_long(val);
		std::cout << val << std::endl;
	}
	co_return val;
}

static future_t<std::string&> async_get_string(std::string & ref_string)
{
	awaitable_t<std::string&> awaitable;
	callback_get_long(std::stoi(ref_string), [awaitable, &ref_string](int64_t val)
		{
			ref_string = std::to_string(val);
			awaitable.set_value(ref_string);
		});
	return awaitable.get_future();
}

static future_t<std::string&> resumable_get_string(std::string& val)
{
	std::cout << val << std::endl;
	val = co_await async_get_string(val);
	std::cout << val << std::endl;
	val = co_await async_get_string(val);
	std::cout << val << std::endl;
	val = co_await async_get_string(val);
	std::cout << val << std::endl;
	co_return static_cast<std::string&>(val);
}

void resumable_main_cb()
{
	std::cout << __FUNCTION__ << std::endl;
	//由于使用者可能不能明确的区分是resume function返回的awaitor还是awaitable function返回的awaitor
	//导致均有可能加入到协程里去调度。
	//所以，协程调度器应该需要能处理这种情况。
	go async_get_long(3);
	this_scheduler()->run_until_notask();

	std::string ref_string{"2"};
	go resumable_get_string(ref_string);
	this_scheduler()->run_until_notask();

	GO
	{
		auto val = co_await resumable_get_long(2);
		std::cout << "GO:" << val << std::endl;
	};

	go loop_get_long(3);
	this_scheduler()->run_until_notask();
}

#if LIBRF_TUTORIAL_STAND_ALONE
int main()
{
	resumable_main_cb();
	return 0;
}
#endif