cat.nyaa.nyaacore.component.IMessageQueue Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of rpgitems Show documentation
The legacy RPGItems continues. (fork)
There is a newer version: 3.12.2
package cat.nyaa.nyaacore.component;

import cat.nyaa.nyaacore.Message;
import com.google.common.collect.LinkedListMultimap;
import com.google.common.collect.Multimap;
import org.bukkit.OfflinePlayer;
import org.bukkit.event.EventHandler;
import org.bukkit.event.HandlerList;
import org.bukkit.event.Listener;
import org.bukkit.event.player.PlayerJoinEvent;

import java.text.DateFormat;
import java.util.Collections;
import java.util.HashMap;
import java.util.Map;

/**
 * Send a message to a offline player so he will receive it when he login
 */
public interface IMessageQueue extends IComponent {

    /**
     * Send a message to a offline player
     * If player is online, behavior is implementation-defined
     *
     * @param player  recipient of the message
     * @param message content of the message
     */
    void send(OfflinePlayer player, Message message);

    /**
     * Send a message to a offline player, with a set
     * If player is online, behavior is implementation-defined
     *
     * @param player  recipient of the message
     * @param message content of the message
     */
    void send(OfflinePlayer player, Message message, long time);

    /**
     * Default in-memory message queue, which does not persist across reboot.
     */
    class DefaultMessageQueue implements IMessageQueue, Listener {
        /* Map[Recipient, Pairs[SystemMilliTimeWhenTheMessageIsSent, TheMessage]] */
        static final Map> messageStorage = new HashMap<>();

        @Override
        public boolean canReplaceMe(IComponent successor) {
            HandlerList.unregisterAll(this);
            messageStorage.forEach(
                    (player, msgs) -> msgs.forEach(
                            (time, message) -> ((IMessageQueue) successor).send(player, message, time)
                    )
            );
            return true;
        }

        @Override
        public void send(OfflinePlayer player, Message message) {
            send(player, message, System.currentTimeMillis());
        }

        @Override
        public void send(OfflinePlayer player, Message message, long time) {
            if (player.isOnline()) return;
            messageStorage.compute(player, (uuid, msgs) -> {
                if (msgs == null) msgs = LinkedListMultimap.create();
                msgs.put(time, message);
                return msgs;
            });
        }

        @EventHandler
        public void onPlayerJoin(PlayerJoinEvent event) {
            Multimap msgs = messageStorage.remove(event.getPlayer());
            if (msgs == null) return;
            msgs.forEach((time, msgJson) -> {
                Message message = new Message("").append(DateFormat.getDateTimeInstance(DateFormat.LONG, DateFormat.LONG).format(time) + ": {message}", Collections.singletonMap("{message}", msgJson.inner));
                message.send(event.getPlayer());
            });
        }
    }
}