// SPDX-FileCopyrightText: 2020 Florian Märkl <info@florianmaerkl.de>
// SPDX-License-Identifier: LGPL-3.0-only

#include <rz_core.h>
#include "minunit.h"

static void *my_function(RzCore *core, void *user) {
	size_t val = (size_t)user;
	int i;
	for (i = 0; i < 5; i++) {
		rz_cons_printf("%u, %d\n", (unsigned int)val, i);
		rz_core_task_yield(&core->tasks);
	}
	return rz_cons_get_buffer_dup();
}

static bool test_core_task(void) {
	RzCore *core = rz_core_new();
	rz_config_set_i(core->config, "scr.interactive", 0);
	rz_core_task_sync_begin(&core->tasks);

	RzCoreTask *a = rz_core_cmd_task_new(core, "echo hello; echo world; echo from; echo a; echo task", NULL, NULL);
	rz_core_task_enqueue(&core->tasks, a);

	RzCoreTask *b = rz_core_function_task_new(core, my_function, (void *)(size_t)1337);
	rz_core_task_enqueue(&core->tasks, b);

	rz_cons_printf("Hello\n");
	rz_cons_printf("this\n");
	rz_cons_printf("is\n");
	rz_cons_printf("the\n");
	rz_cons_printf("main\n");
	rz_cons_printf("task!\n");

	rz_core_task_join(&core->tasks, rz_core_task_self(&core->tasks), a->id);
	rz_core_task_join(&core->tasks, rz_core_task_self(&core->tasks), b->id);

	void *fcn_result = rz_core_function_task_get_result(a);
	mu_assert_null(fcn_result, "fcn result");
	const char *cmd_result = rz_core_cmd_task_get_result(a);
	mu_assert_streq(cmd_result, "hello\nworld\nfrom\na\ntask\n", "cmd result");

	fcn_result = rz_core_function_task_get_result(b);
	mu_assert_streq((const char *)fcn_result, "1337, 0\n1337, 1\n1337, 2\n1337, 3\n1337, 4\n", "fcn result");
	free(fcn_result);
	cmd_result = rz_core_cmd_task_get_result(b);
	mu_assert_null(cmd_result, "cmd result");

	rz_core_task_del(&core->tasks, a->id);
	rz_core_task_del(&core->tasks, b->id);

	mu_assert_streq(rz_cons_get_buffer(), "Hello\nthis\nis\nthe\nmain\ntask!\n", "main buffer");

	rz_core_task_sync_end(&core->tasks);
	rz_core_free(core);
	mu_end;
}

static void finished_cb(const char *res, void *user) {
	*(char **)user = strdup(res);
}

static bool test_core_task_finished_cb(void) {
	RzCore *core = rz_core_new();
	rz_config_set_i(core->config, "scr.interactive", 0);
	rz_core_task_sync_begin(&core->tasks);

	char *res_indir = NULL; // finished_cb puts the result in here too
	RzCoreTask *a = rz_core_cmd_task_new(core, "echo amor; echo vincit; echo omnia", finished_cb, &res_indir);
	rz_core_task_enqueue(&core->tasks, a);

	RzCoreTaskJoinErr join_err = rz_core_task_join(&core->tasks, rz_core_task_self(&core->tasks), a->id);
	mu_assert_eq(join_err, RZ_CORE_TASK_JOIN_ERR_SUCCESS, "task joined");

	const char *cmd_result = rz_core_cmd_task_get_result(a);
	mu_assert_streq(cmd_result, "amor\nvincit\nomnia\n", "cmd result");
	mu_assert_streq(res_indir, "amor\nvincit\nomnia\n", "cmd result");
	free(res_indir);

	rz_core_task_del(&core->tasks, a->id);

	rz_core_task_sync_end(&core->tasks);
	rz_core_free(core);
	mu_end;
}

// This test is best served with helgrind
static int all_tests(void) {
	mu_run_test(test_core_task);
	mu_run_test(test_core_task_finished_cb);
	return tests_passed != tests_run;
}

mu_main(all_tests)