io.gravitee.management.rest.enhancer.ApplicationEnhancer Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of gravitee-management-api-rest Show documentation
There is a newer version: 1.30.31
Show newest version
/**
 * Copyright (C) 2015 The Gravitee team (http://gravitee.io)
 *
 * 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 io.gravitee.management.rest.enhancer;

import io.gravitee.management.model.*;
import io.gravitee.management.service.ApiService;
import io.gravitee.management.service.MembershipService;
import io.gravitee.management.service.UserService;
import io.gravitee.repository.management.model.MembershipReferenceType;
import org.springframework.stereotype.Component;

import javax.inject.Inject;
import javax.ws.rs.core.SecurityContext;
import java.util.Collection;
import java.util.Optional;
import java.util.function.Function;

/**
 * @author David BRASSELY (david.brassely at graviteesource.com
 * @author Nicolas GERAUD (nicolas.geraud at graviteesource.com)
 * @author GraviteeSource Team
 */
@Component
public class ApplicationEnhancer {

    @Inject
    private MembershipService membershipService;
    
    @Inject
    private ApiService apiService;

    @Inject
    private UserService userService;

    public Function enhance(SecurityContext securityContext) {
        return application -> {
            // Add primary owner
            Collection members = membershipService.getMembers(MembershipReferenceType.APPLICATION, application.getId());
//            Optional primaryOwnerOpt = members.stream().filter(m -> MembershipType.PRIMARY_OWNER.equals(m.getType())).findFirst();
//            if (application..isPresent()) {
//                MemberEntity primaryOwner = primaryOwnerOpt.get();
//                UserEntity user = userService.findByName(primaryOwner.getUsername());
//
//                PrimaryOwnerEntity owner = new PrimaryOwnerEntity();
//                owner.setUsername(user.getUsername());
//                owner.setEmail(user.getEmail());
//                owner.setFirstname(user.getFirstname());
//                owner.setLastname(user.getLastname());
//                application.setPrimaryOwner(owner);
//            }
            
            // Add Members size
            application.setMembersSize(members.size());

            // Add permission for current user (if authenticated)
//            if (securityContext.isUserInRole(DefaultRole.ADMIN.name())) {
//                application.setPermission(MembershipType.PRIMARY_OWNER);
//            } else if(securityContext.getUserPrincipal() != null) {
//                MemberEntity member = membershipService.getMember(
//                        MembershipReferenceType.APPLICATION,
//                        application.getId(),
//                        securityContext.getUserPrincipal().getName());
//                if (member != null) {
//                    application.setPermission(member.getType());
//                } else if (application.getGroup() != null && application.getGroup().getId() != null) {
//                    member = membershipService.getMember(
//                            MembershipReferenceType.APPLICATION_GROUP,
//                            application.getGroup().getId(),
//                            securityContext.getUserPrincipal().getName());
//                    if (member != null) {
//                        application.setPermission(member.getType());
//                    }
//                }
//            }

            return application;
        };
    }
}