/*
 * Copyright (C) 2003-2011 eXo Platform SAS.
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as published by
 * the Free Software Foundation, either version 3 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 Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 */

package org.exoplatform.social.core.storage.impl;

import org.apache.commons.lang.StringEscapeUtils;
import org.chromattic.api.ChromatticSession;
import org.chromattic.api.query.Ordering;
import org.chromattic.api.query.Query;
import org.chromattic.api.query.QueryBuilder;
import org.chromattic.api.query.QueryResult;
import org.chromattic.core.api.ChromatticSessionImpl;

import org.exoplatform.container.PortalContainer;
import org.exoplatform.services.log.ExoLogger;
import org.exoplatform.services.log.Log;
import org.exoplatform.social.core.chromattic.entity.IdentityEntity;
import org.exoplatform.social.core.chromattic.entity.ProviderEntity;
import org.exoplatform.social.core.chromattic.entity.SpaceEntity;
import org.exoplatform.social.core.chromattic.entity.SpaceListEntity;
import org.exoplatform.social.core.chromattic.entity.SpaceRef;
import org.exoplatform.social.core.chromattic.entity.SpaceRootEntity;
import org.exoplatform.social.core.identity.model.Identity;
import org.exoplatform.social.core.identity.model.Profile;
import org.exoplatform.social.core.identity.provider.OrganizationIdentityProvider;
import org.exoplatform.social.core.identity.provider.SpaceIdentityProvider;
import org.exoplatform.social.core.search.Sorting;
import org.exoplatform.social.core.service.LinkProvider;
import org.exoplatform.social.core.space.SpaceFilter;
import org.exoplatform.social.core.space.SpaceUtils;
import org.exoplatform.social.core.space.model.Space;
import org.exoplatform.social.core.storage.SpaceStorageException;
import org.exoplatform.social.core.storage.api.SpaceStorage;
import org.exoplatform.social.core.storage.exception.NodeNotFoundException;
import org.exoplatform.social.core.storage.query.JCRProperties;
import org.exoplatform.social.core.storage.query.QueryFunction;
import org.exoplatform.social.core.storage.query.WhereExpression;
import org.exoplatform.social.core.storage.streams.StreamInvocationHelper;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collection;
import java.util.Collections;
import java.util.HashSet;
import java.util.Iterator;
import java.util.LinkedHashMap;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Set;

import javax.jcr.Node;
import javax.jcr.Property;
import javax.jcr.RepositoryException;

/**
 * Space storage layer.
 *
 * @author <a href="mailto:alain.defrance@exoplatform.com">Alain Defrance</a>
 * @version $Revision$
 */
public class SpaceStorageImpl extends AbstractStorage implements SpaceStorage {

  /** Logger */
  private static final Log LOG = ExoLogger.getLogger(SpaceStorageImpl.class);

  /**
   * The identity storage
   */
  private IdentityStorageImpl identityStorage;

  private static final int TWO_SECONDS = 2000;
  
  /**
   * Constructor.
   *
   * @param identityStorage the identity storage
   */
  public SpaceStorageImpl(IdentityStorageImpl identityStorage) {
   this.identityStorage = identityStorage;
 }

  public void setIdentityStorage(IdentityStorageImpl identityStorage) {
    this.identityStorage = identityStorage;
  }

  /**
   * Fills {@link Space}'s properties to {@link SpaceEntity}'s.
   *
   * @param entity the space entity from chromattic
   * @param space  the space pojo for services
   */
  private void fillSpaceFromEntity(SpaceEntity entity, Space space) {

    space.setApp(entity.getApp());
    space.setId(entity.getId());
    space.setDisplayName(entity.getDisplayName());
    space.setPrettyName(entity.getPrettyName());
    space.setRegistration(entity.getRegistration());
    space.setDescription(entity.getDescription());
    space.setType(entity.getType());
    space.setVisibility(entity.getVisibility());
    space.setPriority(entity.getPriority());
    space.setGroupId(entity.getGroupId());
    space.setUrl(entity.getURL());
    space.setPendingUsers(entity.getPendingMembersId());
    space.setInvitedUsers(entity.getInvitedMembersId());
    space.setCreatedTime(entity.getCreatedTime());

    //
    String[] members = entity.getMembersId();
    String[] managers = entity.getManagerMembersId();

    //
    Set<String> membersList = new HashSet<String>();
    if (members != null) membersList.addAll(Arrays.asList(members));
    if (managers != null) membersList.addAll(Arrays.asList(managers));

    //
    space.setMembers(membersList.toArray(new String[]{}));
    space.setManagers(entity.getManagerMembersId());


    if (entity.getAvatarLastUpdated() != null) {
      try {
        PortalContainer container = PortalContainer.getInstance();
        ChromatticSession chromatticSession = getSession();
        String url = String.format("/%s/jcr/%s/%s/production/soc:providers/soc:space/soc:%s/soc:profile/soc:avatar/?upd=%d",
            container.getRestContextName(),
            lifeCycle.getRepositoryName(),
            chromatticSession.getJCRSession().getWorkspace().getName(),
            entity.getPrettyName(),
            entity.getAvatarLastUpdated());
        space.setAvatarUrl(LinkProvider.escapeJCRSpecialCharacters(url));
      } catch (Exception e) {
        LOG.warn("Failed to build avatar url: " + e.getMessage());
      }
    }
    space.setAvatarLastUpdated(entity.getAvatarLastUpdated());
  }
  
  /**
   * Fills {@link Space}'s properties to {@link SpaceEntity}'s.
   *
   * @param entity the space entity from chromattic
   * @param space  the space pojo for services
   */
  private void fillSpaceSimpleFromEntity(SpaceEntity entity, Space space) {

    space.setId(entity.getId());
    space.setDisplayName(entity.getDisplayName());
    space.setPrettyName(entity.getPrettyName());
    space.setDescription(entity.getDescription());
    space.setGroupId(entity.getGroupId());
    space.setUrl(entity.getURL());
    space.setCreatedTime(entity.getCreatedTime());

    if (entity.getAvatarLastUpdated() != null) {
      try {
        PortalContainer container = PortalContainer.getInstance();
        ChromatticSession chromatticSession = getSession();
        String url = String.format("/%s/jcr/%s/%s/production/soc:providers/soc:space/soc:%s/soc:profile/soc:avatar/?upd=%d",
            container.getRestContextName(),
            lifeCycle.getRepositoryName(),
            chromatticSession.getJCRSession().getWorkspace().getName(),
            entity.getPrettyName(),
            entity.getAvatarLastUpdated());
        space.setAvatarUrl(LinkProvider.escapeJCRSpecialCharacters(url));
      } catch (Exception e) {
        LOG.warn("Failed to build avatar url: " + e.getMessage());
      }
    }
    space.setAvatarLastUpdated(entity.getAvatarLastUpdated());
  }

  /**
   * Fills {@link SpaceEntity}'s properties from {@link Space}'s.
   *
   * @param space the space pojo for services
   * @param entity the space entity from chromattic
   */
  private void fillEntityFromSpace(Space space, SpaceEntity entity) {
    
    entity.setName(space.getPrettyName());
    
    entity.setApp(space.getApp());
    entity.setPrettyName(space.getPrettyName());
    entity.setDisplayName(space.getDisplayName());
    entity.setRegistration(space.getRegistration());
    entity.setDescription(space.getDescription());
    entity.setType(space.getType());
    entity.setVisibility(space.getVisibility());
    entity.setPriority(space.getPriority());
    entity.setGroupId(space.getGroupId());
    entity.setURL(space.getUrl());
    entity.setMembersId(space.getMembers());
    entity.setManagerMembersId(space.getManagers());
    entity.setPendingMembersId(space.getPendingUsers());
    entity.setInvitedMembersId(space.getInvitedUsers());
    entity.setAvatarLastUpdated(space.getAvatarLastUpdated());
    entity.setCreatedTime(space.getCreatedTime() != 0 ? space.getCreatedTime() : System.currentTimeMillis());
  }

  private void applyOrder(QueryBuilder builder, SpaceFilter spaceFilter) {

    //
    Sorting sorting;
    if (spaceFilter == null) {
      sorting = new Sorting(Sorting.SortBy.TITLE, Sorting.OrderBy.ASC);
    } else {
      sorting = spaceFilter.getSorting();
    }

    //
    Ordering ordering = Ordering.valueOf(sorting.orderBy.toString());
    switch (sorting.sortBy) {
      case DATE:
        builder.orderBy(SpaceEntity.createdTime.getName(), ordering);
        break;
      case RELEVANCY:
        builder.orderBy(JCRProperties.JCR_RELEVANCY.getName(), ordering);
      case TITLE:
        builder.orderBy(SpaceEntity.name.getName(), ordering);
        break;
    }
  }

