org.graylog2.database.entities.EntityScopeService Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of graylog2-server Show documentation
Graylog
There is a newer version: 6.1.4
/*
 * Copyright (C) 2020 Graylog, Inc.
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the Server Side Public License, version 1,
 * as published by MongoDB, Inc.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * Server Side Public License for more details.
 *
 * You should have received a copy of the Server Side Public License
 * along with this program. If not, see
 * .
 */
package org.graylog2.database.entities;


import javax.inject.Inject;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.Locale;
import java.util.Map;
import java.util.Objects;
import java.util.Set;
import java.util.stream.Collectors;

public final class EntityScopeService {

    private final Map entityScopes;

    @Inject
    public EntityScopeService(Set entityScopes) {
        this.entityScopes = Objects.requireNonNull(entityScopes)
                .stream()
                .collect(Collectors.toMap(e -> e.getName().toUpperCase(Locale.ROOT), e -> e));
    }

    public List getEntityScopes() {
        return Collections.unmodifiableList(new ArrayList<>(entityScopes.values()));
    }

    public boolean isMutable(ScopedEntity scopedEntity) {
        Objects.requireNonNull(scopedEntity, "Entity must not be null");
        String scope = scopedEntity.scope();
        if (scope == null || scope.isEmpty()) {
            return true;
        }

        EntityScope entityScope = entityScopes.get(scope.toUpperCase(Locale.ROOT));
        if (entityScope == null) {
            throw new IllegalArgumentException("Entity Scope does not exist: " + scope);
        }

        return entityScope.isMutable();
    }


    public boolean isDeletable(ScopedEntity scopedEntity) {
        Objects.requireNonNull(scopedEntity, "Entity must not be null");
        String scope = scopedEntity.scope();
        if (scope == null || scope.isEmpty()) {
            return true;
        }

        EntityScope entityScope = entityScopes.get(scope.toUpperCase(Locale.ROOT));
        if (entityScope == null) {
            throw new IllegalArgumentException("Entity Scope does not exist: " + scope);
        }

        return entityScope.isDeletable();
    }

    public boolean hasValidScope(ScopedEntity scopedEntity) {
        Objects.requireNonNull(scopedEntity, "Entity must not be null");
        String scope = scopedEntity.scope();

        return scope != null && entityScopes.containsKey(scope);
    }
}