APIv3/src/main/java/net/frozenorb/apiv3/models/Grant.java

package net.frozenorb.apiv3.models;

import com.google.common.collect.Collections2;
import com.mongodb.async.client.MongoCollection;
import com.mongodb.client.result.DeleteResult;
import com.mongodb.client.result.UpdateResult;
import eu.dozd.mongo.annotation.Entity;
import eu.dozd.mongo.annotation.Id;
import lombok.AllArgsConstructor;
import lombok.Getter;
import net.frozenorb.apiv3.APIv3;
import net.frozenorb.apiv3.unsorted.BlockingCallback;
import net.frozenorb.apiv3.utils.SyncUtils;
import org.bson.Document;
import org.bson.types.ObjectId;

import java.util.*;

@Entity
@AllArgsConstructor
public final class Grant {

    private static final MongoCollection<Grant> grantsCollection = APIv3.getDatabase().getCollection("grants", Grant.class);

    @Getter @Id private String id;
    @Getter private UUID user;
    @Getter private String reason;
    @Getter private Set<String> scopes = new HashSet<>(); // So on things w/o scopes we still load properly (Morphia drops empty sets)
    @Getter private String rank;
    @Getter private Date expiresAt;

    @Getter private UUID addedBy;
    @Getter private Date addedAt;

    @Getter private UUID removedBy;
    @Getter private Date removedAt;
    @Getter private String removalReason;

    public static List<Grant> findAll() {
        return SyncUtils.blockMulti(grantsCollection.find());
    }

    public static List<Grant> findByRank(Iterable<Rank> ranks) {
        return SyncUtils.blockMulti(grantsCollection.find(new Document("rank", new Document("$in", ranks))));
    }

    public static Grant findById(String id) {
        return SyncUtils.blockOne(grantsCollection.find(new Document("_id", id)));
    }

    public static List<Grant> findByUser(User user) {
        return findByUser(user.getId());
    }

    public static List<Grant> findByUser(UUID user) {
        return SyncUtils.blockMulti(grantsCollection.find(new Document("user", user)));
    }

    public Grant() {} // For Morphia

    public Grant(User user, String reason, Set<ServerGroup> scopes, Rank rank, Date expiresAt, User addedBy) {
        this.id = new ObjectId().toString();
        this.user = user.getId();
        this.reason = reason;
        this.scopes = new HashSet<>(Collections2.transform(scopes, ServerGroup::getId));
        this.rank = rank.getId();
        this.expiresAt = expiresAt;
        this.addedBy = addedBy == null ? null : addedBy.getId();
        this.addedAt = new Date();
    }

    public boolean isActive() {
        return !(isExpired() || isRemoved());
    }

    public boolean isExpired() {
        if (expiresAt == null) {
            return false; // Never expires
        } else {
            return expiresAt.before(new Date());
        }
    }

    public boolean isRemoved() {
        return removedBy != null;
    }

    public boolean appliesOn(ServerGroup serverGroup) {
        return isGlobal() || scopes.contains(serverGroup.getId());
    }

    public boolean isGlobal() {
        return scopes.isEmpty();
    }

    public void insert() {
        BlockingCallback<Void> callback = new BlockingCallback<>();
        grantsCollection.insertOne(this, callback);
        callback.get();
    }

    public void delete(User removedBy, String reason) {
        this.removedBy = removedBy.getId();
        this.removedAt = new Date();
        this.removalReason = reason;

        BlockingCallback<DeleteResult> callback = new BlockingCallback<>();
        grantsCollection.deleteOne(new Document("_id", id), callback);
        callback.get();
    }

}