Resources.java

/*
 * 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 static org.apache.commons.io.IOUtils.copy;

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;

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);
			copy(is, resp.getOutputStream());
		} else {
			resp.sendError(404);
		}

	}
}