  /**
   * The reference types.
   */
  public enum RefType {
    MEMBER() {
      @Override
      public SpaceListEntity refsOf(IdentityEntity identityEntity) {
        return identityEntity.getSpaces();
      }
      @Override
      public String[] idsOf(Space space) {
        return space.getMembers();
      }
      @Override
      public void setIds(Space space, String[] ids) {
        space.setMembers(ids);
      }
    },
    MANAGER() {
      @Override
      public SpaceListEntity refsOf(IdentityEntity identityEntity) {
        return identityEntity.getManagerSpaces();
      }
      @Override
      public String[] idsOf(Space space) {
        return space.getManagers();
      }
      @Override
      public void setIds(Space space, String[] ids) {
        space.setManagers(ids);
      }
    },
    PENDING() {
      @Override
      public SpaceListEntity refsOf(IdentityEntity identityEntity) {
        return identityEntity.getPendingSpaces();
      }
      @Override
      public String[] idsOf(Space space) {
        return space.getPendingUsers();
      }
      @Override
      public void setIds(Space space, String[] ids) {
        space.setPendingUsers(ids);
      }
    },
    INVITED() {
      @Override
      public SpaceListEntity refsOf(IdentityEntity identityEntity) {
        return identityEntity.getInvitedSpaces();
      }
      @Override
      public String[] idsOf(Space space) {
        return space.getInvitedUsers();
      }
      @Override
      public void setIds(Space space, String[] ids) {
        space.setInvitedUsers(ids);
      }
    };

    public abstract SpaceListEntity refsOf(IdentityEntity identityEntity);
    public abstract String[] idsOf(Space space);
    public abstract void setIds(Space space, String[] ids);
  }

  private class UpdateContext {
    private String[] added;
    private String[] removed;

    private UpdateContext(String[] added, String[] removed) {
      this.added = added;
      this.removed = removed;
    }

    public String[] getAdded() {
      return added;
    }

    public String[] getRemoved() {
      return removed;
    }
  }

  private String[] sub(String[] l1, String[] l2) {

    if (l1 == null) {
      return new String[]{};
    }

    if (l2 == null) {
      return l1;
    }

    List<String> l = new ArrayList(Arrays.asList(l1));
    l.removeAll(Arrays.asList(l2));
    return l.toArray(new String[]{});
  }

  private void createRefs(SpaceEntity spaceEntity, Space space) throws NodeNotFoundException {

    String[] removedMembers = sub(spaceEntity.getMembersId(), space.getMembers());
    String[] removedManagers = sub(spaceEntity.getManagerMembersId(), space.getManagers());
    String[] removedInvited = sub(spaceEntity.getInvitedMembersId(), space.getInvitedUsers());
    String[] removedPending = sub(spaceEntity.getPendingMembersId(), space.getPendingUsers());

    String[] addedMembers = sub(space.getMembers(), spaceEntity.getMembersId());
    String[] addedManagers = sub(space.getManagers(), spaceEntity.getManagerMembersId());
    String[] addedInvited = sub(space.getInvitedUsers(), spaceEntity.getInvitedMembersId());
    String[] addedPending = sub(space.getPendingUsers(), spaceEntity.getPendingMembersId());

    manageRefList(new UpdateContext(addedMembers, removedMembers), spaceEntity, RefType.MEMBER);
    manageActivityRefList(new UpdateContext(addedMembers, removedMembers), spaceEntity, RefType.MEMBER);
    
    manageRefList(new UpdateContext(addedManagers, removedManagers), spaceEntity, RefType.MANAGER);
    manageRefList(new UpdateContext(addedInvited, removedInvited), spaceEntity, RefType.INVITED);
    manageRefList(new UpdateContext(addedPending, removedPending), spaceEntity, RefType.PENDING);

  }

  private void changeSpaceRef(SpaceEntity spaceEntity, Space space, RefType type) {
    String []listUserNames = null;
    
    if (RefType.MEMBER.equals(type)) {
      listUserNames = spaceEntity.getMembersId();
    } else if (RefType.MANAGER.equals(type)) {
      listUserNames = spaceEntity.getManagerMembersId();
    } else if (RefType.INVITED.equals(type)) {
      listUserNames = spaceEntity.getInvitedMembersId();
    } else {
      listUserNames = spaceEntity.getPendingMembersId();
    }
    
    if (listUserNames != null) {
      for (String userName : listUserNames) {
        try {
          IdentityEntity identityEntity = identityStorage._findIdentityEntity(OrganizationIdentityProvider.NAME, userName);
          SpaceListEntity listRef = type.refsOf(identityEntity);
          SpaceRef ref = listRef.getRef(spaceEntity.getName());
          ref.setName(space.getPrettyName());
        } catch (NodeNotFoundException e) {
          LOG.warn(e.getMessage(), e);
        }
      }
    }
  }
  
  private void changeSpaceRef(String remoteId, SpaceEntity spaceEntity, Space space, RefType type) {
    try {
      IdentityEntity identityEntity = identityStorage._findIdentityEntity(OrganizationIdentityProvider.NAME, remoteId);
      SpaceListEntity listRef = type.refsOf(identityEntity);
      SpaceRef ref = listRef.getRef(spaceEntity.getName());
      ref.setName(space.getPrettyName());
    } catch (NodeNotFoundException e) {
      LOG.warn(e.getMessage(), e);
    }
  }
  
  private void manageRefList(UpdateContext context, SpaceEntity spaceEntity, RefType type) {

    if (context.getAdded() != null) {
      for (String userName : context.getAdded()) {
        try {
          IdentityEntity identityEntity = identityStorage._findIdentityEntity(OrganizationIdentityProvider.NAME, userName);
          SpaceListEntity listRef = type.refsOf(identityEntity);
          SpaceRef ref = listRef.getRef(spaceEntity.getName());
          if (!ref.getName().equals(spaceEntity.getName())) {
            ref.setName(spaceEntity.getName());
          }
          ref.setSpaceRef(spaceEntity);
        }
        catch (NodeNotFoundException e) {
          LOG.warn(e.getMessage(), e);
        }
      }

      for (String userName : context.getRemoved()) {
        try {
          IdentityEntity identityEntity = identityStorage._findIdentityEntity(OrganizationIdentityProvider.NAME, userName);
          SpaceListEntity listRef = type.refsOf(identityEntity);
          SpaceRef ref = listRef.getRef(spaceEntity.getName());
          getSession().remove(ref);
        }
        catch (NodeNotFoundException e) {
          LOG.warn(e.getMessage(), e);
        }
      }
    }
  }
  
  private void manageActivityRefList(UpdateContext context, SpaceEntity spaceEntity, RefType type) {

    Identity spaceIdentity = identityStorage.findIdentity(SpaceIdentityProvider.NAME,
                                                          spaceEntity.getPrettyName());
    if (context.getAdded() != null) {
      for (String userName : context.getAdded()) {
        Identity identity = identityStorage.findIdentity(OrganizationIdentityProvider.NAME, userName);
        //streamStorage.addSpaceMember(identity, spaceIdentity);
        StreamInvocationHelper.addSpaceMember(identity, spaceIdentity);
      }
    }

    if (context.getRemoved() != null) {
      for (String userName : context.getRemoved()) {
        Identity identity = identityStorage.findIdentity(OrganizationIdentityProvider.NAME, userName);
        //streamStorage.removeSpaceMember(identity, spaceIdentity);
        StreamInvocationHelper.removeSpaceMember(identity, spaceIdentity);
      }
    }
  }

  private boolean validateFilter(SpaceFilter filter) {
    if (filter == null) {
      return false;
    }

    if (filter.getSpaceNameSearchCondition() != null &&
        filter.getSpaceNameSearchCondition().length() != 0) {
      return isValidInput(filter.getSpaceNameSearchCondition());
    }
    else if (!Character.isDigit(filter.getFirstCharacterOfSpaceName())) {
      return true;
    }
    return false;

  }

