andreacavalli/TelegramBots
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Be able to add AbilityExtensions in Bot constructor

Browse Source
This commit is contained in:
Christian Blos 2020-10-05 18:05:25 +02:00
parent 203b26587f
commit b715f2a154
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
telegrambots-abilities/src/main/java/org/telegram/abilitybots/api/bot/BaseAbilityBot.java
View File

@ -106,6 +106,7 @@ public abstract class BaseAbilityBot extends DefaultAbsSender implements Ability
private final String botUsername; private final String botUsername;
// Ability registry // Ability registry
protected final List<AbilityExtension> extensions = new ArrayList<>();
private Map<String, Ability> abilities; private Map<String, Ability> abilities;
private Map<String, Stats> stats; private Map<String, Stats> stats;
@ -280,10 +281,10 @@ public abstract class BaseAbilityBot extends DefaultAbsSender implements Ability
private void registerAbilities() { private void registerAbilities() {
try { try {
// Collect all classes that implement AbilityExtension declared in the bot // Collect all classes that implement AbilityExtension declared in the bot
List<AbilityExtension> extensions = stream(getClass().getMethods()) extensions.addAll(stream(getClass().getMethods())
.filter(checkReturnType(AbilityExtension.class)) .filter(checkReturnType(AbilityExtension.class))
.map(returnExtension(this)) .map(returnExtension(this))
.collect(Collectors.toList()); .collect(Collectors.toList()));
// Add the bot itself as it is an AbilityExtension // Add the bot itself as it is an AbilityExtension
extensions.add(this); extensions.add(this);