Account.java 3.03 KB
Newer Older
Melledy's avatar
Melledy committed
1
2
package emu.grasscutter.game;

3
import dev.morphia.annotations.*;
Melledy's avatar
Melledy committed
4
5
6
7
import emu.grasscutter.database.DatabaseHelper;
import emu.grasscutter.utils.Crypto;
import emu.grasscutter.utils.Utils;

Melledy's avatar
Melledy committed
8
import java.util.ArrayList;
9
10
import java.util.List;

11
12
import org.bson.Document;

13
14
import com.mongodb.DBObject;

KingRainbow44's avatar
KingRainbow44 committed
15
@Entity(value = "accounts", useDiscriminator = false)
Melledy's avatar
Melledy committed
16
17
18
19
20
21
22
23
public class Account {
	@Id private String id;
	
	@Indexed(options = @IndexOptions(unique = true))
	@Collation(locale = "simple", caseLevel = true)
	private String username;
	private String password; // Unused for now
	
24
	@AlsoLoad("playerUid") private int playerId;
Melledy's avatar
Melledy committed
25
26
27
28
	private String email;
	
	private String token;
	private String sessionKey; // Session token for dispatch server
29
	private List<String> permissions;
30
	
Melledy's avatar
Melledy committed
31
	@Deprecated
Melledy's avatar
Melledy committed
32
33
34
	public Account() {
		this.permissions = new ArrayList<>();
	}
Melledy's avatar
Melledy committed
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

	public String getId() {
		return id;
	}

	public void setId(String id) {
		this.id = id;
	}

	public String getUsername() {
		return username;
	}

	public void setUsername(String username) {
		this.username = username;
	}

	public String getPassword() {
		return password;
	}

	public void setPassword(String password) {
		this.password = password;
	}

	public String getToken() {
		return token;
	}

	public void setToken(String token) {
		this.token = token;
	}

68
	public int getPlayerUid() {
Melledy's avatar
Melledy committed
69
70
71
72
73
74
75
76
		return this.playerId;
	}

	public void setPlayerId(int playerId) {
		this.playerId = playerId;
	}
	
	public String getEmail() {
4Benj_'s avatar
4Benj_ committed
77
78
79
80
81
		if(email != null && !email.isEmpty()) {
			return email;
		} else {
			return "";
		}
Melledy's avatar
Melledy committed
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
	}

	public void setEmail(String email) {
		this.email = email;
	}

	public String getSessionKey() {
		return this.sessionKey;
	}

	public String generateSessionKey() {
		this.sessionKey = Utils.bytesToHex(Crypto.createSessionKey(32));
		this.save();
		return this.sessionKey;
	}
97
98
99
100
101
102
103
104
105
106
107
108

	/**
	 * The collection of a player's permissions.
	 */
	public List<String> getPermissions() {
		return this.permissions;
	}
	
	public boolean addPermission(String permission) {
		if(this.permissions.contains(permission)) return false;
		this.permissions.add(permission); return true;
	}
109
110

	public boolean hasPermission(String permission) {
lhhxxxxx's avatar
lhhxxxxx committed
111
112
113
114
		return this.permissions.contains(permission) ||
                this.permissions.contains("*") ||
                (this.permissions.contains("player") || this.permissions.contains("player.*")) && permission.startsWith("player.") ||
                (this.permissions.contains("server") || this.permissions.contains("server.*")) && permission.startsWith("server.");
115
	}
116
117
118
119
	
	public boolean removePermission(String permission) {
		return this.permissions.remove(permission);
	}
Melledy's avatar
Melledy committed
120
121
122
123
124
125
126
127
128
129
130
	
	// TODO make unique
	public String generateLoginToken() {
		this.token = Utils.bytesToHex(Crypto.createSessionKey(32));
		this.save();
		return this.token;
	}
	
	public void save() {
		DatabaseHelper.saveAccount(this);
	}
131

132
	@PreLoad
133
	public void onLoad(Document document) {
134
		// Grant the superuser permissions to accounts created before the permissions update
135
		if (!document.containsKey("permissions")) {
136
137
138
			this.addPermission("*");
		}
	}
Melledy's avatar
Melledy committed
139
}