  private SpaceEntity createSpace(Space space) throws SpaceStorageException {

    SpaceRootEntity spaceRootEntity = getSpaceRoot();
    SpaceEntity spaceEntity = spaceRootEntity.getSpace(space.getPrettyName());
    space.setId(spaceEntity.getId());

    return spaceEntity;

  }

  private SpaceEntity _saveSpace(Space space) throws NodeNotFoundException {

    return _findById(SpaceEntity.class, space.getId());

  }

  private void _applyFilter(WhereExpression whereExpression, SpaceFilter spaceFilter) {

    String spaceNameSearchCondition = spaceFilter.getSpaceNameSearchCondition();
    char firstCharacterOfName = spaceFilter.getFirstCharacterOfSpaceName();

    if (spaceNameSearchCondition != null && spaceNameSearchCondition.length() != 0) {
      if (this.isValidInput(spaceNameSearchCondition)) {

        spaceNameSearchCondition = new StringBuilder().append(StorageUtils.PERCENT_STR)
            .append(spaceNameSearchCondition.replace(StorageUtils.ASTERISK_STR, StorageUtils.PERCENT_STR).toLowerCase()).append(StorageUtils.PERCENT_STR).toString();

        whereExpression.startGroup();
        whereExpression
            .like(whereExpression.callFunction(QueryFunction.LOWER, SpaceEntity.name), spaceNameSearchCondition)
            .or()
            .like(whereExpression.callFunction(QueryFunction.LOWER, SpaceEntity.displayName), spaceNameSearchCondition)
            .or()
            .like(whereExpression.callFunction(QueryFunction.LOWER, SpaceEntity.description), StringEscapeUtils.escapeHtml(spaceNameSearchCondition));
        whereExpression.endGroup();

        List<Space> exclusions = spaceFilter.getExclusions();
        if (exclusions != null) {
          for (Space space : exclusions) {
            whereExpression.and().not().equals(SpaceEntity.name, space.getPrettyName());
          }
        }
      }
    }
    else if (!Character.isDigit(firstCharacterOfName)) {
      String firstCharacterOfNameString = Character.toString(firstCharacterOfName);
      String firstCharacterOfNameLowerCase = firstCharacterOfNameString.toLowerCase() + StorageUtils.PERCENT_STR;
      whereExpression
          .like(whereExpression.callFunction(QueryFunction.LOWER, SpaceEntity.name), firstCharacterOfNameLowerCase);
    }
  }
  
  private boolean isValidInput(String input) {
    if (input == null || input.length() == 0) {
      return false;
    }
    String cleanString = input.replaceAll("\\*", "");
    cleanString = cleanString.replaceAll("\\%", "");
    if (cleanString.length() == 0) {
       return false;
    }
    return true;
  }

  private List<String> processUnifiedSearchCondition(String searchCondition) {
    String[] spaceConditions = searchCondition.split(" ");
    List<String> result = new ArrayList<String>(spaceConditions.length);
    for (String conditionValue : spaceConditions) {
      result.add(conditionValue.toString());
    }
    return result;
  }

  /*
    Filter query
   */

  private Query<SpaceEntity> _getSpacesByFilterQuery(String userId, SpaceFilter spaceFilter) {

    QueryBuilder<SpaceEntity> builder = getSession().createQueryBuilder(SpaceEntity.class);
    WhereExpression whereExpression = new WhereExpression();

    if (validateFilter(spaceFilter)) {
      _applyFilter(whereExpression, spaceFilter);
    }

    if (userId != null && validateFilter(spaceFilter)) {
      whereExpression
          .and()
          .equals(SpaceEntity.membersId, userId);
    } else if (userId != null && !validateFilter(spaceFilter)) {
      whereExpression
          .equals(SpaceEntity.membersId, userId);
    }

    if (whereExpression.toString().length() > 0) {
      builder.where(whereExpression.toString());
    }

    applyOrder(builder, spaceFilter);

    return builder.get();

  }

  private Query<SpaceEntity> getAllSpacesQuery(SpaceFilter spaceFilter) {
    QueryBuilder<SpaceEntity> builder = getSession().createQueryBuilder(SpaceEntity.class);
    WhereExpression whereExpression = new WhereExpression();

    if (validateFilter(spaceFilter)) {
      _applyFilter(whereExpression, spaceFilter);
      whereExpression.and();
      whereExpression.startGroup();
    }

    if (whereExpression.toString().length() > 0) {
      builder.where(whereExpression.toString());
    }

    applyOrder(builder, spaceFilter);

    return builder.get();
    
  }
  
  private Query<SpaceEntity> getAccessibleSpacesByFilterQuery(String userId, SpaceFilter spaceFilter) {

    QueryBuilder<SpaceEntity> builder = getSession().createQueryBuilder(SpaceEntity.class);
    WhereExpression whereExpression = new WhereExpression();

    if (validateFilter(spaceFilter)) {
      _applyFilter(whereExpression, spaceFilter);
      whereExpression.and();
      
      //
      if (spaceFilter.getAppId() != null) {
        whereExpression.contains(SpaceEntity.app, spaceFilter.getAppId());
        whereExpression.and();
      }
      
      //
      whereExpression.startGroup();
    } else if (spaceFilter != null && spaceFilter.getAppId() != null) {
      //
      whereExpression.contains(SpaceEntity.app, spaceFilter.getAppId());
      whereExpression.and();
      
      //
      whereExpression.startGroup();
    }

    whereExpression
        .equals(SpaceEntity.membersId, userId)
        .or()
        .equals(SpaceEntity.managerMembersId, userId);

    whereExpression.endAllGroup();

    if (whereExpression.toString().length() > 0) {
      builder.where(whereExpression.toString());
    }

    applyOrder(builder, spaceFilter);

    return builder.get();

  }

  private Query<SpaceEntity> getPublicSpacesQuery(String userId) {
    return getPublicSpacesQuery(userId, null);
  }

  private Query<SpaceEntity> getPublicSpacesQuery(String userId, SpaceFilter spaceFilter) {

    QueryBuilder<SpaceEntity> builder = getSession().createQueryBuilder(SpaceEntity.class);
    WhereExpression whereExpression = new WhereExpression();

    if (validateFilter(spaceFilter)) {
      _applyFilter(whereExpression, spaceFilter);
      whereExpression.and();
    }

    builder.where(whereExpression
        .not().equals(SpaceEntity.membersId, userId)
        .and().not().equals(SpaceEntity.managerMembersId, userId)
        .and().not().equals(SpaceEntity.invitedMembersId, userId)
        .and().not().equals(SpaceEntity.pendingMembersId, userId)
        .toString()
    );

    if (whereExpression.toString().length() > 0) {
      builder.where(whereExpression.toString());
    }

    applyOrder(builder, spaceFilter);

    return builder.get();

  }
  
  private Query<SpaceEntity> getPublicSpacesOfMemberQuery(String userId) {

    QueryBuilder<SpaceEntity> builder = getSession().createQueryBuilder(SpaceEntity.class);
    WhereExpression whereExpression = new WhereExpression();
    
    builder.where(whereExpression
        .equals(SpaceEntity.membersId, userId)
        .and().equals(SpaceEntity.visibility, Space.PUBLIC)
        .and().not().equals(SpaceEntity.managerMembersId, userId)
        .and().not().equals(SpaceEntity.invitedMembersId, userId)
        .and().not().equals(SpaceEntity.pendingMembersId, userId)
        .toString()
    );

    if (whereExpression.toString().length() > 0) {
      builder.where(whereExpression.toString());
    }

    builder.orderBy(SpaceEntity.name.getName(), Ordering.ASC);

    return builder.get();

  }

  private Query<SpaceEntity> getSpacesOfMemberQuery(String userId) {

    QueryBuilder<SpaceEntity> builder = getSession().createQueryBuilder(SpaceEntity.class);
    WhereExpression whereExpression = new WhereExpression();
    
    builder.where(whereExpression
        .equals(SpaceEntity.membersId, userId)
        .and().not().equals(SpaceEntity.visibility, Space.HIDDEN)
        .and().not().equals(SpaceEntity.invitedMembersId, userId)
        .and().not().equals(SpaceEntity.pendingMembersId, userId)
        .toString()
    );

    if (whereExpression.toString().length() > 0) {
      builder.where(whereExpression.toString());
    }

    builder.orderBy(SpaceEntity.name.getName(), Ordering.ASC);

    return builder.get();

  }
  
