org.dspace.webmvc.controller.BrowseHierarchyController.java Source code

Introduction

Here is the source code for org.dspace.webmvc.controller.BrowseHierarchyController.java

Source

/**
 * $Id: $
 * $URL: $
 * *************************************************************************
 * Copyright (c) 2002-2009, DuraSpace.  All rights reserved
 * Licensed under the DuraSpace License.
 *
 * A copy of the DuraSpace License has been included in this
 * distribution and is available at: http://scm.dspace.org/svn/repo/licenses/LICENSE.txt
 */

package org.dspace.webmvc.controller;

import org.apache.log4j.Logger;
import org.dspace.content.Collection;
import org.dspace.content.Community;
import org.dspace.core.Context;
import org.dspace.webmvc.bind.annotation.RequestAttribute;
import org.springframework.stereotype.Controller;
import org.springframework.ui.ModelMap;
import org.springframework.web.bind.annotation.RequestMapping;
import org.dspace.authorize.AuthorizeManager;

import javax.servlet.http.HttpServletRequest;
import java.sql.SQLException;
import java.util.HashMap;
import java.util.Map;

@Controller
public class BrowseHierarchyController {

    @RequestMapping
    protected String handleRequestInternal(@RequestAttribute Context context, ModelMap model,
            HttpServletRequest request) throws Exception {
        BrowseHierarchyRequestProcessor bhrp = new BrowseHierarchyRequestProcessor(context, request);

        model.addAttribute("collectionMap", bhrp.getCollectionMap());
        model.addAttribute("communityMap", bhrp.getCommunityMap());
        model.addAttribute("topLevelCommunities", bhrp.getTopLevelCommunities());

        if (AuthorizeManager.isAdmin(context)) {
            model.addAttribute("admin_button", Boolean.TRUE);
        }

        //return "pages/hierarchy";
        return "pages/community-list";
    }

    static class BrowseHierarchyRequestProcessor {
        private static Logger log = Logger.getLogger(BrowseHierarchyRequestProcessor.class);

        private Context context;
        private HttpServletRequest request;

        private Community[] topLevelCommunities;

        private Map<Integer, Collection[]> collectionMap = new HashMap<Integer, Collection[]>();
        private Map<Integer, Community[]> communityMap = new HashMap<Integer, Community[]>();

        BrowseHierarchyRequestProcessor(Context pContext, HttpServletRequest pRequest) {
            context = pContext;
            request = pRequest;

        }

        private void initializeMappings() throws SQLException {
            if (topLevelCommunities == null) {
                topLevelCommunities = Community.findAllTop(context);
                for (Community comm : topLevelCommunities) {
                    Integer commId = Integer.valueOf(comm.getID());

                    // Find collections in community
                    collectionMap.put(commId, comm.getCollections());

                    // Find subcommunties in community
                    communityMap.put(commId, comm.getSubcommunities());
                }
            }
        }

        Map<Integer, Collection[]> getCollectionMap() throws SQLException {
            initializeMappings();
            return collectionMap;
        }

        Map<Integer, Community[]> getCommunityMap() throws SQLException {
            initializeMappings();
            return communityMap;
        }

        Community[] getTopLevelCommunities() throws SQLException {
            initializeMappings();
            return topLevelCommunities;
        }
    }
}