com.alvexcore.repo.AlvexVersionableAspect.java Source code

Java tutorial

Introduction

Here is the source code for com.alvexcore.repo.AlvexVersionableAspect.java

Source

/*
 * Copyright (C) 2005-2012 Alfresco Software Limited.
 * Copyright (C) 2014 ITD Systems LLC.
 *
 * This file is part of Alvex
 *
 * Alvex is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * Alvex 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 Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with Alvex. If not, see <http://www.gnu.org/licenses/>.
 */
package com.alvexcore.repo;

import java.io.Serializable;
import java.util.Collections;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;

import org.alfresco.model.ContentModel;
import org.alfresco.repo.content.ContentServicePolicies;
import org.alfresco.repo.copy.CopyBehaviourCallback;
import org.alfresco.repo.copy.CopyDetails;
import org.alfresco.repo.copy.CopyServicePolicies;
import org.alfresco.repo.copy.DefaultCopyBehaviourCallback;
import org.alfresco.repo.dictionary.DictionaryDAO;
import org.alfresco.repo.dictionary.DictionaryListener;
import org.alfresco.repo.lock.LockUtils;
import org.alfresco.repo.node.NodeServicePolicies;
import org.alfresco.repo.node.NodeServicePolicies.OnUpdatePropertiesPolicy;
import org.alfresco.repo.policy.Behaviour;
import org.alfresco.repo.policy.JavaBehaviour;
import org.alfresco.repo.policy.PolicyComponent;
import org.alfresco.repo.security.authentication.AuthenticationUtil;
import org.alfresco.repo.security.authentication.AuthenticationUtil.RunAsWork;
import org.alfresco.repo.transaction.AlfrescoTransactionSupport;
import org.alfresco.repo.version.VersionServicePolicies;
import org.alfresco.repo.version.VersionModel;
import org.alfresco.service.cmr.lock.LockService;
import org.alfresco.service.cmr.repository.ChildAssociationRef;
import org.alfresco.service.cmr.repository.NodeRef;
import org.alfresco.service.cmr.repository.NodeService;
import org.alfresco.service.cmr.version.Version;
import org.alfresco.service.cmr.version.VersionService;
import org.alfresco.service.cmr.version.VersionType;
import org.alfresco.service.namespace.NamespacePrefixResolver;
import org.alfresco.service.namespace.NamespaceService;
import org.alfresco.service.namespace.QName;
import org.alfresco.util.EqualsHelper;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.springframework.extensions.surf.util.I18NUtil;

/**
 * Alvex custom implementaion extending default versionable aspect
 * 
 */
