org.jahia.modules.portal.widgets.action.DocTreeAction.java Source code

Introduction

Here is the source code for org.jahia.modules.portal.widgets.action.DocTreeAction.java

Source

/**
 * ==========================================================================================
 * =                   JAHIA'S DUAL LICENSING - IMPORTANT INFORMATION                       =
 * ==========================================================================================
 *
 *                                 http://www.jahia.com
 *
 *     Copyright (C) 2002-2016 Jahia Solutions Group SA. All rights reserved.
 *
 *     THIS FILE IS AVAILABLE UNDER TWO DIFFERENT LICENSES:
 *     1/GPL OR 2/JSEL
 *
 *     1/ GPL
 *     ==================================================================================
 *
 *     IF YOU DECIDE TO CHOOSE THE GPL LICENSE, YOU MUST COMPLY WITH THE FOLLOWING TERMS:
 *
 *     This program is free software: you can redistribute it and/or modify
 *     it under the terms of the GNU 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/>.
 *
 *
 *     2/ JSEL - Commercial and Supported Versions of the program
 *     ===================================================================================
 *
 *     IF YOU DECIDE TO CHOOSE THE JSEL LICENSE, YOU MUST COMPLY WITH THE FOLLOWING TERMS:
 *
 *     Alternatively, commercial and supported versions of the program - also known as
 *     Enterprise Distributions - must be used in accordance with the terms and conditions
 *     contained in a separate written agreement between you and Jahia Solutions Group SA.
 *
 *     If you are unsure which license is appropriate for your use,
 *     please contact the sales department at sales@jahia.com.
 */
package org.jahia.modules.portal.widgets.action;

import org.jahia.bin.Action;
import org.jahia.bin.ActionResult;
import org.jahia.services.content.JCRNodeIteratorWrapper;
import org.jahia.services.content.JCRNodeWrapper;
import org.jahia.services.content.JCRSessionWrapper;
import org.jahia.services.render.RenderContext;
import org.jahia.services.render.Resource;
import org.jahia.services.render.URLResolver;
import org.json.JSONArray;
import org.json.JSONException;
import org.json.JSONObject;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.stereotype.Component;

import javax.jcr.RepositoryException;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.util.List;
import java.util.Map;

/**
 * Created by kevan on 31/01/14.
 */
@Component
public class DocTreeAction extends Action {
    private static Logger logger = LoggerFactory.getLogger(DocTreeAction.class);

    @Override
    public ActionResult doExecute(HttpServletRequest req, RenderContext renderContext, Resource resource,
            JCRSessionWrapper session, Map<String, List<String>> parameters, URLResolver urlResolver)
            throws Exception {
        String path = "";
        if (parameters.containsKey("path")) {
            path = parameters.get("path").get(0);
            return new ActionResult(HttpServletResponse.SC_OK, null, buildTree(session.getNode(path), true));
        } else {
            return new ActionResult(HttpServletResponse.SC_OK, null, buildRoot(session));
        }
    }

    private JSONObject buildRoot(JCRSessionWrapper sessionWrapper) throws JSONException, RepositoryException {
        JSONObject item = new JSONObject();
        item.put("title", "/");
        item.put("nodeType", "/");
        item.put("url", "");
        item.put("path", "/");

        JSONArray itemChilds = new JSONArray();
        itemChilds.put(buildTree(sessionWrapper.getNode("/sites"), false));
        itemChilds.put(buildTree(sessionWrapper.getNode("/mounts"), false));
        item.put("childs", itemChilds);
        return item;
    }

    private JSONObject buildTree(JCRNodeWrapper node, boolean recursive) throws JSONException, RepositoryException {
        JSONObject item = new JSONObject();
        item.put("title", node.getDisplayableName());
        item.put("nodeType", node.getPrimaryNodeTypeName());
        item.put("url", node.getUrl());
        item.put("path", node.getPath());

        JSONArray itemChilds = new JSONArray();
        if (recursive) {
            JCRNodeIteratorWrapper childs = node.getNodes();
            while (childs.hasNext()) {
                JCRNodeWrapper child = (JCRNodeWrapper) childs.next();
                if ((child.isNodeType("jnt:virtualsite") && !child.getName().equals("systemsite"))
                        || child.isNodeType("jnt:folder") || child.isNodeType("jnt:file")) {
                    itemChilds.put(buildTree(child, false));
                }
            }
        }
        item.put("childs", itemChilds);
        return item;
    }
}