  private Query<SpaceEntity> getPendingSpacesFilterQuery(String userId, SpaceFilter spaceFilter) {

    QueryBuilder<SpaceEntity> builder = getSession().createQueryBuilder(SpaceEntity.class);
    WhereExpression whereExpression = new WhereExpression();

    if (validateFilter(spaceFilter)) {
      _applyFilter(whereExpression, spaceFilter);
      whereExpression.and();
    }

    builder.where(whereExpression
        .equals(SpaceEntity.pendingMembersId, userId)
        .toString()
    );

    if (whereExpression.toString().length() > 0) {
      builder.where(whereExpression.toString());
    }

    applyOrder(builder, spaceFilter);

    return builder.get();

  }

  private Query<SpaceEntity> getInvitedSpacesFilterQuery(String userId, SpaceFilter spaceFilter) {

    QueryBuilder<SpaceEntity> builder = getSession().createQueryBuilder(SpaceEntity.class);
    WhereExpression whereExpression = new WhereExpression();

    if (validateFilter(spaceFilter)) {
      _applyFilter(whereExpression, spaceFilter);
      whereExpression.and();
    }

    builder.where(whereExpression
        .equals(SpaceEntity.invitedMembersId, userId)
        .toString()
    );

    if (whereExpression.toString().length() > 0) {
      builder.where(whereExpression.toString());
    }

    applyOrder(builder, spaceFilter);

    return builder.get();

  }

  private Query<SpaceEntity> getEditableSpacesFilterQuery(String userId, SpaceFilter spaceFilter) {

    QueryBuilder<SpaceEntity> builder = getSession().createQueryBuilder(SpaceEntity.class);
    WhereExpression whereExpression = new WhereExpression();

    if (validateFilter(spaceFilter)) {
      _applyFilter(whereExpression, spaceFilter);
      whereExpression.and();
    }

    builder.where(whereExpression
        .equals(SpaceEntity.managerMembersId, userId)
        .toString()
    );

    if (whereExpression.toString().length() > 0) {
      builder.where(whereExpression.toString());
    }
    
    applyOrder(builder, spaceFilter);

    return builder.get();

  }

  private Query<SpaceEntity> getSpacesByFilterQuery(SpaceFilter spaceFilter) {
    return _getSpacesByFilterQuery(null, spaceFilter);
  }

  /*
    Public
   */

  /**
   * {@inheritDoc}
   */
  public Space getSpaceByDisplayName(String spaceDisplayName) throws SpaceStorageException {
    Space space = null;

    QueryBuilder<SpaceEntity> builder = getSession().createQueryBuilder(SpaceEntity.class);
    WhereExpression whereExpression = new WhereExpression();

    whereExpression.equals(SpaceEntity.displayName, spaceDisplayName);

    QueryResult<SpaceEntity> result = builder.where(whereExpression.toString()).get().objects();
    
    if (result.hasNext()) {
      space = new Space();
      fillSpaceFromEntity(result.next(), space);
    }

    return space;
  }

  /**
   * {@inheritDoc}
   */
  public void saveSpace(Space space, boolean isNew) throws SpaceStorageException {

    SpaceEntity entity;

    try {
      
      if (isNew) {
        entity = createSpace(space);
      }
      else {
        entity = _saveSpace(space);
      }

      //
      createRefs(entity, space);
      
      fillEntityFromSpace(space, entity);
      
      if (!isNew) {
        getSession().save();
      }
      
      LOG.debug(String.format(
          "Space %s (%s) saved",
          space.getPrettyName(),
          space.getId()
      ));

    }
    catch (NodeNotFoundException e) {
      throw new SpaceStorageException(SpaceStorageException.Type.FAILED_TO_SAVE_SPACE, e.getMessage(), e);
    }

  }

  /**
   * {@inheritDoc}
   */
  public void renameSpace(Space space, String newDisplayName) {
    renameSpace(null, space, newDisplayName);
  }
  
  /**
   * {@inheritDoc}
   */
  public void renameSpace(String remoteId, Space space, String newDisplayName) {
    SpaceEntity entity;

    try {
      String oldPrettyName = space.getPrettyName();
      
      space.setDisplayName(newDisplayName);
      space.setPrettyName(space.getDisplayName());
      space.setUrl(SpaceUtils.cleanString(newDisplayName));
      
      entity = _saveSpace(space);
        
      //change space ref
      this.changeSpaceRef(entity, space, RefType.MEMBER);
      this.changeSpaceRef(entity, space, RefType.MANAGER);
      this.changeSpaceRef(entity, space, RefType.INVITED);
      this.changeSpaceRef(entity, space, RefType.PENDING);
      
      if (remoteId != null) {
        this.changeSpaceRef(remoteId, entity, space, RefType.MEMBER);
      }
      
      fillEntityFromSpace(space, entity);

      //
      getSession().save();

      //change profile of space
      Identity identitySpace = identityStorage.findIdentity(SpaceIdentityProvider.NAME, oldPrettyName);
      
      if (identitySpace != null) {
        Profile profileSpace = identitySpace.getProfile();
        profileSpace.setProperty(Profile.FIRST_NAME, space.getDisplayName());
        profileSpace.setProperty(Profile.USERNAME, space.getPrettyName());
        //profileSpace.setProperty(Profile.AVATAR_URL, space.getAvatarUrl());
        profileSpace.setProperty(Profile.URL, space.getUrl());
        
        identityStorage.saveProfile(profileSpace);
        
        identitySpace.setRemoteId(space.getPrettyName());
        renameIdentity(identitySpace);
      }
      
      //
      LOG.debug(String.format(
          "Space %s (%s) saved",
          space.getPrettyName(),
          space.getId()
      ));

    } catch (NodeNotFoundException e) {
      throw new SpaceStorageException(SpaceStorageException.Type.FAILED_TO_RENAME_SPACE, e.getMessage(), e);
    }
  }
  
  /**
   * Add this method to resolve SOC-3439
   * @param identity
   * @throws NodeNotFoundException
   */
  private void renameIdentity(Identity identity) throws NodeNotFoundException {
    ProviderEntity providerEntity = getProviderRoot().getProvider(identity.getProviderId());
    // Move identity
    IdentityEntity identityEntity = _findById(IdentityEntity.class, identity.getId());
    providerEntity.getIdentities().put(identity.getRemoteId(), identityEntity);
    
    identityEntity.setRemoteId(identity.getRemoteId());
  }
  
  /**
   * {@inheritDoc}
   */
  public void deleteSpace(String id) throws SpaceStorageException {

    String name;

    //
    try {
      name = _findById(SpaceEntity.class, id).getPrettyName();
    }
    catch (NodeNotFoundException e) {
      throw new SpaceStorageException(SpaceStorageException.Type.FAILED_TO_DELETE_SPACE, e.getMessage());
    }

    //
    _removeById(SpaceEntity.class, id);

    //
    getSession().save();

    //
    LOG.debug(String.format(
        "Space %s removed",
        name)
    );
  }

  /**
   * {@inheritDoc}
   */
  @Override
  public void ignoreSpace(String spaceId, String userId) {
    return;
  }

  /**
   * {@inheritDoc}
   */
  @Override
  public boolean isSpaceIgnored(String spaceId, String userId) {
    return false;
  }

  /*
    Member spaces
   */

  /**
   * {@inheritDoc}
   */
  public int getMemberSpacesCount(String userId) throws SpaceStorageException {
    try {
       return identityStorage._findIdentityEntity(OrganizationIdentityProvider.NAME, userId).getSpaces().getRefs().size();
    }
    catch (NodeNotFoundException e){
       return 0;
    }
  }

  /**
   * {@inheritDoc}
   */
  public int getMemberSpacesByFilterCount(String userId, SpaceFilter spaceFilter) {
    return _getSpacesByFilterQuery(userId, spaceFilter).objects().size();
  }

  /**
   * {@inheritDoc}
   */
  public List<Space> getMemberSpaces(String userId) throws SpaceStorageException {

    List<Space> spaces = new ArrayList<Space>();

    //
    QueryResult<SpaceEntity> results = _getSpacesByFilterQuery(userId, null).objects();

    while (results.hasNext()) {
      SpaceEntity currentSpace = results.next();
      Space space = new Space();
      fillSpaceFromEntity(currentSpace, space);
      spaces.add(space);
    }

    return spaces;
  }

