issue-157 follow references support

Author: Jeff Bornemann

src/main/groovy/com/twcable/grabbit/client/batch/ClientBatchJobExecutionListener.groovy

@@ -16,7 +16,7 @@
 
 package com.twcable.grabbit.client.batch
 
-import com.twcable.grabbit.jcr.JcrUtil
+import com.twcable.grabbit.jcr.JCRUtil
 import groovy.transform.CompileStatic
 import groovy.util.logging.Slf4j
 import org.apache.sling.jcr.api.SlingRepository
@@ -58,7 +58,7 @@ class ClientBatchJobExecutionListener implements JobExecutionListener {
     void beforeJob(JobExecution jobExecution) {
         log.debug "SlingRepository : ${slingRepository}"
         final clientUsername = jobExecution.jobParameters.getString(ClientBatchJob.CLIENT_USERNAME)
-        final Session session = JcrUtil.getSession(slingRepository, clientUsername)
+        final Session session = JCRUtil.getSession(slingRepository, clientUsername)
 
         ClientBatchJobContext.setSession(session)
         log.info "Starting job : ${jobExecution}\n\n"

src/main/groovy/com/twcable/grabbit/jcr/ACLProtoNodeDecorator.groovy

@@ -166,7 +166,7 @@ class ACLProtoNodeDecorator extends ProtoNodeDecorator {
         final ProtoPropertyDecorator privilegeProperty = node.propertiesList.collect { new ProtoPropertyDecorator(it) }.find { ProtoPropertyDecorator property ->
             property.isPrivilege()
         }
-        return privilegeProperty.getPropertyValues().collect { Value value -> value.string }.toArray() as String[]
+        return privilegeProperty.getStringValues().toArray() as String[]
     }
 
 

src/main/groovy/com/twcable/grabbit/jcr/AuthorizableProtoNodeDecorator.groovy

@@ -25,7 +25,6 @@ import groovy.util.logging.Slf4j
 import java.lang.reflect.Field
 import java.lang.reflect.Method
 import java.lang.reflect.ReflectPermission
-import java.util.regex.Pattern
 import javax.annotation.Nonnull
 import javax.jcr.Session
 import org.apache.jackrabbit.api.security.user.Authorizable
@@ -35,7 +34,6 @@ import org.apache.jackrabbit.api.security.user.UserManager
 import org.apache.jackrabbit.value.StringValue
 import org.apache.sling.jcr.base.util.AccessControlUtil
 
-
 /**
  * This class wraps a serialized node that represents an Authorizable. Authorizables are special system protected nodes, that can only be written under certain
  * trees, and can not be written directly by a client.
@@ -59,8 +57,8 @@ class AuthorizableProtoNodeDecorator extends ProtoNodeDecorator {
         }
         Authorizable existingAuthorizable = findAuthorizable(session)
         Authorizable newAuthorizable = existingAuthorizable ? updateAuthorizable(existingAuthorizable, session) : createNewAuthorizable(session)
-        writeAuthorizablePieces(newAuthorizable, session)
-        return new JCRNodeDecorator(session.getNode(newAuthorizable.getPath()))
+
+        return new JCRNodeDecorator(session.getNode(newAuthorizable.getPath()), getID())
     }
 
 
@@ -86,11 +84,20 @@ class AuthorizableProtoNodeDecorator extends ProtoNodeDecorator {
                 setPasswordForUser(newUser, session)
             }
             session.save()
+            writeMandatoryPieces(session, newUser.getPath())
             return newUser
         }
-        final Group group = userManager.createGroup(authorizableID, new AuthorizablePrincipal(authorizableID), getParentPath())
+        final Group newGroup = userManager.createGroup(authorizableID, new AuthorizablePrincipal(authorizableID), getParentPath())
+        /*
+         * Write all mandatory pieces, and find those that are authorizables. We then need to see if any of them have membership in this group, and add them.
+         */
+        final Collection<JCRNodeDecorator> authorizablePieces = writeMandatoryPieces(session, newGroup.getPath()).findAll { it.isAuthorizableType() }
+        final Collection<JCRNodeDecorator> members = authorizablePieces.findAll { getMembershipIDs().contains(it.getTransferredID()) }
+        members.each { JCRNodeDecorator member ->
+            newGroup.addMember(member as Authorizable)
+        }
         session.save()
-        return group
+        return newGroup
     }
 
 
@@ -100,22 +107,19 @@ class AuthorizableProtoNodeDecorator extends ProtoNodeDecorator {
      * @return new instance of updated authorizable
      */
     private Authorizable updateAuthorizable(final Authorizable authorizable, final Session session) {
+        //We get all the declared groups of this authorizable so that we can add them back to the new, updated authorizable
+        final Collection<Group> declaredGroups = authorizable.declaredMemberOf().toList()
+        for(Group group : declaredGroups) {
+            group.removeMember(authorizable)
+        }
         authorizable.remove()
         session.save()
-        createNewAuthorizable(session)
-    }
-
-
-    /**
-     * Authorizable pieces (nodes that live under Authorizables - profile, preferences, etc) get sent with the authorizable node instead of streamed independently because we do not know the client's new
-     * authorizable UUID node name at runtime. In other words, authorizables can live under different node names from server to server
-     */
-    private void writeAuthorizablePieces(final Authorizable authorizable, final Session session) {
-        innerProtoNode.mandatoryChildNodeList.each {
-            //We replace the incoming server authorizable path, with the new authorizable path
-            createFrom(it, it.name.replaceFirst(Pattern.quote(getName()), authorizable.getPath())).writeToJcr(session)
+        final Authorizable newAuthorizable = createNewAuthorizable(session)
+        for(Group group: declaredGroups) {
+            group.addMember(newAuthorizable)
         }
         session.save()
+        return newAuthorizable
     }
 
 
@@ -135,6 +139,11 @@ class AuthorizableProtoNodeDecorator extends ProtoNodeDecorator {
     }
 
 
+    private Collection<String> getMembershipIDs() {
+        return hasProperty('rep:members') ? getStringValuesFrom('rep:members') : []
+    }
+
+
     /**
      * Some JVM's have a SecurityManager set, which based on configuration, can potentially inhibit our hack {@code setPasswordForUser(User, Session)} from working.
      * We need to check security permissions before proceeding
@@ -245,13 +254,4 @@ class AuthorizableProtoNodeDecorator extends ProtoNodeDecorator {
         */
         setPasswordMethod.invoke(userManagerDelegate, getTreeMethod.invoke(authorizable), getAuthorizableID(), getStringValueFrom('rep:password'), false)
     }
-
-
-    /**
-     * An instance wrapper for ease of mocking
-     * @see super.createFrom
-     */
-    ProtoNodeDecorator createFrom(final ProtoNode protoNode, final String nameOverride) {
-        super.createFrom(protoNode, nameOverride)
-    }
 }

src/main/groovy/com/twcable/grabbit/jcr/DefaultProtoNodeDecorator.groovy

@@ -19,11 +19,12 @@ import com.twcable.grabbit.proto.NodeProtos.Node as ProtoNode
 import com.twcable.grabbit.proto.NodeProtos.Value as ProtoValue
 import groovy.transform.CompileStatic
 import groovy.util.logging.Slf4j
-import java.util.regex.Pattern
 import javax.annotation.Nonnull
 import javax.jcr.Node as JCRNode
 import javax.jcr.Session
+import javax.jcr.Value
 import org.apache.jackrabbit.commons.JcrUtils
+import org.apache.jackrabbit.value.StringValue
 
 
 import static org.apache.jackrabbit.JcrConstants.JCR_MIXINTYPES
@@ -49,16 +50,34 @@ class DefaultProtoNodeDecorator extends ProtoNodeDecorator {
         if(mixinProperty) {
             addMixins(mixinProperty, jcrNode)
         }
-        //Then add other properties
-        writableProperties.each { it.writeToNode(jcrNode) }
 
-        if(innerProtoNode.mandatoryChildNodeList && innerProtoNode.mandatoryChildNodeList.size() > 0) {
-            for(ProtoNode childNode: innerProtoNode.mandatoryChildNodeList) {
-                //Mandatory children must inherit any name overrides from their parent (if they exist)
-                createFrom(childNode, childNode.getName().replaceFirst(Pattern.quote(innerProtoNode.name), getName())).writeToJcr(session)
+        final Collection<ProtoPropertyDecorator> referenceTypeProperties = writableProperties.findAll { it.isReferenceType() }
+        final Collection<ProtoPropertyDecorator> nonReferenceTypeProperties = writableProperties.findAll { !it.isReferenceType() }
+
+        //These can be written now. Reference properties can be written after we write the referenced nodes
+        nonReferenceTypeProperties.each { it.writeToNode(jcrNode) }
+
+        final Collection<JCRNodeDecorator> referenceables = writeMandatoryPieces(session, getName()).findAll { it.isReferenceable() }
+
+        /*
+         * Nodes that are referenced from reference properties are written above in writeMandatoryPieces(). We can now map each
+         * reference pointer to a transferred id from a node above, and write the pointer with a mapped nodes new identifier.
+         * The transferred id is what the identifier was on sending server, and the current identifier is what it is now that it is
+          * written to this server. Identifiers only apply to referenceable nodes (i.e nodes with mix:referenceable)
+         */
+        referenceTypeProperties.each { ProtoPropertyDecorator property ->
+            final Collection<Value> newReferenceIDValues = property.getStringValues().findResults { String referenceID ->
+                final JCRNodeDecorator match = referenceables.find { it.transferredID == referenceID }
+                if(match) {
+                    return new StringValue(match.getIdentifier())
+                }
+            } as Collection<Value>
+            if(!newReferenceIDValues.isEmpty()) {
+                property.writeToNode(jcrNode, (newReferenceIDValues.toArray() as Value[]))
             }
         }
-        return new JCRNodeDecorator(jcrNode)
+
+        return new JCRNodeDecorator(jcrNode, getID())
     }