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

package net.frozenorb.apiv3.model;

import com.google.common.collect.ImmutableSet;
import lombok.Getter;
import lombok.ToString;
import net.frozenorb.apiv3.LiteFullJson;
import org.bson.Document;

import java.time.Instant;
import java.util.Collection;
import java.util.Set;
import java.util.UUID;

@ToString
public final class Grant implements LiteFullJson {

    @Getter private String id;
    @Getter private UUID target;
    @Getter private String reason;
    @Getter private Set<String> scopes;
    @Getter private String rank;
    @Getter private UUID addedBy;
    @Getter private Instant addedAt;

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

    public Grant(Document json) {
        this.id = json.getString("_id");
        this.target = UUID.fromString(json.getString("target"));
        this.reason = json.getString("reason");
        this.scopes = ImmutableSet.copyOf((Collection) json.get("scopes")); // This is a safe cast, the collection's type is always String
        this.rank = json.getString("rank");
        this.addedBy = UUID.fromString(json.getString("addedBy"));
        this.addedAt = (Instant) json.get("addedAt");

        if (json.containsKey("removedBy")) {
            this.removedBy = UUID.fromString(json.getString("removedBy"));
            this.removedAt = (Instant) json.get("removedAt");
            this.removalReason = json.getString("removalReason");
        }
    }

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

        json.put("id", id);
        json.put("target", target.toString());
        json.put("reason", reason);
        json.put("scopes", scopes);
        json.put("rank", rank);
        json.put("addedBy", addedBy.toString());
        json.put("addedAt", addedAt.toString());

        if (removedBy != null) {
            json.put("removedBy", removedBy.toString());
            json.put("removedAt", removedAt.toString());
            json.put("removalReason", removalReason);
        }

        return json;
    }

}