io.polyglotted.pgmodel.ac.Subject Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of pg-model Show documentation
Standard set of models for geo coding, access control and elastic search abstraction
The newest version!
package io.polyglotted.pgmodel.ac;

import com.google.common.collect.ImmutableMap;
import lombok.AccessLevel;
import lombok.RequiredArgsConstructor;
import lombok.Setter;
import lombok.ToString;
import lombok.experimental.Accessors;

import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import java.util.Objects;

import static com.google.common.base.Preconditions.checkNotNull;
import static com.google.common.base.Predicates.not;
import static com.google.common.base.Strings.isNullOrEmpty;
import static com.google.common.collect.ImmutableList.of;
import static com.google.common.collect.Maps.filterKeys;
import static io.polyglotted.pgmodel.ac.SubjectAttribute.ACCESS_TOKEN;
import static io.polyglotted.pgmodel.ac.SubjectAttribute.CREDENTIAL;
import static io.polyglotted.pgmodel.ac.SubjectAttribute.DISPLAY_NAME;
import static io.polyglotted.pgmodel.ac.SubjectAttribute.GROUPS;
import static io.polyglotted.pgmodel.ac.SubjectAttribute.ROLES;
import static io.polyglotted.pgmodel.ac.SubjectAttribute.TRUSTED;
import static io.polyglotted.pgmodel.util.MapRetriever.deepRetrieve;
import static io.polyglotted.pgmodel.util.ModelUtil.jsonEquals;

@ToString(includeFieldNames = false)
@RequiredArgsConstructor
public final class Subject {
    public final String principal;
    public final ImmutableMap attributes;

    public final  T attribute(String path) { return deepRetrieve(attributes, path); }

    public final  T attribute(SubjectAttribute attribute) { return attribute(attribute.name()); }

    public String credential() { return attribute(CREDENTIAL); }

    public String token() { return attribute(ACCESS_TOKEN); }

    public boolean hasAttribute(SubjectAttribute attribute) { return hasAttribute(attribute.name()); }

    public boolean hasAttribute(String property) { return attributes.containsKey(property); }

    public String user() {
        String displayName = attribute(DISPLAY_NAME);
        return isNullOrEmpty(displayName) ? principal : displayName;
    }

    public boolean isTrusted() { return Boolean.TRUE.equals(attribute(TRUSTED)); }

    public List groups() {
        return attributes.containsKey(GROUPS.name()) ? attribute(GROUPS) : of();
    }

    public List roles() {
        return attributes.containsKey(ROLES.name()) ? attribute(ROLES) : of();
    }

    @Override
    public boolean equals(Object o) {
        return jsonEquals(this, o);
    }

    @Override
    public int hashCode() {
        return Objects.hash(principal, attributes);
    }

    public static Builder subjectBuilder() {
        return new Builder();
    }

    @Setter
    @Accessors(fluent = true, chain = true)
    @RequiredArgsConstructor(access = AccessLevel.PRIVATE)
    public static class Builder {
        private String principal;
        private final Map attributes = new LinkedHashMap<>();

        public Builder attribute(SubjectAttribute attr, Object value) {
            return attribute(attr.name(), value);
        }

        public Builder attribute(String key, Object value) {
            this.attributes.put(key, value);
            return this;
        }

        public Builder attributes(Map map) {
            this.attributes.putAll(map);
            return this;
        }

        public Subject build() {
            return new Subject(checkNotNull(principal, "principal is required"),
               ImmutableMap.copyOf(filterKeys(attributes, not("principal"::equals))));
        }
    }
}