APIv3/src/main/java/net/frozenorb/apiv3/collections/Server.java

package net.frozenorb.apiv3.collections;

import io.vertx.core.AsyncResult;
import io.vertx.core.Handler;
import lombok.Getter;
import lombok.ToString;
import net.frozenorb.apiv3.APIv3;
import net.frozenorb.apiv3.LiteFullJson;
import net.frozenorb.apiv3.utils.MongoUtils;
import org.bson.Document;

import java.time.Instant;
import java.util.*;
import java.util.stream.Collectors;

@ToString(exclude={ "secret" })
public final class Server implements LiteFullJson {

    private static final String COLLECTION_NAME = "server";

    @Getter private String id;
    @Getter private String bungeeId;
    @Getter private String displayName;
    @Getter private String secret;
    @Getter private String group;
    @Getter private String internalIp;
    @Getter private Instant lastUpdate;
    @Getter private double lastTps;
    @Getter private List<UUID> players;

    public static void findAll(Handler<AsyncResult<Collection<Server>>> callback) {
        MongoUtils.findAndTransform(COLLECTION_NAME, new Document(), Server::new, callback);
    }

    public static void findById(String id, Handler<AsyncResult<Server>> callback) {
        MongoUtils.findOneAndTransform(COLLECTION_NAME, new Document("_id", id), Server::new, callback);
    }

    public static void findByGroup(String groupId, Handler<AsyncResult<Collection<Server>>> callback) {
        MongoUtils.findAndTransform(COLLECTION_NAME, new Document("group", groupId), Server::new, callback);
    }

    private Server(Document json) {
        this.id = json.getString("_id");
        this.bungeeId = json.getString("bungeeId");
        this.displayName = json.getString("displayName");
        this.secret = json.getString("secret");
        this.group = json.getString("group");
        this.internalIp = json.getString("internalIp");
        this.lastUpdate = json.get("lastUpdate", Instant.class);
        this.lastTps = json.get("lastTps", Number.class).doubleValue();
        this.players = new ArrayList<>();

        for (Object uuidString : json.get("players", Collection.class)) {
            players.add(UUID.fromString((String) uuidString));
        }
    }

    public void update() {
        APIv3.getMongo().getCollection(COLLECTION_NAME).updateOne(new Document("_id", id), new Document("$set", new Document("lastUpdate", new Date())), (result, error) -> {});
    }

    public Document toLiteJson() {
        Document json = new Document();

        json.put("_id", id);
        json.put("bungeeId", bungeeId);
        json.put("displayName", displayName);
        json.put("group", group);
        json.put("internalIp", internalIp);

        return json;
    }

    public Document toFullJson() {
        Document json = toLiteJson();

        json.put("lastUpdate", lastUpdate.toString());
        json.put("lastTps", lastTps);
        json.put("players", players.stream().map(UUID::toString).collect(Collectors.toList()));

        return json;
    }

}