org.craftercms.profile.social.ProfileUsersConnectionRepository Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of crafter-profile-social-medial-integration Show documentation
The newest version!
package org.craftercms.profile.social;

import java.util.ArrayList;
import java.util.Collections;
import java.util.LinkedHashSet;
import java.util.List;
import java.util.Set;

import org.apache.commons.collections4.CollectionUtils;
import org.apache.commons.lang3.StringUtils;
import org.craftercms.commons.crypto.TextEncryptor;
import org.craftercms.profile.api.Profile;
import org.craftercms.profile.api.ProfileConstants;
import org.craftercms.profile.api.exceptions.ProfileException;
import org.craftercms.profile.api.services.ProfileService;
import org.craftercms.profile.social.exceptions.ProfileConnectionRepositoryException;
import org.craftercms.profile.social.utils.ConnectionUtils;
import org.craftercms.profile.social.utils.TenantResolver;
import org.springframework.beans.factory.annotation.Required;
import org.springframework.social.connect.Connection;
import org.springframework.social.connect.ConnectionFactoryLocator;
import org.springframework.social.connect.ConnectionRepository;

/**
 * Implementation of {@link org.craftercms.profile.social.AnonymousAwareUsersConnectionRepository} that uses
 * Crafter Profile for connection persistence.
 *
 * @author avasquez
 */
public class ProfileUsersConnectionRepository implements AnonymousAwareUsersConnectionRepository {

    public static final String FIND_PROFILES_BY_PROVIDER_ID_AND_USER_ID_QUERY = "{attributes." + ConnectionUtils
        .CONNECTIONS_ATTRIBUTE_NAME + ".%s.providerUserId: {$in: [%s]}}";

    protected ProfileService profileService;
    protected TenantResolver tenantResolver;
    protected ConnectionFactoryLocator connectionFactoryLocator;
    protected TextEncryptor encryptor;

    @Required
    public void setProfileService(final ProfileService profileService) {
        this.profileService = profileService;
    }

    @Required
    public void setTenantResolver(final TenantResolver tenantResolver) {
        this.tenantResolver = tenantResolver;
    }

    @Required
    public void setConnectionFactoryLocator(final ConnectionFactoryLocator connectionFactoryLocator) {
        this.connectionFactoryLocator = connectionFactoryLocator;
    }

    @Required
    public void setEncryptor(final TextEncryptor encryptor) {
        this.encryptor = encryptor;
    }

    @Override
    public List findUserIdsWithConnection(Connection connection) {
        String providerId = connection.getKey().getProviderId();
        String providerUserId = connection.getKey().getProviderUserId();
        String query = String.format(FIND_PROFILES_BY_PROVIDER_ID_AND_USER_ID_QUERY, providerId, providerUserId);
        List profiles = findProfilesByQuery(query);

        if (CollectionUtils.isNotEmpty(profiles)) {
            List ids = new ArrayList<>(profiles.size());

            for (Profile profile : profiles) {
                ids.add(profile.getId().toString());
            }

            return ids;
        } else {
            return Collections.emptyList();
        }
    }

    @Override
    public Set findUserIdsConnectedTo(String providerId, Set providerUserIds) {
        String providerUserIdsStr = StringUtils.join(providerUserIds, ", ");
        String query = String.format(FIND_PROFILES_BY_PROVIDER_ID_AND_USER_ID_QUERY, providerId, providerUserIdsStr);
        List profiles = findProfilesByQuery(query);

        if (CollectionUtils.isNotEmpty(profiles)) {
            Set ids = new LinkedHashSet<>(profiles.size());

            for (Profile profile : profiles) {
                ids.add(profile.getId().toString());
            }

            return ids;
        } else {
            return Collections.emptySet();
        }
    }

    @Override
    public ConnectionRepository createConnectionRepository(String userId) {
        Profile profile = findProfile(userId);
        if (profile != null) {
            return new ProfileConnectionRepository(connectionFactoryLocator, profile, profileService, encryptor);
        } else {
            return null;
        }
    }

    @Override
    public ConnectionRepository createConnectionRepository() {
        return new ProfileConnectionRepository(connectionFactoryLocator, profileService, tenantResolver, encryptor);
    }

    protected List findProfilesByQuery(String query) {
        String tenant = tenantResolver.getCurrentTenant();
        try {
            return profileService.getProfilesByQuery(tenant, query, null, null, null, null, ProfileConstants
                .NO_ATTRIBUTE);
        } catch (ProfileException e) {
            throw new ProfileConnectionRepositoryException("Unable to find profiles of tenant '" + tenant +
                "' by query " + query, e);
        }
    }

    protected Profile findProfile(String profileId) {
        try {
            return profileService.getProfile(profileId);
        } catch (ProfileException e) {
            throw new ProfileConnectionRepositoryException("Unable to find profile '" + profileId + "'", e);
        }
    }

}