summaryrefslogtreecommitdiff
path: root/src/main/java/net/uomc/mineshaft/resources/ResourceTable.java
blob: 8cce03eb9aa8f9352a1e1a3c79dd0f2370c063d9 (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
package net.uomc.mineshaft.resources;

import java.util.ArrayList;
import java.util.Collections;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;

import org.json.JSONObject;

import net.dv8tion.jda.api.entities.Member;
import net.dv8tion.jda.api.entities.User;

public class ResourceTable extends JSONObject {
    public ResourceTable(JSONObject jsonObject) {
        super();
        jsonObject.keySet().forEach(k -> put(k, jsonObject.get(k)));
    }

    public ResourceTable() {
        super();
    }

    public void setResource(Resource resource, long value) {
        put(resource.getName(), value);
    }

    public long getResource(Resource resource) {
        String name = resource.getName();
        if (has(name)) {
            return getLong(name);
        }
        return 0l;
    }

    public Map<Resource, Long> getTable(ResourceManager resourceManager) {
        HashMap<Resource, Long> hashMap = new HashMap<Resource, Long>();
        Iterator<String> keys = keys();
        while (keys.hasNext()) {
            String key = keys.next();
            Resource resource = resourceManager.getResource(key);
            if (resource != null) {
                long q = getLong(key);
                hashMap.put(resource, q);
            }
        }

        return Collections.unmodifiableMap(hashMap);
    }

    public void addToUser(Member user, ResourceManager resourceManager) {
        Map<Resource, Long> table = getTable(resourceManager);
        table.forEach((resource, q) -> {
            resource.increment(user, q);
        });
    }

    public void removeFromUser(Member user, ResourceManager resourceManager) {
        Map<Resource, Long> table = getTable(resourceManager);
        table.forEach((resource, q) -> {
            resource.increment(user, -q);
        });
    }

    /**
     * @param user
     * @param resourceManager
     * @return list of resources that are lacking
     */
    public Map<Resource, Long> doesUserHave(Member user, ResourceManager resourceManager) {
        Map<Resource, Long> table = getTable(resourceManager);
        Map<Resource, Long> lacking = new HashMap<Resource, Long>();
        table.forEach((resource, q) -> {
            if (resource.get(user) < q) {
                lacking.put(resource, q - resource.get(user));
            }
        });

        return Collections.unmodifiableMap(lacking);
    }

    public String toStringList(ResourceManager resourceManager) {
        Map<Resource, Long> table = getTable(resourceManager);

        List<String> parts = new ArrayList<String>();
        table.forEach((r, q) -> parts.add(r.prettyValue(q)));

        return String.join(", ", parts);
    }

    public void addResourceTable(ResourceTable table) {
        Iterator<String> keys = table.keys();
        while (keys.hasNext()) {
            String next = keys.next();

            put(next, optLong(next) + table.getLong(next));
        }
    }

    public ResourceTable negatives() {
        ResourceTable negatives = new ResourceTable();
        Iterator<String> keys = keys();
        while (keys.hasNext()) {
            String next = keys.next();
            long long1 = getLong(next);
            if (long1 < 0) {
                negatives.put(next, Math.abs(long1));
            }
        }
        return negatives;
    }

    public ResourceTable postives() {
        ResourceTable positives = new ResourceTable();
        Iterator<String> keys = keys();
        while (keys.hasNext()) {
            String next = keys.next();
            long long1 = getLong(next);
            if (long1 > 0) {
                positives.put(next, long1);
            }
        }
        return positives;
    }

}