/*
 * Copyright (C) 2003-2007 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 General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, see<http://www.gnu.org/licenses/>.
 */
package org.exoplatform.contact.service.impl;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;

import javax.jcr.Node;
import javax.jcr.NodeIterator;
import javax.jcr.ReferentialIntegrityException;
import javax.jcr.Value;
import javax.jcr.query.Query;
import javax.jcr.query.QueryManager;
import javax.jcr.query.QueryResult;

import org.exoplatform.contact.service.Contact;
import org.exoplatform.contact.service.ContactService;
import org.exoplatform.services.jcr.ext.common.SessionProvider;
import org.exoplatform.services.jcr.ext.hierarchy.NodeHierarchyCreator;
import org.exoplatform.services.organization.Membership;
import org.exoplatform.services.organization.MembershipEventListener;

/**
 * Created by The eXo Platform SARL
 * Author : Hung Nguyen Quang
 *          hung.nguyen@exoplatform.com
 * Nov 23, 2007 3:09:21 PM
 */
public class NewMembershipListener extends MembershipEventListener {
  private ContactService cservice_ ;
  private NodeHierarchyCreator nodeHierarchyCreator_ ;
  public NewMembershipListener(ContactService cservice, NodeHierarchyCreator nodeHierarchyCreator) throws Exception {
    cservice_ = cservice ;
    nodeHierarchyCreator_ = nodeHierarchyCreator ;
  }

  public void postSave(Membership m, boolean isNew) throws Exception {
    
    String username = m.getUserName();
    String groupId = m.getGroupId();
    cservice_.addUserContactInAddressBook(username, groupId) ;
    JCRDataStorage storage_ = new JCRDataStorage(nodeHierarchyCreator_) ;
    SessionProvider systemSession = SessionProvider.createSystemProvider() ;
    try {
      Node publicContactHome = storage_.getPublicContactsHome(systemSession) ;      
      String usersPath = nodeHierarchyCreator_.getJcrPath(JCRDataStorage.USERS_PATH) ;
      QueryManager qm = publicContactHome.getSession().getWorkspace().getQueryManager();


      // Add the user's contact to the public address book for the group
      Contact contact = cservice_.getPublicContact(username) ;
      Map<String, String> groups = new LinkedHashMap<String, String>() ;
      for (String group  : contact.getAddressBookIds()) groups.put(group, group) ;
      groups.put(groupId, groupId) ;
      contact.setAddressBookIds(groups.keySet().toArray(new String[] {})) ;
      cservice_.saveContact(username, contact, false) ;

      // lookup address books to this group
      StringBuffer queryString = new StringBuffer("/jcr:root" + usersPath 
          + "//element(*,exo:contactGroup)[@exo:viewPermissionGroups='").append( groupId + "']") ;        
      Query query = qm.createQuery(queryString.toString(), Query.XPATH);
      QueryResult result = query.execute();
      NodeIterator nodes = result.getNodes() ;
      List<String> to = Arrays.asList(new String []{username});
      while (nodes.hasNext()) {
        Node address = nodes.nextNode() ;
        String from = address.getProperty("exo:sharedUserId").getString();
        String addressBookId = address.getProperty("exo:id").getString();
        storage_.shareAddressBook(from, addressBookId, to) ;
      }

      // lookup single contacts shared to this group
      queryString = new StringBuffer("/jcr:root" + usersPath 
          + "//element(*,exo:contact)[@exo:viewPermissionGroups='").append(groupId + "']") ;        
      query = qm.createQuery(queryString.toString(), Query.XPATH);
      result = query.execute();
      nodes = result.getNodes() ;
      while (nodes.hasNext()) {
        Node contactNode = nodes.nextNode() ;
        String split = "/" ;
        String temp = contactNode.getPath().split(usersPath)[1] ;
        String userId = temp.split(split)[1] ;
        String [] addressBookIds = new String[] {contactNode.getProperty("exo:id").getString()};
        storage_.shareContact(userId, addressBookIds, to) ;
      }
    } catch (Exception e) {
      e.printStackTrace() ;
    } finally {
      systemSession.close() ;
    }
  }

