summaryrefslogtreecommitdiff
path: root/src/main/java/net/uomc/mineshaft/PlayerStats.java
blob: 40ef349090b2c74571e755363ddd6c5155b4e1e0 (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
package net.uomc.mineshaft;

import java.io.File;
import java.util.HashMap;
import java.util.Map;
import net.dv8tion.jda.api.entities.Member;

public class PlayerStats {

    public static final int MAX_PICKAXE_LEVEL = 5;

    public static final String DIMENSION_TABLE = "dimension";
    public static final String FARM_LEVEL_TABLE = "farms";

    private String playersDb;

    private Map<String, SavedStat> stats;

    public PlayerStats(Mineshaft mineshaft) {
        File playersDbFile = new File(mineshaft.getDatadir(), "players.db");

        // migrate old file if exists
        if (!playersDbFile.exists()) {
            File oldplayersDbFile = new File(mineshaft.getDatadir(), "pickaxes.db");
            if (oldplayersDbFile.exists()) {
                oldplayersDbFile.renameTo(playersDbFile);
            }
        }

        stats = new HashMap<>();

        playersDb = playersDbFile.getPath();
        initStats();
    }

    private void initStats() {
        createStat(DIMENSION_TABLE);
        createStat(FARM_LEVEL_TABLE);
    }

    public SavedStat createStat(String name) {
        SavedStat savedStat = new SavedStat(playersDb, name);
        stats.put(name, savedStat);
        return savedStat;
    }


    public SavedStat getStat(String name) {
        if (stats.containsKey(name))
            return stats.get(name);

        SavedStat savedStat = new SavedStat(playersDb, name);
        stats.put(name, savedStat);
        return savedStat;
    }

    public void setFarmLevel(Member member, long value) {
        getStat(FARM_LEVEL_TABLE).set(member, value);
    }

    public long getFarmLevel(String member, String guild) {
        return getStat(FARM_LEVEL_TABLE).get(member, guild);
    }

    public long getFarmLevel(Member member) {
        return getFarmLevel(member.getId(), member.getGuild().getId());
    }

    public void setNether(Member member, boolean nether) {
        getStat(DIMENSION_TABLE).set(member, nether ? 1 : 0);
    }

    public boolean isNether(Member member) {
        return getStat(DIMENSION_TABLE).get(member) == 1;
    }

	public String getDbPath() {
		return playersDb;
	}
}