com.thinkbiganalytics.security.service.user.UserModelTransform.java Source code

Introduction

Here is the source code for com.thinkbiganalytics.security.service.user.UserModelTransform.java

Source

package com.thinkbiganalytics.security.service.user;

/*-
 * #%L
 * thinkbig-security-controller
 * %%
 * Copyright (C) 2017 ThinkBig Analytics
 * %%
 * 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.
 * #L%
 */

import com.google.common.collect.Iterables;
import com.thinkbiganalytics.security.rest.model.UserGroup;
import com.thinkbiganalytics.security.rest.model.User;

import java.util.function.Function;
import java.util.stream.Collectors;

import javax.annotation.Nonnull;

/**
 * Transforms users and groups between Security and Metadata objects.
 */
public class UserModelTransform {

    /**
     * Instances of {@code UserModelTransform} should not be constructed.
     *
     * @throws UnsupportedOperationException always
     */
    private UserModelTransform() {
        throw new UnsupportedOperationException();
    }

    /**
     * Transforms Metadata groups to Security groups.
     *
     * @return the Security groups
     */
    @Nonnull
    public static Function<com.thinkbiganalytics.metadata.api.user.UserGroup, UserGroup> toGroupPrincipal() {
        return (com.thinkbiganalytics.metadata.api.user.UserGroup group) -> {
            final UserGroup principal = new UserGroup();
            principal.setDescription(group.getDescription());
            principal.setMemberCount(Iterables.size(group.getGroups()) + Iterables.size(group.getUsers()));
            principal.setTitle(group.getTitle());
            principal.setSystemName(group.getSystemName());
            return principal;
        };
    }

    /**
     * Transforms Metadata users to Security users.
     *
     * @return the Security users
     */
    @Nonnull
    public static Function<com.thinkbiganalytics.metadata.api.user.User, User> toUserPrincipal() {
        return (com.thinkbiganalytics.metadata.api.user.User domain) -> {
            final User principal = new User();
            principal.setDisplayName(domain.getDisplayName());
            principal.setEmail(domain.getEmail());
            principal.setEnabled(domain.isEnabled());
            principal.setGroups(domain.getGroups().stream()
                    .map(com.thinkbiganalytics.metadata.api.user.UserGroup::getSystemName)
                    .collect(Collectors.toSet()));
            principal.setSystemName(domain.getSystemName());
            return principal;
        };
    }
}