'Alfresco: Calculate folder size using Java based WebScript'

package org.dedunu.alfresco;

import java.util.HashMap;
import java.util.List;
import java.util.Map;

import org.alfresco.model.ContentModel;
import org.alfresco.service.cmr.repository.ChildAssociationRef;
import org.alfresco.service.cmr.repository.ContentData;
import org.alfresco.service.cmr.repository.NodeRef;
import org.alfresco.service.cmr.repository.NodeService;
import org.springframework.extensions.webscripts.Cache;
import org.springframework.extensions.webscripts.DeclarativeWebScript;
import org.springframework.extensions.webscripts.Status;
import org.springframework.extensions.webscripts.WebScriptRequest;

public class FileSizeWebScript extends DeclarativeWebScript {

    /**
     * Defining node service.
     */
    private NodeService nodeService;

    /**
     * @param nodeService
     */
    public final void setNodeService(final NodeService nodeService) {
        this.nodeService = nodeService;
    }

    @Override
    protected Map<String, Object> executeImpl(WebScriptRequest req,
             Status status, Cache cache) {
        Map<String, Object> model = new HashMap<String, Object>();
        String nodeRefId = req.getParameter("nodeRef");
        NodeRef nodeRef = new NodeRef(nodeRefId);

        String fileName = (String) nodeService.getProperty(nodeRef,
                ContentModel.PROP_NAME);

        long size = getNodeSize(nodeRef);

        model.put("fileName", fileName);
        model.put("size", Long.toString(size));

        return model;
    }

    private long getNodeSize(NodeRef nodeRef) {
        long size = 0;

        // Collecting current node size
        ContentData contentData = (ContentData) nodeService.getProperty(
                nodeRef, ContentModel.PROP_CONTENT);
        try {
            size = contentData.getSize();
        } catch (Exception e) {
            size = 0;
        }

        // Collecting child nodes' sizes
        List<ChildAssociationRef> chilAssocsList = nodeService
                .getChildAssocs(nodeRef);

        for (ChildAssociationRef childAssociationRef : chilAssocsList) {
            NodeRef childNodeRef = childAssociationRef.getChildRef();
            size = size + getNodeSize(childNodeRef);
        }

        return size;
    }

}