#1555 Add RegisterEvent and AuthMeAsyncPreRegisterEvent (#1559)

* #1555 Add RegisterEvent and AuthMeAsyncPreRegisterEvent

* Add missing javadoc

src/main/java/fr/xephi/authme/events/AuthMeAsyncPreRegisterEvent.java

@@ -0,0 +1,70 @@
+package fr.xephi.authme.events;
+
+import org.bukkit.entity.Player;
+import org.bukkit.event.Event;
+import org.bukkit.event.HandlerList;
+
+/**
+ * This event is called when a player uses the register command,
+ * it's fired even when a user does a /register with invalid arguments.
+ * {@link #setCanRegister(boolean) event.setCanRegister(false)} prevents the player from registering.
+ */
+public class AuthMeAsyncPreRegisterEvent extends CustomEvent {
+
+    private static final HandlerList handlers = new HandlerList();
+    private final Player player;
+    private boolean canRegister = true;
+
+    /**
+     * Constructor.
+     *
+     * @param player The player
+     * @param isAsync True if the event is async, false otherwise
+     */
+    public AuthMeAsyncPreRegisterEvent(Player player, boolean isAsync) {
+        super(isAsync);
+        this.player = player;
+    }
+
+    /**
+     * Return the player concerned by this event.
+     *
+     * @return The player who executed a valid {@code /login} command
+     */
+    public Player getPlayer() {
+        return player;
+    }
+
+    /**
+     * Return whether the player is allowed to register.
+     *
+     * @return True if the player can log in, false otherwise
+     */
+    public boolean canRegister() {
+        return canRegister;
+    }
+
+    /**
+     * Define whether or not the player may register.
+     *
+     * @param canRegister True to allow the player to log in; false to prevent him
+     */
+    public void setCanRegister(boolean canRegister) {
+        this.canRegister = canRegister;
+    }
+
+    /**
+     * Return the list of handlers, equivalent to {@link #getHandlers()} and required by {@link Event}.
+     *
+     * @return The list of handlers
+     */
+    public static HandlerList getHandlerList() {
+        return handlers;
+    }
+
+    @Override
+    public HandlerList getHandlers() {
+        return handlers;
+    }
+
+}

src/main/java/fr/xephi/authme/events/RegisterEvent.java

@@ -0,0 +1,47 @@
+package fr.xephi.authme.events;
+
+import org.bukkit.entity.Player;
+import org.bukkit.event.Event;
+import org.bukkit.event.HandlerList;
+
+/**
+ * Event fired when a player has successfully registered.
+ */
+public class RegisterEvent extends CustomEvent {
+
+    private static final HandlerList handlers = new HandlerList();
+    private final Player player;
+
+    /**
+     * Constructor.
+     *
+     * @param player The player
+     */
+    public RegisterEvent(Player player) {
+        this.player = player;
+    }
+
+    /**
+     * Return the player that has successfully logged in or registered.
+     *
+     * @return The player
+     */
+    public Player getPlayer() {
+        return player;
+    }
+
+    /**
+     * Return the list of handlers, equivalent to {@link #getHandlers()} and required by {@link Event}.
+     *
+     * @return The list of handlers
+     */
+    public static HandlerList getHandlerList() {
+        return handlers;
+    }
+
+    @Override
+    public HandlerList getHandlers() {
+        return handlers;
+    }
+
+}

src/main/java/fr/xephi/authme/process/register/AsyncRegister.java

@@ -4,14 +4,17 @@ import ch.jalu.injector.factory.SingletonStore;
 import fr.xephi.authme.data.auth.PlayerAuth;
 import fr.xephi.authme.data.auth.PlayerCache;
 import fr.xephi.authme.datasource.DataSource;
+import fr.xephi.authme.events.AuthMeAsyncPreRegisterEvent;
 import fr.xephi.authme.message.MessageKey;
 import fr.xephi.authme.process.AsynchronousProcess;
 import fr.xephi.authme.process.register.executors.RegistrationExecutor;
 import fr.xephi.authme.process.register.executors.RegistrationMethod;
 import fr.xephi.authme.process.register.executors.RegistrationParameters;
+import fr.xephi.authme.service.BukkitService;
 import fr.xephi.authme.service.CommonService;
 import fr.xephi.authme.service.bungeecord.BungeeSender;
 import fr.xephi.authme.service.bungeecord.MessageType;
+import fr.xephi.authme.settings.properties.PluginSettings;
 import fr.xephi.authme.settings.properties.RegistrationSettings;
 import fr.xephi.authme.settings.properties.RestrictionSettings;
 import fr.xephi.authme.util.PlayerUtils;
