com.epam.eco.kafkamanager.PrincipalPermissionsDeleteParams Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of kafka-manager-commons Show documentation
Common models, utilities, etc
The newest version!
/*******************************************************************************
 *  Copyright 2022 EPAM Systems
 *
 *  Licensed under the Apache License, Version 2.0 (the "License"); you may not
 *  use this file except in compliance with the License.  You may obtain a copy
 *  of the License at
 *
 *    http://www.apache.org/licenses/LICENSE-2.0
 *
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
 *  WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.  See the
 *  License for the specific language governing permissions and limitations under
 *  the License.
 *******************************************************************************/
package com.epam.eco.kafkamanager;

import java.util.Map;
import java.util.Objects;
import java.util.Set;

import org.apache.commons.collections4.CollectionUtils;
import org.apache.commons.lang3.Validate;
import org.apache.kafka.common.resource.ResourceType;
import org.apache.kafka.common.security.auth.KafkaPrincipal;
import org.apache.kafka.common.utils.SecurityUtils;

import com.fasterxml.jackson.annotation.JsonCreator;
import com.fasterxml.jackson.annotation.JsonIgnore;
import com.fasterxml.jackson.annotation.JsonProperty;

import com.epam.eco.kafkamanager.utils.MapperUtils;

import static java.util.Objects.isNull;

/**
 * @author Andrei_Tytsik
 */
public class PrincipalPermissionsDeleteParams {

    private final String principal;
    private final KafkaPrincipal principalObject;
    private final Set excludes;
    private final Set resourceTypes;

    @JsonCreator
    public PrincipalPermissionsDeleteParams(
            @JsonProperty("principal") String principal) {
        this(
                principal,
                SecurityUtils.parseKafkaPrincipal(principal),
                null,
                null);
    }

    public PrincipalPermissionsDeleteParams(KafkaPrincipal principal) {
        this(principal != null ? principal.toString() : null, principal, null, null);
    }

    public PrincipalPermissionsDeleteParams(KafkaPrincipal principal,
                                            Set resourceTypes,
                                            Set excludeds) {
        this(principal != null ? principal.toString() : null, principal, resourceTypes,  excludeds);
    }

    private PrincipalPermissionsDeleteParams(
            String principal,
            KafkaPrincipal principalObject,
            Set resourceTypes,
            Set excludes) {
        Validate.notBlank(principal, "Principal is blank");
        Validate.notNull(principalObject, "Principal Object is null");

        this.principal = principal;
        this.principalObject = principalObject;
        this.resourceTypes = resourceTypes;
        this.excludes = excludes;
    }

    public String getPrincipal() {
        return principal;
    }
    @JsonIgnore
    public KafkaPrincipal getPrincipalObject() {
        return principalObject;
    }

    public Set getResourceTypes() {
        return resourceTypes;
    }

    public Set getExcludes() {
        return excludes;
    }

    public boolean containsInExcludes(String resourceName, ResourceType resourceType) {
        if(isNull(this.excludes)) {
            return false;
        }
        return this.excludes.contains(new ResourceExcludes(resourceName,resourceType));
    }

    public boolean contains(ResourceType type) {
        if(isNull(this.resourceTypes)) {
            return true;
        }
        return this.resourceTypes.contains(type);

    }

    @Override
    public boolean equals(Object obj) {
        if (this == obj) {
            return true;
        }
        if (obj == null || getClass() != obj.getClass()) {
            return false;
        }

        PrincipalPermissionsDeleteParams that = (PrincipalPermissionsDeleteParams) obj;
        return
                Objects.equals(principal, that.principal);
    }

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

    @Override
    public String toString() {
        return
                "{principal: " + principal + "}";
    }

    public record ResourceExcludes(String resourceName, ResourceType resourceType) {}

    public Builder toBuilder() {
        return builder(this);
    }

    public static Builder builder() {
        return builder(null);
    }

    public static Builder builder(PrincipalPermissionsDeleteParams origin) {
        return new Builder(origin);
    }

    public static PrincipalPermissionsDeleteParams fromJson(Map map) {
        Validate.notNull(map, "JSON map is null");

        return MapperUtils.convert(map, PrincipalPermissionsDeleteParams.class);
    }

    public static PrincipalPermissionsDeleteParams fromJson(String json) {
        Validate.notNull(json, "JSON is null");

        return MapperUtils.jsonToBean(json, PrincipalPermissionsDeleteParams.class);
    }

    public static class Builder {

        private String principal;
        private KafkaPrincipal principalObject;
        private Set resourceTypes;
        private Set excludes;

        private Builder(PrincipalPermissionsDeleteParams origin) {
            if (origin == null) {
                return;
            }
            this.principal = origin.principal;
        }

        public Builder principal(String principal) {
            this.principal = principal;
            this.principalObject = principal != null ? SecurityUtils.parseKafkaPrincipal(principal) : null;
            return this;
        }

        public Builder principal(KafkaPrincipal principal) {
            this.principalObject = principal;
            this.principal = principal != null ? principal.toString() : null;
            return this;
        }

        public Builder excludes(Set excludes) {
            this.excludes = excludes;
            return this;
        }

        public Builder resourceTypes(Set resourceTypes) {
            this.resourceTypes = resourceTypes;
            return this;
        }

        public PrincipalPermissionsDeleteParams build() {
            return new PrincipalPermissionsDeleteParams(
                    principal,
                    principalObject,
                    resourceTypes,
                    excludes);
        }
    }

}