  /**
   * {@inheritDoc}
   */
  public List<Space> getMemberSpaces(String userId, long offset, long limit) throws SpaceStorageException {

    List<Space> spaces = new ArrayList<Space>();

    //
    QueryResult<SpaceEntity> results = _getSpacesByFilterQuery(userId, null).objects(offset, limit);

    while (results.hasNext()) {
      SpaceEntity currentSpace = results.next();
      Space space = new Space();
      fillSpaceFromEntity(currentSpace, space);
      spaces.add(space);
    }

    return spaces;
  }

  /**
   * {@inheritDoc}
   */
  public List<Space> getMemberSpacesByFilter(String userId, SpaceFilter spaceFilter, long offset, long limit) {

    List<Space> spaces = new ArrayList<Space>();

    //
    QueryResult<SpaceEntity> results = _getSpacesByFilterQuery(userId, spaceFilter).objects(offset, limit);

    while (results.hasNext()) {
      SpaceEntity currentSpace = results.next();
      Space space = new Space();
      fillSpaceFromEntity(currentSpace, space);
      spaces.add(space);
    }

    return spaces;

  }

  /*
    Pending spaces
   */

  /**
   * {@inheritDoc}
   */
  public int getPendingSpacesCount(String userId) throws SpaceStorageException {
    try {
      IdentityEntity identityEntity = identityStorage._findIdentityEntity(OrganizationIdentityProvider.NAME, userId);
      Collection<SpaceRef> spaceEntities = identityEntity.getPendingSpaces().getRefs().values();
      return spaceEntities.size();
    }
    catch (NodeNotFoundException e) {
      return 0;
    }
  }

  /**
   * {@inheritDoc}
   */
  public int getPendingSpacesByFilterCount(String userId, SpaceFilter spaceFilter) {
    return getPendingSpacesFilterQuery(userId, spaceFilter).objects().size();
  }

  /**
   * {@inheritDoc}
   */
  public List<Space> getPendingSpaces(String userId) throws SpaceStorageException {

    List<Space> spaces = new ArrayList<Space>();

    //
    QueryResult<SpaceEntity> results = getPendingSpacesFilterQuery(userId, null).objects();

    while (results.hasNext()) {
      SpaceEntity currentSpace = results.next();
      Space space = new Space();
      fillSpaceFromEntity(currentSpace, space);
      spaces.add(space);
    }

    return spaces;
  }

  /**
   * {@inheritDoc}
   */
  public List<Space> getPendingSpaces(String userId, long offset, long limit) throws SpaceStorageException {

    List<Space> spaces = new ArrayList<Space>();

    //
    QueryResult<SpaceEntity> results = getPendingSpacesFilterQuery(userId, null).objects(offset, limit);

    while (results.hasNext()) {
      SpaceEntity currentSpace = results.next();
      Space space = new Space();
      fillSpaceFromEntity(currentSpace, space);
      spaces.add(space);
    }

    return spaces;
  }

  /**
   * {@inheritDoc}
   */
  public List<Space> getPendingSpacesByFilter(String userId, SpaceFilter spaceFilter, long offset, long limit) {

    List<Space> spaces = new ArrayList<Space>();

    //
    QueryResult<SpaceEntity> results = getPendingSpacesFilterQuery(userId, spaceFilter).objects(offset, limit);

    while (results.hasNext()) {
      SpaceEntity currentSpace = results.next();
      Space space = new Space();
      fillSpaceFromEntity(currentSpace, space);
      spaces.add(space);
    }

    return spaces;
  }

  /*
    Invited spaces
   */

  /**
   * {@inheritDoc}
   */
  public int getInvitedSpacesCount(String userId) throws SpaceStorageException {

    try {
      IdentityEntity identityEntity = identityStorage._findIdentityEntity(OrganizationIdentityProvider.NAME, userId);
      Collection<SpaceRef> spaceEntities = identityEntity.getInvitedSpaces().getRefs().values();
      return spaceEntities.size();
    }
    catch (NodeNotFoundException e) {
      return 0;
    }
  }

  /**
   * {@inheritDoc}
   */
  public int getInvitedSpacesByFilterCount(String userId, SpaceFilter spaceFilter) {

    if (validateFilter(spaceFilter)) {
      return getInvitedSpacesFilterQuery(userId, spaceFilter).objects().size();
    }
    else {
      return 0;
    }
  }

  /**
   * {@inheritDoc}
   */
  public List<Space> getInvitedSpaces(String userId) throws SpaceStorageException {

    List<Space> spaces = new ArrayList<Space>();

    //
    QueryResult<SpaceEntity> results = getInvitedSpacesFilterQuery(userId, null).objects();

    while (results.hasNext()) {
      SpaceEntity currentSpace = results.next();
      Space space = new Space();
      fillSpaceFromEntity(currentSpace, space);
      spaces.add(space);
    }

    return spaces;
  }

  /**
   * {@inheritDoc}
   */
  public List<Space> getInvitedSpaces(String userId, long offset, long limit) throws SpaceStorageException {
    
    List<Space> spaces = new ArrayList<Space>();

    //
    QueryResult<SpaceEntity> results = getInvitedSpacesFilterQuery(userId, null).objects(offset, limit);

    while (results.hasNext()) {
      SpaceEntity currentSpace = results.next();
      Space space = new Space();
      fillSpaceFromEntity(currentSpace, space);
      spaces.add(space);
    }

    return spaces;
  }

  /**
   * {@inheritDoc}
   */
  public List<Space> getInvitedSpacesByFilter(String userId, SpaceFilter spaceFilter, long offset, long limit) {

    List<Space> spaces = new ArrayList<Space>();

    //
    QueryResult<SpaceEntity> results = getInvitedSpacesFilterQuery(userId, spaceFilter).objects(offset, limit);

    while (results.hasNext()) {
      SpaceEntity currentSpace = results.next();
      Space space = new Space();
      fillSpaceFromEntity(currentSpace, space);
      spaces.add(space);
    }

    return spaces;
  }

  /*
    Public spaces
   */

  /**
   * {@inheritDoc}
   */
  public int getPublicSpacesCount(String userId) throws SpaceStorageException {
    return getPublicSpacesQuery(userId).objects().size();
  }

  /**
   * {@inheritDoc}
   */
  public int getPublicSpacesByFilterCount(String userId, SpaceFilter spaceFilter) {
    if (validateFilter(spaceFilter)) {
      return getPublicSpacesQuery(userId, spaceFilter).objects().size();
    }
    else {
      return 0;
    }
  }

  /**
   * {@inheritDoc}
   */
  public List<Space> getPublicSpacesByFilter(String userId, SpaceFilter spaceFilter, long offset, long limit) {

    try {
      identityStorage._findIdentityEntity(OrganizationIdentityProvider.NAME, userId);
    }
    catch (NodeNotFoundException e) {
      userId = null;
    }

    List<Space> spaces = new ArrayList<Space>();

    //
    QueryResult<SpaceEntity> results = getPublicSpacesQuery(userId, spaceFilter).objects(offset, limit);

    while (results.hasNext()) {
      SpaceEntity currentSpace = results.next();
      Space space = new Space();
      fillSpaceFromEntity(currentSpace, space);
      spaces.add(space);
    }

    return spaces;
  }

  /**
   * {@inheritDoc}
   */
  public List<Space> getPublicSpaces(String userId) throws SpaceStorageException {
    List<Space> spaces = new ArrayList<Space>();

    //
    QueryResult<SpaceEntity> results = getPublicSpacesQuery(userId).objects();

    while (results.hasNext()) {
      SpaceEntity currentSpace = results.next();
      Space space = new Space();
      fillSpaceFromEntity(currentSpace, space);
      spaces.add(space);
    }

    return spaces;
  }

  /**
   * {@inheritDoc}
   */
  public List<Space> getPublicSpaces(String userId, long offset, long limit) throws SpaceStorageException {

    List<Space> spaces = new ArrayList<Space>();

    //
    QueryResult<SpaceEntity> results = getPublicSpacesQuery(userId).objects(offset, limit);

    while (results.hasNext()) {
      SpaceEntity currentSpace = results.next();
      Space space = new Space();
      fillSpaceFromEntity(currentSpace, space);
      spaces.add(space);
    }

    return spaces;
  }
  
  
  /*
    Accessible spaces
   */
  /**
   * {@inheritDoc}
   */
  public int getAccessibleSpacesCount(String userId) throws SpaceStorageException {
    return getAccessibleSpacesByFilterQuery(userId, null).objects().size();
  }

