org.fao.geonet.services.ownership.OwnershipUtils.java Source code

Introduction

Here is the source code for org.fao.geonet.services.ownership.OwnershipUtils.java

Source

//=============================================================================
//===   Copyright (C) 2001-2007 Food and Agriculture Organization of the
//===   United Nations (FAO-UN), United Nations World Food Programme (WFP)
//===   and United Nations Environment Programme (UNEP)
//===
//===   This program is free software; you can redistribute it and/or modify
//===   it under the terms of the GNU General Public License as published by
//===   the Free Software Foundation; either version 2 of the License, or (at
//===   your option) any later version.
//===
//===   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 GNU
//===   General Public License for more details.
//===
//===   You should have received a copy of the GNU General Public License
//===   along with this program; if not, write to the Free Software
//===   Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
//===
//===   Contact: Jeroen Ticheler - FAO - Viale delle Terme di Caracalla 2,
//===   Rome - Italy. email: geonetwork@osgeo.org
//==============================================================================

package org.fao.geonet.services.ownership;

import com.google.common.base.Function;
import com.google.common.collect.Lists;
import jeeves.server.UserSession;
import jeeves.server.context.ServiceContext;
import org.fao.geonet.domain.Profile;
import org.fao.geonet.domain.User;
import org.fao.geonet.domain.UserGroup;
import org.fao.geonet.repository.UserGroupRepository;
import org.fao.geonet.repository.UserRepository;
import org.fao.geonet.repository.specification.UserGroupSpecs;
import org.fao.geonet.repository.specification.UserSpecs;
import org.jdom.Element;
import org.springframework.data.jpa.domain.Specifications;

import javax.annotation.Nonnull;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.HashSet;
import java.util.List;
import java.util.Set;

//=============================================================================

public class OwnershipUtils {

    public static List<Element> getOwnerUsers(ServiceContext context, UserSession us) throws SQLException {
        if (!us.isAuthenticated())
            return new ArrayList<Element>();

        final List<User> allUsersThatOwnMetadata = context.getBean(UserRepository.class)
                .findAllUsersThatOwnMetadata();

        return getUsers(context, us, allUsersThatOwnMetadata);
    }

    public static List<Element> getEditorUsers(ServiceContext context, UserSession us) throws SQLException {
        if (!us.isAuthenticated())
            return new ArrayList<Element>();

        List<User> users = context.getBean(UserRepository.class)
                .findAll(Specifications.not(UserSpecs.hasProfile(Profile.RegisteredUser)));
        return getUsers(context, us, users);
    }

    public static List<Element> getUsers(ServiceContext context, UserSession us, List<User> users)
            throws SQLException {

        int id = us.getUserIdAsInt();

        if (us.getProfile() == Profile.Administrator) {
            final List<Element> userXml = Lists.transform(users, new Function<User, Element>() {
                @Override
                @Nonnull
                public Element apply(@Nonnull User input) {
                    return input.asXml();

                }
            });
            return userXml;
        }

        //--- we have a user admin

        Set<String> hsMyGroups = getUserGroups(context, id);

        Set<Profile> profileSet = us.getProfile().getAll();

        //--- now filter them

        List<Element> newList = new ArrayList<Element>();

        for (User elRec : users) {
            int userId = elRec.getId();
            Profile profile = elRec.getProfile();

            if (profileSet.contains(profile)) {
                if (hsMyGroups.containsAll(getUserGroups(context, userId))) {
                    newList.add(elRec.asXml());
                }
            }
        }

        //--- return result

        return newList;
    }

    //--------------------------------------------------------------------------
    //---
    //--- Private methods
    //---
    //--------------------------------------------------------------------------

    private static Set<String> getUserGroups(ServiceContext context, int id) throws SQLException {
        HashSet<String> groupIds = new HashSet<String>();

        final List<UserGroup> users = context.getBean(UserGroupRepository.class)
                .findAll(UserGroupSpecs.hasUserId(id));
        for (UserGroup el : users) {
            groupIds.add("" + el.getId().getGroupId());
        }

        return groupIds;
    }
}

//=============================================================================