net.sf.appstatus.web.pages.Resources.java Source code

Introduction

Here is the source code for net.sf.appstatus.web.pages.Resources.java

Source

/*
 * Copyright 2010-2013 Capgemini 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 net.sf.appstatus.web.pages;

import java.io.IOException;
import java.io.InputStream;
import java.util.HashMap;
import java.util.Map;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import net.sf.appstatus.web.StatusWebHandler;

import org.apache.commons.io.IOUtils;

public class Resources {

    public static class ResourceDefinition {
        private final String location;
        private final String mimeType;

        protected ResourceDefinition(String location, String mimeType) {
            this.location = location;
            this.mimeType = mimeType;
        }

        public String getLocation() {
            return location;
        }

        public String getMimeType() {
            return mimeType;
        }
    }

    public static final String LOGO = "logo";
    private static Map<String, ResourceDefinition> resources = new HashMap<String, ResourceDefinition>();
    public static final String STATUS_ERROR = "error";
    public static final String STATUS_JOB = "job";
    public static final String STATUS_JOB_ERROR = "job-error";
    public static final String STATUS_JOB_WARNING = "job-warning";
    public static final String STATUS_OK = "ok";
    public static final String STATUS_PROP = "prop";

    public static final String STATUS_WARN = "warn";

    static {
        addResource(STATUS_OK, "/org/freedesktop/tango/22x22/status/weather-clear.png", "image/png");
        addResource(STATUS_WARN, "/org/freedesktop/tango/22x22/status/weather-overcast.png", "image/png");
        addResource(STATUS_ERROR, "/org/freedesktop/tango/22x22/status/weather-severe-alert.png", "image/png");
        addResource(STATUS_PROP, "/org/freedesktop/tango/22x22/actions/format-justify-fill.png", "image/png");
        addResource(STATUS_JOB, "/org/freedesktop/tango/22x22/emblems/emblem-system.png", "image/png");
        addResource(STATUS_JOB_ERROR, "/org/freedesktop/tango/22x22/status/dialog-error.png", "image/png");
        addResource(STATUS_JOB_WARNING, "/org/freedesktop/tango/22x22/status/dialog-warning.png", "image/png");
        addResource(LOGO, "/assets/img/appstatus-logo.png", "image/png");
    }

    public static void addResource(String id, String location, String mimeType) {
        resources.put(id, new ResourceDefinition(location, mimeType));
    }

    public static void doGet(StatusWebHandler webHandler, HttpServletRequest req, HttpServletResponse resp)
            throws IOException {

        String location = null;
        String id = req.getParameter("resource");
        if (id == null) {
            id = req.getParameter("icon");
        }

        if (resources.containsKey(id)) {
            resp.setContentType(resources.get(id).getMimeType());
            location = resources.get(id).getLocation();
            InputStream is = Resources.class.getResourceAsStream(location);
            IOUtils.copy(is, resp.getOutputStream());
        } else {
            resp.sendError(404);
        }

    }
}