  public void preDelete(Membership m) throws Exception {
    /*OrganizationService organizationService = 
      (OrganizationService)PortalContainer.getComponent(OrganizationService.class) ;
    Object[] objGroupIds = organizationService.getGroupHandler().findGroupsOfUser(m.getUserName()).toArray() ;*/

    //  remove group of public contact
    Contact contact = cservice_.getPublicContact(m.getUserName()) ;
    Map<String, String> groupIds = new LinkedHashMap<String, String>() ;
    for (String group  : contact.getAddressBook()) groupIds.put(group, group) ;
    groupIds.remove(m.getGroupId()) ;
    contact.setAddressBook(groupIds.keySet().toArray(new String[] {})) ;
    SessionProvider systemSession = SessionProvider.createSystemProvider();
    try {
      cservice_.saveContact(m.getUserName(), contact, false) ;
      JCRDataStorage storage_ = new JCRDataStorage(nodeHierarchyCreator_) ;
      Node publicContactHome = storage_.getPublicContactsHome(systemSession) ;      
      String usersPath = nodeHierarchyCreator_.getJcrPath(JCRDataStorage.USERS_PATH) ;
      QueryManager qm = publicContactHome.getSession().getWorkspace().getQueryManager();
      StringBuffer queryString = new StringBuffer("/jcr:root" + usersPath 
          + "//element(*,exo:contactGroup)[@exo:viewPermissionGroups='").append( m.getGroupId() + "']") ;        
      Query query = qm.createQuery(queryString.toString(), Query.XPATH);
      QueryResult result = query.execute();
      NodeIterator nodes = result.getNodes() ;
      while (nodes.hasNext()) {
        Node address = nodes.nextNode() ;
        storage_.unshareAddressBook(
            address.getProperty("exo:sharedUserId")
            .getString(), address.getProperty("exo:id").getString(), m.getUserName()) ;

//      user shared if belong another groups shared
        for (Value groupShared : address.getProperty("exo:viewPermissionGroups").getValues()) {
          if (groupIds.keySet().contains(groupShared.getString())) {
            List<String> reciever = new ArrayList<String>() ;
            reciever.add(m.getUserName()) ;
            storage_.shareAddressBook(address.getProperty("exo:sharedUserId")
                .getString(), address.getProperty("exo:id").getString(), reciever) ;
          }
        }
      }
//    lookup shared contacts
      queryString = new StringBuffer("/jcr:root" + usersPath 
          + "//element(*,exo:contact)[@exo:viewPermissionGroups='").append(m.getGroupId() + "']") ;        
      query = qm.createQuery(queryString.toString(), Query.XPATH);
      result = query.execute();
      nodes = result.getNodes() ;
      while (nodes.hasNext()) {
        Node contactNode = nodes.nextNode() ;
        String split = "/" ;
        String temp = contactNode.getPath().split(usersPath)[1] ;
        String userId = temp.split(split)[1] ;
        storage_.removeUserShareContact(userId, contactNode.getProperty("exo:id").getString(), m.getUserName()) ;

        // user shared if belong another groups shared
        for (Value groupShared : contactNode.getProperty("exo:viewPermissionGroups").getValues()) {
          if (groupIds.keySet().contains(groupShared.getString())) {
            List<String> reciever = new ArrayList<String>() ;
            reciever.add(m.getUserName()) ;
            storage_.shareContact(userId, new String [] {contactNode.getProperty("exo:id").getString()}, reciever) ;
          }
        }        
      }      
    } catch (ReferentialIntegrityException e) {
      
    } catch (Exception ex) {
      ex.printStackTrace();
    } finally {
      systemSession.close() ;
    }
  }
}