@@ -32,6 +35,8 @@ public class AsyncRegister implements AsynchronousProcess {
     @Inject
     private PlayerCache playerCache;
     @Inject
+    private BukkitService bukkitService;
+    @Inject
     private CommonService service;
     @Inject
     private SingletonStore<RegistrationExecutor> registrationExecutorFactory;
@@ -44,9 +49,9 @@ public class AsyncRegister implements AsynchronousProcess {
     /**
      * Performs the registration process for the given player.
      *
-     * @param variant the registration method
+     * @param variant    the registration method
      * @param parameters the parameters
-     * @param <P> parameters type
+     * @param <P>        parameters type
      */
     public <P extends RegistrationParameters> void register(RegistrationMethod<P> variant, P parameters) {
         if (preRegisterCheck(parameters.getPlayer())) {
@@ -57,6 +62,13 @@ public class AsyncRegister implements AsynchronousProcess {
         }
     }
 
+    /**
+     * Checks if the player is able to register, in that case the {@link AuthMeAsyncPreRegisterEvent} is invoked.
+     *
+     * @param player the player which is trying to register.
+     *
+     * @return true if the checks are successful and the event hasn't marked the action as denied, false otherwise.
+     */
     private boolean preRegisterCheck(Player player) {
         final String name = player.getName().toLowerCase();
         if (playerCache.isAuthenticated(name)) {
@@ -70,6 +82,13 @@ public class AsyncRegister implements AsynchronousProcess {
             return false;
         }
 
+        boolean isAsync = service.getProperty(PluginSettings.USE_ASYNC_TASKS);
+        AuthMeAsyncPreRegisterEvent event = new AuthMeAsyncPreRegisterEvent(player, isAsync);
+        bukkitService.callEvent(event);
+        if (!event.canRegister()) {
+            return false;
+        }
+
         return isPlayerIpAllowedToRegister(player);
     }
 
@@ -77,11 +96,11 @@ public class AsyncRegister implements AsynchronousProcess {
      * Executes the registration.
      *
      * @param parameters the registration parameters
-     * @param executor the executor to perform the registration process with
-     * @param <P> registration params type
+     * @param executor   the executor to perform the registration process with
+     * @param <P>        registration params type
      */
     private <P extends RegistrationParameters>
-            void executeRegistration(P parameters, RegistrationExecutor<P> executor) {
+    void executeRegistration(P parameters, RegistrationExecutor<P> executor) {
         PlayerAuth auth = executor.buildPlayerAuth(parameters);
         if (database.saveAuth(auth)) {
             executor.executePostPersistAction(parameters);
@@ -95,6 +114,7 @@ public class AsyncRegister implements AsynchronousProcess {
      * Checks whether the registration threshold has been exceeded for the given player's IP address.
      *
      * @param player the player to check
+     *
      * @return true if registration may take place, false otherwise (IP check failed)
      */
     private boolean isPlayerIpAllowedToRegister(Player player) {

src/main/java/fr/xephi/authme/process/register/ProcessSyncEmailRegister.java

@@ -2,8 +2,10 @@ package fr.xephi.authme.process.register;
 
 import fr.xephi.authme.ConsoleLogger;
 import fr.xephi.authme.data.limbo.LimboService;
+import fr.xephi.authme.events.RegisterEvent;
 import fr.xephi.authme.message.MessageKey;
 import fr.xephi.authme.process.SynchronousProcess;
+import fr.xephi.authme.service.BukkitService;
 import fr.xephi.authme.service.CommonService;
 import fr.xephi.authme.util.PlayerUtils;
 import org.bukkit.entity.Player;
@@ -15,6 +17,9 @@ import javax.inject.Inject;
  */
 public class ProcessSyncEmailRegister implements SynchronousProcess {
 
+    @Inject
+    private BukkitService bukkitService;
+
     @Inject
     private CommonService service;
 
@@ -34,6 +39,7 @@ public class ProcessSyncEmailRegister implements SynchronousProcess {
         limboService.replaceTasksAfterRegistration(player);
 
         player.saveData();
+        bukkitService.callEvent(new RegisterEvent(player));
         ConsoleLogger.fine(player.getName() + " registered " + PlayerUtils.getPlayerIp(player));
     }
 

src/main/java/fr/xephi/authme/process/register/ProcessSyncPasswordRegister.java

@@ -2,8 +2,10 @@ package fr.xephi.authme.process.register;
 
 import fr.xephi.authme.ConsoleLogger;
 import fr.xephi.authme.data.limbo.LimboService;
+import fr.xephi.authme.events.RegisterEvent;
 import fr.xephi.authme.message.MessageKey;
 import fr.xephi.authme.process.SynchronousProcess;
+import fr.xephi.authme.service.BukkitService;
 import fr.xephi.authme.service.CommonService;
 import fr.xephi.authme.service.bungeecord.BungeeSender;
 import fr.xephi.authme.settings.commandconfig.CommandManager;
@@ -31,6 +33,9 @@ public class ProcessSyncPasswordRegister implements SynchronousProcess {
     @Inject
     private CommandManager commandManager;
 
+    @Inject
+    private BukkitService bukkitService;
+
     ProcessSyncPasswordRegister() {
     }
 
@@ -60,6 +65,7 @@ public class ProcessSyncPasswordRegister implements SynchronousProcess {
         }
 
         player.saveData();
+        bukkitService.callEvent(new RegisterEvent(player));
         ConsoleLogger.fine(player.getName() + " registered " + PlayerUtils.getPlayerIp(player));
 
         // Kick Player after Registration is enabled, kick the player