org.etudes.ambrosia.util.AmbrosiaServlet.java Source code

Java tutorial

Introduction

Here is the source code for org.etudes.ambrosia.util.AmbrosiaServlet.java

Source

/**********************************************************************************
 * $URL$
 * $Id$
 ***********************************************************************************
 *
 * Copyright (c) 2008, 2009, 2010 Etudes, Inc.
 * 
 * Portions completed before September 1, 2008
 * Copyright (c) 2007, 2008 The Regents of the University of Michigan & Foothill College, ETUDES Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 
 *      http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
 **********************************************************************************/

package org.etudes.ambrosia.util;

import java.io.IOException;
import java.io.PrintWriter;
import java.util.HashSet;
import java.util.Set;

import javax.servlet.ServletConfig;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.etudes.ambrosia.api.Context;
import org.etudes.ambrosia.api.Controller;
import org.etudes.ambrosia.api.UiService;
import org.sakaiproject.component.cover.ComponentManager;
import org.sakaiproject.tool.api.ToolManager;
import org.sakaiproject.util.ResourceLoader;
import org.sakaiproject.util.StringUtil;
import org.sakaiproject.util.Web;

/**
 * Base class for an Ambrosia application's servlet.
 */
public class AmbrosiaServlet extends HttpServlet {
    /** Our log (commons). */
    private static Log M_log = LogFactory.getLog(AmbrosiaServlet.class);

    /** Localized messages. */
    protected static ResourceLoader messages = new ResourceLoader("mneme-test-tool");

    /** The default view. */
    protected String defaultView = null;

    /** The error view. */
    protected String errorView = "error";

    /** Set of static resource paths. */
    protected Set<String> resourcePaths = new HashSet<String>();

    /** The tool id for this tool. */
    protected String toolId = null;

    /** Our self-injected ToolManager reference. */
    protected ToolManager toolManager = null;

    /** Our self-injected ui service reference. */
    protected UiService ui = null;

    /**
     * Shutdown the servlet.
     */
    public void destroy() {
        M_log.info("destroy()");

        super.destroy();
    }

    /**
     * Access the Servlet's information display.
     * 
     * @return servlet information.
     */
    public String getServletInfo() {
        return "Ambrosia";
    }

    /**
     * Initialize the servlet.
     * 
     * @param config
     *        The servlet config.
     * @throws ServletException
     */
    public void init(ServletConfig config) throws ServletException {
        super.init(config);

        // toolId, resourcePaths, defaultView
        this.toolId = config.getInitParameter("toolId");

        String resourcePaths = config.getInitParameter("resourcePaths");
        if (resourcePaths != null)
            this.resourcePaths.add(resourcePaths);

        this.defaultView = config.getInitParameter("defaultView");

        String errorView = config.getInitParameter("errorView");
        if (errorView != null)
            this.errorView = errorView;

        // self-inject
        ui = (UiService) ComponentManager.get(UiService.class);
        toolManager = (ToolManager) ComponentManager.get(ToolManager.class);

        M_log.info("init()");
    }

    /**
     * Respond to requests.
     * 
     * @param req
     *        The servlet request.
     * @param res
     *        The servlet response.
     * @throws ServletException.
     * @throws IOException.
     */
    protected void doGet(HttpServletRequest req, HttpServletResponse res) throws ServletException, IOException {
        // handle static resource requests
        if (ui.dispatchResource(req, res, getServletContext(), this.resourcePaths))
            return;

        // // special courier handling
        // if ((req.getPathInfo() != null) && (req.getPathInfo().startsWith("/" + Destinations.courier.toString())))
        // {
        // courierGet(req, res);
        // return;
        // }

        // handle pathless requests
        if (ui.redirectToCurrentDestination(req, res, getDefaultView()))
            return;

        Context context = this.ui.prepareGet(req, res, getDefaultView());

        // get and split up the tool destination: 0 parts means must "/", otherwise parts[0] = "", parts[1] = the first part, etc.
        String[] parts = context.getDestination().split("/");

        // handle a top redirect
        if ("!portal!".equals(parts[1])) {
            String url = "/portal/directtool/" + StringUtil.unsplit(parts, 2, parts.length - 2, "/");
            sendTopRedirect(res, url);
            return;
        } else if ("!PORTAL!".equals(parts[1])) {
            String url = "/" + StringUtil.unsplit(parts, 2, parts.length - 2, "/");
            sendTopRedirect(res, url);
            return;
        }

        // which destination?
        Controller destination = ui.getController(parts[1], this.toolId);
        if (destination == null) {
            M_log.warn("doGet: no view registered for: " + parts[1] + " tool: " + this.toolId);
            redirectError(req, res);
        }

        else {
            try {
                context.addMessages(ui.getMessages());
                context.addMessages(destination.getSharedMessages());
                context.addMessages(destination.getMessages());

                String toolContext = this.toolManager.getCurrentPlacement().getContext();
                String docsPath = destination.getDocsPath();
                if (docsPath != null) {
                    docsPath = docsPath.replace("{CONTEXT}", toolContext);
                    context.setDocsPath(docsPath);
                }

                destination.get(req, res, context, parts);
            } catch (IllegalArgumentException e) {
                redirectError(req, res);
            }
        }
    }

    /**
     * Respond to requests.
     * 
     * @param req
     *        The servlet request.
     * @param res
     *        The servlet response.
     * @throws ServletException.
     * @throws IOException.
     */
    protected void doPost(HttpServletRequest req, HttpServletResponse res) throws ServletException, IOException {
        Context context = ui.preparePost(req, res, getDefaultView());

        // get and split up the tool destination: 0 parts means must "/", otherwise parts[0] = "", parts[1] = the first part, etc.
        String[] parts = context.getDestination().split("/");

        // which destination?
        Controller destination = ui.getController(parts[1], this.toolId);
        if (destination == null) {
            redirectError(req, res);
        }

        else {
            try {
                context.addMessages(ui.getMessages());
                context.addMessages(destination.getSharedMessages());
                context.addMessages(destination.getMessages());

                destination.post(req, res, context, parts);
            } catch (IllegalArgumentException e) {
                // redirect to error
                redirectError(req, res);
            }
        }
    }

    /**
     * Get the default view.
     * 
     * @return The default view.
     */
    protected String getDefaultView() {
        return this.defaultView;
    }

    /**
     * Send a redirect to the error destination.
     * 
     * @param req
     * @param res
     * @throws IOException
     */
    protected void redirectError(HttpServletRequest req, HttpServletResponse res) throws IOException {
        String error = Web.returnUrl(req, "/" + this.errorView);
        res.sendRedirect(res.encodeRedirectURL(error));
    }

    /**
     * Send a redirect so our "top" ends up at the url, via javascript.
     * 
     * @param url
     *        The redirect url
     */
    protected void sendTopRedirect(HttpServletResponse res, String url) throws IOException {
        res.setContentType("text/html; charset=UTF-8");
        res.addDateHeader("Expires", System.currentTimeMillis() - (1000L * 60L * 60L * 24L * 365L));
        res.addDateHeader("Last-Modified", System.currentTimeMillis());
        res.addHeader("Cache-Control", "no-store, no-cache, must-revalidate, max-age=0, post-check=0, pre-check=0");
        res.addHeader("Pragma", "no-cache");

        // get the writer
        PrintWriter out = res.getWriter();

        // we are on deep under the main portal window
        out.println("<html><head></head><body><script language=\"JavaScript\">parent.location.replace('" + url
                + "');</script></body></html>");
    }
}