The-pack-family
/
APIv3
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Create an ip ban for the user's last used ip address if their current ip address is not provided when banning. Closes #46

This commit is contained in:
Colin McDonald 2016-07-17 16:56:51 -04:00
parent c30f11ead4
commit 612fe880dc
2 changed files with 13 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/main/java/net/frozenorb/apiv3/model/IpLogEntry.java
View File

@ -46,6 +46,10 @@ public final class IpLogEntry {
ipLogCollection.find(new Document("user", user)).sort(new Document("lastSeenAt", -1)).into(new LinkedList<>(), SyncUtils.vertxWrap(callback));
}
public static void findLatestByUser(UUID user, SingleResultCallback<IpLogEntry> callback) {
ipLogCollection.find(new Document("user", user)).sort(new Document("lastSeenAt", -1)).limit(1).first(SyncUtils.vertxWrap(callback));
}
public static void findByIp(String userIp, SingleResultCallback<List<IpLogEntry>> callback) {
ipLogCollection.find(new Document("userIp", userIp)).into(new LinkedList<>(), SyncUtils.vertxWrap(callback));
}

9
src/main/java/net/frozenorb/apiv3/route/punishments/POSTPunishments.java
View File

@ -9,6 +9,7 @@ import net.frozenorb.apiv3.APIv3;
import net.frozenorb.apiv3.auditLog.AuditLog;
import net.frozenorb.apiv3.auditLog.AuditLogActionType;
import net.frozenorb.apiv3.model.IpBan;
import net.frozenorb.apiv3.model.IpLogEntry;
import net.frozenorb.apiv3.model.Punishment;
import net.frozenorb.apiv3.model.User;
import net.frozenorb.apiv3.unsorted.Permissions;
@ -89,6 +90,14 @@ public final class POSTPunishments implements Handler<RoutingContext> {
String accessDenialReason = punishment.getAccessDenialReason();
String userIp = requestBody.getString("userIp");
if (userIp == null) {
IpLogEntry latestIpLogEntry = SyncUtils.runBlocking(v -> IpLogEntry.findLatestByUser(target.getId(), v));
if (latestIpLogEntry != null) {
userIp = latestIpLogEntry.getUserIp();
}
}
if ((type == Punishment.PunishmentType.BAN || type == Punishment.PunishmentType.BLACKLIST) && userIp != null) {
IpBan ipBan = new IpBan(userIp, punishment);
SyncUtils.<Void>runBlocking(v -> ipBan.insert(v));