  /**
   * {@inheritDoc}
   */
  public int getAccessibleSpacesByFilterCount(String userId, SpaceFilter spaceFilter) {
    return getAccessibleSpacesByFilterQuery(userId, spaceFilter).objects().size();
  }
  
  /**
   * {@inheritDoc}
   */
  public int getLastAccessedSpaceCount(SpaceFilter filter) {
    
    try {
      IdentityEntity identityEntity = identityStorage._findIdentityEntity(OrganizationIdentityProvider.NAME, filter.getRemoteId());
      SpaceListEntity listRef = RefType.MEMBER.refsOf(identityEntity);
      Map<String, SpaceRef> mapRefs = listRef.getRefs();
      //
      int counter = 0;
      
      //
      for(Map.Entry<String, SpaceRef> entry :  mapRefs.entrySet()) {
        SpaceRef ref = entry.getValue();

        // Lazy clean up
        if (ref.getSpaceRef() == null) {
          listRef.removeRef(entry.getKey());
          continue;
        }

        if (filter.getAppId() == null) {
          counter++;
        } else {
          if (ref.getSpaceRef().getApp().toLowerCase().indexOf(filter.getAppId().toLowerCase()) > 0) {
            counter++;
          }
        }
      }

      
      
      return counter;
      } catch (NodeNotFoundException e) {
        LOG.warn(e.getMessage(), e);
        return 0;
      }
  }

  /**
   * {@inheritDoc}
   */
  public List<Space> getAccessibleSpaces(String userId) throws SpaceStorageException {

    List<Space> spaces = new ArrayList<Space>();

    //
    QueryResult<SpaceEntity> results = getAccessibleSpacesByFilterQuery(userId, null).objects();

    while (results.hasNext()) {
      SpaceEntity currentSpace = results.next();
      Space space = new Space();
      fillSpaceFromEntity(currentSpace, space);
      spaces.add(space);
    }

    return spaces;
  }
  
  /*
   * Visible spaces
   */

  /**
   * {@inheritDoc}
   */
  public int getVisibleSpacesCount(String userId, SpaceFilter spaceFilter) throws SpaceStorageException {
    return _getVisibleSpaces(userId, spaceFilter).objects().size();
  }
  
  /**
   * {@inheritDoc}
   */
  public int getUnifiedSearchSpacesCount(String userId, SpaceFilter spaceFilter) throws SpaceStorageException {
    return _getUnifiedSearchSpaces(userId, spaceFilter).objects().size();
  }

  
  /**
   * {@inheritDoc}
   */
  public List<Space> getVisibleSpaces(String userId, SpaceFilter spaceFilter) throws SpaceStorageException {

    List<Space> spaces = new ArrayList<Space>();

    //
    QueryResult<SpaceEntity> results = _getVisibleSpaces(userId, spaceFilter).objects();

    while (results.hasNext()) {
      SpaceEntity currentSpace = results.next();
      Space space = new Space();
      fillSpaceFromEntity(currentSpace, space);
      spaces.add(space);
    }

    return spaces;
  }
  
  /**
   * {@inheritDoc}
   */
  public List<Space> getVisibleSpaces(String userId, SpaceFilter spaceFilter, long offset, long limit)
                                      throws SpaceStorageException {
    List<Space> spaces = new ArrayList<Space>();

    //
    QueryResult<SpaceEntity> results = _getVisibleSpaces(userId, spaceFilter).objects(offset, limit);

    while (results.hasNext()) {
      SpaceEntity currentSpace = results.next();
      Space space = new Space();
      fillSpaceFromEntity(currentSpace, space);
      spaces.add(space);
    }

    return spaces;
  }
  
  /**
   * {@inheritDoc}
   */
  public List<Space> getUnifiedSearchSpaces(String userId, SpaceFilter spaceFilter, long offset, long limit)
                                      throws SpaceStorageException {
    List<Space> spaces = new ArrayList<Space>();
    
    //
    QueryResult<SpaceEntity> results = _getUnifiedSearchSpaces(userId, spaceFilter).objects(offset, limit);

    while (results.hasNext()) {
      SpaceEntity currentSpace = results.next();
      Space space = new Space();
      fillSpaceFromEntity(currentSpace, space);
      spaces.add(space);
    }

    return spaces;
  }
  
  private Query<SpaceEntity> _getUnifiedSearchSpaces(String userId, SpaceFilter spaceFilter) {
    QueryBuilder<SpaceEntity> builder = getSession().createQueryBuilder(SpaceEntity.class);
    WhereExpression whereExpression = new WhereExpression();

    _applyUnifiedSearchFilter(whereExpression, spaceFilter);

    if (whereExpression.toString().trim().length() > 0) {
      builder.where(whereExpression.toString());
    }
    applyOrder(builder, spaceFilter);
    
    return builder.get();
  }
  
  private void _applyUnifiedSearchFilter(WhereExpression whereExpression, SpaceFilter spaceFilter) {

    if (spaceFilter == null) return;

    String spaceNameSearchCondition = StorageUtils.escapeSpecialCharacter(spaceFilter.getSpaceNameSearchCondition());

    char firstCharacterOfName = spaceFilter.getFirstCharacterOfSpaceName();

    if (spaceNameSearchCondition != null && spaceNameSearchCondition.length() != 0) {

        List<String> unifiedSearchConditions = this.processUnifiedSearchCondition(spaceNameSearchCondition);
        
        if (unifiedSearchConditions.size() > 0) {
          whereExpression.startGroup();
        }
        
        boolean first = true;
        for(String condition : unifiedSearchConditions) {
          //
          if (first == false) {
            whereExpression.and();
          }
          whereExpression.startGroup();
          whereExpression
             .contains(SpaceEntity.name, condition.toLowerCase())
             .or()
             .contains(SpaceEntity.displayName, condition.toLowerCase())
             .or()
             .contains(SpaceEntity.description, StringEscapeUtils.escapeHtml(condition).toLowerCase());
          whereExpression.endGroup();
          
          first = false;
        } //end for
        
        if (unifiedSearchConditions.size() > 0) {
          whereExpression.endGroup();
        }
    }
    else if (!Character.isDigit(firstCharacterOfName)) {
      String firstCharacterOfNameString = Character.toString(firstCharacterOfName);
      String firstCharacterOfNameLowerCase = firstCharacterOfNameString.toLowerCase() + StorageUtils.PERCENT_STR;
      whereExpression
          .like(whereExpression.callFunction(QueryFunction.LOWER, SpaceEntity.name), firstCharacterOfNameLowerCase);
    }
  }

  private Query<SpaceEntity> _getVisibleSpaces(String userId, SpaceFilter spaceFilter) {

    QueryBuilder<SpaceEntity> builder = getSession().createQueryBuilder(SpaceEntity.class);
    WhereExpression whereExpression = new WhereExpression();

    if (validateFilter(spaceFilter)) {
      _applyFilter(whereExpression, spaceFilter);
      whereExpression.and();
      whereExpression.startGroup();
    }
    
    //visibility::(soc:visibily like 'private') 
    whereExpression.startGroup();
    whereExpression.like(SpaceEntity.visibility, Space.PRIVATE);
    whereExpression.endGroup();
    
    //(soc:visibily like 'private' AND (soc:registration like 'open' OR soc:registration like 'validate'))
    // OR
    //(soc:membersId like '' OR managerMembersId like '' OR soc:invitedMembersId like '')
    whereExpression.or(); 
    whereExpression.startGroup(); 

    whereExpression.equals(SpaceEntity.membersId, userId)
                   .or()
                   .equals(SpaceEntity.managerMembersId, userId)
                   .or()
                   .equals(SpaceEntity.invitedMembersId, userId);

    whereExpression.endGroup();
    whereExpression.endAllGroup();


    builder.where(whereExpression.toString());
    applyOrder(builder, spaceFilter);
    
    return builder.get();

  }


  /**
   * {@inheritDoc}
   */
  public List<Space> getAccessibleSpaces(String userId, long offset, long limit) throws SpaceStorageException {
    List<Space> spaces = new ArrayList<Space>();

    //
    QueryResult<SpaceEntity> results = getAccessibleSpacesByFilterQuery(userId, null).objects(offset, limit);

    while (results.hasNext()) {
      SpaceEntity currentSpace = results.next();
      Space space = new Space();
      fillSpaceFromEntity(currentSpace, space);
      spaces.add(space);
    }

    return spaces;
  }

