limboai/util/limbo_task_db.h

/**
 * limbo_task_db.h
 * =============================================================================
 * Copyright 2021-2024 Serhii Snitsaruk
 *
 * Use of this source code is governed by an MIT-style
 * license that can be found in the LICENSE file or at
 * https://opensource.org/licenses/MIT.
 * =============================================================================
 */

#ifndef LIMBO_TASK_DB_H
#define LIMBO_TASK_DB_H

#ifdef LIMBOAI_MODULE
#include "core/object/class_db.h"
#include "core/templates/hash_map.h"
#include "core/templates/list.h"
#endif // LIMBOAI_MODULE

#ifdef LIMBOAI_GDEXTENSION
#include <godot_cpp/core/class_db.hpp>
#include <godot_cpp/templates/hash_map.hpp>
#include <godot_cpp/templates/list.hpp>
#include <godot_cpp/variant/string.hpp>
using namespace godot;
#endif // LIMBOAI_GDEXTENSION

class LimboTaskDB {
private:
	static HashMap<String, List<String>> core_tasks;
	static HashMap<String, List<String>> tasks_cache;

	struct ComparatorByTaskName {
		bool operator()(const String &p_left, const String &p_right) const {
			return get_task_name(p_left) < get_task_name(p_right);
		}
	};

public:
	template <class T>
	static void register_task() {
		GDREGISTER_CLASS(T);
		HashMap<String, List<String>>::Iterator E = core_tasks.find(T::get_task_category());
		if (E) {
			E->value.push_back(T::get_class_static());
		} else {
			List<String> tasks;
			tasks.push_back(T::get_class_static());
			core_tasks.insert(T::get_task_category(), tasks);
		}
	}

	static void scan_user_tasks();
	static _FORCE_INLINE_ String get_misc_category() { return "Misc"; }
	static List<String> get_categories();
	static List<String> get_tasks_in_category(const String &p_category);
	static _FORCE_INLINE_ String get_task_name(String p_class_or_script_path) {
		if (p_class_or_script_path.begins_with("res:")) {
			return p_class_or_script_path.get_file().get_basename().trim_prefix("BT").to_pascal_case();
		} else {
			return p_class_or_script_path.trim_prefix("BT");
		}
	}
};

#ifdef LIMBOAI_MODULE
#define LIMBO_REGISTER_TASK(m_class)             \
	if (m_class::_class_is_enabled) {            \
		::LimboTaskDB::register_task<m_class>(); \
	}
#elif LIMBOAI_GDEXTENSION
#define LIMBO_REGISTER_TASK(m_class) LimboTaskDB::register_task<m_class>();
#endif

#define TASK_CATEGORY(m_cat)                           \
public:                                                \
	static _FORCE_INLINE_ String get_task_category() { \
		return String(#m_cat);                         \
	}                                                  \
                                                       \
private:

#endif // LIMBO_TASK_DB_H