public class AlvexVersionableAspect
        implements ContentServicePolicies.OnContentUpdatePolicy, NodeServicePolicies.BeforeAddAspectPolicy,
        NodeServicePolicies.OnAddAspectPolicy, NodeServicePolicies.OnRemoveAspectPolicy,
        NodeServicePolicies.OnDeleteNodePolicy, NodeServicePolicies.OnUpdatePropertiesPolicy,
        VersionServicePolicies.AfterCreateVersionPolicy, CopyServicePolicies.OnCopyNodePolicy, DictionaryListener {
    protected static Log logger = LogFactory.getLog(AlvexVersionableAspect.class);

    /** The i18n'ized messages */
    private static final String MSG_INITIAL_VERSION = "create_version.initial_version";
    private static final String MSG_AUTO_VERSION = "create_version.auto_version";
    private static final String MSG_AUTO_VERSION_PROPS = "create_version.auto_version_props";

    /** Transaction resource key */
    private static final String KEY_VERSIONED_NODEREFS = "versioned_noderefs";

    /** The policy component */
    private PolicyComponent policyComponent;

    /** The node service */
    private NodeService nodeService;

    private LockService lockService;

    /** The Version service */
    private VersionService versionService;

    /** The dictionary DAO. */
    private DictionaryDAO dictionaryDAO;

    /** The Namespace Prefix Resolver. */
    private NamespacePrefixResolver namespacePrefixResolver;

    /** Behaviours */
    JavaBehaviour onUpdatePropertiesBehaviour;

    /** 
     * Optional list of excluded props 
     * - only applies if cm:autoVersionOnUpdateProps=true (and cm:autoVersion=true)
     * - if any one these props changes then "auto version on prop update" does not occur (even if there are other property changes)
     */
    private List<String> excludedOnUpdateProps = Collections.emptyList();

    private Set<QName> excludedOnUpdatePropQNames = Collections.emptySet();

    /** flag indicating whether auto-versioning should be enabled or not */
    private boolean enableAutoVersioning = true;

    /**
     * Set the policy component
     * 
     * @param policyComponent   the policy component
     */
    public void setPolicyComponent(PolicyComponent policyComponent) {
        this.policyComponent = policyComponent;
    }

    /**
     * Set the version service
     * 
     * @param versionService    the version service
     */
    public void setVersionService(VersionService versionService) {
        this.versionService = versionService;
    }

    /**
     * Set the node service
     * 
     * @param nodeService   the node service
     */
    public void setNodeService(NodeService nodeService) {
        this.nodeService = nodeService;
    }

    /**
     * Set the lock service
     * 
     * @param lockService   the lock service
     */
    public void setLockService(LockService lockService) {
        this.lockService = lockService;
    }

    /**
     * Sets the dictionary DAO.
     * 
     * @param dictionaryDAO
     *            the dictionary DAO
     */
    public void setDictionaryDAO(DictionaryDAO dictionaryDAO) {
        this.dictionaryDAO = dictionaryDAO;
    }

    /**
     * Sets the namespace prefix resolver.
     * 
     * @param namespacePrefixResolver
     *            the namespace prefix resolver
     */
    public void setNamespacePrefixResolver(NamespacePrefixResolver namespacePrefixResolver) {
        this.namespacePrefixResolver = namespacePrefixResolver;
    }

    /**
      * @return              Returns the current list of properties that <b>do not</b> trigger versioning
      */
    public List<String> getExcludedOnUpdateProps() {
        return excludedOnUpdateProps;
    }

    /**
     * @param excludedOnUpdateProps         the list of properties that force versioning to ignore changes
     */
    public void setExcludedOnUpdateProps(List<String> excludedOnUpdateProps) {
        this.excludedOnUpdateProps = Collections.unmodifiableList(excludedOnUpdateProps);
    }

    /**
     * Set whether the aspect-associated behaviour should be enabled or disabled.  This is only used
     * during {@link #init() initialization}.
     * 
     * @param enableAutoVersioning          <tt>true</tt> to enable the aspect behaviour otherwise <tt>false</tt>
     */
    public void setEnableAutoVersioning(boolean enableAutoVersioning) {
        this.enableAutoVersioning = enableAutoVersioning;
    }

    /**
     * Initialise the versionable aspect policies
     */
    public void init() {
        if (!enableAutoVersioning) {
            return;
        }

        this.policyComponent.bindClassBehaviour(QName.createQName(NamespaceService.ALFRESCO_URI, "beforeAddAspect"),
                ContentModel.ASPECT_VERSIONABLE,
                new JavaBehaviour(this, "beforeAddAspect", Behaviour.NotificationFrequency.EVERY_EVENT));

        this.policyComponent.bindClassBehaviour(QName.createQName(NamespaceService.ALFRESCO_URI, "onAddAspect"),
                ContentModel.ASPECT_VERSIONABLE,
                new JavaBehaviour(this, "onAddAspect", Behaviour.NotificationFrequency.TRANSACTION_COMMIT));

        this.policyComponent.bindClassBehaviour(QName.createQName(NamespaceService.ALFRESCO_URI, "onRemoveAspect"),
                ContentModel.ASPECT_VERSIONABLE,
                new JavaBehaviour(this, "onRemoveAspect", Behaviour.NotificationFrequency.TRANSACTION_COMMIT));

        this.policyComponent.bindClassBehaviour(QName.createQName(NamespaceService.ALFRESCO_URI, "onDeleteNode"),
                ContentModel.ASPECT_VERSIONABLE,
                new JavaBehaviour(this, "onDeleteNode", Behaviour.NotificationFrequency.TRANSACTION_COMMIT));

        this.policyComponent.bindClassBehaviour(
                QName.createQName(NamespaceService.ALFRESCO_URI, "afterCreateVersion"),
                ContentModel.ASPECT_VERSIONABLE,
                new JavaBehaviour(this, "afterCreateVersion", Behaviour.NotificationFrequency.EVERY_EVENT));

        this.policyComponent.bindClassBehaviour(ContentServicePolicies.OnContentUpdatePolicy.QNAME,
                ContentModel.ASPECT_VERSIONABLE,
                new JavaBehaviour(this, "onContentUpdate", Behaviour.NotificationFrequency.TRANSACTION_COMMIT));

        onUpdatePropertiesBehaviour = new JavaBehaviour(this, "onUpdateProperties",
                Behaviour.NotificationFrequency.TRANSACTION_COMMIT);
        this.policyComponent.bindClassBehaviour(OnUpdatePropertiesPolicy.QNAME, ContentModel.ASPECT_VERSIONABLE,
                onUpdatePropertiesBehaviour);

        this.policyComponent.bindClassBehaviour(QName.createQName(NamespaceService.ALFRESCO_URI, "getCopyCallback"),
                ContentModel.ASPECT_VERSIONABLE, new JavaBehaviour(this, "getCopyCallback"));

        this.dictionaryDAO.register(this);
    }

    /**
     * @see org.alfresco.repo.node.NodeServicePolicies.OnDeleteNodePolicy#onDeleteNode(org.alfresco.service.cmr.repository.ChildAssociationRef, boolean)
     */
    public void onDeleteNode(ChildAssociationRef childAssocRef, boolean isNodeArchived) {
        if (isNodeArchived == false) {
            // If we are perminantly deleting the node then we need to remove the associated version history
            this.versionService.deleteVersionHistory(childAssocRef.getChildRef());
        }
        // otherwise we do nothing since we need to hold onto the version history in case the node is restored later
    }

    /**
     * @return          Returns the {@link VersionableAspectCopyBehaviourCallback}
     */
    public CopyBehaviourCallback getCopyCallback(QName classRef, CopyDetails copyDetails) {
        return VersionableAspectCopyBehaviourCallback.INSTANCE;
    }

    /**
     * Copy behaviour for the <b>cm:versionable</b> aspect
     * 
     * @author Derek Hulley
     * @since 3.2
     */
    private static class VersionableAspectCopyBehaviourCallback extends DefaultCopyBehaviourCallback {
        private static final CopyBehaviourCallback INSTANCE = new VersionableAspectCopyBehaviourCallback();

        /**
         * Copy the aspect, but only the {@link ContentModel#PROP_AUTO_VERSION} and {@link ContentModel#PROP_AUTO_VERSION_PROPS} properties
         */
        @Override
        public Map<QName, Serializable> getCopyProperties(QName classQName, CopyDetails copyDetails,
                Map<QName, Serializable> properties) {
            Serializable value1 = properties.get(ContentModel.PROP_AUTO_VERSION);
            Serializable value2 = properties.get(ContentModel.PROP_AUTO_VERSION_PROPS);

            if ((value1 != null) || (value2 != null)) {
                Map<QName, Serializable> newProperties = new HashMap<QName, Serializable>(2);

                if (value1 != null) {
                    newProperties.put(ContentModel.PROP_AUTO_VERSION, value1);
                }

                if (value2 != null) {
                    newProperties.put(ContentModel.PROP_AUTO_VERSION_PROPS, value2);
                }

                return newProperties;
            } else {
                return Collections.emptyMap();
            }
        }
    }

    /**
     * Before add aspect policy behaviour
     *
     * @param nodeRef
     * @param aspectTypeQName
     */
    public void beforeAddAspect(final NodeRef nodeRef, QName aspectTypeQName) {
        AuthenticationUtil.runAsSystem(new RunAsWork<Void>() {
            @Override
            public Void doWork() throws Exception {
                if (nodeService.hasAspect(nodeRef, ContentModel.ASPECT_VERSIONABLE) == false
                        && versionService.getVersionHistory(nodeRef) != null) {
                    versionService.deleteVersionHistory(nodeRef);
                    logger.warn("The version history of node " + nodeRef
                            + " that doesn't have versionable aspect was deleted");
                }
                return null;
            }
        });
    }

    /**
     * On add aspect policy behaviour
     * 
     * @param nodeRef
     * @param aspectTypeQName
     */
    public void onAddAspect(NodeRef nodeRef, QName aspectTypeQName) {
        if (this.nodeService.exists(nodeRef) == true
                && this.nodeService.hasAspect(nodeRef, ContentModel.ASPECT_VERSIONABLE) == true
                && aspectTypeQName.equals(ContentModel.ASPECT_VERSIONABLE) == true) {
            boolean initialVersion = true;
            Boolean value = (Boolean) this.nodeService.getProperty(nodeRef, ContentModel.PROP_INITIAL_VERSION);
            if (value != null) {
                initialVersion = value.booleanValue();
            }
            // else this means that the default value has not been set the versionable aspect we applied pre-1.2

            if (initialVersion == true) {
                @SuppressWarnings("unchecked")
                Map<NodeRef, NodeRef> versionedNodeRefs = (Map<NodeRef, NodeRef>) AlfrescoTransactionSupport
                        .getResource(KEY_VERSIONED_NODEREFS);
                if (versionedNodeRefs == null || versionedNodeRefs.containsKey(nodeRef) == false) {
                    // Create the initial-version
                    Map<String, Serializable> versionProperties = new HashMap<String, Serializable>(1);

                    // If a major version is requested, indicate it in the versionProperties map
                    String versionType = (String) nodeService.getProperty(nodeRef, ContentModel.PROP_VERSION_TYPE);
                    if (versionType == null || !versionType.equals(VersionType.MINOR.toString())) {
                        versionProperties.put(VersionModel.PROP_VERSION_TYPE, VersionType.MAJOR);
                    }

                    versionProperties.put(Version.PROP_DESCRIPTION, I18NUtil.getMessage(MSG_INITIAL_VERSION));

                    createVersionImpl(nodeRef, versionProperties);
                }
            }
        }
    }

    /**
     * @see org.alfresco.repo.node.NodeServicePolicies.OnRemoveAspectPolicy#onRemoveAspect(org.alfresco.service.cmr.repository.NodeRef, org.alfresco.service.namespace.QName)
     */
    public void onRemoveAspect(NodeRef nodeRef, QName aspectTypeQName) {
        // When the versionable aspect is removed from a node, then delete the associated version history
        this.versionService.deleteVersionHistory(nodeRef);
    }

    /**
     * On content update policy behaviour
     * 
     * If applicable and "cm:autoVersion" is TRUE then version the node on content update (even if no property updates)
     */
    @SuppressWarnings({ "unchecked", "rawtypes" })
    public void onContentUpdate(NodeRef nodeRef, boolean newContent) {
        if (this.nodeService.exists(nodeRef) == true
                && this.nodeService.hasAspect(nodeRef, ContentModel.ASPECT_VERSIONABLE) == true
                && this.nodeService.hasAspect(nodeRef, ContentModel.ASPECT_TEMPORARY) == false) {
            Map<NodeRef, NodeRef> versionedNodeRefs = (Map) AlfrescoTransactionSupport
                    .getResource(KEY_VERSIONED_NODEREFS);
            if (versionedNodeRefs == null || versionedNodeRefs.containsKey(nodeRef) == false) {
                // Determine whether the node is auto versionable (for content updates) or not
                boolean autoVersion = false;
                Boolean value = (Boolean) this.nodeService.getProperty(nodeRef, ContentModel.PROP_AUTO_VERSION);
                if (value != null) {
                    // If the value is not null then 
                    autoVersion = value.booleanValue();
                }
                // else this means that the default value has not been set and the versionable aspect was applied pre-1.1

                if (autoVersion == true) {
                    // Create the auto-version
                    Map<String, Serializable> versionProperties = new HashMap<String, Serializable>(1);
                    versionProperties.put(Version.PROP_DESCRIPTION, I18NUtil.getMessage(MSG_AUTO_VERSION));

                    createVersionImpl(nodeRef, versionProperties);
                }
            }
        }
    }

    /**
     * On update properties policy behaviour 
     * 
     * If applicable and "cm:autoVersionOnUpdateProps" is TRUE then version the node on properties update (even if no content updates)
     * 
     * @since 3.2
     */
    @SuppressWarnings({ "unchecked", "rawtypes" })
    public void onUpdateProperties(NodeRef nodeRef, Map<QName, Serializable> before,
            Map<QName, Serializable> after) {
        if ((this.nodeService.exists(nodeRef) == true) && !LockUtils.isLockedAndReadOnly(nodeRef, lockService)
                && (this.nodeService.hasAspect(nodeRef, ContentModel.ASPECT_VERSIONABLE) == true)
                && (this.nodeService.hasAspect(nodeRef, ContentModel.ASPECT_TEMPORARY) == false)) {
            onUpdatePropertiesBehaviour.disable();
            try {
                Map<NodeRef, NodeRef> versionedNodeRefs = (Map) AlfrescoTransactionSupport
                        .getResource(KEY_VERSIONED_NODEREFS);
                if (versionedNodeRefs == null || versionedNodeRefs.containsKey(nodeRef) == false) {
                    // Determine whether the node is auto versionable (for property only updates) or not
                    boolean autoVersion = false;
                    Boolean value = (Boolean) this.nodeService.getProperty(nodeRef, ContentModel.PROP_AUTO_VERSION);
                    if (value != null) {
                        // If the value is not null then 
                        autoVersion = value.booleanValue();
                    }

                    boolean autoVersionProps = false;
                    value = (Boolean) this.nodeService.getProperty(nodeRef, ContentModel.PROP_AUTO_VERSION_PROPS);
                    if (value != null) {
                        // If the value is not null then 
                        autoVersionProps = value.booleanValue();
                    }

                    if ((autoVersion == true) && (autoVersionProps == true)) {
                        // Check for explicitly excluded props.
                        // If only they change, then do not auto-version.
                        // (It's exactly the difference with default VersionableAspect.)
                        if (excludedOnUpdatePropQNames.size() > 0) {
                            int diffSize = findChangedProps(before, after);
                            // return if no difference actually (after excludedOnUpdatePropQNames are removed)
                            if (diffSize == 0)
                                return;
                            // drop through and auto-version
                        }

                        // Commenting it out, 
                        // since it results in 2 identical node copies on the first props update.
                        // Older version should keep 'before' props.
                        // nodeService.setProperties(nodeRef, before);

                        // Create the auto-version
                        Map<String, Serializable> versionProperties = new HashMap<String, Serializable>(4);
                        versionProperties.put(Version.PROP_DESCRIPTION,
                                I18NUtil.getMessage(MSG_AUTO_VERSION_PROPS));
                        versionProperties.put(VersionModel.PROP_VERSION_TYPE, VersionType.MINOR);

                        createVersionImpl(nodeRef, versionProperties);

                        // Commenting it out together with the line above.
                        // New version should get 'after' props.
                        // nodeService.setProperties(nodeRef, after);
                    }
                }
            } finally {
                onUpdatePropertiesBehaviour.enable();
            }
        }
    }

    // TODO: may be it should be converted into "ifAtLeastOnePropChanged"
    private int findChangedProps(Map<QName, Serializable> before, Map<QName, Serializable> after) {
        int diffSize = 0;
        //Set<QName> changedNames = new HashSet<QName>();
        Set<QName> propNames = new HashSet<QName>(after.size() * 2);
        propNames.addAll(after.keySet());
        propNames.addAll(before.keySet());
        propNames.removeAll(excludedOnUpdatePropQNames);
        for (QName prop : propNames) {
            Serializable beforeValue = before.get(prop);
            Serializable afterValue = after.get(prop);

            if (EqualsHelper.nullSafeEquals(beforeValue, afterValue) != true) {
                diffSize++;
                //changedNames.add(prop);
            }
        }
        return diffSize;
    }

    /**
     * On create version implementation method
     * 
     * @param nodeRef
     * @param versionProperties
     */
    private void createVersionImpl(NodeRef nodeRef, Map<String, Serializable> versionProperties) {
        final VersionService vs = this.versionService;
        final NodeRef nf = nodeRef;
        final Map<String, Serializable> vp = versionProperties;

        AuthenticationUtil.runAs(new RunAsWork<Void>() {

            @Override
            public Void doWork() throws Exception {
                recordCreateVersion(nf, null);
                vs.createVersion(nf, vp);
                return null;
            }
        }, AuthenticationUtil.getRunAsUser());

    }

    /**
     * @see org.alfresco.repo.version.VersionServicePolicies.OnCreateVersionPolicy#onCreateVersion(org.alfresco.service.namespace.QName, org.alfresco.service.cmr.repository.NodeRef, java.util.Map, org.alfresco.repo.policy.PolicyScope)
     */
    public void afterCreateVersion(NodeRef versionableNode, Version version) {
        recordCreateVersion(versionableNode, version);
    }

    @SuppressWarnings("unchecked")
    private void recordCreateVersion(NodeRef versionableNode, Version version) {
        Map<NodeRef, NodeRef> versionedNodeRefs = (Map<NodeRef, NodeRef>) AlfrescoTransactionSupport
                .getResource(KEY_VERSIONED_NODEREFS);
        if (versionedNodeRefs == null) {
            versionedNodeRefs = new HashMap<NodeRef, NodeRef>();
            AlfrescoTransactionSupport.bindResource(KEY_VERSIONED_NODEREFS, versionedNodeRefs);
        }
        versionedNodeRefs.put(versionableNode, versionableNode);
    }

    /*
     * (non-Javadoc)
     * @see org.alfresco.repo.dictionary.DictionaryListener#onDictionaryInit()
     */
    @Override
    public void onDictionaryInit() {
    }

    /*
     * (non-Javadoc)
     * @see org.alfresco.repo.dictionary.DictionaryListener#afterDictionaryInit()
     */
    @Override
    public void afterDictionaryInit() {
        this.excludedOnUpdatePropQNames = new HashSet<QName>(this.excludedOnUpdateProps.size() * 2);
        for (String prefixString : this.excludedOnUpdateProps) {
            try {
                this.excludedOnUpdatePropQNames.add(QName.createQName(prefixString, this.namespacePrefixResolver));
            } catch (Exception e) {
                // An unregistered prefix. Ignore and continue
            }
        }
    }

    /*
     * (non-Javadoc)
     * @see org.alfresco.repo.dictionary.DictionaryListener#afterDictionaryDestroy()
     */
    @Override
    public void afterDictionaryDestroy() {
    }
}