  /**
   * {@inheritDoc}
   */
  public List<Space> getAccessibleSpacesByFilter(String userId, SpaceFilter spaceFilter, long offset, long limit) {

    List<Space> spaces = new ArrayList<Space>();

    //
    QueryResult<SpaceEntity> results = getAccessibleSpacesByFilterQuery(userId, spaceFilter).objects(offset, limit);

    while (results.hasNext()) {
      SpaceEntity currentSpace = results.next();
      Space space = new Space();
      fillSpaceFromEntity(currentSpace, space);
      spaces.add(space);
    }

    return spaces;
  }

  /*
    Editable spaces
   */

  /**
   * {@inheritDoc}
   */
  public int getEditableSpacesCount(String userId) throws SpaceStorageException {
    try {
      IdentityEntity identityEntity = identityStorage._findIdentityEntity(OrganizationIdentityProvider.NAME, userId);
      return identityEntity.getManagerSpaces().getRefs().size();
    }
    catch (NodeNotFoundException e) {
      return 0;
    }
  }

  /**
   * {@inheritDoc}
   */
  public int getEditableSpacesByFilterCount(String userId, SpaceFilter spaceFilter) {
    return getEditableSpacesFilterQuery(userId, spaceFilter).objects().size();
  }

  /**
   * {@inheritDoc}
   */
  public List<Space> getEditableSpaces(String userId) throws SpaceStorageException {

    List<Space> spaces = new ArrayList<Space>();

    //
    QueryResult<SpaceEntity> results = getEditableSpacesFilterQuery(userId, null).objects();

    while (results.hasNext()) {
      SpaceEntity currentSpace = results.next();
      Space space = new Space();
      fillSpaceFromEntity(currentSpace, space);
      spaces.add(space);
    }

    return spaces;
  }

  /**
   * {@inheritDoc}
   */
  public List<Space> getEditableSpaces(String userId, long offset, long limit) throws SpaceStorageException {

    List<Space> spaces = new ArrayList<Space>();

    //
    QueryResult<SpaceEntity> results = getEditableSpacesFilterQuery(userId, null).objects(offset, limit);

    while (results.hasNext()) {
      SpaceEntity currentSpace = results.next();
      Space space = new Space();
      fillSpaceFromEntity(currentSpace, space);
      spaces.add(space);
    }

    return spaces;
  }

  /**
   * {@inheritDoc}
   */
  public List<Space> getEditableSpacesByFilter(String userId, SpaceFilter spaceFilter, long offset, long limit) {

    List<Space> spaces = new ArrayList<Space>();

    //
    QueryResult<SpaceEntity> results = getEditableSpacesFilterQuery(userId, spaceFilter).objects(offset, limit);

    while (results.hasNext()) {
      SpaceEntity currentSpace = results.next();
      Space space = new Space();
      fillSpaceFromEntity(currentSpace, space);
      spaces.add(space);
    }

    return spaces;
  }

  /*
    All spaces
   */

  /**
   * {@inheritDoc}
   */
  public int getAllSpacesCount() throws SpaceStorageException {

    // TODO : use property to improve the perfs

    return getSpaceRoot().getSpaces().size();

  }

  /**
   * {@inheritDoc}
   */
  public List<Space> getAllSpaces() throws SpaceStorageException {

    List<Space> spaces = new ArrayList<Space>();

    QueryResult<SpaceEntity> results = getSpacesByFilterQuery(null).objects();

    while (results.hasNext()) {
      SpaceEntity currentSpace = results.next();
      Space space = new Space();
      fillSpaceFromEntity(currentSpace, space);
      spaces.add(space);
    }
    return spaces;
  }

  /**
   * {@inheritDoc}
   */
  public int getAllSpacesByFilterCount(SpaceFilter spaceFilter) {

    if (validateFilter(spaceFilter)) {
      return getSpacesByFilterQuery(spaceFilter).objects().size();
    }
    else {
      return 0;
    }

  }


  /*
    Get spaces
   */

  /**
   * {@inheritDoc}
   */
  public List<Space> getSpaces(long offset, long limit) throws SpaceStorageException {
    List<Space> spaces = new ArrayList<Space>();

    QueryResult<SpaceEntity> results = getSpacesByFilterQuery(null).objects(offset, limit);

    while (results.hasNext()) {
      SpaceEntity currentSpace = results.next();
      Space space = new Space();
      fillSpaceFromEntity(currentSpace, space);
      spaces.add(space);
    }
    return spaces;
  }

  /**
   * {@inheritDoc}
   */
  public List<Space> getSpacesByFilter(SpaceFilter spaceFilter, long offset, long limit) {

    List<Space> spaces = new ArrayList<Space>();

    QueryResult<SpaceEntity> results = getSpacesByFilterQuery(spaceFilter).objects(offset, limit);

    while (results.hasNext()) {
      SpaceEntity currentSpace = results.next();
      Space space = new Space();
      fillSpaceFromEntity(currentSpace, space);
      spaces.add(space);
    }
    return spaces;
  }

  /**
   * {@inheritDoc}
   */
  public Space getSpaceById(String id) throws SpaceStorageException {

    try {

      SpaceEntity spaceEntity = _findById(SpaceEntity.class, id);

      Space space = new Space();

      fillSpaceFromEntity(spaceEntity, space);

      return space;

    }
    catch (NodeNotFoundException e) {
      return null;
    }

  }
  
  /**
   * {@inheritDoc}
   */
  public Space getSpaceSimpleById(String id) throws SpaceStorageException {

    try {

      SpaceEntity spaceEntity = _findById(SpaceEntity.class, id);

      Space space = new Space();

      fillSpaceSimpleFromEntity(spaceEntity, space);

      return space;

    }
    catch (NodeNotFoundException e) {
      return null;
    }

  }

  /**
   * {@inheritDoc}
   */
  public Space getSpaceByPrettyName(String spacePrettyName) throws SpaceStorageException {

    try {

      SpaceEntity entity = _findByPath(SpaceEntity.class, String.format("/production/soc:spaces/soc:%s", spacePrettyName));

      Space space = new Space();
      fillSpaceFromEntity(entity, space);

      return space;

    }
    catch (NodeNotFoundException e) {
      return null;
    }
  }

  /**
   * {@inheritDoc}
   */
  public Space getSpaceByGroupId(String groupId) throws SpaceStorageException {

    // TODO : avoid JCR query ?

    QueryBuilder<SpaceEntity> builder = getSession().createQueryBuilder(SpaceEntity.class);
    WhereExpression whereExpression = new WhereExpression();

    builder.where(whereExpression.equals(SpaceEntity.groupId, groupId).toString());

    QueryResult<SpaceEntity> result = builder.get().objects();

    if (result.hasNext()) {
      SpaceEntity entity =  result.next();
      Space space = new Space();

      fillSpaceFromEntity(entity, space);

      return space;
    }
    else {
      return null;
    }

  }

  /**
   * {@inheritDoc}
   */
  public Space getSpaceByUrl(String url) throws SpaceStorageException {

    // TODO : avoid JCR query ?

    QueryBuilder<SpaceEntity> builder = getSession().createQueryBuilder(SpaceEntity.class);

    if (url != null) {
      WhereExpression whereExpression = new WhereExpression();
      whereExpression.equals(SpaceEntity.url, url);
      builder.where(whereExpression.toString());
    }

    QueryResult<SpaceEntity> result = builder.get().objects();

    if (result.hasNext()) {

      Space space = new Space();
      SpaceEntity entity =  builder.get().objects().next();

      fillSpaceFromEntity(entity, space);

      return space;

    }
    else {
      return null;
    }

  }

  @Override
  public void updateSpaceAccessed(String remoteId, Space space) throws SpaceStorageException {
    try {
      IdentityEntity identityEntity = identityStorage._findIdentityEntity(OrganizationIdentityProvider.NAME, remoteId);
      SpaceListEntity listRef = RefType.MEMBER.refsOf(identityEntity);
      
      SpaceEntity spaceEntity = _findById(SpaceEntity.class, space.getId());

      SpaceRef ref = listRef.getRef(spaceEntity.getName());
      if (!ref.getName().equals(spaceEntity.getName())) {
        ref.setName(spaceEntity.getName());
      }
      ref.setSpaceRef(spaceEntity);

//      getSession().save();

    } catch (NodeNotFoundException e) {
      LOG.warn(e.getMessage(), e);
    }
  }

