summaryrefslogtreecommitdiff
path: root/src/main/java/net/uomc/mineshaft/DropUtil.java
blob: dce4a16c6844ff10921030ff3864a3bc2f006454 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
package net.uomc.mineshaft;

import java.util.ArrayList;
import java.util.Comparator;
import java.util.HashMap;
import java.util.Map;
import java.util.Random;
import java.util.concurrent.TimeUnit;
import java.util.stream.Collectors;

import com.mouldycheerio.dbot.util.EventWaiter;
import com.mouldycheerio.dbot.util.PeelingUtils;

import net.dv8tion.jda.api.EmbedBuilder;
import net.dv8tion.jda.api.entities.Member;
import net.dv8tion.jda.api.entities.Message;
import net.dv8tion.jda.api.entities.MessageEmbed;
import net.dv8tion.jda.api.entities.channel.middleman.MessageChannel;
import net.dv8tion.jda.api.events.message.MessageReceivedEvent;

public class DropUtil {
    private static final String FOLLOWING_STRING = "The following items have dropped:";
    private static final int TO_PICK = 3;
    private static final double MAX_AMOUNT_PERCENT = 0.35;
    private static final long DROP_DURATION_SECONDS = 90;
    public static final String LOOT_STRING = "loot";

    public static Map<MineshaftItem, Long> getDeathDrops(Mineshaft bot, Member member) {
        Map<MineshaftItem, Long> drop = new HashMap<>();

        Map<MineshaftItem, Long> inventory = bot.getInventory(member);

        int items = TO_PICK;
        if (bot.getItem(member, MineshaftItem.XP) > 0) {
            drop.put(MineshaftItem.XP,
                    (long) Math.ceil(bot.getItem(member, MineshaftItem.XP) * (Math.random() * MAX_AMOUNT_PERCENT)));
            items--;
        }

        int random = new Random().nextInt();
        inventory.entrySet().stream().filter(e -> e.getValue() > 0 && e.getKey() != MineshaftItem.XP)
                .sorted(Comparator.comparingInt(o -> System.identityHashCode(o) ^ random))
                .limit(items).collect(Collectors.toList()).forEach(e -> {
                    drop.put(e.getKey(),
                            (long) Math.ceil(e.getValue() * (Math.random() * MAX_AMOUNT_PERCENT)));
                });

        return drop;
    }

    public static void spawnDropEdit(Mineshaft bot, Message message, MessageEmbed embed, Map<MineshaftItem, Long> items) {
        message.editMessageEmbeds(embed).setContent("").queue(m -> {
            waitForPickUp(bot, embed, items, m);
        });
    }

    private static void waitForPickUp(Mineshaft bot, MessageEmbed embed,
            Map<MineshaftItem, Long> items, Message m) {
        EventWaiter eventWaiter = new EventWaiter();
        m.getChannel().getJDA().addEventListener(eventWaiter);

        eventWaiter.waitForEvent(MessageReceivedEvent.class,
                e -> {
                    if (!e.getMessage().getContentRaw().equalsIgnoreCase(LOOT_STRING))
                        return false;

                    return true;
                }, e -> {
                    bot.addItems(e.getMember(), items);
                    EmbedBuilder em = new EmbedBuilder(embed);
                    String desc = embed.getDescription().split(FOLLOWING_STRING)[0];
                    em.setDescription( desc + "\n" + e.getMember().getAsMention() + " claimed the loot!\n**"
                            + bot.createItemList(items, "+%s") + "**");
                    m.editMessageEmbeds(em.build()).queue();
                    cleanupDrop(m.getChannel(), m);
                }, DROP_DURATION_SECONDS, TimeUnit.SECONDS, () -> {
                    m.delete().queue();
                    cleanupDrop(m.getChannel(), m);
                });
    }

    public static void spawnDrop(Mineshaft bot, MessageChannel channel, MessageEmbed embed,
            Map<MineshaftItem, Long> items) {
        channel.sendMessageEmbeds(embed).queue(m -> {
            waitForPickUp(bot, embed, items, m);
        });
    }

    public static MessageEmbed getNoDropEmbed(Mineshaft bot, String reason, String title) {
        EmbedBuilder eb = new EmbedBuilder();
        eb.setTitle(title);
        eb.getDescriptionBuilder().append(reason);
        eb.getDescriptionBuilder().append("\nThey dropped nothing...");
        eb.setColor(bot.color);
        return eb.build();
    }

    public static MessageEmbed getDropEmbed(Mineshaft bot, Map<MineshaftItem, Long> items, MessageEmbed embed) {
        EmbedBuilder eb = new EmbedBuilder(embed);
        if (!bot.isEmpty(items)) {
            eb.getDescriptionBuilder().append("\n" + FOLLOWING_STRING + "\n\n");
            eb.getDescriptionBuilder().append(bot.createItemList(items));
            eb.getDescriptionBuilder().append("\nSay `" + LOOT_STRING + "` to loot these items.\nHurry you only have "
                    + PeelingUtils.formatTimeRelativeFromNow(DROP_DURATION_SECONDS * 1000l) + " to collect them!");
        }
        return eb.build();
    }

    public static MessageEmbed getDropEmbed(Mineshaft bot, Map<MineshaftItem, Long> items, String description,
            String title) {
        EmbedBuilder eb = new EmbedBuilder();
        eb.setTitle(title);
        eb.getDescriptionBuilder().append(description);
        if (!bot.isEmpty(items)) {
            eb.getDescriptionBuilder().append("\n" + FOLLOWING_STRING + "\n\n");
            eb.getDescriptionBuilder().append(bot.createItemList(items));
            eb.getDescriptionBuilder().append("\n\nSay `" + LOOT_STRING + "` to loot these items.\nHurry you only have "
                    + PeelingUtils.formatTimeRelativeFromNow(DROP_DURATION_SECONDS * 1000l) + " to collect them!");
        }
        eb.setColor(bot.color);
        return eb.build();
    }

    public static void cleanupDrop(MessageChannel channel, Message message) {
        channel.getHistoryAfter(message, 100).queueAfter(1, TimeUnit.SECONDS, h -> {
            ArrayList<Message> msgs = new ArrayList<Message>();
            h.getRetrievedHistory().forEach(m -> {
                if (m.getContentRaw().equalsIgnoreCase(LOOT_STRING)) {
                    msgs.add(m);
                }
            });
            PeelingUtils.bulkDelte(channel, msgs);
        });
        ;
    }
}