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

package net.frozenorb.apiv3.models;

import com.google.common.collect.Collections2;
import lombok.AllArgsConstructor;
import lombok.Getter;
import net.frozenorb.apiv3.APIv3;
import org.bson.types.ObjectId;
import org.mongodb.morphia.annotations.Entity;
import org.mongodb.morphia.annotations.Id;
import org.mongodb.morphia.annotations.Indexed;

import java.util.Date;
import java.util.HashSet;
import java.util.Set;
import java.util.UUID;

@Entity(value = "grants", noClassnameStored = true)
@AllArgsConstructor
public final class Grant {

    @Getter @Id private String id;
    @Getter @Indexed 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 @Indexed private String rank;
    @Getter private Date expiresAt;

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

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

    public static Grant byId(String id) {
        return APIv3.getDatastore().createQuery(Grant.class).field("id").equal(id).get();
    }

    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 void delete(User removedBy, String reason) {
        this.removedBy = removedBy.getId();
        this.removedAt = new Date();
        this.removalReason = reason;

        APIv3.getDatastore().save(this);
    }

    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();
    }

}