uk.ac.ebi.fg.annotare2.db.model.EffectiveAcl.java Source code

Introduction

Here is the source code for uk.ac.ebi.fg.annotare2.db.model.EffectiveAcl.java

Source

/*
 * Copyright 2009-2016 European Molecular Biology Laboratory
 *
 * 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 uk.ac.ebi.fg.annotare2.db.model;

import com.google.common.base.Function;
import com.google.common.base.Optional;
import com.google.common.collect.ImmutableList;
import uk.ac.ebi.fg.annotare2.db.model.enums.Permission;
import uk.ac.ebi.fg.annotare2.db.model.enums.Role;

import javax.annotation.Nullable;
import java.util.Collection;

import static com.google.common.collect.ImmutableList.builder;
import static com.google.common.collect.Lists.transform;

/**
 * @author Olga Melnichuk
 */
public class EffectiveAcl {

    private Acl acl;

    private Optional<User> creator;
    private Optional<User> owner;

    public EffectiveAcl(Acl acl, Optional<User> creator, Optional<User> owner) {
        this.acl = acl;
        this.creator = creator;
        this.owner = owner;
    }

    public boolean hasPermission(User user, Permission permission) {
        return acl != null && acl.hasPermission(getEffectiveRoles(user), permission);
    }

    private Collection<? extends Role> getEffectiveRoles(User user) {
        ImmutableList.Builder<Role> roles = builder();
        roles.addAll(transform(user.getRoles(), new Function<UserRole, Role>() {
            @Nullable
            @Override
            public Role apply(@Nullable UserRole input) {
                return input.getRole();
            }
        }));
        if (creator.isPresent() && user.equals(creator.get())) {
            roles.add(Role.CREATOR);
        }
        if (owner.isPresent() && user.equals(owner.get())) {
            roles.add(Role.OWNER);
        }
        return roles.build();
    }
}