PlotSquared/src/main/java/com/plotsquared/bukkit/object/comment/CommentManager.java

package com.plotsquared.bukkit.object.comment;

import com.intellectualcrafters.plot.config.C;
import com.intellectualcrafters.plot.config.Settings;
import com.intellectualcrafters.plot.object.Plot;
import com.intellectualcrafters.plot.object.PlotPlayer;
import com.intellectualcrafters.plot.object.RunnableVal;
import com.intellectualcrafters.plot.object.comment.*;
import com.intellectualcrafters.plot.util.TaskManager;
import com.plotsquared.bukkit.titles.AbstractTitle;
import org.bukkit.ChatColor;

import java.util.ArrayList;
import java.util.Collection;
import java.util.HashMap;
import java.util.concurrent.atomic.AtomicInteger;


public class CommentManager {
    public static HashMap<String, CommentInbox> inboxes = new HashMap<>();
    
    public static void sendTitle(final PlotPlayer player, final Plot plot) {
        if (!Settings.COMMENT_NOTIFICATIONS) {
            return;
        }
        if (!plot.isOwner(player.getUUID())) {
            return;
        }
        TaskManager.runTaskLaterAsync(new Runnable() {
            @Override
            public void run() {
                Collection<CommentInbox> boxes = CommentManager.inboxes.values();
                final AtomicInteger count = new AtomicInteger(0);
                final AtomicInteger size = new AtomicInteger(boxes.size());
                for (final CommentInbox inbox : inboxes.values()) {
                    inbox.getComments(plot, new RunnableVal() {
                        @Override
                        public void run() {
                            int total;
                            if (value != null) {
                                int num = 0;
                                for (PlotComment comment : (ArrayList<PlotComment>) value) {
                                    if (comment.timestamp > getTimestamp(player, inbox.toString())) {
                                        num++;
                                    }
                                }
                                total = count.addAndGet(num);
                            }
                            else {
                                total = count.get();
                            }
                            if (size.decrementAndGet() == 0 && total > 0) {
                                AbstractTitle.sendTitle(player, "", C.INBOX_NOTIFICATION.s().replaceAll("%s", "" + total), ChatColor.GOLD, ChatColor.valueOf(C.TITLE_ENTERED_PLOT_SUB_COLOR.s()));
                            }
                        }
                    });
                }
            }
        }, 20);
    }
    
    public static long getTimestamp(PlotPlayer player, String inbox) {
        Object meta = player.getMeta("inbox:"+inbox);
        if (meta == null) {
            return player.getPreviousLogin();
        }
        return (Long) meta;
    }
    
    public static void addInbox(CommentInbox inbox) {
        inboxes.put(inbox.toString().toLowerCase(), inbox);
    }
    
    public static void registerDefaultInboxes() {
        addInbox(new InboxReport());
        addInbox(new InboxPublic());
        addInbox(new InboxOwner());
    }
}