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

package net.frozenorb.apiv3.models;

import lombok.Getter;
import net.frozenorb.apiv3.APIv3;
import net.frozenorb.apiv3.actors.Actor;
import net.frozenorb.apiv3.actors.ActorType;
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.UUID;

@Entity(value = "punishments", noClassnameStored = true)
public final class Punishment {

    @Getter @Id private String id;
    @Getter @Indexed private UUID target;
    @Getter private String reason;
    @Getter @Indexed private PunishmentType type; // Type is indexed for the rank dump
    @Getter private Date expiresAt;

    @Getter private UUID addedBy;
    @Getter @Indexed private Date addedAt;
    @Getter private String actorName;
    @Getter private ActorType actorType;

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

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

    public Punishment() {} // For Morphia

    public Punishment(User target, String reason, PunishmentType type, Date expiresAt, User addedBy, Actor actor) {
        this.id = new ObjectId().toString();
        this.target = target.getId();
        this.reason = reason;
        this.type = type;
        this.expiresAt = (Date) expiresAt.clone();
        this.addedBy = addedBy.getId();
        this.addedAt = new Date();
        this.actorName = actor.getName();
        this.actorType = actor.getType();
    }

    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.after(new Date());
        }
    }

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

    public enum PunishmentType {

        BLACKLIST, BAN, MUTE, WARN

    }

}