entity_spell_system/ai/ai_spell_action.h

#ifndef AI_SPELL_ACTION_H
#define AI_SPELL_ACTION_H

#include "core/ustring.h"

#include "ai_action.h"

class Spell;

class AISpellAction : public AIAction {
	GDCLASS(AISpellAction, AIAction);

public:
	enum AISpellActionType {
		AI_SPELL_ACTION_TYPE_NONE,
		AI_SPELL_ACTION_TYPE_ATTACK,
		AI_SPELL_ACTION_TYPE_CHARGE,
		AI_SPELL_ACTION_TYPE_CC,
		AI_SPELL_ACTION_TYPE_DPS,
		AI_SPELL_ACTION_TYPE_REGENERATION,
		AI_SPELL_ACTION_TYPE_STUNBREAK
	};

	static const String BINDING_STRING_SPELL_ACTION_TYPE;

public:
	AISpellActionType get_type() const;
	void set_type(const AISpellActionType type);

	Ref<Spell> get_spell();
	void set_spell(Ref<Spell> spell);

	AISpellAction();

protected:
	static void _bind_methods();

private:
	AISpellActionType _type;
	Ref<Spell> _spell;
};

VARIANT_ENUM_CAST(AISpellAction::AISpellActionType);

#endif