  @Override
  public List<Space> getLastAccessedSpace(SpaceFilter filter, int offset, int limit) throws SpaceStorageException {
    try {
      IdentityEntity identityEntity = identityStorage._findIdentityEntity(OrganizationIdentityProvider.NAME, filter.getRemoteId());
      SpaceListEntity listRef = RefType.MEMBER.refsOf(identityEntity);
      Map<String, SpaceRef> mapRefs = listRef.getRefs();
      
      //
      ChromatticSessionImpl chromatticSession = (ChromatticSessionImpl) getSession();
      Map<SpaceRef, Long> spaceRefs = new LinkedHashMap<SpaceRef, Long>();
      Space space = null;
      
      //
      for(Map.Entry<String, SpaceRef> entry :  mapRefs.entrySet()) {
        SpaceRef ref = entry.getValue();
        Node node = chromatticSession.getNode(ref);
        Property p = getProperty(node, JCRProperties.JCR_LAST_MODIFIED_DATE.getName());
        long lastModifedDate = p == null ? 0 : p.getDate().getTimeInMillis();

        // Lazy clean up
        if (ref.getSpaceRef() == null) {
          listRef.removeRef(entry.getKey());
          continue;
        }

        if (filter.getAppId() == null) {
          spaceRefs.put(ref, lastModifedDate);
        } else {
          if (ref.getSpaceRef().getApp().toLowerCase().indexOf(filter.getAppId().toLowerCase()) > 0) {
            spaceRefs.put(ref, lastModifedDate);
          }
        }
      }
      spaceRefs = StorageUtils.sortMapByValue(spaceRefs, false);
      
      List<Space> got = new LinkedList<Space>();
      Iterator<SpaceRef> it1 = spaceRefs.keySet().iterator();
      _skip(it1, offset);

      //
      int numberOfSpace = 0;
      while (it1.hasNext()) {
        space = new Space();
        fillSpaceSimpleFromEntity(it1.next().getSpaceRef(), space);
        got.add(space);
        //
        if (++numberOfSpace == limit) {
          break;
        }
      }

      return got;
    } catch (NodeNotFoundException e) {
      LOG.warn("Get last accessed spaces failure.", e);
    } catch (RepositoryException e) {
      LOG.warn("Get last accessed spaces failure.", e);
    }
    
    //
    return Collections.emptyList();
  }

  public List<Space> getLastSpaces(final int limit) {
    QueryBuilder<SpaceEntity> builder = getSession().createQueryBuilder(SpaceEntity.class);

    Ordering ordering = Ordering.valueOf(Sorting.OrderBy.DESC.toString());

    builder.orderBy(SpaceEntity.createdTime.getName(), ordering);

    QueryResult<SpaceEntity> result = builder.get().objects(0L, (long)limit);

    List<Space> got = new LinkedList<Space>();
    while (result.hasNext()) {
      SpaceEntity entity =  result.next();
      Space space = new Space();

      fillSpaceFromEntity(entity, space);

      got.add(space);
    }
    return got;
  }

  public int getNumberOfMemberPublicSpaces(String userId) {
    return getSpacesOfMemberQuery(userId).objects().size();
  }
  
  @Override
  public List<Space> getVisitedSpaces(SpaceFilter filter, int offset, int limit) throws SpaceStorageException {

    try {
      IdentityEntity identityEntity = identityStorage._findIdentityEntity(OrganizationIdentityProvider.NAME, filter.getRemoteId());
      SpaceListEntity listRef = RefType.MEMBER.refsOf(identityEntity);
      Map<String, SpaceRef> mapRefs = listRef.getRefs();
      
      //
      Map<SpaceRef, Long> visitedSpaceRefs = new LinkedHashMap<SpaceRef, Long>();
      List<SpaceRef> neverVisitedSpaceRefs = new LinkedList<SpaceRef>();
      
      visitedSpaceRefs = getSpaceRefs(mapRefs, visitedSpaceRefs, neverVisitedSpaceRefs, filter.getAppId());
      Iterator<SpaceRef> spaceRefs = visitedSpaceRefs.keySet().iterator();
      
      if (offset < visitedSpaceRefs.size()) {
        _skip(spaceRefs, offset);
        offset = 0;
      } else {
        _skip(spaceRefs, offset);
        offset = offset - (visitedSpaceRefs.size());
      }
      
      //
      List<Space> got = new LinkedList<Space>();
      //priority for visited spaces to return
      getSpacesFromSpaceRefs(spaceRefs, got, limit);
      
      // process the spaces which are never be visited
      int remain = limit - got.size();
      if (neverVisitedSpaceRefs.isEmpty() || (remain == 0)) {
        return got;
      }

      spaceRefs = neverVisitedSpaceRefs.iterator();
      _skip(spaceRefs, offset);
      //
      List<Space> neverVisitedSpaces = new LinkedList<Space>();
      getSpacesFromSpaceRefs(spaceRefs, neverVisitedSpaces, -1);
      neverVisitedSpaces = StorageUtils.sortSpaceByName(neverVisitedSpaces, true);

      //
      got.addAll(neverVisitedSpaces.subList(0, Math.min(remain, neverVisitedSpaces.size())));
      
      return got;
    } catch (NodeNotFoundException e) {
      LOG.warn(e.getMessage(), e);
    } catch (RepositoryException e) {
      LOG.warn(e.getMessage(), e);
    }
    //
    return Collections.emptyList();
  }
  
  private Map<SpaceRef, Long> getSpaceRefs(Map<String, SpaceRef> spaceRefs, Map<SpaceRef, Long> visitedSpaceRefs, List<SpaceRef> neverVisitedSpaceRefs, String appId) throws RepositoryException {
    //
    ChromatticSessionImpl chromatticSession = (ChromatticSessionImpl) getSession();
    
    for (Entry<String, SpaceRef> entry : spaceRefs.entrySet()) {
      SpaceRef ref = entry.getValue();
      Node node = chromatticSession.getNode(ref);
      Property p1 = getProperty(node, JCRProperties.JCR_LAST_CREATED_DATE.getName());
      Property p2 = getProperty(node, JCRProperties.JCR_LAST_MODIFIED_DATE.getName());
      long createdTime = p1 == null ? 0 : p1.getDate().getTimeInMillis();
      long lastModifedDate = p2 == null ? 0 : p2.getDate().getTimeInMillis();
      
      boolean isValid = false;
      if (appId == null) {
        isValid = true;
      } else {
        if (ref.getSpaceRef().getApp().toLowerCase().indexOf(appId.toLowerCase()) > 0) {
          isValid = true;
        }

      }
      
      //The never visited spaces which have last modified date different with the created time less than 2 seconds
      if (lastModifedDate - createdTime < TWO_SECONDS && isValid) {
        neverVisitedSpaceRefs.add(ref);
      } else if (isValid) {
        visitedSpaceRefs.put(ref, lastModifedDate);
      }
    }
    
    //sort visited space by modified date
    return StorageUtils.sortMapByValue(visitedSpaceRefs, false);
  }
  
  private void getSpacesFromSpaceRefs(Iterator<SpaceRef> it, List<Space> list, int limit) {
    Space space = null;
    //
    int numberOfSpace = 0;
    while (it.hasNext()) {
      space = new Space();
      fillSpaceFromEntity(it.next().getSpaceRef(), space);
      list.add(space);
      //
      if (++numberOfSpace == limit) {
        break;
      }
    }
  }

  private Property getProperty(Node node, String propertyName) {
    try {
      return node.getProperty(propertyName);
    } catch (RepositoryException e) {
      LOG.error(String.format("Get property %s failed", propertyName));
    }
    return null;
  }

  @Override
  public List<String> getMemberSpaceIds(String identityId, int offset, int limit) throws SpaceStorageException {
    List<String> identitiesId = new ArrayList<String>();
    try {
      IdentityEntity identityEntity = _findById(IdentityEntity.class, identityId);
      Set<String> spaceNames = identityEntity.getSpaces().getRefs().keySet();
      ProviderEntity providerEntity = getProviderRoot().getProvider(SpaceIdentityProvider.NAME);
      for (String spacePrettyName : spaceNames) {
        IdentityEntity spaceIdentity = providerEntity.getIdentities().get(spacePrettyName);
        if (spaceIdentity != null) {
          identitiesId.add(spaceIdentity.getId());
        }
      }
    } catch (Exception e) {
      LOG.error("Failed to get list of space identity of current user");
    }
    return identitiesId;
  }
}
